compute post response vars and add them to env variables

Author: jsajal

src/components/molecules/flow/graph/Graph.js

@@ -1,5 +1,6 @@
 // assumption is that apis are giving json as output
 
+import { cloneDeep } from 'lodash';
 import useCanvasStore from 'stores/CanvasStore';
 import useCollectionStore from 'stores/CollectionStore';
 import { useTabStore } from 'stores/TabStore';
@@ -9,6 +10,10 @@ import { computeRequestNode } from './compute/requestnode';
 
 class Graph {
   constructor(nodes, edges, collectionId, onGraphComplete) {
+    const activeEnv = useCollectionStore
+      .getState()
+      .collections.find((c) => c.id === collectionId)
+      ?.environments.find((e) => e.name === useTabStore.getState().selectedEnv);
     this.nodes = nodes;
     this.edges = edges;
     this.onGraphComplete = onGraphComplete;
@@ -17,10 +22,7 @@ class Graph {
     this.startTime = Date.now();
     this.graphRunNodeOutput = {};
     this.auth = undefined;
-    this.env = useCollectionStore
-      .getState()
-      .collections.find((c) => c.id === collectionId)
-      ?.environments.find((e) => e.name === useTabStore.getState().selectedEnv);
+    this.envVariables = activeEnv ? cloneDeep(activeEnv.variables) : undefined;
   }
 
   #checkTimeout() {
@@ -95,12 +97,19 @@ class Graph {
       }
 
       if (node.type === 'authNode') {
-        this.auth = node.data.type ? computeAuthNode(node.data, this.env) : undefined;
+        this.auth = node.data.type ? computeAuthNode(node.data, this.envVariables) : undefined;
         result = ['Success', node, prevNodeOutput];
       }
 
       if (node.type === 'requestNode') {
-        result = await computeRequestNode(node, prevNodeOutputData, this.env, this.auth, this.logs);
+        result = await computeRequestNode(node, prevNodeOutputData, this.envVariables, this.auth, this.logs);
+        // add post response variables if any
+        if (result[3]) {
+          this.envVariables = {
+            ...this.envVariables,
+            ...result[3],
+          };
+        }
       }
 
       if (this.#checkTimeout()) {

src/components/molecules/flow/graph/compute/authnode.js

@@ -1,9 +1,9 @@
 import { computeVariables } from './utils';
 
-export const computeAuthNode = (auth, env) => {
+export const computeAuthNode = (auth, envVariables) => {
   if (auth.type === 'basic-auth') {
-    auth.username = computeVariables(auth.username, env?.variables);
-    auth.password = computeVariables(auth.password, env?.variables);
+    auth.username = computeVariables(auth.username, envVariables);
+    auth.password = computeVariables(auth.password, envVariables);
 
     return auth;
   } else if (auth.type === 'no-auth') {

src/components/molecules/flow/graph/compute/requestnode.js

@@ -48,12 +48,12 @@ const formulateRequest = (node, finalUrl, variablesDict, auth, logs) => {
   return options;
 };
 
-export const computeRequestNode = async (node, prevNodeOutputData, env, auth, logs) => {
+export const computeRequestNode = async (node, prevNodeOutputData, envVariables, auth, logs) => {
   // step1 evaluate variables of this node
   const evalVariables = computeNodeVariables(node.data.preReqVars, prevNodeOutputData);
 
   const variablesDict = {
-    ...env?.variables,
+    ...envVariables,
     ...evalVariables,
   };
 
@@ -76,6 +76,10 @@ export const computeRequestNode = async (node, prevNodeOutputData, env, auth, lo
   } else {
     logs.push(`Request successful: ${JSON.stringify(res)}`);
     console.debug('Response: ', JSON.stringify(res));
+    if (node.data.postRespVars) {
+      const evalPostRespVars = computeNodeVariables(node.data.postRespVars, res.data);
+      return ['Success', node, res, evalPostRespVars];
+    }
     return ['Success', node, res];
   }
 };