From 99f0b803e9b81ce0cffaf50840121edaaba400b2 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Sat, 24 Jun 2023 16:32:48 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A8=20refactor(tabsContext.tsx):=20imp?=
 =?UTF-8?q?rove=20code=20readability=20by=20adding=20whitespace=20and=20co?=
 =?UTF-8?q?nsistent=20formatting=20The=20code=20has=20been=20refactored=20?=
 =?UTF-8?q?to=20improve=20readability=20by=20adding=20whitespace=20and=20c?=
 =?UTF-8?q?onsistent=20formatting.=20The=20function=20names=20have=20been?=
 =?UTF-8?q?=20updated=20to=20use=20camelCase=20for=20consistency.=20The=20?=
 =?UTF-8?q?updateDisplay=5Fname=20function=20has=20been=20updated=20to=20u?=
 =?UTF-8?q?se=20a=20logical=20OR=20operator=20instead=20of=20a=20ternary?=
 =?UTF-8?q?=20operator=20for=20better=20readability.=20The=20updateNodeDoc?=
 =?UTF-8?q?umentation=20function=20has=20been=20added=20to=20update=20the?=
 =?UTF-8?q?=20node=20documentation.=20The=20updateNodeBaseClasses,=20updat?=
 =?UTF-8?q?eNodeEdges,=20updateNodeDescription,=20and=20updateNodeTemplate?=
 =?UTF-8?q?=20functions=20have=20been=20updated=20to=20use=20consistent=20?=
 =?UTF-8?q?parameter=20names=20and=20whitespace.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/frontend/src/contexts/tabsContext.tsx | 30 +++++++++++++++--------
 1 file changed, 20 insertions(+), 10 deletions(-)

diff --git a/src/frontend/src/contexts/tabsContext.tsx b/src/frontend/src/contexts/tabsContext.tsx
index 7637699e3..8a2e17465 100644
--- a/src/frontend/src/contexts/tabsContext.tsx
+++ b/src/frontend/src/contexts/tabsContext.tsx
@@ -192,39 +192,49 @@ export function TabsProvider({ children }: { children: ReactNode }) {
   }
 
   function processFlowEdges(flow) {
-    if(!flow.data || !flow.data.edges) return;
+    if (!flow.data || !flow.data.edges) return;
     flow.data.edges.forEach((edge) => {
       edge.className = "";
       edge.style = { stroke: "#555555" };
     });
   }
-  function updateDisplay_name(node:NodeType,template:APIClassType) {
-    node.data.node.display_name = template["display_name"]?template["display_name"]:node.data.type;
+
+  function updateDisplay_name(node: NodeType, template: APIClassType) {
+    node.data.node.display_name = template["display_name"] || node.data.type;
+  }
+
+  function updateNodeDocumentation(node: NodeType, template: APIClassType) {
+    node.data.node.documentation = template["documentation"];
   }
 
   function processFlowNodes(flow) {
-    if(!flow.data || !flow.data.nodes) return;
-    flow.data.nodes.forEach((node:NodeType) => {
+    if (!flow.data || !flow.data.nodes) return;
+    flow.data.nodes.forEach((node: NodeType) => {
       const template = templates[node.data.type];
       if (!template) {
         setErrorData({ title: `Unknown node type: ${node.data.type}` });
         return;
       }
       if (Object.keys(template["template"]).length > 0) {
-        updateDisplay_name(node,template);
+        updateDisplay_name(node, template);
         updateNodeBaseClasses(node, template);
         updateNodeEdges(flow, node, template);
         updateNodeDescription(node, template);
         updateNodeTemplate(node, template);
+        updateNodeDocumentation(node, template);
       }
     });
   }
 
-  function updateNodeBaseClasses(node:NodeType,template:APIClassType) {
+  function updateNodeBaseClasses(node: NodeType, template: APIClassType) {
     node.data.node.base_classes = template["base_classes"];
   }
 
-  function updateNodeEdges(flow:FlowType, node:NodeType,template:APIClassType) {
+  function updateNodeEdges(
+    flow: FlowType,
+    node: NodeType,
+    template: APIClassType
+  ) {
     flow.data.edges.forEach((edge) => {
       if (edge.source === node.id) {
         edge.sourceHandle = edge.sourceHandle
@@ -236,11 +246,11 @@ export function TabsProvider({ children }: { children: ReactNode }) {
     });
   }
 
-  function updateNodeDescription(node:NodeType,template:APIClassType) {
+  function updateNodeDescription(node: NodeType, template: APIClassType) {
     node.data.node.description = template["description"];
   }
 
-  function updateNodeTemplate(node:NodeType,template:APIClassType) {
+  function updateNodeTemplate(node: NodeType, template: APIClassType) {
     node.data.node.template = updateTemplate(
       template["template"] as unknown as APITemplateType,
       node.data.node.template as APITemplateType