Homarr/packages/request-handler/src/smart-home-entity-state.ts

import dayjs from "dayjs";

import type { IntegrationKindByCategory } from "@homarr/definitions";
import { createIntegrationAsync } from "@homarr/integrations";

import { createCachedIntegrationRequestHandler } from "./lib/cached-integration-request-handler";

export const smartHomeEntityStateRequestHandler = createCachedIntegrationRequestHandler<
  string,
  IntegrationKindByCategory<"smartHomeServer">,
  { entityId: string }
>({
  async requestAsync(integration, input) {
    const integrationInstance = await createIntegrationAsync(integration);
    const result = await integrationInstance.getEntityStateAsync(input.entityId);

    if (!result.success) {
      throw new Error(`Unable to fetch data from Home Assistant error='${result.error as string}'`);
    }

    return result.data.state;
  },
  cacheDuration: dayjs.duration(1, "minute"),
  queryKey: "smartHome-entityState",
});