@agentscope-ai/flow 0.0.1

package/dist/hooks/use-nodes-interaction.js

@@ -0,0 +1,777 @@
+function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
+function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/facebook/regenerator/blob/main/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return e; }; var t, e = {}, r = Object.prototype, n = r.hasOwnProperty, o = Object.defineProperty || function (t, e, r) { t[e] = r.value; }, i = "function" == typeof Symbol ? Symbol : {}, a = i.iterator || "@@iterator", c = i.asyncIterator || "@@asyncIterator", u = i.toStringTag || "@@toStringTag"; function define(t, e, r) { return Object.defineProperty(t, e, { value: r, enumerable: !0, configurable: !0, writable: !0 }), t[e]; } try { define({}, ""); } catch (t) { define = function define(t, e, r) { return t[e] = r; }; } function wrap(t, e, r, n) { var i = e && e.prototype instanceof Generator ? e : Generator, a = Object.create(i.prototype), c = new Context(n || []); return o(a, "_invoke", { value: makeInvokeMethod(t, r, c) }), a; } function tryCatch(t, e, r) { try { return { type: "normal", arg: t.call(e, r) }; } catch (t) { return { type: "throw", arg: t }; } } e.wrap = wrap; var h = "suspendedStart", l = "suspendedYield", f = "executing", s = "completed", y = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var p = {}; define(p, a, function () { return this; }); var d = Object.getPrototypeOf, v = d && d(d(values([]))); v && v !== r && n.call(v, a) && (p = v); var g = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(p); function defineIteratorMethods(t) { ["next", "throw", "return"].forEach(function (e) { define(t, e, function (t) { return this._invoke(e, t); }); }); } function AsyncIterator(t, e) { function invoke(r, o, i, a) { var c = tryCatch(t[r], t, o); if ("throw" !== c.type) { var u = c.arg, h = u.value; return h && "object" == _typeof(h) && n.call(h, "__await") ? e.resolve(h.__await).then(function (t) { invoke("next", t, i, a); }, function (t) { invoke("throw", t, i, a); }) : e.resolve(h).then(function (t) { u.value = t, i(u); }, function (t) { return invoke("throw", t, i, a); }); } a(c.arg); } var r; o(this, "_invoke", { value: function value(t, n) { function callInvokeWithMethodAndArg() { return new e(function (e, r) { invoke(t, n, e, r); }); } return r = r ? r.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg(); } }); } function makeInvokeMethod(e, r, n) { var o = h; return function (i, a) { if (o === f) throw new Error("Generator is already running"); if (o === s) { if ("throw" === i) throw a; return { value: t, done: !0 }; } for (n.method = i, n.arg = a;;) { var c = n.delegate; if (c) { var u = maybeInvokeDelegate(c, n); if (u) { if (u === y) continue; return u; } } if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) { if (o === h) throw o = s, n.arg; n.dispatchException(n.arg); } else "return" === n.method && n.abrupt("return", n.arg); o = f; var p = tryCatch(e, r, n); if ("normal" === p.type) { if (o = n.done ? s : l, p.arg === y) continue; return { value: p.arg, done: n.done }; } "throw" === p.type && (o = s, n.method = "throw", n.arg = p.arg); } }; } function maybeInvokeDelegate(e, r) { var n = r.method, o = e.iterator[n]; if (o === t) return r.delegate = null, "throw" === n && e.iterator.return && (r.method = "return", r.arg = t, maybeInvokeDelegate(e, r), "throw" === r.method) || "return" !== n && (r.method = "throw", r.arg = new TypeError("The iterator does not provide a '" + n + "' method")), y; var i = tryCatch(o, e.iterator, r.arg); if ("throw" === i.type) return r.method = "throw", r.arg = i.arg, r.delegate = null, y; var a = i.arg; return a ? a.done ? (r[e.resultName] = a.value, r.next = e.nextLoc, "return" !== r.method && (r.method = "next", r.arg = t), r.delegate = null, y) : a : (r.method = "throw", r.arg = new TypeError("iterator result is not an object"), r.delegate = null, y); } function pushTryEntry(t) { var e = { tryLoc: t[0] }; 1 in t && (e.catchLoc = t[1]), 2 in t && (e.finallyLoc = t[2], e.afterLoc = t[3]), this.tryEntries.push(e); } function resetTryEntry(t) { var e = t.completion || {}; e.type = "normal", delete e.arg, t.completion = e; } function Context(t) { this.tryEntries = [{ tryLoc: "root" }], t.forEach(pushTryEntry, this), this.reset(!0); } function values(e) { if (e || "" === e) { var r = e[a]; if (r) return r.call(e); if ("function" == typeof e.next) return e; if (!isNaN(e.length)) { var o = -1, i = function next() { for (; ++o < e.length;) if (n.call(e, o)) return next.value = e[o], next.done = !1, next; return next.value = t, next.done = !0, next; }; return i.next = i; } } throw new TypeError(_typeof(e) + " is not iterable"); } return GeneratorFunction.prototype = GeneratorFunctionPrototype, o(g, "constructor", { value: GeneratorFunctionPrototype, configurable: !0 }), o(GeneratorFunctionPrototype, "constructor", { value: GeneratorFunction, configurable: !0 }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, u, "GeneratorFunction"), e.isGeneratorFunction = function (t) { var e = "function" == typeof t && t.constructor; return !!e && (e === GeneratorFunction || "GeneratorFunction" === (e.displayName || e.name)); }, e.mark = function (t) { return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, define(t, u, "GeneratorFunction")), t.prototype = Object.create(g), t; }, e.awrap = function (t) { return { __await: t }; }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, c, function () { return this; }), e.AsyncIterator = AsyncIterator, e.async = function (t, r, n, o, i) { void 0 === i && (i = Promise); var a = new AsyncIterator(wrap(t, r, n, o), i); return e.isGeneratorFunction(r) ? a : a.next().then(function (t) { return t.done ? t.value : a.next(); }); }, defineIteratorMethods(g), define(g, u, "Generator"), define(g, a, function () { return this; }), define(g, "toString", function () { return "[object Generator]"; }), e.keys = function (t) { var e = Object(t), r = []; for (var n in e) r.push(n); return r.reverse(), function next() { for (; r.length;) { var t = r.pop(); if (t in e) return next.value = t, next.done = !1, next; } return next.done = !0, next; }; }, e.values = values, Context.prototype = { constructor: Context, reset: function reset(e) { if (this.prev = 0, this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(resetTryEntry), !e) for (var r in this) "t" === r.charAt(0) && n.call(this, r) && !isNaN(+r.slice(1)) && (this[r] = t); }, stop: function stop() { this.done = !0; var t = this.tryEntries[0].completion; if ("throw" === t.type) throw t.arg; return this.rval; }, dispatchException: function dispatchException(e) { if (this.done) throw e; var r = this; function handle(n, o) { return a.type = "throw", a.arg = e, r.next = n, o && (r.method = "next", r.arg = t), !!o; } for (var o = this.tryEntries.length - 1; o >= 0; --o) { var i = this.tryEntries[o], a = i.completion; if ("root" === i.tryLoc) return handle("end"); if (i.tryLoc <= this.prev) { var c = n.call(i, "catchLoc"), u = n.call(i, "finallyLoc"); if (c && u) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } else if (c) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); } else { if (!u) throw new Error("try statement without catch or finally"); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } } } }, abrupt: function abrupt(t, e) { for (var r = this.tryEntries.length - 1; r >= 0; --r) { var o = this.tryEntries[r]; if (o.tryLoc <= this.prev && n.call(o, "finallyLoc") && this.prev < o.finallyLoc) { var i = o; break; } } i && ("break" === t || "continue" === t) && i.tryLoc <= e && e <= i.finallyLoc && (i = null); var a = i ? i.completion : {}; return a.type = t, a.arg = e, i ? (this.method = "next", this.next = i.finallyLoc, y) : this.complete(a); }, complete: function complete(t, e) { if ("throw" === t.type) throw t.arg; return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && e && (this.next = e), y; }, finish: function finish(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.finallyLoc === t) return this.complete(r.completion, r.afterLoc), resetTryEntry(r), y; } }, catch: function _catch(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.tryLoc === t) { var n = r.completion; if ("throw" === n.type) { var o = n.arg; resetTryEntry(r); } return o; } } throw new Error("illegal catch attempt"); }, delegateYield: function delegateYield(e, r, n) { return this.delegate = { iterator: values(e), resultName: r, nextLoc: n }, "next" === this.method && (this.arg = t), y; } }, e; }
+function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
+function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
+function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
+function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
+function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
+function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : String(i); }
+function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
+function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }
+function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
+function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
+function _createForOfIteratorHelper(o, allowArrayLike) { var it = typeof Symbol !== "undefined" && o[Symbol.iterator] || o["@@iterator"]; if (!it) { if (Array.isArray(o) || (it = _unsupportedIterableToArray(o)) || allowArrayLike && o && typeof o.length === "number") { if (it) o = it; var i = 0; var F = function F() {}; return { s: F, n: function n() { if (i >= o.length) return { done: true }; return { done: false, value: o[i++] }; }, e: function e(_e) { throw _e; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var normalCompletion = true, didErr = false, err; return { s: function s() { it = it.call(o); }, n: function n() { var step = it.next(); normalCompletion = step.done; return step; }, e: function e(_e2) { didErr = true; err = _e2; }, f: function f() { try { if (!normalCompletion && it.return != null) it.return(); } finally { if (didErr) throw err; } } }; }
+function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
+function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
+import { ITERATION_PADDING, NEW_NODE_PADDING } from "../constant";
+import { useStore } from "../flow/context";
+import $i18n from "../i18n";
+import { copyNodeConfig, copySubFlowNodeConfig, generateEdge, generateUniqueName, getIteratorNodeSize } from "../utils";
+import uniqueId from "../utils/unique-id";
+import { applyNodeChanges, getOutgoers, useReactFlow, useStoreApi } from '@xyflow/react';
+import { message } from 'antd';
+import { useCallback } from 'react';
+import { useFlowInteraction } from "./use-flow-interaction";
+import { useFlowSave } from "./use-flow-save";
+export var useNodesInteraction = function useNodesInteraction() {
+  var _useFlowSave = useFlowSave(),
+    handleSaveFlowDraft = _useFlowSave.handleSaveFlowDraft;
+  var store = useStoreApi();
+  var setSelectedNode = useStore(function (state) {
+    return state.setSelectedNode;
+  });
+  var selectedNode = useStore(function (state) {
+    return state.selectedNode;
+  });
+  var nodeSchemaMap = useStore(function (state) {
+    return state.nodeSchemaMap;
+  });
+  var onAddCustomNode = useStore(function (state) {
+    return state.onAddCustomNode;
+  });
+  var _useReactFlow = useReactFlow(),
+    screenToFlowPosition = _useReactFlow.screenToFlowPosition,
+    getNode = _useReactFlow.getNode;
+  var _useFlowInteraction = useFlowInteraction(),
+    focusElement = _useFlowInteraction.focusElement;
+  var setIsDragging = useStore(function (state) {
+    return state.setIsDragging;
+  });
+  var onConnect = useCallback(function (connection) {
+    var _store$getState = store.getState(),
+      edges = _store$getState.edges,
+      nodes = _store$getState.nodes,
+      setEdges = _store$getState.setEdges;
+    var targetNode = nodes.find(function (node) {
+      return node.id === connection.target;
+    });
+    var sourceNode = nodes.find(function (node) {
+      return node.id === connection.source;
+    });
+    if ((sourceNode !== null && sourceNode !== void 0 && sourceNode.parentId || targetNode !== null && targetNode !== void 0 && targetNode.parentId) && (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.parentId) !== (targetNode === null || targetNode === void 0 ? void 0 : targetNode.parentId)) {
+      message.warning($i18n.get({
+        id: 'spark-flow.hooks.useNodesInteraction.sameSubCanvasOnly',
+        dm: '只能同一子画布中的节点相连'
+      }));
+      return;
+    }
+    if ((targetNode === null || targetNode === void 0 ? void 0 : targetNode.id) === connection.source) {
+      message.warning($i18n.get({
+        id: 'spark-flow.hooks.useNodesInteraction.cannotConnectToSelf',
+        dm: '不能与自身相连'
+      }));
+      return;
+    }
+    var hasCycle = function hasCycle(node) {
+      var visited = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : new Set();
+      if (visited.has(node.id)) return false;
+      visited.add(node.id);
+      var _iterator = _createForOfIteratorHelper(getOutgoers(node, nodes, edges)),
+        _step;
+      try {
+        for (_iterator.s(); !(_step = _iterator.n()).done;) {
+          var outgoer = _step.value;
+          if (outgoer.id === connection.source) return true;
+          if (hasCycle(outgoer, visited)) return true;
+        }
+      } catch (err) {
+        _iterator.e(err);
+      } finally {
+        _iterator.f();
+      }
+    };
+    if (targetNode && hasCycle(targetNode)) {
+      message.warning($i18n.get({
+        id: 'spark-flow.hooks.useNodesInteraction.cannotFormLoop',
+        dm: '不能形成环路'
+      }));
+      return;
+    }
+    var newEdge = generateEdge(connection, !!(sourceNode !== null && sourceNode !== void 0 && sourceNode.parentId));
+    setEdges([].concat(_toConsumableArray(edges), [newEdge]));
+    handleSaveFlowDraft();
+  }, [store, handleSaveFlowDraft]);
+  var generateNewNode = useCallback(function (_ref, position) {
+    var type = _ref.type,
+      parentId = _ref.parentId,
+      _ref$size = _ref.size,
+      size = _ref$size === void 0 ? 'normal' : _ref$size,
+      _ref$disableScreenToF = _ref.disableScreenToFlowPosition,
+      disableScreenToFlowPosition = _ref$disableScreenToF === void 0 ? false : _ref$disableScreenToF,
+      _ref$disableUniqueNam = _ref.disableUniqueName,
+      disableUniqueName = _ref$disableUniqueNam === void 0 ? false : _ref$disableUniqueNam;
+    var newPosition = _objectSpread({}, position);
+    var _store$getState2 = store.getState(),
+      nodes = _store$getState2.nodes;
+    if (!disableScreenToFlowPosition) {
+      newPosition = screenToFlowPosition(newPosition);
+      if (parentId) {
+        var parentNode = nodes.find(function (item) {
+          return item.id === parentId;
+        });
+        if (parentNode) {
+          newPosition.x = newPosition.x - parentNode.position.x;
+          newPosition.y = newPosition.y - parentNode.position.y;
+        }
+      }
+    }
+    var nodeInfo = nodeSchemaMap[type];
+    var boundsParams = type === 'Iterator' || type === 'Parallel' ? {
+      width: 1080,
+      height: 360
+    } : {
+      width: size === 'normal' ? 320 : 200
+    };
+    return _objectSpread(_objectSpread({
+      id: "".concat(type, "_").concat(uniqueId(4)),
+      parentId: parentId,
+      type: type,
+      position: newPosition,
+      zIndex: parentId ? 1002 : void 0
+    }, boundsParams), {}, {
+      data: _objectSpread({
+        label: disableUniqueName ? nodeInfo.title : generateUniqueName(nodeInfo.title, nodes.map(function (item) {
+          return item.data.label;
+        }))
+      }, nodeInfo.defaultParams || {})
+    });
+  }, [store, nodeSchemaMap]);
+  var handleNodeClick = useCallback(function (node) {
+    if (!nodeSchemaMap[node.type].notAllowConfig) {
+      setSelectedNode(node);
+      setTimeout(function () {
+        focusElement({
+          nodeId: node.id
+        });
+      }, 200);
+    }
+  }, [nodeSchemaMap, focusElement, setSelectedNode]);
+  var handleNodeClickByNodeId = useCallback(function (nodeId) {
+    var node = store.getState().nodes.find(function (item) {
+      return item.id === nodeId;
+    });
+    if (node) {
+      handleNodeClick(node);
+    }
+  }, [handleNodeClick, store]);
+  var addNewNodeCallback = useCallback(function (_ref2) {
+    var newNodes = _ref2.newNodes,
+      parentId = _ref2.parentId;
+    var _store$getState3 = store.getState(),
+      nodes = _store$getState3.nodes,
+      setNodes = _store$getState3.setNodes;
+    if (!parentId) {
+      setNodes([].concat(_toConsumableArray(nodes.map(function (item) {
+        return _objectSpread(_objectSpread({}, item), {}, {
+          selected: false
+        });
+      })), _toConsumableArray(newNodes.map(function (item, index) {
+        return _objectSpread(_objectSpread({}, item), {}, {
+          selected: index === 0
+        });
+      }))));
+    } else {
+      var subFlowNodes = nodes.filter(function (item) {
+        return item.parentId === parentId;
+      });
+      var _getIteratorNodeSize = getIteratorNodeSize([].concat(_toConsumableArray(subFlowNodes), _toConsumableArray(newNodes))),
+        width = _getIteratorNodeSize.width,
+        height = _getIteratorNodeSize.height;
+      setNodes([].concat(_toConsumableArray(nodes.map(function (item) {
+        if (item.id === parentId) {
+          return _objectSpread(_objectSpread({}, item), {}, {
+            width: item.width < width ? width : item.width,
+            height: item.height < height ? height : item.height,
+            selected: false
+          });
+        }
+        return _objectSpread(_objectSpread({}, item), {}, {
+          selected: false
+        });
+      })), _toConsumableArray(newNodes.map(function (item, index) {
+        return _objectSpread(_objectSpread({}, item), {}, {
+          selected: index === 0
+        });
+      }))));
+    }
+    setSelectedNode(newNodes[0]);
+    handleNodeClick(newNodes[0]);
+  }, [store, handleNodeClick]);
+  var onAddNewNode = useCallback( /*#__PURE__*/function () {
+    var _ref4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(_ref3, position) {
+      var type, parentId, parentNodeType, nodeInfo, newNodes, newNode;
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            type = _ref3.type, parentId = _ref3.parentId, parentNodeType = _ref3.parentNodeType;
+            nodeInfo = nodeSchemaMap[type];
+            newNodes = [];
+            newNode = generateNewNode({
+              type: type,
+              parentId: parentId
+            }, position);
+            if (!parentId) {
+              _context.next = 11;
+              break;
+            }
+            if (!(typeof nodeInfo.disableInGroup === 'boolean' && nodeInfo.disableInGroup)) {
+              _context.next = 8;
+              break;
+            }
+            message.warning($i18n.get({
+              id: 'spark-flow.hooks.useNodesInteraction.nodeCannotBeAddedToGroup',
+              dm: '该节点不能添加到子画布中'
+            }));
+            return _context.abrupt("return");
+          case 8:
+            if (!(parentNodeType && Array.isArray(nodeInfo.disableInGroup) && nodeInfo.disableInGroup.includes(parentNodeType))) {
+              _context.next = 11;
+              break;
+            }
+            message.warning($i18n.get({
+              id: 'spark-flow.hooks.useNodesInteraction.nodeCannotBeAddedToGroupWithNodeType',
+              dm: '该节点不能添加到该类型子画布中'
+            }));
+            return _context.abrupt("return");
+          case 11:
+            if (!nodeInfo.isGroup) {
+              _context.next = 17;
+              break;
+            }
+            if (!parentId) {
+              _context.next = 15;
+              break;
+            }
+            message.warning($i18n.get({
+              id: 'spark-flow.hooks.useNodesInteraction.subCanvasCannotNest',
+              dm: '子画布不能互相嵌套'
+            }));
+            return _context.abrupt("return");
+          case 15:
+            newNodes.push(generateNewNode({
+              type: "".concat(type, "Start"),
+              size: 'small',
+              disableUniqueName: true,
+              parentId: newNode.id,
+              disableScreenToFlowPosition: true
+            }, {
+              x: 50,
+              y: 152
+            }));
+            newNodes.push(generateNewNode({
+              type: "".concat(type, "End"),
+              size: 'small',
+              disableUniqueName: true,
+              parentId: newNode.id,
+              disableScreenToFlowPosition: true
+            }, {
+              x: 700,
+              y: 152
+            }));
+          case 17:
+            _context.prev = 17;
+            if (!nodeInfo.customAdd) {
+              _context.next = 22;
+              break;
+            }
+            _context.next = 21;
+            return onAddCustomNode(newNode);
+          case 21:
+            newNode = _context.sent;
+          case 22:
+            _context.next = 26;
+            break;
+          case 24:
+            _context.prev = 24;
+            _context.t0 = _context["catch"](17);
+          case 26:
+            if (newNode) {
+              _context.next = 28;
+              break;
+            }
+            return _context.abrupt("return");
+          case 28:
+            addNewNodeCallback({
+              newNodes: [newNode].concat(newNodes),
+              parentId: parentId
+            });
+            handleSaveFlowDraft();
+          case 30:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee, null, [[17, 24]]);
+    }));
+    return function (_x, _x2) {
+      return _ref4.apply(this, arguments);
+    };
+  }(), [store, handleSaveFlowDraft, nodeSchemaMap, onAddCustomNode, addNewNodeCallback]);
+  var onDrop = useCallback(function (event, parentId) {
+    setIsDragging(false);
+    event.preventDefault();
+    var type = event.dataTransfer.getData('application/reactflow');
+    if (typeof type === 'undefined' || !type) {
+      return;
+    }
+    onAddNewNode({
+      type: type,
+      parentId: parentId
+    }, {
+      x: event.clientX,
+      y: event.clientY
+    });
+  }, [onAddNewNode]);
+  var onDragOver = useCallback(function (event) {
+    event.preventDefault();
+    event.dataTransfer.dropEffect = 'move';
+  }, []);
+  var handleSelectNode = useCallback(function (nodeId) {
+    var _store$getState4 = store.getState(),
+      nodes = _store$getState4.nodes,
+      setNodes = _store$getState4.setNodes;
+    var newNodes = nodes.map(function (item) {
+      if (item.id === nodeId) return _objectSpread(_objectSpread({}, item), {}, {
+        selected: true
+      });
+      return item;
+    });
+    setNodes(newNodes);
+  }, [store]);
+  var onAddNewNodeWithSource = useCallback( /*#__PURE__*/function () {
+    var _ref6 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(_ref5, source, target) {
+      var type, parentId, _store$getState5, nodes, setNodes, sourceNode, nodeInfo, newNodes, newNode, _tempNodes, targetNode, x, y, _newNode, _store$getState6, edges, setEdges, newEdges, tempNodes, sourceHandle;
+      return _regeneratorRuntime().wrap(function _callee2$(_context2) {
+        while (1) switch (_context2.prev = _context2.next) {
+          case 0:
+            type = _ref5.type, parentId = _ref5.parentId;
+            _store$getState5 = store.getState(), nodes = _store$getState5.nodes, setNodes = _store$getState5.setNodes;
+            sourceNode = nodes.find(function (item) {
+              return item.id === source.id;
+            });
+            nodeInfo = nodeSchemaMap[type];
+            newNodes = [];
+            if (target) {
+              _context2.next = 24;
+              break;
+            }
+            newNode = generateNewNode({
+              type: type,
+              parentId: parentId,
+              disableScreenToFlowPosition: true
+            }, {
+              x: (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.position.x) + (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.width) + NEW_NODE_PADDING.x,
+              y: (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.position.y) || 0
+            });
+            if (nodeInfo.isGroup) {
+              /* Iteration start and batch body nodes need to add loop start/end nodes by default; */
+              newNodes.push(generateNewNode({
+                type: "".concat(type, "Start"),
+                size: 'small',
+                parentId: newNode.id,
+                disableScreenToFlowPosition: true,
+                disableUniqueName: true
+              }, {
+                x: 50,
+                y: 152
+              }));
+              newNodes.push(generateNewNode({
+                type: "".concat(type, "End"),
+                size: 'small',
+                parentId: newNode.id,
+                disableScreenToFlowPosition: true,
+                disableUniqueName: true
+              }, {
+                x: 700,
+                y: 152
+              }));
+            }
+            _context2.prev = 8;
+            if (!nodeInfo.customAdd) {
+              _context2.next = 13;
+              break;
+            }
+            _context2.next = 12;
+            return onAddCustomNode(newNode);
+          case 12:
+            newNode = _context2.sent;
+          case 13:
+            _context2.next = 17;
+            break;
+          case 15:
+            _context2.prev = 15;
+            _context2.t0 = _context2["catch"](8);
+          case 17:
+            if (newNode) {
+              _context2.next = 19;
+              break;
+            }
+            return _context2.abrupt("return");
+          case 19:
+            newNode.selected = true;
+            addNewNodeCallback({
+              newNodes: [newNode].concat(newNodes),
+              parentId: parentId
+            });
+            onConnect({
+              source: source.id,
+              sourceHandle: source.handleId,
+              target: newNode.id,
+              targetHandle: newNode.id
+            });
+            _context2.next = 54;
+            break;
+          case 24:
+            targetNode = nodes.find(function (item) {
+              return item.id === target.id;
+            });
+            if (targetNode) {
+              _context2.next = 27;
+              break;
+            }
+            return _context2.abrupt("return");
+          case 27:
+            x = (targetNode.position.x - (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.position.x)) / 2 + (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.position.x);
+            y = (targetNode.position.y - (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.position.y)) / 2 + (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.position.y);
+            if (x <= (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.position.x) + (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.width) + NEW_NODE_PADDING.x) {
+              x = (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.position.x) + (sourceNode === null || sourceNode === void 0 ? void 0 : sourceNode.width) + NEW_NODE_PADDING.x;
+            }
+            _newNode = generateNewNode({
+              type: type,
+              parentId: parentId,
+              disableScreenToFlowPosition: true
+            }, {
+              x: x,
+              y: y
+            });
+            _context2.prev = 31;
+            if (!nodeInfo.customAdd) {
+              _context2.next = 36;
+              break;
+            }
+            _context2.next = 35;
+            return onAddCustomNode(_newNode);
+          case 35:
+            _newNode = _context2.sent;
+          case 36:
+            _context2.next = 40;
+            break;
+          case 38:
+            _context2.prev = 38;
+            _context2.t1 = _context2["catch"](31);
+          case 40:
+            if (nodeInfo.isGroup) {
+              /* Iteration start and batch body nodes need to add loop start/end nodes by default; */
+              newNodes.push(generateNewNode({
+                type: "".concat(type, "Start"),
+                size: 'small',
+                parentId: _newNode.id,
+                disableScreenToFlowPosition: true,
+                disableUniqueName: true
+              }, {
+                x: 50,
+                y: 152
+              }));
+              newNodes.push(generateNewNode({
+                type: "".concat(type, "End"),
+                size: 'small',
+                parentId: _newNode.id,
+                disableScreenToFlowPosition: true,
+                disableUniqueName: true
+              }, {
+                x: 700,
+                y: 152
+              }));
+            }
+            _store$getState6 = store.getState(), edges = _store$getState6.edges, setEdges = _store$getState6.setEdges;
+            /* Remove the edge between source and target */
+            newEdges = edges.filter(function (item) {
+              return item.sourceHandle !== source.handleId || item.targetHandle !== target.handleId;
+            });
+            tempNodes = nodes.map(function (item) {
+              if (item.id === target.id) {
+                var newX = item.position.x;
+                /* Check if there are overlapping nodes */
+                if (_newNode.position.x + _newNode.width + NEW_NODE_PADDING.x > item.position.x) {
+                  newX = _newNode.position.x + _newNode.width + NEW_NODE_PADDING.x;
+                }
+                return _objectSpread(_objectSpread({}, item), {}, {
+                  selected: false,
+                  position: {
+                    x: newX,
+                    y: item.position.y
+                  }
+                });
+              }
+              return _objectSpread(_objectSpread({}, item), {}, {
+                selected: false
+              });
+            });
+            tempNodes = (_tempNodes = tempNodes).concat.apply(_tempNodes, [_newNode].concat(newNodes));
+            if (parentId) {
+              tempNodes = tempNodes.map(function (item) {
+                if (item.id === parentId) {
+                  var newBounds = getIteratorNodeSize(tempNodes.filter(function (vItem) {
+                    return vItem.parentId === parentId;
+                  }));
+                  return _objectSpread(_objectSpread({}, item), {}, {
+                    width: newBounds.width > item.width ? newBounds.width : item.width,
+                    height: newBounds.height > item.height ? newBounds.height : item.height
+                  });
+                }
+                return item;
+              });
+            }
+            sourceHandle = _newNode.id;
+            if (_newNode.type === 'Classifier') {
+              sourceHandle = "".concat(_newNode.id, "_").concat(_newNode.data.node_param.conditions[0].id);
+            }
+            if (_newNode.type === 'Judge') {
+              sourceHandle = "".concat(_newNode.id, "_").concat(_newNode.data.node_param.branches[0].id);
+            }
+            newEdges.push(generateEdge({
+              source: source.id,
+              sourceHandle: source.handleId,
+              target: _newNode.id,
+              targetHandle: _newNode.id
+            }, !!parentId));
+            if (sourceHandle) {
+              newEdges.push(generateEdge({
+                source: _newNode.id,
+                sourceHandle: sourceHandle,
+                target: target.id,
+                targetHandle: target.handleId
+              }, !!parentId));
+            }
+            setNodes(tempNodes);
+            setEdges(_toConsumableArray(newEdges));
+            handleNodeClick(_newNode);
+          case 54:
+            handleSaveFlowDraft();
+          case 55:
+          case "end":
+            return _context2.stop();
+        }
+      }, _callee2, null, [[8, 15], [31, 38]]);
+    }));
+    return function (_x3, _x4, _x5) {
+      return _ref6.apply(this, arguments);
+    };
+  }(), [store, getNode, handleSaveFlowDraft]);
+  var onNodeDrag = useCallback(function (e, _, dragNodes) {
+    e.stopPropagation();
+    var _store$getState7 = store.getState(),
+      nodes = _store$getState7.nodes,
+      setNodes = _store$getState7.setNodes;
+    var needFreshParentNodeIdsMap = {};
+    var newNodes = nodes.map(function (item) {
+      var targetNode = dragNodes.find(function (vItem) {
+        return vItem.id === item.id;
+      });
+      if (!targetNode) return item;
+      if (targetNode.parentId && !needFreshParentNodeIdsMap[targetNode.parentId]) {
+        needFreshParentNodeIdsMap[targetNode.parentId] = true;
+      }
+      return _objectSpread(_objectSpread({}, item), {}, {
+        position: item.parentId ? {
+          x: targetNode.position.x <= ITERATION_PADDING.left ? ITERATION_PADDING.left : targetNode.position.x,
+          y: targetNode.position.y <= ITERATION_PADDING.top ? ITERATION_PADDING.top : targetNode.position.y
+        } : targetNode.position
+      });
+    });
+    newNodes = newNodes.map(function (item) {
+      if (needFreshParentNodeIdsMap[item.id]) {
+        var subFlowNodes = newNodes.filter(function (vItem) {
+          return vItem.parentId === item.id;
+        });
+        var _getIteratorNodeSize2 = getIteratorNodeSize(subFlowNodes),
+          width = _getIteratorNodeSize2.width,
+          height = _getIteratorNodeSize2.height;
+        return _objectSpread(_objectSpread({}, item), {}, {
+          width: width,
+          height: height
+        });
+      }
+      return item;
+    });
+    setNodes(newNodes);
+    handleSaveFlowDraft();
+  }, [store, handleSaveFlowDraft]);
+  var onNodeClick = useCallback(function (_, node) {
+    handleNodeClick(node);
+  }, [handleNodeClick]);
+  var onNodeResize = useCallback(function (resizeBound, nodeId) {
+    var _store$getState8 = store.getState(),
+      nodes = _store$getState8.nodes,
+      setNodes = _store$getState8.setNodes;
+    var minBoundSize = getIteratorNodeSize(nodes.filter(function (item) {
+      return item.parentId === nodeId;
+    }));
+    var newWidth = resizeBound.width < minBoundSize.width ? minBoundSize.width : resizeBound.width;
+    var newHeight = resizeBound.height < minBoundSize.height ? minBoundSize.height : resizeBound.height;
+    var newNodes = nodes.map(function (item) {
+      if (item.id === nodeId) return _objectSpread(_objectSpread({}, item), {}, {
+        width: newWidth,
+        height: newHeight,
+        position: {
+          x: resizeBound.x,
+          y: resizeBound.y
+        }
+      });
+      return item;
+    });
+    setNodes(newNodes);
+  }, [store]);
+  var onNodesChange = useCallback(function (changes) {
+    if (!['remove', 'select', 'dimensions'].includes(changes[0].type)) return;
+    var _store$getState9 = store.getState(),
+      nodes = _store$getState9.nodes,
+      setNodes = _store$getState9.setNodes;
+    if (changes[0].type === 'dimensions') {
+      /* Override the dimensions update method; */
+      return setNodes(nodes.map(function (item) {
+        /* @ts-ignore */
+        var targetChange = changes.find(function (vItem) {
+          return vItem.id === item.id;
+        });
+        /* @ts-ignore */
+        if (!targetChange || !(targetChange !== null && targetChange !== void 0 && targetChange.dimensions)) return item;
+        return _objectSpread(_objectSpread({}, item), {}, {
+          measured: {
+            /* @ts-ignore */
+            width: targetChange.dimensions.width,
+            /* @ts-ignore */
+            height: targetChange.dimensions.height
+          }
+        });
+      }));
+    }
+    var newNodes = applyNodeChanges(changes, nodes);
+    setNodes(newNodes);
+    if (changes[0].type === 'remove') handleSaveFlowDraft();
+  }, [store, handleSaveFlowDraft, applyNodeChanges]);
+  var onNodeDelete = useCallback(function (nodeId) {
+    var _store$getState10 = store.getState(),
+      nodes = _store$getState10.nodes,
+      setNodes = _store$getState10.setNodes,
+      setEdges = _store$getState10.setEdges,
+      edges = _store$getState10.edges;
+    var newNodes = nodes.filter(function (item) {
+      return item.id !== nodeId && item.parentId !== nodeId;
+    });
+    /* If the selected node is deleted, deselect it */
+    if ((selectedNode === null || selectedNode === void 0 ? void 0 : selectedNode.id) === nodeId) setSelectedNode(null);
+    setNodes(newNodes);
+    /* Remove disconnected edges */
+    setEdges(edges.filter(function (item) {
+      return item.source !== nodeId && item.target !== nodeId;
+    }));
+    handleSaveFlowDraft();
+  }, [store, selectedNode, handleSaveFlowDraft]);
+  var onNodeCopy = useCallback(function (nodeId) {
+    var nodes = store.getState().nodes;
+    var _store$getState11 = store.getState(),
+      setNodes = _store$getState11.setNodes,
+      edges = _store$getState11.edges,
+      setEdges = _store$getState11.setEdges;
+    var node = nodes.find(function (item) {
+      return item.id === nodeId;
+    });
+    if (node) {
+      var newNodes = [];
+      var newEdges = [];
+      /* Generate a new node */
+      var newNode = copyNodeConfig(node, nodes);
+      newNodes.push(newNode);
+      if (nodeSchemaMap[node.type].isGroup) {
+        /* When copying group nodes, sub-canvas information needs to be copied in bulk */
+        var subNodes = nodes.filter(function (item) {
+          return item.parentId === nodeId;
+        });
+        var newSubNodesMap = {};
+        var subEdges = edges.filter(function (item) {
+          return subNodes.some(function (vItem) {
+            return [item.source, item.target].includes(vItem.id);
+          });
+        });
+        subNodes.forEach(function (item) {
+          var newSubNode = copySubFlowNodeConfig(_objectSpread(_objectSpread({}, item), {}, {
+            parentId: newNode.id
+          }), [].concat(_toConsumableArray(nodes), _toConsumableArray(subNodes)));
+          newSubNodesMap[item.id] = newSubNode;
+          newNodes.push(newSubNode);
+        });
+        subEdges.forEach(function (item) {
+          var source = item.source,
+            target = item.target,
+            sourceHandle = item.sourceHandle;
+          var newSourceNode = newSubNodesMap[source];
+          var newTargetNode = newSubNodesMap[target];
+          var newEdge = generateEdge({
+            source: newSourceNode.id,
+            sourceHandle: (sourceHandle || '').replace(source, newSourceNode.id),
+            target: newTargetNode.id,
+            targetHandle: newTargetNode.id
+          }, true);
+          newEdges.push(newEdge);
+        });
+      }
+      setNodes([].concat(_toConsumableArray(nodes.map(function (item) {
+        return _objectSpread(_objectSpread({}, item), {}, {
+          selected: false
+        });
+      })), newNodes));
+      if (newEdges.length) setEdges([].concat(_toConsumableArray(edges), newEdges));
+      handleNodeClick(newNode);
+      handleSaveFlowDraft();
+      message.success($i18n.get({
+        id: 'spark-flow.hooks.useNodesInteraction.copySuccess',
+        dm: '复制成功'
+      }));
+    }
+  }, [store, nodeSchemaMap, handleSaveFlowDraft]);
+  var updateParentNodeSize = useCallback(function (nodeId) {
+    var _store$getState12 = store.getState(),
+      nodes = _store$getState12.nodes,
+      setNodes = _store$getState12.setNodes;
+    setNodes(nodes.map(function (item) {
+      if (item.id === nodeId) {
+        var parentNodeBounds = getIteratorNodeSize(nodes.filter(function (vItem) {
+          return vItem.parentId === item.id;
+        }));
+        return _objectSpread(_objectSpread({}, item), {}, {
+          width: parentNodeBounds.width,
+          height: parentNodeBounds.height
+        });
+      }
+      return item;
+    }));
+  }, [store]);
+  return {
+    onConnect: onConnect,
+    onAddNewNode: onAddNewNode,
+    onNodeDrag: onNodeDrag,
+    onAddNewNodeWithSource: onAddNewNodeWithSource,
+    onNodesChange: onNodesChange,
+    onNodeClick: onNodeClick,
+    handleNodeClick: handleNodeClick,
+    handleSelectNode: handleSelectNode,
+    onNodeResize: onNodeResize,
+    onNodeDelete: onNodeDelete,
+    onNodeCopy: onNodeCopy,
+    handleNodeClickByNodeId: handleNodeClickByNodeId,
+    updateParentNodeSize: updateParentNodeSize,
+    onDrop: onDrop,
+    onDragOver: onDragOver
+  };
+};