@bian-womp/spark-workbench 0.1.22 → 0.1.23

package/lib/cjs/index.cjs

@@ -873,7 +873,7 @@ function preformatValueForDisplay(typeId, value, registry) {
         return preformatValueForDisplay(sparkGraph.getTypedOutputTypeId(value), sparkGraph.getTypedOutputValue(value), registry);
     }
     // Enums
-    if (typeId && typeId.includes("enum:") && registry) {
+    if (typeId && typeId.startsWith("enum:") && registry) {
         const n = Number(value);
         const label = registry.enums.get(typeId)?.valueToLabel.get(n);
         if (label)
@@ -1640,7 +1640,7 @@ function Inspector({ debug, autoScroll, hideWorkbench, onAutoScrollChange, onHid
                                         const orig = originals[h] ?? safeToString(typeId, current);
                                         setDrafts((d) => ({ ...d, [h]: orig }));
                                     };
-                                    const isEnum = typeId?.includes("enum:");
+                                    const isEnum = typeId?.startsWith("enum:");
                                     const inIssues = selectedNodeHandleValidation.inputs.filter((m) => m.handle === h);
                                     const hasValidation = inIssues.length > 0;
                                     const hasErr = inIssues.some((m) => m.level === "error");
@@ -1756,16 +1756,18 @@ function DefaultContextMenu({ open, clientPos, onAdd, onClose, }) {
     const { registry } = useWorkbenchContext();
     const rf = ReactFlow.useReactFlow();
     const ids = Array.from(registry.nodes.keys());
-    // Group node ids by the segment before the first '.'
-    const grouped = {};
+    const root = { __children: {} };
     for (const id of ids) {
         const parts = id.split(".");
-        const cat = parts.length > 1 ? parts[0] : "other";
-        const label = parts.length > 1 ? parts.slice(1).join(".") : id;
-        (grouped[cat] = grouped[cat] || []).push({ id, label });
+        let node = root;
+        for (let i = 0; i < parts.length; i++) {
+            const key = parts[i];
+            node.__children[key] = node.__children[key] || { __children: {} };
+            node = node.__children[key];
+            if (i === parts.length - 1)
+                node.__self = id;
+        }
     }
-    const cats = Object.keys(grouped).sort((a, b) => a.localeCompare(b));
-    cats.forEach((c) => grouped[c].sort((a, b) => a.label.localeCompare(b.label)));
     const totalCount = ids.length;
     // Ref for focus/outside click handling
     const ref = React.useRef(null);
@@ -1809,10 +1811,19 @@ function DefaultContextMenu({ open, clientPos, onAdd, onClose, }) {
         onAdd(typeId, p);
         onClose();
     };
+    const renderTree = (tree, path = []) => {
+        const entries = Object.entries(tree.__children).sort((a, b) => a[0].localeCompare(b[0]));
+        return (jsxRuntime.jsx("div", { children: entries.map(([key, child]) => {
+                const label = key;
+                const hasChildren = Object.keys(child.__children).length > 0;
+                !!child.__self && !hasChildren;
+                return (jsxRuntime.jsxs("div", { children: [jsxRuntime.jsx("div", { className: "px-2 py-1 text-[11px] uppercase tracking-wide text-gray-400", children: label }), child.__self && (jsxRuntime.jsx("button", { onClick: () => handleClick(child.__self), className: "block w-full text-left px-3 py-1 hover:bg-gray-100 cursor-pointer", title: child.__self, children: child.__self.split(".").slice(-1)[0] })), hasChildren && (jsxRuntime.jsx("div", { className: "pl-2 border-l border-gray-200 ml-2", children: renderTree(child, [...path, key]) }))] }, [...path, key].join(".")));
+            }) }));
+    };
     return (jsxRuntime.jsxs("div", { ref: ref, tabIndex: -1, className: "fixed z-[1000] bg-white border border-gray-300 rounded-none shadow-lg p-1 min-w-[180px] text-sm text-gray-700", style: { left: x, top: y }, onClick: (e) => e.stopPropagation(), onMouseDown: (e) => e.stopPropagation(), onWheel: (e) => e.stopPropagation(), onContextMenu: (e) => {
             e.preventDefault();
             e.stopPropagation();
-        }, children: [jsxRuntime.jsxs("div", { className: "px-2 py-1 font-semibold text-gray-700", children: ["Add Node ", jsxRuntime.jsxs("span", { className: "text-gray-500 font-normal", children: ["(", totalCount, ")"] })] }), jsxRuntime.jsx("div", { className: "max-h-60 overflow-auto", children: cats.map((cat) => (jsxRuntime.jsxs("div", { className: "py-1", children: [jsxRuntime.jsxs("div", { className: "px-2 py-1 text-[11px] uppercase tracking-wide text-gray-400", children: [cat, " ", jsxRuntime.jsxs("span", { className: "opacity-60 normal-case", children: ["(", grouped[cat].length, ")"] })] }), grouped[cat].map(({ id, label }) => (jsxRuntime.jsx("button", { onClick: () => handleClick(id), className: "block w-full text-left px-3 py-1 hover:bg-gray-100 cursor-pointer", title: id, children: label }, id)))] }, cat))) })] }));
+        }, children: [jsxRuntime.jsxs("div", { className: "px-2 py-1 font-semibold text-gray-700", children: ["Add Node ", jsxRuntime.jsxs("span", { className: "text-gray-500 font-normal", children: ["(", totalCount, ")"] })] }), jsxRuntime.jsx("div", { className: "max-h-60 overflow-auto", children: renderTree(root) })] }));
 }
 
 function NodeContextMenu({ open, clientPos, nodeId, onClose, }) {
@@ -2347,7 +2358,7 @@ function WorkbenchStudioCanvas({ setRegistry, autoScroll, onAutoScrollChange, ex
             // value.ts handles data URL formatting
             return title || url.slice(0, 32) + (url.length > 32 ? "…" : "");
         }
-        if (typeId && typeId.includes("enum:")) {
+        if (typeId && typeId.startsWith("enum:")) {
             const n = Number(value);
             const label = registry.enums.get(typeId)?.valueToLabel.get(n);
             return label ?? String(n);