y-mxgraph 0.1.6 → 0.2.1

package/y-mxgraph.umd.js

@@ -1453,32 +1453,148 @@
       cleanupAwareness == null ? void 0 : cleanupAwareness();
     };
   }
+  const defaultApplyFileData = (file, xml) => {
+    file.ui.setFileData(xml);
+  };
+  function mergeFileIntoDoc(doc, fileXml, strategy) {
+    let parsed;
+    try {
+      parsed = parse$4(fileXml);
+    } catch (err) {
+      console.warn(
+        "[y-mxgraph] 合并失败，file XML 解析异常，回退到 replace：",
+        err
+      );
+      return false;
+    }
+    const mxfileObj = parsed == null ? void 0 : parsed.mxfile;
+    if (!mxfileObj || !Array.isArray(mxfileObj.diagram)) {
+      console.warn(
+        "[y-mxgraph] 合并失败，file XML 不是合法 mxfile，回退到 replace"
+      );
+      return false;
+    }
+    const mxfileMap = doc.getMap(key);
+    const diagramMap = mxfileMap.get(key$1);
+    const diagramOrder = mxfileMap.get(diagramOrderKey);
+    if (!diagramMap || !diagramOrder) {
+      console.warn("[y-mxgraph] 合并失败，doc 结构不完整，回退到 replace");
+      return false;
+    }
+    doc.transact(() => {
+      var _a;
+      for (const diagram of mxfileObj.diagram) {
+        const id = ((_a = diagram._attributes) == null ? void 0 : _a.id) || "";
+        if (!id)
+          continue;
+        const docHas = diagramMap.has(id);
+        if (docHas && strategy === "merge-remote") {
+          continue;
+        }
+        const yDiagram = parse$1(
+          diagram
+        );
+        diagramMap.set(id, yDiagram);
+        if (!docHas) {
+          diagramOrder.push([id]);
+        }
+      }
+    });
+    return true;
+  }
+  function reconcileInitialContent(doc, file, strategy, applyFileData) {
+    const mxfileMap = doc.getMap(key);
+    const docHasData = mxfileMap.size > 0;
+    const fileHasAnyData = !!file.data;
+    const fileHasDiagrams = fileHasAnyData && file.data.includes("<diagram");
+    if (strategy === "replace") {
+      if (docHasData) {
+        const xml2 = doc2xml(doc);
+        if (xml2 && xml2.includes("<diagram")) {
+          applyFileData(file, xml2);
+        } else if (!fileHasAnyData) {
+          applyFileData(file, Binding.generateFileTemplate("diagram-0"));
+        }
+      } else if (!fileHasAnyData) {
+        applyFileData(file, Binding.generateFileTemplate("diagram-0"));
+      }
+      return mxfileMap.size > 0;
+    }
+    if (!docHasData && !fileHasDiagrams) {
+      if (!fileHasAnyData) {
+        applyFileData(file, Binding.generateFileTemplate("diagram-0"));
+      }
+      return false;
+    }
+    if (!docHasData && fileHasDiagrams) {
+      try {
+        doc.transact(() => {
+          xml2doc(file.data, doc);
+        });
+        return true;
+      } catch (err) {
+        console.warn(
+          "[y-mxgraph] merge 模式下 xml2doc 失败，回退 replace：",
+          err
+        );
+        applyFileData(file, Binding.generateFileTemplate("diagram-0"));
+        return false;
+      }
+    }
+    if (docHasData && !fileHasDiagrams) {
+      const xml2 = doc2xml(doc);
+      if (xml2 && xml2.includes("<diagram")) {
+        applyFileData(file, xml2);
+      } else if (!fileHasAnyData) {
+        applyFileData(file, Binding.generateFileTemplate("diagram-0"));
+      }
+      return mxfileMap.size > 0;
+    }
+    const ok = mergeFileIntoDoc(doc, file.data, strategy);
+    if (!ok) {
+      const xml2 = doc2xml(doc);
+      if (xml2 && xml2.includes("<diagram"))
+        applyFileData(file, xml2);
+      return mxfileMap.size > 0;
+    }
+    const xml = doc2xml(doc);
+    if (xml && xml.includes("<diagram"))
+      applyFileData(file, xml);
+    return true;
+  }
   class Binding {
     constructor(file, options) {
       this.suppressLocalApply = false;
       this.docInitialized = false;
-      const { doc, awareness, undoManager, mouseMoveThrottle, cursor } = options;
+      const {
+        doc,
+        awareness,
+        undoManager,
+        mouseMoveThrottle,
+        cursor,
+        initialContent = "replace",
+        applyFileData = defaultApplyFileData
+      } = options;
       this.doc = doc;
+      this.initialContentStrategy = initialContent;
       const ui = file.getUi();
       const graph = ui.editor.graph;
       this.mxGraphModel = graph.model;
-      const mxfileMap = doc.getMap(key);
-      const docHasData = mxfileMap.size > 0;
-      this.docInitialized = docHasData;
-      file.setShadowPages(file.ui.clonePages(file.ui.pages));
-      if (docHasData) {
-        initDocSnapshot(doc, false);
-        const fullPatch = generatePatch([], doc);
-        if (Object.keys(fullPatch).length > 0) {
-          this.suppressLocalApply = true;
-          try {
-            file.patch([fullPatch]);
-          } finally {
-            this.suppressLocalApply = false;
-          }
+      this.suppressLocalApply = true;
+      try {
+        this.docInitialized = reconcileInitialContent(
+          doc,
+          file,
+          initialContent,
+          applyFileData
+        );
+        if (this.docInitialized) {
+          initDocSnapshot(doc, false);
         }
-        file.setShadowPages(file.ui.clonePages(file.ui.pages));
+      } finally {
+        this.suppressLocalApply = false;
       }
+      file.setShadowPages(file.ui.clonePages(file.ui.pages));
       this.mxListener = () => {
         if (this.suppressLocalApply)
           return;
@@ -1501,13 +1617,36 @@
       };
       this.mxGraphModel.addListener("change", this.mxListener);
       this.docObserver = (events, transaction) => {
-        if (!this.docInitialized) {
-          this.docInitialized = true;
-        }
         if (transaction.local && transaction.origin === LOCAL_ORIGIN) {
           generatePatch(events);
           return;
         }
+        if (this.shouldReplaceWhenDocHasData) {
+          const mxfileMap = doc.getMap(key);
+          const diagramMap = mxfileMap.get(key$1);
+          if (diagramMap && diagramMap.size > 0) {
+            const xml = doc2xml(doc);
+            if (xml && xml.includes("<diagram")) {
+              this.suppressLocalApply = true;
+              try {
+                applyFileData(file, xml);
+                file.setShadowPages(file.ui.clonePages(file.ui.pages));
+                initDocSnapshot(doc, false);
+                const ui2 = file.getUi();
+                const editor = ui2.editor;
+                editor.setStatus("");
+                editor.setModified(false);
+              } finally {
+                this.suppressLocalApply = false;
+              }
+              this.docInitialized = true;
+              return;
+            }
+          }
+        }
+        if (!this.docInitialized) {
+          this.docInitialized = true;
+        }
         const patch = generatePatch(events);
         if (Object.keys(patch).length === 0)
           return;
@@ -1519,7 +1658,7 @@
           this.suppressLocalApply = false;
         }
       };
-      mxfileMap.observeDeep(this.docObserver);
+      doc.getMap(key).observeDeep(this.docObserver);
       if (awareness) {
         this.cleanupCollaborator = bindCollaborator(file, {
           awareness,
@@ -1532,6 +1671,10 @@
         this.cleanupUndoManager = bindUndoManager(doc, file, undoManager);
       }
     }
+    /** replace 策略下，构造时 doc 为空，现在 doc 有数据时需要强制替换 */
+    get shouldReplaceWhenDocHasData() {
+      return this.initialContentStrategy === "replace" && !this.docInitialized;
+    }
     /**
      * 销毁绑定，解除所有监听器
      * @param deep - 是否深度清理（包括 awareness/undoManager），默认 false