Homarr/src/server/api/routers/app.ts

import { TRPCError } from '@trpc/server';
import axios, { AxiosError } from 'axios';
import Consola from 'consola';
import https from 'https';
import { z } from 'zod';
import { isStatusOk } from '~/components/Dashboard/Tiles/Apps/AppPing';
import { getConfig } from '~/tools/config/getConfig';
import { AppType } from '~/types/app';

import { createTRPCRouter, publicProcedure } from '../trpc';

export const appRouter = createTRPCRouter({
  ping: publicProcedure.input(z.object({
    id: z.string(),
    configName: z.string()
  })).query(async ({ input }) => {
    const agent = new https.Agent({ rejectUnauthorized: false });
    const config = getConfig(input.configName);
    const app = config.apps.find((app) => app.id === input.id);

    if (!app?.url) {
      Consola.error(`App ${input} not found`);
      throw new TRPCError({
        code: 'NOT_FOUND',
        cause: input,
        message: `App ${input.id} was not found`,
      });
    }
    const res = await axios
      .get(app.url, { httpsAgent: agent, timeout: 10000 })
      .then((response) => ({
        status: response.status,
        statusText: response.statusText,
        state: isStatusOk(app as AppType, response.status) ? 'online' : 'offline'
      }))
      .catch((error: AxiosError) => {
        if (error.response) {
          return {
            state: isStatusOk(app as AppType, error.response.status) ? 'online' : 'offline',
            status: error.response.status,
            statusText: error.response.statusText,
          };
        }

        if (error.code === 'ECONNABORTED') {
          Consola.error(`Ping timed out for app with id : ${input} (url: ${app.url})`);
          throw new TRPCError({
            code: 'TIMEOUT',
            cause: input,
            message: `Ping timed out`,
          });
        }

        Consola.error(`Unexpected response: ${error.message}`);
        throw new TRPCError({
          code: 'UNPROCESSABLE_CONTENT',
          cause: input,
          message: `Unexpected response: ${error.message}`,
        });
      });
    return res;
  }),
});