fix: add auth health check

Author: arcoraven

src/server/routes/system/health.ts

@@ -5,7 +5,6 @@ import { isDatabaseReachable } from "../../../shared/db/client";
 import { env } from "../../../shared/utils/env";
 import { isRedisReachable } from "../../../shared/utils/redis/redis";
 import { thirdwebClientId } from "../../../shared/utils/sdk";
-import { createCustomError } from "../../middleware/error";
 
 type EngineFeature =
   | "KEYPAIR_AUTH"
@@ -15,7 +14,9 @@ type EngineFeature =
   | "SMART_BACKEND_WALLETS";
 
 const ReplySchemaOk = Type.Object({
-  status: Type.String(),
+  db: Type.Boolean(),
+  redis: Type.Boolean(),
+  auth: Type.Boolean(),
   engineVersion: Type.Optional(Type.String()),
   engineTier: Type.Optional(Type.String()),
   features: Type.Array(
@@ -54,29 +55,22 @@ export async function healthCheck(fastify: FastifyInstance) {
       },
     },
     handler: async (_, res) => {
-      if (!(await isDatabaseReachable())) {
-        throw createCustomError(
-          "The database is unreachable.",
-          StatusCodes.SERVICE_UNAVAILABLE,
-          "FAILED_HEALTHCHECK",
-        );
-      }
+      const db = await isDatabaseReachable();
+      const redis = await isRedisReachable();
+      const auth = await isAuthValid();
+      const isHealthy = db && redis && auth;
 
-      if (!(await isRedisReachable())) {
-        throw createCustomError(
-          "Redis is unreachable.",
-          StatusCodes.SERVICE_UNAVAILABLE,
-          "FAILED_HEALTHCHECK",
-        );
-      }
-
-      res.status(StatusCodes.OK).send({
-        status: "OK",
-        engineVersion: env.ENGINE_VERSION,
-        engineTier: env.ENGINE_TIER ?? "SELF_HOSTED",
-        features: getFeatures(),
-        clientId: thirdwebClientId,
-      });
+      res
+        .status(isHealthy ? StatusCodes.OK : StatusCodes.SERVICE_UNAVAILABLE)
+        .send({
+          db,
+          redis,
+          auth,
+          engineVersion: env.ENGINE_VERSION,
+          engineTier: env.ENGINE_TIER ?? "SELF_HOSTED",
+          features: getFeatures(),
+          clientId: thirdwebClientId,
+        });
     },
   });
 }
@@ -95,3 +89,16 @@ const getFeatures = (): EngineFeature[] => {
 
   return features;
 };
+
+async function isAuthValid() {
+  try {
+    const resp = await fetch("https://api.thirdweb.com/v2/keys/use", {
+      headers: {
+        "x-secret-key": env.THIRDWEB_API_SECRET_KEY,
+      },
+    });
+    return resp.ok;
+  } catch {
+    return false;
+  }
+}