homarr/packages/api/src/router/widgets/calendar.ts

import { observable } from "@trpc/server/observable";
import { z } from "zod/v4";

import type { Modify } from "@homarr/common/types";
import { createLogger } from "@homarr/core/infrastructure/logs";
import { ErrorWithMetadata } from "@homarr/core/infrastructure/logs/error";
import type { Integration } from "@homarr/db/schema";
import type { IntegrationKindByCategory } from "@homarr/definitions";
import { getIntegrationKindsByCategory } from "@homarr/definitions";
import type { CalendarEvent } from "@homarr/integrations/types";
import { radarrReleaseTypes } from "@homarr/integrations/types";
import { calendarMonthRequestHandler } from "@homarr/request-handler/calendar";

import { createManyIntegrationMiddleware } from "../../middlewares/integration";
import { createTRPCRouter, publicProcedure } from "../../trpc";

const logger = createLogger({ module: "calendarRouter" });

export const calendarRouter = createTRPCRouter({
  findAllEvents: publicProcedure
    .input(
      z.object({
        year: z.number(),
        month: z.number(),
        releaseType: z.array(z.enum(radarrReleaseTypes)),
        showUnmonitored: z.boolean(),
      }),
    )
    .concat(createManyIntegrationMiddleware("query", ...getIntegrationKindsByCategory("calendar")))
    .query(async ({ ctx, input }) => {
      const settled = await Promise.allSettled(
        ctx.integrations.map(async (integration) => {
          const { integrationIds: _integrationIds, ...handlerInput } = input;
          const innerHandler = calendarMonthRequestHandler.handler(integration, handlerInput);
          const { data } = await innerHandler.getCachedOrUpdatedDataAsync({ forceUpdate: false });

          return {
            events: data,
            integration: {
              id: integration.id,
              name: integration.name,
              kind: integration.kind,
            },
          };
        }),
      );

      return settled.flatMap((result, index) => {
        if (result.status === "fulfilled") {
          return [result.value];
        }
        const integration = ctx.integrations[index];
        logger.warn(
          new ErrorWithMetadata(
            "Calendar integration request failed; skipping events for this integration",
            {
              integrationId: integration?.id,
              integrationKind: integration?.kind,
            },
            { cause: result.reason },
          ),
        );
        return [];
      });
    }),
  subscribeToEvents: publicProcedure
    .input(
      z.object({
        year: z.number(),
        month: z.number(),
        releaseType: z.array(z.enum(radarrReleaseTypes)),
        showUnmonitored: z.boolean(),
      }),
    )
    .concat(createManyIntegrationMiddleware("query", ...getIntegrationKindsByCategory("calendar")))
    .subscription(({ ctx, input }) => {
      return observable<{
        integration: Modify<Integration, { kind: IntegrationKindByCategory<"calendar"> }>;
        events: CalendarEvent[];
      }>((emit) => {
        const unsubscribes: (() => void)[] = [];
        for (const integrationWithSecrets of ctx.integrations) {
          const { decryptedSecrets: _, ...integration } = integrationWithSecrets;
          const { integrationIds: _integrationIds, ...handlerInput } = input;
          const innerHandler = calendarMonthRequestHandler.handler(integrationWithSecrets, handlerInput);
          const unsubscribe = innerHandler.subscribe((events) => {
            emit.next({
              integration,
              events,
            });
          });
          unsubscribes.push(unsubscribe);
        }
        return () => {
          unsubscribes.forEach((unsubscribe) => {
            unsubscribe();
          });
        };
      });
    }),
});