support set response for decorateProxyReqOpt

- now it's possible to set cookie for response
reservio · Oct 2, 2019 · 85af365 · 85af365
1 parent 6007a04
commit 85af365
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/app/steps/decorateProxyReqOpts.js b/app/steps/decorateProxyReqOpts.js
@@ -5,14 +5,15 @@ function defaultDecorator(proxyReqOptBuilder /*, userReq */) {
 }
 
 function decorateProxyReqOpt(container) {
-  var resolverFn = container.options.proxyReqOptDecorator || defaultDecorator;
+    var resolverFn = container.options.proxyReqOptDecorator || defaultDecorator;
 
-  return Promise
-    .resolve(resolverFn(container.proxy.reqBuilder, container.user.req))
-    .then(function(processedReqOpts) {
-        delete processedReqOpts.params;
-        container.proxy.reqBuilder = processedReqOpts;
-        return Promise.resolve(container);
+    return Promise
+        .resolve(resolverFn(container.proxy.reqBuilder, container.user.req, container.user.res))
+        .then(function(data) {
+            delete data.proxyReq.params;
+            container.proxy.reqBuilder = data.proxyReq;
+            container.user.res = data.res;
+            return Promise.resolve(container);
     });
 }