Skip to content

Commit

Permalink
Add preview support to binary sensor group (home-assistant#17682)
Browse files Browse the repository at this point in the history
  • Loading branch information
emontnemery committed Aug 23, 2023
1 parent 5ce31f3 commit ed92958
Show file tree
Hide file tree
Showing 2 changed files with 105 additions and 0 deletions.
17 changes: 17 additions & 0 deletions src/data/group.ts
Original file line number Diff line number Diff line change
Expand Up @@ -43,3 +43,20 @@ export const subscribePreviewGroupSensor = (
flow_type,
user_input,
});

export const subscribePreviewGroupBinarySensor = (
hass: HomeAssistant,
flow_id: string,
flow_type: "config_flow" | "options_flow",
user_input: Record<string, any>,
callback: (preview: {
state: string;
attributes: Record<string, any>;
}) => void
): Promise<UnsubscribeFunc> =>
hass.connection.subscribeMessage(callback, {
type: "group/binary_sensor/start_preview",
flow_id,
flow_type,
user_input,
});
Original file line number Diff line number Diff line change
@@ -0,0 +1,88 @@
import { HassEntity, UnsubscribeFunc } from "home-assistant-js-websocket";
import { LitElement, html } from "lit";
import { customElement, property, state } from "lit/decorators";
import { FlowType } from "../../../data/data_entry_flow";
import { subscribePreviewGroupBinarySensor } from "../../../data/group";
import { HomeAssistant } from "../../../types";
import "./entity-preview-row";

@customElement("flow-preview-group_binary_sensor")
class FlowPreviewGroupBinarySensor extends LitElement {
@property({ attribute: false }) public hass!: HomeAssistant;

@property() public flowType!: FlowType;

public handler!: string;

public stepId!: string;

@property() public flowId!: string;

@property() public stepData!: Record<string, any>;

@state() private _preview?: HassEntity;

private _unsub?: Promise<UnsubscribeFunc>;

disconnectedCallback(): void {
super.disconnectedCallback();
if (this._unsub) {
this._unsub.then((unsub) => unsub());
this._unsub = undefined;
}
}

willUpdate(changedProps) {
if (changedProps.has("stepData")) {
this._subscribePreview();
}
}

protected render() {
return html`<entity-preview-row
.hass=${this.hass}
.stateObj=${this._preview}
></entity-preview-row>`;
}

private _setPreview = (preview: {
state: string;
attributes: Record<string, any>;
}) => {
const now = new Date().toISOString();
this._preview = {
entity_id: "binary_sensor.flow_preview",
last_changed: now,
last_updated: now,
context: { id: "", parent_id: null, user_id: null },
...preview,
};
};

private async _subscribePreview() {
if (this._unsub) {
(await this._unsub)();
this._unsub = undefined;
}
if (this.flowType === "repair_flow") {
return;
}
try {
this._unsub = subscribePreviewGroupBinarySensor(
this.hass,
this.flowId,
this.flowType,
this.stepData,
this._setPreview
);
} catch (err) {
this._preview = undefined;
}
}
}

declare global {
interface HTMLElementTagNameMap {
"flow-preview-group_binary_sensor": FlowPreviewGroupBinarySensor;
}
}

0 comments on commit ed92958

Please sign in to comment.