@rkmodules/rules 0.0.163 → 0.0.165

package/dist/index.esm.mjs

@@ -587,6 +587,30 @@ function mergeTrees() {
     return out;
 }
 
+function mergeWithOrder() {
+    var objects = [];
+    for (var _i = 0; _i < arguments.length; _i++) {
+        objects[_i] = arguments[_i];
+    }
+    var merged = {};
+    var keys = [];
+    objects.forEach(function (obj) {
+        Object.keys(obj).forEach(function (key) {
+            if (!keys.includes(key)) {
+                keys.push(key);
+            }
+        });
+    });
+    keys.forEach(function (key) {
+        for (var i = objects.length - 1; i >= 0; i--) {
+            if (key in objects[i]) {
+                merged[key] = objects[i][key];
+                break;
+            }
+        }
+    });
+    return merged;
+}
 function isReference(value) {
     return (typeof value === "string" && !!value.match(/^<([\w\.]+)>$/)
     // value.startsWith("<") &&
@@ -7057,16 +7081,22 @@ OutputNode.displayName = "OutputNode";
 
 var DynamicOutput = React.memo(function (_a) {
     var id = _a.id, data = _a.data, selected = _a.selected;
-    var _b = __read(React.useState({}), 2), outputs = _b[0], setOutputs = _b[1];
+    var outputDefs = data.outputDefs || {};
+    Object.keys(data.outputs || {}).forEach(function (key) {
+        if (!outputDefs[key]) {
+            outputDefs[key] = { type: "any", label: key };
+        }
+    });
+    var _b = __read(React.useState(outputDefs), 2), outputs = _b[0], setOutputs = _b[1];
     var engine = useEngine();
     React.useEffect(function () {
         return engine.subscribe("functionResult", function (event) {
             if (event.name === id) {
                 var result = event.result || {};
-                setOutputs(Object.fromEntries(Object.keys(result).map(function (name, i) { return [
+                setOutputs(mergeWithOrder(Object.fromEntries(Object.keys(result).map(function (name, i) { return [
                     name,
                     { type: "any", label: name },
-                ]; })));
+                ]; })), outputs));
             }
         });
     }, []);
@@ -7293,6 +7323,7 @@ function createNode(engine, id, fncall, options) {
             description: calledFn === null || calledFn === void 0 ? void 0 : calledFn.description,
             inputs: fncall.inputs,
             params: fncall.params,
+            outputs: {},
             inputDefs: __assign(__assign({}, fncall.inputDefs), calledFn === null || calledFn === void 0 ? void 0 : calledFn.inputs),
             paramDefs: (calledFn === null || calledFn === void 0 ? void 0 : calledFn.params) || {},
             outputDefs: (calledFn === null || calledFn === void 0 ? void 0 : calledFn.outputs) || {},
@@ -7314,6 +7345,16 @@ function createEdge(nodeId, ref) {
         targetHandle: "".concat(nodeId, "-").concat(safeName(ref.name)),
     };
 }
+function addOutputToNode(nodes, ref, targetId) {
+    var _a;
+    var node = nodes[ref.refNode];
+    if (!node) {
+        console.warn("Node not found: " + ref.refNode);
+        console.log(nodes, ref, targetId);
+        return;
+    }
+    node.data.outputs = __assign(__assign({}, node.data.outputs), (_a = {}, _a[ref.refField] = "".concat(targetId, ".").concat(ref.name), _a));
+}
 /**
  * flow state stores nodes and edges for the flow graph these can be updated when the function changes
  * using setFn
@@ -7344,6 +7385,7 @@ var useFlowState = create$1(function (set, get) { return ({
         }
         var oldKeys = Object.keys((oldFn === null || oldFn === void 0 ? void 0 : oldFn.body) || {});
         var newKeys = Object.keys(fn.body || {});
+        var edgesToAdd = [];
         // remove all the nodes that were removed
         Object.keys(nodes).forEach(function (key) {
             if (!newKeys.includes(key)) {
@@ -7375,7 +7417,7 @@ var useFlowState = create$1(function (set, get) { return ({
             // add edges
             var refs = getReferences(fncall.inputs || {});
             refs.forEach(function (ref) {
-                edges.push(createEdge(key_1, ref));
+                edgesToAdd.push([key_1, ref]);
             });
         }
         // make sure there is an input node if inputs are defined
@@ -7400,7 +7442,7 @@ var useFlowState = create$1(function (set, get) { return ({
             // add edges
             var refs = getReferences(fncall.inputs || {});
             refs.forEach(function (ref) {
-                edges.push(createEdge(key, ref));
+                edgesToAdd.push([key, ref]);
             });
         });
         // update the changed nodes
@@ -7420,11 +7462,17 @@ var useFlowState = create$1(function (set, get) { return ({
             var refs = getReferences(fncall.inputs || {});
             // remove all incoming edges
             edges = edges === null || edges === void 0 ? void 0 : edges.filter(function (e) { return e.target !== key; });
+            edgesToAdd = edgesToAdd.filter(function (e) { return e[0] !== key; });
             // add the edge
             refs.forEach(function (ref) {
-                edges.push(createEdge(key, ref));
+                edgesToAdd.push([key, ref]);
             });
         });
+        edgesToAdd.forEach(function (_a) {
+            var _b = __read(_a, 2), nodeId = _b[0], ref = _b[1];
+            edges.push(createEdge(nodeId, ref));
+            addOutputToNode(nodes, ref, nodeId);
+        });
         // console.log("final graph", nodes, edges);
         // update the function, nodes and edges
         set({ fn: fn, nodes: nodes, edges: edges });
@@ -8695,5 +8743,5 @@ function DDContext(_a) {
     return React.createElement(DndProvider, { backend: HTML5Backend }, children);
 }
 
-export { DDContext, DISCARD, Engine, Flow, FlowContext, GenericNode, Input, InputBoolean, InputNumber, InputSelect, InputString, InputText, Lib, Output, Param, binaryOnTree, binaryOnTreeBranch, broadCast, deRefEdgeHandle, ensureBranch, expandTree, forEachBranch, forEachItem, getBranch, getBranches, getItem, getPaths, getPositions, getReferences, getValue$1 as getValue, getVariable, graftTree, hasReference, interpolate, isEmpty, isEmptyTree, isReference, isSingleTon, isTree, mapTree, mapTreeBranch, nAryOnTree, nAryOnTreeBranch, normalizePaths, normalizeVarDef, parseReference, primitives, pushItem, safeName, sameShape, simplifyTree$1 as simplifyTree, toArray, topSort, treeSize, treeStats, trimTree, uid$1 as uid, useControls, useDraggableNode, useEngine, useFlow, useFunction, useGraphedFunction, usePositions, usePrompts, useUpdatePositions, useVariable, variableStore };
+export { DDContext, DISCARD, Engine, Flow, FlowContext, GenericNode, Input, InputBoolean, InputNumber, InputSelect, InputString, InputText, Lib, Output, Param, binaryOnTree, binaryOnTreeBranch, broadCast, deRefEdgeHandle, ensureBranch, expandTree, forEachBranch, forEachItem, getBranch, getBranches, getItem, getPaths, getPositions, getReferences, getValue$1 as getValue, getVariable, graftTree, hasReference, interpolate, isEmpty, isEmptyTree, isReference, isSingleTon, isTree, mapTree, mapTreeBranch, mergeWithOrder, nAryOnTree, nAryOnTreeBranch, normalizePaths, normalizeVarDef, parseReference, primitives, pushItem, safeName, sameShape, simplifyTree$1 as simplifyTree, toArray, topSort, treeSize, treeStats, trimTree, uid$1 as uid, useControls, useDraggableNode, useEngine, useFlow, useFunction, useGraphedFunction, usePositions, usePrompts, useUpdatePositions, useVariable, variableStore };
 //# sourceMappingURL=index.esm.mjs.map