⚡️ Improve ping function

src/server/api/routers/app.ts

@@ -1,46 +1,58 @@
-import { z } from 'zod';
-import axios, { AxiosError } from 'axios';
-import https from 'https';
-import Consola from 'consola';
 import { TRPCError } from '@trpc/server';
+import axios, { AxiosError } from 'axios';
+import Consola from 'consola';
+import { getCookie } from 'cookies-next';
+import https from 'https';
+import { z } from 'zod';
+import { getIsOk } 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({
-        url: z.string(),
-      })
-    )
-    .query(async ({ input }) => {
-      const agent = new https.Agent({ rejectUnauthorized: false });
-      const res = await axios
-        .get(input.url, { httpsAgent: agent, timeout: 2000 })
-        .then((response) => ({
-          status: response.status,
-          statusText: response.statusText,
-        }))
-        .catch((error: AxiosError) => {
-          if (error.response) {
-            Consola.warn(`Unexpected response: ${error.message}`);
+  ping: publicProcedure.input(z.string()).query(async ({ input }) => {
+    const agent = new https.Agent({ rejectUnauthorized: false });
+    const configName = getCookie('config-name');
+    const config = getConfig(configName?.toString() ?? 'default');
+    const app = config.apps.find((app) => app.id === input);
+    const url = app?.url;
+    if (url === undefined || !app) {
+      throw new TRPCError({
+        code: 'NOT_FOUND',
+        message: 'App or url not found',
+      });
+    }
+    const res = await axios
+      .get(url, { httpsAgent: agent, timeout: 2000 })
+      .then((response) => ({
+        status: response.status,
+        statusText: response.statusText,
+      }))
+      .catch((error: AxiosError) => {
+        if (error.response) {
+          if (getIsOk(app as AppType, error.response.status)) {
             return {
+              state: 'offline',
               status: error.response.status,
               statusText: error.response.statusText,
             };
           }
-          if (error.code === 'ECONNABORTED') {
-            throw new TRPCError({
-              code: 'TIMEOUT',
-              message: 'Request Timeout',
-            });
-          }
-
-          Consola.error(`Unexpected error: ${error.message}`);
+        }
+        if (error.code === 'ECONNABORTED') {
           throw new TRPCError({
-            code: 'INTERNAL_SERVER_ERROR',
-            message: 'Internal Server Error',
+            code: 'TIMEOUT',
+            message: 'Request Timeout',
           });
+        }
+
+        Consola.error(`Unexpected response: ${error.message}`);
+        throw new TRPCError({
+          code: 'INTERNAL_SERVER_ERROR',
+          cause: app.id,
+          message: error.message,
         });
-      return res;
-    }),
+      });
+    return res;
+  }),
 });