Global modal system

src/lib/stores/workflowState.ts

@@ -1,5 +1,5 @@
 import type { SerializedGraphCanvasState } from '$lib/ComfyGraphCanvas';
-import { clamp, LGraphNode, type LGraphCanvas, type NodeID, type SerializedLGraph, type UUID, LGraph } from '@litegraph-ts/core';
+import { clamp, LGraphNode, type LGraphCanvas, type NodeID, type SerializedLGraph, type UUID, LGraph, LiteGraph } from '@litegraph-ts/core';
 import { get, writable } from 'svelte/store';
 import type { Readable, Writable } from 'svelte/store';
 import { defaultWorkflowAttributes, type SerializedLayoutState, type WritableLayoutStateStore } from './layoutStates';
@@ -76,6 +76,11 @@ export class ComfyWorkflow {
      */
     isModified: boolean = false;
 
+    /*
+     * Missing node types encountered when deserializing the graph
+     */
+    missingNodeTypes: string[];
+
     get layout(): WritableLayoutStateStore | null {
         return layoutStates.getLayout(this.id)
     }
@@ -170,6 +175,18 @@ export class ComfyWorkflow {
     }
 
     deserialize(layoutState: WritableLayoutStateStore, data: SerializedWorkflowState) {
+        this.missingNodeTypes = []
+
+        for (let n of data.graph.nodes) {
+            // Patch T2IAdapterLoader to ControlNetLoader since they are the same node now
+            if (n.type == "T2IAdapterLoader") n.type = "ControlNetLoader";
+
+            // Find missing node types
+            if (!(n.type in LiteGraph.registered_node_types)) {
+                this.missingNodeTypes.push(n.type);
+            }
+        }
+
         // Ensure loadGraphData does not trigger any state changes in layoutState
         // (isConfiguring is set to true here)
         // lGraph.configure will add new nodes, triggering onNodeAdded, but we