@foresthubai/workflow-builder 0.4.4 → 0.4.6

package/src/stores/canvasStore.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import { create, UseBoundStore, StoreApi } from "zustand";
 import { Node, Edge, Viewport } from "@xyflow/react";
 import { NodeCategory, type NodeData } from "@foresthubai/workflow-core/node";

package/src/stores/debugStore.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import { create } from "zustand";
 
 // ---------------------------------------------------------------------------

package/src/stores/diagnosticsStore.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 /**
  * diagnosticsStore.ts
  *

package/src/stores/editorStore.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import { create } from "zustand";
 import { getCanvasStore, getOrCreateCanvasStore, MAIN_CANVAS_ID } from "./canvasStore";
 import type { Channel } from "@foresthubai/workflow-core/channel";
@@ -5,7 +9,7 @@ import type { Memory } from "@foresthubai/workflow-core/memory";
 import type { Model, ModelInfo } from "@foresthubai/workflow-core/model";
 import type { FunctionDeclaration } from "@foresthubai/workflow-core/function";
 
-import type { BuilderMode } from "../WorkflowBuilder";
+import type { BuilderMode } from "../mode";
 // Type-only (erased) — the active left-sidebar tab lives here so non-sidebar code
 // (e.g. validation navigation) can open a specific panel. No runtime cycle.
 import type { SidebarTab } from "../panels/BuilderSidebar";

package/src/utils/categoryConstants.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 // Shared category icons and colors for node categories in the workflow builder.
 
 import { NodeCategory } from "@foresthubai/workflow-core/node";

package/src/utils/channelOperations.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import { CHANNEL_DEFINITION, type ChannelType, type Channel } from "@foresthubai/workflow-core/channel";
 import { isParameterActive } from "@foresthubai/workflow-core/parameter";
 import { useEditorStore } from "../stores/editorStore";

package/src/utils/connectionRules.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 // Editor connection rules — which ports may connect, which node types are
 // offered from a port, and whether a node can take another outgoing edge.
 //
@@ -33,24 +37,23 @@ function hasControlFlowEdge(nodeId: string, nodeData: NodeData, edges: Edge[]):
 /**
  * Check whether an output port can accept at least one more outgoing edge.
  * Used by the contextual "+" button on output ports.
+ * Takes the node's own data rather than the nodes array so per-node components
+ * (BaseNode) can call it without subscribing to all nodes.
  */
-export function canPortAcceptEdge(nodeId: string, handleId: string, nodes: Node<NodeData>[], edges: Edge[]): boolean {
-  const node = nodes.find((n) => n.id === nodeId);
-  if (!node) return false;
-
-  const ports = getPorts(node.data);
+export function canPortAcceptEdge(nodeData: NodeData, handleId: string, edges: Edge[]): boolean {
+  const ports = getPorts(nodeData);
   const port = ports.output.find((p) => p.id === handleId);
   if (!port) return false;
 
   // Tool output ports always accept multiple edges (an agent wires up many tools).
   // A control output port accepts a single edge unless the node can branch.
-  if (port.type === "control" && !NodeRegistry.getByType(node.data.type)?.canBranch) {
-    if (edges.some((e) => e.source === nodeId && e.sourceHandle === handleId)) return false;
+  if (port.type === "control" && !NodeRegistry.getByType(nodeData.type)?.canBranch) {
+    if (edges.some((e) => e.source === nodeData.id && e.sourceHandle === handleId)) return false;
   }
 
   // Mutual exclusion: control output blocked when node has tool-input edges
   // (Tool output is exempt — never blocked)
-  if (port.type === "control" && hasToolInputEdge(nodeId, node.data, edges)) return false;
+  if (port.type === "control" && hasToolInputEdge(nodeData.id, nodeData, edges)) return false;
 
   return true;
 }
@@ -107,7 +110,7 @@ export const isValidConnection = (
   if (!srcNode || !tgtNode) return false;
 
   // Source-side checks via canPortAcceptEdge (multiple-outgoing + mutual exclusion)
-  if (!canPortAcceptEdge(sourceId, sourceHandleId, nodes, edges)) return false;
+  if (!canPortAcceptEdge(srcNode.data, sourceHandleId, edges)) return false;
 
   // Get ports using centralized dispatcher
   const sourcePorts = getPorts(srcNode.data);

package/src/utils/functionOperations.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import type { Expression } from "@foresthubai/workflow-core";
 import type { DataType } from "@foresthubai/workflow-core/parameter";
 import type { FunctionDeclaration } from "@foresthubai/workflow-core/function";

package/src/utils/graphOperations.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import {
   NodeBase,
   NodeCategory,

package/src/utils/history.test.ts

@@ -0,0 +1,92 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
+import { describe, it, expect, beforeEach } from "vitest";
+import { create } from "zustand";
+import { history } from "./history";
+
+interface CounterState {
+  count: number;
+  label: string;
+  setCount: (n: number) => void;
+}
+
+// A store on DEFAULT config — no custom equality/partialize — so the default
+// equality path is what's under test.
+function makeStore() {
+  return create(
+    history<CounterState>()((set) => ({
+      count: 0,
+      label: "a",
+      setCount: (n) => set((s) => ({ ...s, count: n })),
+    })),
+  );
+}
+
+describe("history middleware (default config)", () => {
+  let store: ReturnType<typeof makeStore>;
+  beforeEach(() => {
+    store = makeStore();
+  });
+
+  it("withCheckpoint records a checkpoint when the operation changes state", () => {
+    store.getState().withCheckpoint(() => store.getState().setCount(1));
+    expect(store.getState().count).toBe(1);
+    expect(store.getState().canUndo()).toBe(true);
+
+    store.getState().undo();
+    expect(store.getState().count).toBe(0);
+    expect(store.getState().canRedo()).toBe(true);
+
+    store.getState().redo();
+    expect(store.getState().count).toBe(1);
+  });
+
+  it("withCheckpoint does NOT record a checkpoint for a no-op", () => {
+    store.getState().withCheckpoint(() => {});
+    expect(store.getState().canUndo()).toBe(false);
+  });
+
+  it("a no-op withCheckpoint preserves the redo stack", () => {
+    store.getState().withCheckpoint(() => store.getState().setCount(1));
+    store.getState().undo();
+    expect(store.getState().canRedo()).toBe(true);
+
+    store.getState().withCheckpoint(() => {});
+    expect(store.getState().canRedo()).toBe(true);
+  });
+
+  it("a real change clears the redo stack", () => {
+    store.getState().withCheckpoint(() => store.getState().setCount(1));
+    store.getState().undo();
+    store.getState().withCheckpoint(() => store.getState().setCount(7));
+    expect(store.getState().canRedo()).toBe(false);
+  });
+
+  it("withCheckpoint returns the operation's result and propagates throws without a checkpoint", () => {
+    expect(store.getState().withCheckpoint(() => 42)).toBe(42);
+    expect(() =>
+      store.getState().withCheckpoint(() => {
+        throw new Error("boom");
+      }),
+    ).toThrow("boom");
+    expect(store.getState().canUndo()).toBe(false);
+  });
+
+  it("mutationCount ticks on checkpointed changes only", () => {
+    const before = store.getState().mutationCount;
+    store.getState().withCheckpoint(() => {});
+    expect(store.getState().mutationCount).toBe(before);
+    store.getState().withCheckpoint(() => store.getState().setCount(5));
+    expect(store.getState().mutationCount).toBe(before + 1);
+  });
+
+  it("takeCheckpoint always records, and clearHistory empties both stacks", () => {
+    store.getState().takeCheckpoint();
+    expect(store.getState().canUndo()).toBe(true);
+    store.getState().clearHistory();
+    expect(store.getState().canUndo()).toBe(false);
+    expect(store.getState().canRedo()).toBe(false);
+  });
+});

package/src/utils/history.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import { StateCreator, StoreMutatorIdentifier } from "zustand";
 
 // Opaque type for exporting/importing history state
@@ -99,7 +103,7 @@ export const history =
     const {
       limit = 50,
       partialize = (state) => state,
-      equality = (before, after) => JSON.stringify(before) !== JSON.stringify(after),
+      equality = (before, after) => JSON.stringify(before) === JSON.stringify(after),
     } = config;
 
     return (set, get, store) => {

package/src/utils/memoryOperations.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import { MemoryRegistry, type MemoryType, type Memory } from "@foresthubai/workflow-core/memory";
 import { useEditorStore } from "../stores/editorStore";
 import { generateId } from "@foresthubai/workflow-core/id";

package/src/utils/migrateFunctionNodes.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import type { OutputBinding, FunctionCallNode } from "@foresthubai/workflow-core/node";
 import type { Expression } from "@foresthubai/workflow-core";
 import { toast } from "../hooks/use-toast";

package/src/utils/modelOperations.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import { ModelRegistry, type ModelType, type Model } from "@foresthubai/workflow-core/model";
 import { useEditorStore } from "../stores/editorStore";
 import { generateId } from "@foresthubai/workflow-core/id";

package/src/utils/paramDisplay.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 // Presentation-only formatters for inline node display. These render domain
 // values to human-readable text; they live in the builder (not headless
 // workflow-core) because formatting for display is an editor concern.

package/src/utils/resourceHelpers.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import type { Parameter } from "@foresthubai/workflow-core/parameter";
 
 /**

package/src/utils/translation.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import type { TFunction } from "i18next";
 import type { NodeDefinition } from "@foresthubai/workflow-core/node";
 import type { Parameter } from "@foresthubai/workflow-core/parameter";

package/src/utils/variableOperations.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import { declaredVarKey, type DeclaredVariable } from "@foresthubai/workflow-core/variable";
 import type { DataType } from "@foresthubai/workflow-core";
 import { getOrCreateCanvasStore } from "../stores/canvasStore";

package/tailwind-preset.ts

@@ -1,3 +1,7 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// Copyright (c) 2026 ForestHub. All rights reserved.
+// For commercial licensing, contact root@foresthub.ai
+
 import type { Config } from "tailwindcss";
 import animate from "tailwindcss-animate";