From 85c9a721ee03887521f9576fb5ff5fbb06521d9d Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Mon, 20 Feb 2023 14:04:02 -0300
Subject: [PATCH] source changed to target

---
 .../src/CustomNodes/ChatInputNode/index.tsx   |  4 ++--
 .../src/CustomNodes/ChatOutputNode/index.tsx  |  2 +-
 .../components/parameterComponent/index.tsx   |  4 ++--
 .../src/CustomNodes/InputNode/index.tsx       |  4 ++--
 space_flow/src/utils.ts                       | 20 +++++++++----------
 5 files changed, 17 insertions(+), 17 deletions(-)

diff --git a/space_flow/src/CustomNodes/ChatInputNode/index.tsx b/space_flow/src/CustomNodes/ChatInputNode/index.tsx
index e6e4a27fd..eeefe8acd 100644
--- a/space_flow/src/CustomNodes/ChatInputNode/index.tsx
+++ b/space_flow/src/CustomNodes/ChatInputNode/index.tsx
@@ -12,7 +12,7 @@ export default function ChatInputNode({ data }) {
     <div className="prompt-node relative rounded-lg solid border flex justify-center align-center py-3 px-6 bg-gray-50" style={{color: nodeColors[types[data.type]]}}>
       <Tooltip title="Prefix: str">
         <Handle
-          type="source"
+          type="target"
           position={Position.Left}
           id={"str|Prefix|" + data.id}
           isValidConnection={(connection) =>
@@ -24,7 +24,7 @@ export default function ChatInputNode({ data }) {
       </Tooltip>
         <Tooltip title={"Message: str"}>
       <Handle
-        type="target"
+        type="source"
         position={Position.Right}
         id={'str|str|'+data.id}
         isValidConnection={(connection) => isValidConnection(data,connection)}
diff --git a/space_flow/src/CustomNodes/ChatOutputNode/index.tsx b/space_flow/src/CustomNodes/ChatOutputNode/index.tsx
index 47bde0392..c85d5b969 100644
--- a/space_flow/src/CustomNodes/ChatOutputNode/index.tsx
+++ b/space_flow/src/CustomNodes/ChatOutputNode/index.tsx
@@ -12,7 +12,7 @@ export default function ChatOutputNode({ data }) {
     <div className="prompt-node relative rounded-lg solid border flex justify-center align-center py-3 px-6 bg-gray-50" style={{color: nodeColors[types[data.type]]}}>
       <Tooltip title="Message: str">
         <Handle
-          type="source"
+          type="target"
           isValidConnection={(connection) => isValidConnection(data,connection)}
           position={Position.Left}
           id={"str|output|"+data.id}
diff --git a/space_flow/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/space_flow/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
index ebf7cdcd4..b834c581f 100644
--- a/space_flow/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
+++ b/space_flow/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
@@ -38,7 +38,7 @@ export default function ParameterComponent({
   }, [data.id, position, updateNodeInternals]);
 
   const [enabled, setEnabled] = useState(data.node.template[name]?.value ?? false);
-  let disabled = data.reactFlowInstance.getEdges().some((e) => (e.sourceHandle === id));
+  let disabled = data.reactFlowInstance.getEdges().some((e) => (e.targetHandle === id));
 
   return (
     <div ref={ref} className="w-full flex flex-wrap justify-between items-center bg-gray-50 mt-1 px-5 py-2">
@@ -46,7 +46,7 @@ export default function ParameterComponent({
         <div className="text-sm truncate">{title}<span className="text-red-600">{required ? " *" : ""}</span></div>
         <Tooltip title={tooltipTitle + (required ? " (required)" : "")}>
           <Handle
-            type={left ? "source" : "target"}
+            type={left ? "target" : "source"}
             position={left ? Position.Left : Position.Right}
             id={id}
             isValidConnection={(connection) =>
diff --git a/space_flow/src/CustomNodes/InputNode/index.tsx b/space_flow/src/CustomNodes/InputNode/index.tsx
index 24a9ed431..195ac29d3 100644
--- a/space_flow/src/CustomNodes/InputNode/index.tsx
+++ b/space_flow/src/CustomNodes/InputNode/index.tsx
@@ -18,7 +18,7 @@ export default function InputNode({ data }) {
     <div className="prompt-node relative bg-white w-96 rounded-lg solid border flex flex-col justify-center">
       <Tooltip title="Prefix: str">
         <Handle
-          type="source"
+          type="target"
           position={Position.Left}
           id={"str|Prefix|" + data.id}
           isValidConnection={(connection) =>
@@ -56,7 +56,7 @@ export default function InputNode({ data }) {
           />
       </div>
       <Handle
-        type="target"
+        type="source"
         position={Position.Right}
         id={data.type}
         isValidConnection={(connection) => isValidConnection(data, connection)}
diff --git a/space_flow/src/utils.ts b/space_flow/src/utils.ts
index ec5b5227b..d9c17191d 100644
--- a/space_flow/src/utils.ts
+++ b/space_flow/src/utils.ts
@@ -329,7 +329,7 @@ export function getConnectedNodes(edge: Edge, nodes: Array<Node>): Array<Node> {
   const sourceId = edge.source;
   const targetId = edge.target;
   const connectedNodes = nodes.filter(
-    (node) => node.id === sourceId || node.id === targetId
+    (node) => node.id === targetId || node.id === sourceId
   );
   return connectedNodes;
 }
@@ -339,25 +339,25 @@ export function isValidConnection(
   { source, target, sourceHandle, targetHandle }
 ) {
   if (
-    targetHandle.split('|')[0] === sourceHandle.split("|")[0] ||
-    targetHandle.split('|').slice(2).some((t) => t === sourceHandle.split("|")[0]) ||
-    sourceHandle.split("|")[0] === "str"
+    sourceHandle.split('|')[0] === targetHandle.split("|")[0] ||
+    sourceHandle.split('|').slice(2).some((t) => t === targetHandle.split("|")[0]) ||
+    targetHandle.split("|")[0] === "str"
   ) {
-    let sourceNode = data.reactFlowInstance.getNode(source).data.node;
-    if (!sourceNode) {
+    let targetNode = data.reactFlowInstance.getNode(target).data.node;
+    if (!targetNode) {
       if (
         !data.reactFlowInstance
           .getEdges()
-          .find((e) => e.sourceHandle === sourceHandle)
+          .find((e) => e.targetHandle === targetHandle)
       ) {
         return true;
       }
     } else if (
-      (!sourceNode.template[sourceHandle.split("|")[1]].list &&
+      (!targetNode.template[targetHandle.split("|")[1]].list &&
         !data.reactFlowInstance
           .getEdges()
-          .find((e) => e.sourceHandle === sourceHandle)) ||
-      sourceNode.template[sourceHandle.split("|")[1]].list
+          .find((e) => e.targetHandle === targetHandle)) ||
+      targetNode.template[targetHandle.split("|")[1]].list
     ) {
       return true;
     }