Homarr/apps/tasks/src/main.ts

// This import has to be the first import in the file so that the agent is overridden before any other modules are imported.
import "./undici-log-agent-override";

import type { FastifyTRPCPluginOptions } from "@trpc/server/adapters/fastify";
import { fastifyTRPCPlugin } from "@trpc/server/adapters/fastify";
import fastify from "fastify";

import type { JobRouter } from "@homarr/cron-job-api";
import { jobRouter } from "@homarr/cron-job-api";
import { CRON_JOB_API_KEY_HEADER, CRON_JOB_API_PATH, CRON_JOB_API_PORT } from "@homarr/cron-job-api/constants";
import { jobGroup } from "@homarr/cron-jobs";
import { db } from "@homarr/db";
import { logger } from "@homarr/log";

import { JobManager } from "./job-manager";

const server = fastify({
  maxParamLength: 5000,
});
server.register(fastifyTRPCPlugin, {
  prefix: CRON_JOB_API_PATH,
  trpcOptions: {
    router: jobRouter,
    createContext: ({ req }) => ({
      manager: new JobManager(db, jobGroup),
      apiKey: req.headers[CRON_JOB_API_KEY_HEADER] as string | undefined,
    }),
    onError({ path, error }) {
      logger.error(new Error(`Error in tasks tRPC handler path="${path}"`, { cause: error }));
    },
  } satisfies FastifyTRPCPluginOptions<JobRouter>["trpcOptions"],
});

void (async () => {
  await jobGroup.initializeAsync();
  await jobGroup.startAllAsync();

  try {
    await server.listen({ port: CRON_JOB_API_PORT });
    logger.info(`Tasks web server started successfully port="${CRON_JOB_API_PORT}"`);
  } catch (err) {
    logger.error(new Error(`Failed to start tasks web server port="${CRON_JOB_API_PORT}"`, { cause: err }));
    process.exit(1);
  }
})();