Merge pull request #138 from FlowTestAI/small-fixes

fix: mulitple output nodes in the graph should be able to show live d…

Author: jsajal

src/components/atoms/Editor.js

@@ -41,6 +41,9 @@ const hideScrollbar = EditorView.theme({
     whiteSpace: 'nowrap',
     scrollbarWidth: 'none' /* For Firefox */,
   },
+  '&': {
+    cursor: 'text',
+  },
 });
 
 export const Editor = ({ ...props }) => {

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

@@ -73,20 +73,14 @@ class Graph {
       if (node.type === 'outputNode') {
         this.logger.add(LogLevel.INFO, '', { type: 'outputNode', data: { output: prevNodeOutputData } });
         if (this.tab) {
-          const updatedNodes = this.nodes.map((nd) => {
-            if (nd.id === node.id) {
-              return {
-                ...nd,
-                data: {
-                  ...nd.data,
-                  output: prevNodeOutputData,
-                },
-              };
-            }
-
-            return nd;
-          });
-          useTabStore.getState().updateFlowTestNodes(this.tab.id, updatedNodes);
+          const updatedNode = {
+            ...node,
+            data: {
+              ...node.data,
+              output: prevNodeOutputData,
+            },
+          };
+          useTabStore.getState().updateFlowTestNode(this.tab.id, updatedNode);
         }
         result = {
           status: 'Success',

src/stores/TabStore.js

@@ -67,6 +67,26 @@ export const useTabStore = create((set, get) => ({
       }),
     );
   },
+  updateFlowTestNode: (tabId, updatedNode) => {
+    set(
+      produce((state) => {
+        if (tabId) {
+          const existingTab = state.tabs.find((t) => t.id === tabId);
+          if (existingTab) {
+            if (!existingTab.flowDataDraft) {
+              existingTab.flowDataDraft = existingTab.flowData ? cloneDeep(existingTab.flowData) : {};
+            }
+            existingTab.flowDataDraft.nodes = existingTab.flowDataDraft.nodes.map((node, index) => {
+              if (node.id === updatedNode.id) {
+                return updatedNode;
+              }
+              return node;
+            });
+          }
+        }
+      }),
+    );
+  },
   updateFlowTestEdges: (tabId, edges) => {
     set(
       produce((state) => {