Merge pull request #12 from shandowc/fix-undo

fix: undo doesnt work
marc0l92 · Jul 19, 2022 · 32ac8d3 · 32ac8d3
2 parents a1603c5 + fb3e691
commit 32ac8d3
Showing 1 changed file with 24 additions and 2 deletions.
diff --git a/src/dialog/DiagramEditor.js b/src/dialog/DiagramEditor.js
@@ -29,6 +29,17 @@ function DiagramEditor(config, ui, done, initialized, urlParams) {
             catch (e) {
                 console.error(e);
             }
+        } else if (self.exportFrame != null && evt.source == self.exportFrame.contentWindow &&
+            evt.data.length > 0) {
+            try {
+                var msg = JSON.parse(evt.data);
+                if (msg != null && msg.event == 'export') {
+                    self.setElementData(self.startElement, msg.data);
+                }
+            }
+            catch (e) {
+                console.error(e);
+            }
         }
     };
 };
@@ -142,6 +153,12 @@ DiagramEditor.prototype.startEditing = function (data, format, title) {
             this.getFrameUrl(),
             this.getFrameStyle());
         document.body.appendChild(this.frame);
+
+        this.exportFrame = this.createFrame(
+            this.getFrameUrl(),
+            'position: absolute; width:0; height:0; border:0;');
+        document.body.appendChild(this.exportFrame);
+
         this.setWaiting(true);
     }
 };
@@ -203,7 +220,11 @@ DiagramEditor.prototype.stopEditing = function () {
  */
 DiagramEditor.prototype.postMessage = function (msg) {
     if (this.frame != null) {
-        this.frame.contentWindow.postMessage(JSON.stringify(msg), '*');
+        if (msg && msg.action == 'export') {
+            this.exportFrame.contentWindow.postMessage(JSON.stringify(msg), '*');
+        } else {
+            this.frame.contentWindow.postMessage(JSON.stringify(msg), '*');
+        } 
     }
 };
 
@@ -292,7 +313,8 @@ DiagramEditor.prototype.handleMessage = function (msg) {
         this.save(msg.xml, true, this.startElement);
     }
     else if (msg.event == 'export') {
-        this.setElementData(this.startElement, msg.data);
+        console.error('export in none export frame')
+        // this.setElementData(this.startElement, msg.data);
         // this.stopEditing();
         // this.xml = null;
     }