feat: introduce new options for proxy

raystack · Mar 5, 2021 · 5c27f3c · 5c27f3c
1 parent 12c9f1b
commit 5c27f3c
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 7 deletions.
diff --git a/plugin/proxy/index.ts b/plugin/proxy/index.ts
@@ -12,6 +12,31 @@ const httpRequestClient = async (method: string, uri: string, options: any) => {
   return Wreck.request(method, uri, proxyOptions);
 };
 
+const onResponse = (route: any) => async (
+  err: any,
+  res: any,
+  request: Hapi.Request,
+  h: Hapi.ResponseToolkit
+) => {
+  const payload = await Wreck.read(res, {
+    json: 'force',
+    gunzip: true
+  });
+
+  // only return the following key from the response
+  const responseKeyToReturn = R.pathOr(
+    '',
+    ['options', 'app', 'proxy', 'responseKeyToReturn'],
+    route
+  );
+  if (!R.hasPath(responseKeyToReturn.split('.'), payload)) {
+    return h.response(payload);
+  }
+
+  const payloadToReturn = R.pathOr({}, responseKeyToReturn.split('.'), payload);
+  return h.response(payloadToReturn);
+};
+
 export const plugin = {
   name: 'proxies',
   dependencies: [],
@@ -26,12 +51,15 @@ export const plugin = {
     const routes = generateRoutes(routesContent);
 
     const ROUTES: any[] = routes;
-    const ROUTES_WITH_PROXY_DATA = ROUTES.map(
-      R.assocPath(
-        ['handler', 'proxy', 'httpClient', 'request'],
-        httpRequestClient
-      )
-    );
+    const ROUTES_WITH_PROXY_DATA: any[] = ROUTES.map((route) => {
+      return R.pipe(
+        R.assocPath(
+          ['handler', 'proxy', 'httpClient', 'request'],
+          httpRequestClient
+        ),
+        R.assocPath(['handler', 'proxy', 'onResponse'], onResponse(route))
+      )(route);
+    });
 
     server.route(ROUTES_WITH_PROXY_DATA);
   }

diff --git a/plugin/proxy/utils.ts b/plugin/proxy/utils.ts
@@ -63,7 +63,8 @@ export const generateRoutes = (contents: Array<YAMLRoute> = []) => {
           iam: {
             permissions: route?.permissions || [],
             hooks: route?.hooks || []
-          }
+          },
+          proxy: route?.proxy?.extraOptions
         }
       }
     };