frontend/src/components/ha-selector/ha-selector-device.ts

import {
  customElement,
  html,
  internalProperty,
  LitElement,
  property,
} from "lit-element";
import { ConfigEntry, getConfigEntries } from "../../data/config_entries";
import { DeviceRegistryEntry } from "../../data/device_registry";
import { DeviceSelector } from "../../data/selector";
import { HomeAssistant } from "../../types";
import "../device/ha-device-picker";

@customElement("ha-selector-device")
export class HaDeviceSelector extends LitElement {
  @property() public hass!: HomeAssistant;

  @property() public selector!: DeviceSelector;

  @property() public value?: any;

  @property() public label?: string;

  @internalProperty() public _configEntries?: ConfigEntry[];

  protected updated(changedProperties) {
    if (changedProperties.has("selector")) {
      const oldSelector = changedProperties.get("selector");
      if (oldSelector !== this.selector && this.selector.device.integration) {
        this._loadConfigEntries();
      }
    }
  }

  protected render() {
    return html`<ha-device-picker
      .hass=${this.hass}
      .value=${this.value}
      .label=${this.label}
      .deviceFilter=${(device) => this._filterDevices(device)}
      .includeDeviceClasses=${this.selector.device.entity?.device_class
        ? [this.selector.device.entity.device_class]
        : undefined}
      .includeDomains=${this.selector.device.entity?.domain
        ? [this.selector.device.entity.domain]
        : undefined}
      allow-custom-entity
    ></ha-device-picker>`;
  }

  private _filterDevices(device: DeviceRegistryEntry): boolean {
    if (
      this.selector.device.manufacturer &&
      device.manufacturer !== this.selector.device.manufacturer
    ) {
      return false;
    }
    if (
      this.selector.device.model &&
      device.model !== this.selector.device.model
    ) {
      return false;
    }
    if (this.selector.device.integration) {
      if (
        this._configEntries &&
        !this._configEntries.some((entry) =>
          device.config_entries.includes(entry.entry_id)
        )
      ) {
        return false;
      }
    }
    return true;
  }

  private async _loadConfigEntries() {
    this._configEntries = (await getConfigEntries(this.hass)).filter(
      (entry) => entry.domain === this.selector.device.integration
    );
  }
}

declare global {
  interface HTMLElementTagNameMap {
    "ha-selector-device": HaDeviceSelector;
  }
}