🐛 fix dom widget mount timing

lenML · May 22, 2024 · 534b890 · 534b890
1 parent d19a081
commit 534b890
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 11 deletions.
diff --git a/packages/core/package.json b/packages/core/package.json
@@ -1,6 +1,6 @@
 {
     "name": "@litegraph-ts/core",
-    "version": "0.2.6",
+    "version": "0.2.7",
     "description": "A graph node editor similar to PD or UDK Blueprints. It works in an HTML5 Canvas and allows to export graphs to be included in applications.",
     "source": "src/index.ts",
     "types": "src/index.ts",

diff --git a/packages/core/src/widgets/DOMWidget.ts b/packages/core/src/widgets/DOMWidget.ts
@@ -1,4 +1,5 @@
 import type { default as IWidget, WidgetCallback } from "../IWidget";
+import LGraph from "../LGraph";
 import LGraphNode, { SerializedLGraphNode } from "../LGraphNode";
 import LiteGraph from "../LiteGraph";
 import { NodeMode, Vector2 } from "../types";
@@ -275,17 +276,24 @@ export class DOMWidget implements IWidget {
             });
         }
 
+        const mountParentElement = (graph: LGraph) => {
+            const parentElement =
+                this.options.parentElement ||
+                graph.list_of_graphcanvas[0].canvas.offsetParent ||
+                document.body;
+            parentElement.appendChild(element);
+        };
+
         if (!element.parentElement) {
-            const onAdded = this.node.onAdded;
-            this.node.onAdded = (graph) => {
-                onAdded?.(graph);
-
-                const parentElement =
-                    this.options.parentElement ||
-                    graph.list_of_graphcanvas[0].canvas.offsetParent ||
-                    document.body;
-                parentElement.appendChild(element);
-            };
+            if (node.graph) {
+                mountParentElement(node.graph);
+            } else {
+                const onAdded = this.node.onAdded;
+                this.node.onAdded = (graph) => {
+                    onAdded?.(graph);
+                    mountParentElement(graph);
+                };
+            }
         }
     }