frontend/src/common/entity/attribute_icon_path.ts

/** Return an icon representing a attribute. */
import { mdiCircleMedium, mdiCreation } from "@mdi/js";
import { HassEntity } from "home-assistant-js-websocket";
import {
  computeFanModeIcon,
  computeHvacModeIcon,
  computePresetModeIcon,
  computeSwingModeIcon,
} from "../../data/climate";
import { computeHumidiferModeIcon } from "../../data/humidifier";
import { computeDomain } from "./compute_domain";

const iconGenerators: Record<string, Record<string, (value: any) => string>> = {
  climate: {
    fan_mode: computeFanModeIcon,
    hvac_mode: computeHvacModeIcon,
    preset_mode: computePresetModeIcon,
    swing_mode: computeSwingModeIcon,
  },
  humidifier: {
    mode: computeHumidiferModeIcon,
  },
  light: {
    effect: () => mdiCreation,
  },
  fan: {
    preset_mode: () => mdiCircleMedium,
  },
};

export const attributeIconPath = (
  state: HassEntity | undefined,
  attribute: string,
  attributeValue?: string
) => {
  if (!state) {
    return undefined;
  }
  const domain = computeDomain(state.entity_id);
  if (iconGenerators[domain]?.[attribute]) {
    return iconGenerators[domain]?.[attribute](
      attributeValue || state.attributes[attribute]
    );
  }
  return undefined;
};