y-mxgraph 0.1.6 → 0.2.1

package/y-mxgraph.es.js

@@ -1436,32 +1436,148 @@ function bindCollaborator(file, options) {
     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;
@@ -1484,13 +1600,36 @@ class Binding {
     };
     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;
@@ -1502,7 +1641,7 @@ class Binding {
         this.suppressLocalApply = false;
       }
     };
-    mxfileMap.observeDeep(this.docObserver);
+    doc.getMap(key).observeDeep(this.docObserver);
     if (awareness) {
       this.cleanupCollaborator = bindCollaborator(file, {
         awareness,
@@ -1515,6 +1654,10 @@ class Binding {
       this.cleanupUndoManager = bindUndoManager(doc, file, undoManager);
     }
   }
+  /** replace 策略下，构造时 doc 为空，现在 doc 有数据时需要强制替换 */
+  get shouldReplaceWhenDocHasData() {
+    return this.initialContentStrategy === "replace" && !this.docInitialized;
+  }
   /**
    * 销毁绑定，解除所有监听器
    * @param deep - 是否深度清理（包括 awareness/undoManager），默认 false