@foresthubai/workflow-builder 0.4.2 → 0.4.4

package/src/graph/CustomEdge.tsx

@@ -1,185 +1,185 @@
-import { Tooltip, TooltipContent, TooltipTrigger } from "../components/ui/tooltip";
-import { BaseEdge, EdgeLabelRenderer, getBezierPath, Position, useEdges } from "@xyflow/react";
-import { AlertTriangle } from "lucide-react";
-import { EdgeData, isControlFlow, isToolFlow, type EdgeType } from "@foresthubai/workflow-core/edge";
-import { useEffect, useMemo } from "react";
-import { useAvailableVariables } from "../hooks/useAvailableVariables";
-import { useDiagnosticsStore } from "../stores/diagnosticsStore";
-import { useEditorStore } from "../stores/editorStore";
-import { isReadOnly } from "../WorkflowBuilder";
-import { computeEdgeDiagnostics } from "@foresthubai/workflow-core/diagnostics";
-
-const EDGE_BASE_COLOR = "hsl(var(--edge-default))";
-const AGENT_COLOR = "hsl(var(--node-agent))";
-const ERROR_COLOR = "hsl(var(--destructive))";
-
-/**
- * Source/target stroke colors per edge type. Plain control and tool edges use
- * the neutral edge base on both ends. Edges that touch an Agent pick up the
- * agent color on that side, so the SVG gradient makes the edge visually
- * announce its agent endpoint (e.g. agentTask fades base → agent toward the
- * target Agent).
- */
-function endColors(type: EdgeType): { source: string; target: string } {
-  switch (type) {
-    case "control":
-    case "tool":
-      return { source: EDGE_BASE_COLOR, target: EDGE_BASE_COLOR };
-    case "agentTask":
-      return { source: EDGE_BASE_COLOR, target: AGENT_COLOR };
-    case "agentChoice":
-      return { source: AGENT_COLOR, target: EDGE_BASE_COLOR };
-    case "agentDelegate":
-      return { source: AGENT_COLOR, target: AGENT_COLOR };
-  }
-}
-
-export const CustomEdge = ({
-  id,
-  source,
-  type,
-  sourceX,
-  sourceY,
-  targetX,
-  targetY,
-  data,
-  selected,
-}: {
-  id: string;
-  source: string;
-  type: string;
-  sourceX: number;
-  sourceY: number;
-  targetX: number;
-  targetY: number;
-  data?: EdgeData;
-  selected?: boolean;
-}) => {
-  const isHighlighted = selected ?? false;
-  const isPreview = useEditorStore((s) => isReadOnly(s.builderMode));
-  const edgeType = (type ?? "control") as EdgeType;
-  const edges = useEdges();
-  const activeCanvasId = useEditorStore.getState().activeCanvasId;
-  const { lookup: availableVariables } = useAvailableVariables(activeCanvasId);
-
-  const sourceControlEdgeCount = edges.filter((e) => e.source === source && isControlFlow(e.type as EdgeType)).length;
-
-  // Compute edge diagnostics via extracted pure function
-  const diagnostics = useMemo(
-    () =>
-      computeEdgeDiagnostics({
-        canvasId: activeCanvasId,
-        edgeId: id,
-        edgeType,
-        edgeData: data,
-        availableVariables,
-        sourceControlEdgeCount,
-      }),
-    [edgeType, sourceControlEdgeCount, data, availableVariables, activeCanvasId, id],
-  );
-
-  const hasErrors = diagnostics.length > 0;
-
-  // Write diagnostics to store (cleanup on unmount; validateAllCanvases handles full-project)
-  const setEdgeDiagnostics = useDiagnosticsStore((s) => s.setEdgeDiagnostics);
-  const clearEdgeDiagnostics = useDiagnosticsStore((s) => s.clearEdgeDiagnostics);
-  useEffect(() => {
-    if (isPreview) return;
-    setEdgeDiagnostics(id, diagnostics);
-    return () => clearEdgeDiagnostics(id);
-  }, [id, diagnostics, setEdgeDiagnostics, clearEdgeDiagnostics, isPreview]);
-
-  const getEdgePath = () => {
-    if (isToolFlow(edgeType)) {
-      // Tool ports are diamonds, and React Flow anchors the wire at the handle
-      // box's edge midpoint — i.e. the diamond's pointed tip — so a wire meeting
-      // it looks pinched/detached. Pull each end into its node (tool output sits
-      // on a node's bottom, tool input on the next node's top) so the wire
-      // overlaps the diamond body (hidden under the z-20 port) and reads solid.
-      const OVERLAP = 3;
-      return getBezierPath({
-        sourceX,
-        sourceY: sourceY - OVERLAP,
-        targetX,
-        targetY: targetY + OVERLAP,
-      });
-    }
-    // Control flow: ends have horizontal slope
-    const OVERLAP = 2;
-    return getBezierPath({
-      sourceX: sourceX - OVERLAP,
-      sourceY,
-      sourcePosition: Position.Right,
-      targetX: targetX + OVERLAP,
-      targetY,
-      targetPosition: Position.Left,
-    });
-  };
-
-  const [edgePath, labelX, labelY] = getEdgePath();
-  const colors = hasErrors ? { source: ERROR_COLOR, target: ERROR_COLOR } : endColors(edgeType);
-  const strokeWidth = 3;
-  const gradientId = `edge-gradient-${id}`;
-
-  // Glow effect for highlighted edges — uniform high-contrast color (or
-  // destructive when the edge itself is errored).
-  const glowColor = hasErrors ? ERROR_COLOR : "hsl(var(--selection-glow))";
-  const glowFilter = isHighlighted ? `drop-shadow(0 0 8px ${glowColor}) drop-shadow(0 0 16px ${glowColor})` : undefined;
-
-  return (
-    <>
-      <g style={{ filter: glowFilter }}>
-        <defs>
-          {/* gradientUnits=userSpaceOnUse anchors the stops to the edge's actual
-              source/target coordinates, so the fade follows the wire regardless
-              of its bounding box (which is what objectBoundingBox would use). */}
-          <linearGradient
-            id={gradientId}
-            gradientUnits="userSpaceOnUse"
-            x1={sourceX}
-            y1={sourceY}
-            x2={targetX}
-            y2={targetY}
-          >
-            <stop offset="0%" stopColor={colors.source} />
-            <stop offset="100%" stopColor={colors.target} />
-          </linearGradient>
-        </defs>
-        <BaseEdge
-          id={id}
-          path={edgePath}
-          style={{
-            stroke: `url(#${gradientId})`,
-            strokeWidth,
-            strokeDasharray: "none",
-          }}
-        />
-      </g>
-      <EdgeLabelRenderer>
-        <div
-          style={{
-            position: "absolute",
-            transform: `translate(-50%, -50%) translate(${labelX}px,${labelY}px)`,
-            pointerEvents: "all",
-          }}
-          className="flex flex-col items-center gap-0.5"
-        >
-          {hasErrors && (
-            <Tooltip delayDuration={300}>
-              <TooltipTrigger asChild>
-                <div className="cursor-help">
-                  <AlertTriangle className="h-5 w-5 text-destructive fill-background" />
-                </div>
-              </TooltipTrigger>
-              <TooltipContent side="top" className="bg-destructive text-destructive-foreground text-xs px-2 py-1">
-                {diagnostics[0]?.message ?? "This edge has errors"}
-              </TooltipContent>
-            </Tooltip>
-          )}
-        </div>
-      </EdgeLabelRenderer>
-    </>
-  );
-};
-
-export default CustomEdge;
+import { Tooltip, TooltipContent, TooltipTrigger } from "../components/ui/tooltip";
+import { BaseEdge, EdgeLabelRenderer, getBezierPath, Position, useEdges } from "@xyflow/react";
+import { AlertTriangle } from "lucide-react";
+import { EdgeData, isControlFlow, isToolFlow, type EdgeType } from "@foresthubai/workflow-core/edge";
+import { useEffect, useMemo } from "react";
+import { useAvailableVariables } from "../hooks/useAvailableVariables";
+import { useDiagnosticsStore } from "../stores/diagnosticsStore";
+import { useEditorStore } from "../stores/editorStore";
+import { isReadOnly } from "../WorkflowBuilder";
+import { computeEdgeDiagnostics } from "@foresthubai/workflow-core/diagnostics";
+
+const EDGE_BASE_COLOR = "hsl(var(--edge-default))";
+const AGENT_COLOR = "hsl(var(--node-agent))";
+const ERROR_COLOR = "hsl(var(--destructive))";
+
+/**
+ * Source/target stroke colors per edge type. Plain control and tool edges use
+ * the neutral edge base on both ends. Edges that touch an Agent pick up the
+ * agent color on that side, so the SVG gradient makes the edge visually
+ * announce its agent endpoint (e.g. agentTask fades base → agent toward the
+ * target Agent).
+ */
+function endColors(type: EdgeType): { source: string; target: string } {
+  switch (type) {
+    case "control":
+    case "tool":
+      return { source: EDGE_BASE_COLOR, target: EDGE_BASE_COLOR };
+    case "agentTask":
+      return { source: EDGE_BASE_COLOR, target: AGENT_COLOR };
+    case "agentChoice":
+      return { source: AGENT_COLOR, target: EDGE_BASE_COLOR };
+    case "agentDelegate":
+      return { source: AGENT_COLOR, target: AGENT_COLOR };
+  }
+}
+
+export const CustomEdge = ({
+  id,
+  source,
+  type,
+  sourceX,
+  sourceY,
+  targetX,
+  targetY,
+  data,
+  selected,
+}: {
+  id: string;
+  source: string;
+  type: string;
+  sourceX: number;
+  sourceY: number;
+  targetX: number;
+  targetY: number;
+  data?: EdgeData;
+  selected?: boolean;
+}) => {
+  const isHighlighted = selected ?? false;
+  const isPreview = useEditorStore((s) => isReadOnly(s.builderMode));
+  const edgeType = (type ?? "control") as EdgeType;
+  const edges = useEdges();
+  const activeCanvasId = useEditorStore.getState().activeCanvasId;
+  const { lookup: availableVariables } = useAvailableVariables(activeCanvasId);
+
+  const sourceControlEdgeCount = edges.filter((e) => e.source === source && isControlFlow(e.type as EdgeType)).length;
+
+  // Compute edge diagnostics via extracted pure function
+  const diagnostics = useMemo(
+    () =>
+      computeEdgeDiagnostics({
+        canvasId: activeCanvasId,
+        edgeId: id,
+        edgeType,
+        edgeData: data,
+        availableVariables,
+        sourceControlEdgeCount,
+      }),
+    [edgeType, sourceControlEdgeCount, data, availableVariables, activeCanvasId, id],
+  );
+
+  const hasErrors = diagnostics.length > 0;
+
+  // Write diagnostics to store (cleanup on unmount; validateAllCanvases handles full-project)
+  const setEdgeDiagnostics = useDiagnosticsStore((s) => s.setEdgeDiagnostics);
+  const clearEdgeDiagnostics = useDiagnosticsStore((s) => s.clearEdgeDiagnostics);
+  useEffect(() => {
+    if (isPreview) return;
+    setEdgeDiagnostics(id, diagnostics);
+    return () => clearEdgeDiagnostics(id);
+  }, [id, diagnostics, setEdgeDiagnostics, clearEdgeDiagnostics, isPreview]);
+
+  const getEdgePath = () => {
+    if (isToolFlow(edgeType)) {
+      // Tool ports are diamonds, and React Flow anchors the wire at the handle
+      // box's edge midpoint — i.e. the diamond's pointed tip — so a wire meeting
+      // it looks pinched/detached. Pull each end into its node (tool output sits
+      // on a node's bottom, tool input on the next node's top) so the wire
+      // overlaps the diamond body (hidden under the z-20 port) and reads solid.
+      const OVERLAP = 3;
+      return getBezierPath({
+        sourceX,
+        sourceY: sourceY - OVERLAP,
+        targetX,
+        targetY: targetY + OVERLAP,
+      });
+    }
+    // Control flow: ends have horizontal slope
+    const OVERLAP = 2;
+    return getBezierPath({
+      sourceX: sourceX - OVERLAP,
+      sourceY,
+      sourcePosition: Position.Right,
+      targetX: targetX + OVERLAP,
+      targetY,
+      targetPosition: Position.Left,
+    });
+  };
+
+  const [edgePath, labelX, labelY] = getEdgePath();
+  const colors = hasErrors ? { source: ERROR_COLOR, target: ERROR_COLOR } : endColors(edgeType);
+  const strokeWidth = 3;
+  const gradientId = `edge-gradient-${id}`;
+
+  // Glow effect for highlighted edges — uniform high-contrast color (or
+  // destructive when the edge itself is errored).
+  const glowColor = hasErrors ? ERROR_COLOR : "hsl(var(--selection-glow))";
+  const glowFilter = isHighlighted ? `drop-shadow(0 0 8px ${glowColor}) drop-shadow(0 0 16px ${glowColor})` : undefined;
+
+  return (
+    <>
+      <g style={{ filter: glowFilter }}>
+        <defs>
+          {/* gradientUnits=userSpaceOnUse anchors the stops to the edge's actual
+              source/target coordinates, so the fade follows the wire regardless
+              of its bounding box (which is what objectBoundingBox would use). */}
+          <linearGradient
+            id={gradientId}
+            gradientUnits="userSpaceOnUse"
+            x1={sourceX}
+            y1={sourceY}
+            x2={targetX}
+            y2={targetY}
+          >
+            <stop offset="0%" stopColor={colors.source} />
+            <stop offset="100%" stopColor={colors.target} />
+          </linearGradient>
+        </defs>
+        <BaseEdge
+          id={id}
+          path={edgePath}
+          style={{
+            stroke: `url(#${gradientId})`,
+            strokeWidth,
+            strokeDasharray: "none",
+          }}
+        />
+      </g>
+      <EdgeLabelRenderer>
+        <div
+          style={{
+            position: "absolute",
+            transform: `translate(-50%, -50%) translate(${labelX}px,${labelY}px)`,
+            pointerEvents: "all",
+          }}
+          className="flex flex-col items-center gap-0.5"
+        >
+          {hasErrors && (
+            <Tooltip delayDuration={300}>
+              <TooltipTrigger asChild>
+                <div className="cursor-help">
+                  <AlertTriangle className="h-5 w-5 text-destructive fill-background" />
+                </div>
+              </TooltipTrigger>
+              <TooltipContent side="top" className="bg-destructive text-destructive-foreground text-xs px-2 py-1">
+                {diagnostics[0]?.message ?? "This edge has errors"}
+              </TooltipContent>
+            </Tooltip>
+          )}
+        </div>
+      </EdgeLabelRenderer>
+    </>
+  );
+};
+
+export default CustomEdge;

package/src/graph/CustomNode.tsx

@@ -1,16 +1,16 @@
-import { NodeData, NodeRegistry } from "@foresthubai/workflow-core/node";
-import { NodeProps } from "@xyflow/react";
-import { memo, useMemo } from "react";
-import { BaseNode } from "./BaseNode";
-
-// Standard node component for all non-FunctionCall nodes
-// Simply resolves the node definition from the registry and delegates to BaseNode
-export const CustomNode = memo((props: NodeProps) => {
-  const nodeData = props.data as NodeData;
-
-  // Get node definition from static registry
-  const nodeDefinition = useMemo(() => {
-    return NodeRegistry.getByType(nodeData.type);
-  }, [nodeData.type]);
-  return <BaseNode {...props} nodeDefinition={nodeDefinition} />;
-});
+import { NodeData, NodeRegistry } from "@foresthubai/workflow-core/node";
+import { NodeProps } from "@xyflow/react";
+import { memo, useMemo } from "react";
+import { BaseNode } from "./BaseNode";
+
+// Standard node component for all non-FunctionCall nodes
+// Simply resolves the node definition from the registry and delegates to BaseNode
+export const CustomNode = memo((props: NodeProps) => {
+  const nodeData = props.data as NodeData;
+
+  // Get node definition from static registry
+  const nodeDefinition = useMemo(() => {
+    return NodeRegistry.getByType(nodeData.type);
+  }, [nodeData.type]);
+  return <BaseNode {...props} nodeDefinition={nodeDefinition} />;
+});

package/src/graph/FunctionCallNode.tsx

@@ -1,30 +1,30 @@
-import { FunctionCallNode as DomainFunctionCallNode } from "@foresthubai/workflow-core/node";
-import { NodeProps } from "@xyflow/react";
-import { memo, useMemo } from "react";
-import { buildFunctionNodeDef } from "../hooks/useNodeDefinitions";
-import { useFunctionRegistry } from "../hooks/useFunctionRegistry";
-import { BaseNode } from "./BaseNode";
-
-// Specialized node component for FunctionCall nodes
-// Renders from node's stored data (usually up-to-date via auto-migration,
-// but may be stale after undo)
-export const FunctionCallNode = memo((props: NodeProps) => {
-  const nodeData = props.data as DomainFunctionCallNode;
-
-  // Get function info for staleness check and live label
-  const { getFunction } = useFunctionRegistry();
-  const registryFunctionInfo = getFunction(nodeData.functionInfo.id);
-
-  const isDeleted = !registryFunctionInfo;
-  const isStale = registryFunctionInfo
-    ? nodeData.functionInfo.version !== registryFunctionInfo.version
-    : false;
-
-  // Build node definition from node's stored functionInfo, using registry name for label
-  const nodeDefinition = useMemo(() => {
-    const name = registryFunctionInfo?.name ?? nodeData.functionInfo.name;
-    return buildFunctionNodeDef({ ...nodeData.functionInfo, name });
-  }, [nodeData.functionInfo, registryFunctionInfo?.name]);
-
-  return <BaseNode {...props} nodeDefinition={nodeDefinition} isStale={isStale} isDeleted={isDeleted} />;
-});
+import { FunctionCallNode as DomainFunctionCallNode } from "@foresthubai/workflow-core/node";
+import { NodeProps } from "@xyflow/react";
+import { memo, useMemo } from "react";
+import { buildFunctionNodeDef } from "../hooks/useNodeDefinitions";
+import { useFunctionRegistry } from "../hooks/useFunctionRegistry";
+import { BaseNode } from "./BaseNode";
+
+// Specialized node component for FunctionCall nodes
+// Renders from node's stored data (usually up-to-date via auto-migration,
+// but may be stale after undo)
+export const FunctionCallNode = memo((props: NodeProps) => {
+  const nodeData = props.data as DomainFunctionCallNode;
+
+  // Get function info for staleness check and live label
+  const { getFunction } = useFunctionRegistry();
+  const registryFunctionInfo = getFunction(nodeData.functionInfo.id);
+
+  const isDeleted = !registryFunctionInfo;
+  const isStale = registryFunctionInfo
+    ? nodeData.functionInfo.version !== registryFunctionInfo.version
+    : false;
+
+  // Build node definition from node's stored functionInfo, using registry name for label
+  const nodeDefinition = useMemo(() => {
+    const name = registryFunctionInfo?.name ?? nodeData.functionInfo.name;
+    return buildFunctionNodeDef({ ...nodeData.functionInfo, name });
+  }, [nodeData.functionInfo, registryFunctionInfo?.name]);
+
+  return <BaseNode {...props} nodeDefinition={nodeDefinition} isStale={isStale} isDeleted={isDeleted} />;
+});