#675 Inject proxy paths into adjacent server responses

Author: tariqksoliman

adjacent-servers/adjacent-servers-proxy.js

@@ -8,83 +8,90 @@ function initAdjacentServersProxy(app, isDocker, ensureAdmin) {
   // Proxies
   //// STAC
   if (process.env.WITH_STAC === "true") {
+    const stacTarget = `http://${isDocker ? "stac-fastapi" : "localhost"}:${
+      process.env.STAC_PORT || 8881
+    }`;
     app.use(
       `${process.env.ROOT_PATH || ""}/stac`,
       ensureAdmin(false, false, true), // true to allow all GETs - others require admin auth
       createProxyMiddleware({
-        target: `http://${isDocker ? "stac-fastapi" : "localhost"}:${
-          process.env.STAC_PORT || 8881
-        }`,
+        target: stacTarget,
         changeOrigin: true,
         pathRewrite: {
           [`^${process.env.ROOT_PATH || ""}/stac`]: "",
         },
         selfHandleResponse: true,
         on: {
-          proxyRes: createSwaggerInterceptor("stac"),
+          proxyRes: createSwaggerInterceptor("stac", stacTarget),
         },
       })
     );
   }
 
   //// Tipg
   if (process.env.WITH_TIPG === "true") {
+    const tipgTarget = `http://${isDocker ? "tipg" : "localhost"}:${
+      process.env.TIPG_PORT || 8882
+    }`;
     app.use(
       `${process.env.ROOT_PATH || ""}/tipg`,
       ensureAdmin(false, false, true), // true to allow all GETs - others require admin auth
       createProxyMiddleware({
-        target: `http://${isDocker ? "tipg" : "localhost"}:${
-          process.env.TIPG_PORT || 8882
-        }`,
+        target: tipgTarget,
         changeOrigin: true,
         pathRewrite: {
           [`^${process.env.ROOT_PATH || ""}/tipg`]: "",
         },
         selfHandleResponse: true,
         on: {
-          proxyRes: createSwaggerInterceptor("tipg"),
+          proxyRes: createSwaggerInterceptor("tipg", tipgTarget),
         },
       })
     );
   }
 
   //// TiTiler
   if (process.env.WITH_TITILER === "true") {
+    const titilerTarget = `http://${isDocker ? "titiler" : "localhost"}:${
+      process.env.TITILER_PORT || 8883
+    }`;
     app.use(
       `${process.env.ROOT_PATH || ""}/titiler`,
       ensureAdmin(false, false, true, ["/cog/stac"]), // true to allow all GETs (except /cog/stac) - others require admin auth
       createProxyMiddleware({
-        target: `http://${isDocker ? "titiler" : "localhost"}:${
-          process.env.TITILER_PORT || 8883
-        }`,
+        target: titilerTarget,
         changeOrigin: true,
         pathRewrite: {
           [`^${process.env.ROOT_PATH || ""}/titiler`]: "",
         },
         selfHandleResponse: true,
         on: {
-          proxyRes: createSwaggerInterceptor("titiler"),
+          proxyRes: createSwaggerInterceptor("titiler", titilerTarget),
         },
       })
     );
   }
 
   /// TiTiler-pgSTAC
   if (process.env.WITH_TITILER_PGSTAC === "true") {
+    const titilerpgstacTarget = `http://${
+      isDocker ? "titiler-pgstac" : "localhost"
+    }:${process.env.TITILER_PGSTAC_PORT || 8884}`;
     app.use(
       `${process.env.ROOT_PATH || ""}/titilerpgstac`,
       ensureAdmin(false, false, true), // true to allow all GETs - others require admin auth
       createProxyMiddleware({
-        target: `http://${isDocker ? "titiler-pgstac" : "localhost"}:${
-          process.env.TITILER_PGSTAC_PORT || 8884
-        }`,
+        target: titilerpgstacTarget,
         changeOrigin: true,
         pathRewrite: {
           [`^${process.env.ROOT_PATH || ""}/titilerpgstac`]: "",
         },
         selfHandleResponse: true,
         on: {
-          proxyRes: createSwaggerInterceptor("titilerpgstac"),
+          proxyRes: createSwaggerInterceptor(
+            "titilerpgstac",
+            titilerpgstacTarget
+          ),
         },
       })
     );
@@ -108,22 +115,26 @@ function initAdjacentServersProxy(app, isDocker, ensureAdmin) {
   }
 }
 
-const createSwaggerInterceptor = (path) => {
+const createSwaggerInterceptor = (path, target) => {
   return responseInterceptor(async (responseBuffer, proxyRes, req, res) => {
+    let finalReturn = responseBuffer;
+    let newResponse;
+
     if (req.originalUrl.endsWith(`/${path}/api`)) {
-      const response = JSON.parse(responseBuffer.toString("utf8")); // convert buffer to string
-      response.servers = [
+      newResponse = newResponse || responseBuffer.toString("utf8");
+      const responseJSON = JSON.parse(newResponse); // convert buffer to string
+      responseJSON.servers = [
         {
           url: `${
             (process.env.EXTERNAL_ROOT_PATH || "") +
             (process.env.ROOT_PATH || "")
           }/${path}`,
         },
       ];
-      return JSON.stringify(response); // manipulate response and return the result
+      newResponse = JSON.stringify(responseJSON); // manipulate response
     } else if (req.originalUrl.endsWith(`/${path}/api.html`)) {
-      const response = responseBuffer.toString("utf8"); // convert buffer to string
-      return response
+      newResponse = newResponse || responseBuffer.toString("utf8");
+      newResponse = newResponse
         .replace(
           "'/api'",
           `'${
@@ -137,9 +148,22 @@ const createSwaggerInterceptor = (path) => {
             (process.env.EXTERNAL_ROOT_PATH || "") +
             (process.env.ROOT_PATH || "")
           }/${path}/docs/oauth2-redirect'`
-        ); // manipulate response and return the result
+        ); // manipulate response
+    }
+
+    if (
+      res.get("Content-Type") &&
+      (res.get("Content-Type").includes("json") ||
+        res.get("Content-Type").includes("html"))
+    ) {
+      newResponse = newResponse || responseBuffer.toString("utf8");
+      newResponse = newResponse.replaceAll(
+        target,
+        `${req.protocol}://${req.get("host")}/${path}`
+      );
     }
-    return responseBuffer;
+
+    return newResponse || finalReturn;
   });
 };