chore: added health-checks

Author: jesusmpc

code/src/api/health-checks/controllers/getLivenessHealthCheck.ts

@@ -0,0 +1,35 @@
+// SPDX-FileCopyrightText: 2025 2025 INDUSTRIA DE DISEÑO TEXTIL S.A. (INDITEX S.A.)
+//
+// SPDX-License-Identifier: Apache-2.0
+
+import { Request, Response } from "express";
+import { getAzureWebPubsubServer, isStorageConnected } from "../../../store.js";
+
+export const getLivenessHealthCheckController =
+  () =>
+  async (req: Request, res: Response): Promise<void> => {
+    const testRoomId = "test-readiness-room";
+
+    if (!getAzureWebPubsubServer) {
+      res
+        .status(500)
+        .json({ error: "Azure Web PubSub store instance not found" });
+      return;
+    }
+
+    const url = await getAzureWebPubsubServer().clientConnect(testRoomId);
+
+    if (!url) {
+      res
+        .status(500)
+        .json({ status: "Azure Web Pubsub store failed to connect." });
+      return;
+    }
+
+    if (!isStorageConnected()) {
+      res.status(500).json({ status: "Azure Blob Storage failed to connect." });
+      return;
+    }
+
+    res.status(200).json({ status: "OK" });
+  };

code/src/api/health-checks/controllers/getReadinessHealthCheck.ts

@@ -0,0 +1,20 @@
+// SPDX-FileCopyrightText: 2025 2025 INDUSTRIA DE DISEÑO TEXTIL S.A. (INDITEX S.A.)
+//
+// SPDX-License-Identifier: Apache-2.0
+
+import { Request, Response } from "express";
+import {
+  getAzureWebPubsubServer,
+  isStorageInitialized,
+} from "../../../store.js";
+
+export const getReadinessHealthCheckController =
+  () =>
+  async (req: Request, res: Response): Promise<void> => {
+    if (!getAzureWebPubsubServer || !isStorageInitialized()) {
+      res.status(500).json({ status: "Not initialized" });
+      return;
+    }
+
+    res.status(200).json({ status: "OK" });
+  };

code/src/api/health-checks/controllers/getStartUpHealthCheck.ts

@@ -0,0 +1,20 @@
+// SPDX-FileCopyrightText: 2025 2025 INDUSTRIA DE DISEÑO TEXTIL S.A. (INDITEX S.A.)
+//
+// SPDX-License-Identifier: Apache-2.0
+
+import { Request, Response } from "express";
+import {
+  getAzureWebPubsubServer,
+  isStorageInitialized,
+} from "../../../store.js";
+
+export const getStartUpHealthCheckController =
+  () =>
+  (req: Request, res: Response): void => {
+    if (!getAzureWebPubsubServer || !isStorageInitialized()) {
+      res.status(500).json({ status: "Not initialized" });
+      return;
+    }
+
+    res.status(200).json({ status: "OK" });
+  };

code/src/api/health-checks/router.ts

@@ -0,0 +1,25 @@
+// SPDX-FileCopyrightText: 2025 2025 INDUSTRIA DE DISEÑO TEXTIL S.A. (INDITEX S.A.)
+//
+// SPDX-License-Identifier: Apache-2.0
+
+import { Express, Router } from "express";
+import { getLivenessHealthCheckController } from "./controllers/getLivenessHealthCheck.js";
+import { getReadinessHealthCheckController } from "./controllers/getReadinessHealthCheck.js";
+import { getStartUpHealthCheckController } from "./controllers/getStartUpHealthCheck.js";
+
+const router: Router = Router();
+
+export function getHealthChecksRouter() {
+  return router;
+}
+
+export function setupHealthChecksRouter(app: Express) {
+  const router: Router = Router();
+
+  // Setup router routes
+  router.get(`/liveness`, getLivenessHealthCheckController());
+  router.get(`/readiness`, getReadinessHealthCheckController());
+  router.get(`/startup`, getStartUpHealthCheckController());
+
+  app.use("/v1", router);
+}

code/src/api/v1/controllers/getHealth.ts

@@ -5,7 +5,6 @@
 import { Express, Router } from "express";
 import multer from "multer";
 import { getServiceConfig } from "../../config/config.js";
-import { getHealthController } from "./controllers/getHealth.js";
 import { getRoomConnectController } from "./controllers/getRoomConnect.js";
 import { getImageController } from "./controllers/getImage.js";
 import { postUploadImageController } from "./controllers/postUploadImage.js";
@@ -37,9 +36,6 @@ export function setupApiV1Router(app: Express) {
   // Setup cors
   const cors = getCorsMiddleware();
 
-  // Setup router routes
-  router.get(`/health`, cors, getHealthController());
-
   // Room handling API
   router.use(getAzureWebPubsubServer().getExpressJsMiddleware());
   router.get(

code/src/api/v1/router.ts

@@ -10,6 +10,7 @@ import { setupHttpResponseHeadersMiddleware } from "./middlewares/http-response-
 import { setupBodyParserMiddleware } from "./middlewares/body-parser.js";
 import { setupApiV1Router } from "./api/v1/router.js";
 import { setupApiV2Router } from "./api/v2/router.js";
+import { setupHealthChecksRouter } from "./api/health-checks/router.js";
 import { setLogLevel } from "@azure/logger";
 import { getLogger } from "./logger/logger.js";
 
@@ -37,6 +38,9 @@ export function setupApp() {
   setupHttpResponseHeadersMiddleware(app);
   setupBodyParserMiddleware(app);
 
+  // Setup Health Checks Router
+  setupHealthChecksRouter(app);
+
   // Setup Routers
   setupApiV1Router(app);
   setupApiV2Router(app);

code/src/app.ts

@@ -9,7 +9,7 @@ import path from "node:path";
 import { fileURLToPath } from "node:url";
 import { getLogger, setupLogger } from "./logger/logger.js";
 import { setupApp } from "./app.js";
-import { setupStore } from "./store.js";
+import { setupStorage, setupStore } from "./store.js";
 import { validateServiceConfig } from "./validate.js";
 
 // __dirname equivalent in ESM
@@ -27,6 +27,8 @@ if (!config) {
   process.exit(1);
 }
 
+// Setup the Azure Web Pubsub store
+await setupStorage();
 // Setup the Azure Web Pubsub store
 setupStore();
 

code/src/server.ts

@@ -41,7 +41,23 @@ function extractImageIdFromNode(images: string[], node: any) {
   }
 }
 
-async function setupStorage() {
+export const isStorageInitialized = () => storageInitialized;
+
+export const isStorageConnected = async () => {
+  if (!storageInitialized) {
+    await setupStorage();
+  }
+
+  if (!containerClient || !blobServiceClient) {
+    return null;
+  }
+
+  const exists = await containerClient.exists();
+
+  return exists;
+};
+
+export async function setupStorage() {
   const config = getServiceConfig();
 
   const {