@flowgram.ai/document 0.2.2 → 0.2.3

package/dist/esm/index.js

@@ -176,13 +176,37 @@ var DEFAULT_FLOW_NODE_META = (nodeType, document) => {
 };
 var FlowNodeRegistry;
 ((FlowNodeRegistry4) => {
+  function mergeChildRegistries(r1 = [], r2 = []) {
+    if (r1.length === 0 || r2.length === 0) {
+      return [...r1, ...r2];
+    }
+    const r1Filter = r1.map((r1Current) => {
+      const r2Current = r2.find((n) => n.type === r1Current.type);
+      if (r2Current) {
+        return merge(r1Current, r2Current, r1Current.type);
+      }
+      return r1Current;
+    });
+    const r2Filter = r2.filter((n) => !r1.some((r) => r.type === n.type));
+    return [...r1Filter, ...r2Filter];
+  }
+  FlowNodeRegistry4.mergeChildRegistries = mergeChildRegistries;
   function merge(registry1, registry2, finalType) {
+    const extendKeys = registry1.__extends__ ? registry1.__extends__.slice() : [];
+    if (registry1.type !== registry2.type) {
+      extendKeys.unshift(registry1.type);
+    }
     return {
       ...registry1,
       ...registry2,
+      extendChildRegistries: mergeChildRegistries(
+        registry1.extendChildRegistries,
+        registry2.extendChildRegistries
+      ),
       meta: { ...registry1.meta, ...registry2.meta },
       extend: void 0,
-      type: finalType
+      type: finalType,
+      __extends__: extendKeys
     };
   }
   FlowNodeRegistry4.merge = merge;
@@ -1768,7 +1792,7 @@ var FlowDocument = class {
    * @param data
    * @param addedNodes
    */
-  addNode(data, addedNodes, ignoreCreateAndUpdateEvent) {
+  addNode(data, addedNodes, ignoreCreateAndUpdateEvent, ignoreBlocks) {
     const { id, type = "block", originParent, parent, meta, hidden, index } = data;
     let node = this.getNode(id);
     let isNew = false;
@@ -1810,7 +1834,7 @@ var FlowDocument = class {
       if (extendNodes && addedNodes) {
         addedNodes.push(...extendNodes);
       }
-    } else if (data.blocks && data.blocks.length > 0) {
+    } else if (data.blocks && data.blocks.length > 0 && !ignoreBlocks) {
       if (!data.blocks[0].type) {
         this.addInlineBlocks(node, data.blocks, addedNodes);
       } else {
@@ -1936,10 +1960,22 @@ var FlowDocument = class {
         meta: {
           ...preRegistry?.meta,
           ...newRegistry?.meta
-        }
+        },
+        extendChildRegistries: FlowNodeRegistry.mergeChildRegistries(
+          preRegistry?.extendChildRegistries,
+          newRegistry?.extendChildRegistries
+        )
       });
     });
   }
+  /**
+   * Check node extend
+   * @param currentType
+   * @param parentType
+   */
+  isExtend(currentType, parentType) {
+    return (this.getNodeRegistry(currentType).__extends__ || []).includes(parentType);
+  }
   /**
    * 导出数据，可以重载
    */