Set known slot in/out types

src/lib/ComfyNodeDef.ts

@@ -17,7 +17,7 @@ export type ComfyNodeDefInputs = {
     optional?: Record<string, ComfyNodeDefInput>
 }
 export type ComfyNodeDefInput = [ComfyNodeDefInputType, ComfyNodeDefInputOptions | null]
-export type ComfyNodeDefInputType = string[] | keyof typeof ComfyWidgets | string
+export type ComfyNodeDefInputType = any[] | keyof typeof ComfyWidgets | string
 export type ComfyNodeDefInputOptions = {
     forceInput?: boolean
 }

src/lib/api.ts

@@ -253,7 +253,7 @@ export default class ComfyAPI {
             postBody = JSON.stringify(body)
         }
         catch (error) {
-            return Promise.reject({ error })
+            return Promise.reject({ error: error.toString() })
         }
 
         return fetch(this.getBackendUrl() + "/prompt", {
@@ -270,7 +270,7 @@ export default class ComfyAPI {
                 return res.json()
             })
             .then(raw => { return { promptID: raw.prompt_id } })
-            .catch(error => { return { error } })
+            .catch(error => { return error })
     }
 
     /**

src/lib/components/ComfyApp.ts

@@ -121,9 +121,6 @@ export default class ComfyApp {
         this.lCanvas = new ComfyGraphCanvas(this, this.canvasEl);
         this.canvasCtx = this.canvasEl.getContext("2d");
 
-        LiteGraph.release_link_on_empty_shows_menu = true;
-        LiteGraph.alt_drag_do_clone_nodes = true;
-
         const uiUnlocked = get(uiState).uiUnlocked;
         this.lCanvas.allow_dragnodes = uiUnlocked;
         this.lCanvas.allow_interaction = uiUnlocked;
@@ -242,13 +239,50 @@ export default class ComfyApp {
 
         for (const [inputName, [inputType, _inputOpts]] of iterateNodeDefInputs(nodeDef)) {
             if (isBackendNodeDefInputType(inputName, inputType)) {
-                LiteGraph.slot_types_default_out[inputType] ||= []
+                LiteGraph.slot_types_default_out[inputType] ||= ["utils/reroute"]
                 LiteGraph.slot_types_default_out[inputType].push(nodeTypeSpec)
+
+                // Input types have to be stored as lower case
+                // Store each node that can handle this input type
+                const lowerType = inputType.toLocaleLowerCase();
+                if (!(lowerType in LiteGraph.registered_slot_in_types)) {
+                    LiteGraph.registered_slot_in_types[lowerType] = { nodes: [] };
+                }
+                LiteGraph.registered_slot_in_types[lowerType].nodes.push(nodeId);
+
+                if (!LiteGraph.slot_types_in.includes(lowerType)) {
+                    LiteGraph.slot_types_in.push(lowerType);
+                }
+            }
+            else {
+                // inputType is an array of combo box entries (["euler", "karras", ...])
+                // or a widget input type ("INT").
             }
         }
+
         for (const output of iterateNodeDefOutputs(nodeDef)) {
-            LiteGraph.slot_types_default_in[output.type] ||= []
+            LiteGraph.slot_types_default_in[output.type] ||= ["utils/reroute"]
             LiteGraph.slot_types_default_in[output.type].push(nodeTypeSpec)
+
+            // Store each node that can handle this output type
+            if (!(output.type in LiteGraph.registered_slot_out_types)) {
+                LiteGraph.registered_slot_out_types[output.type] = { nodes: [] };
+            }
+            LiteGraph.registered_slot_out_types[output.type].nodes.push(nodeId);
+
+            if (!LiteGraph.slot_types_out.includes(output.type)) {
+                LiteGraph.slot_types_out.push(output.type);
+            }
+        }
+
+        const maxNodeSuggestions = 5 // TODO config
+
+        // TODO config beforeChanged
+        for (const key of Object.keys(LiteGraph.slot_types_default_in)) {
+            LiteGraph.slot_types_default_in[key] = LiteGraph.slot_types_default_in[key].slice(0, maxNodeSuggestions)
+        }
+        for (const key of Object.keys(LiteGraph.slot_types_default_out)) {
+            LiteGraph.slot_types_default_out[key] = LiteGraph.slot_types_default_out[key].slice(0, maxNodeSuggestions)
         }
     }
 

src/lib/init.ts

@@ -5,12 +5,23 @@ import { get } from 'svelte/store';
 
 export function configureLitegraph(isMobile: boolean = false) {
     LiteGraph.catch_exceptions = false;
+
+    // Must be enabled, otherwise subgraphs won't work (because of non-unique node/link IDs)
     LiteGraph.use_uuids = true;
+
+
+    LiteGraph.search_filter_enabled = true;
+    LiteGraph.release_link_on_empty_shows_menu = true;
+    LiteGraph.alt_drag_do_clone_nodes = true;
+    LiteGraph.middle_click_slot_add_default_node = true;
+    LiteGraph.dialog_close_on_mouse_leave = false;
+    LiteGraph.search_hide_on_mouse_leave = false;
+    LiteGraph.graph_inputs_outputs_use_combo_widget = true;
+    LiteGraph.search_box_refresh_interval_ms = 150;
+
     LiteGraph.CANVAS_GRID_SIZE = 32;
 
     if (isMobile) {
-        LiteGraph.dialog_close_on_mouse_leave = false;
-        LiteGraph.search_hide_on_mouse_leave = false;
         LiteGraph.pointerevents_method = "pointer";
     }
 

src/lib/stores/layoutState.ts

@@ -771,7 +771,7 @@ function addContainer(parent: ContainerLayout | null, attrs: Partial<Attributes>
 
     console.debug("[layoutState] addContainer", state)
     store.set(state)
-    runOnChangedForWidgetDefaults(dragItem)
+    // runOnChangedForWidgetDefaults(dragItem)
     return dragItem;
 }
 
@@ -786,7 +786,6 @@ function addWidget(parent: ContainerLayout, node: ComfyWidgetNode, attrs: Partia
         attrs: {
             ...defaultWidgetAttributes,
             title: widgetName,
-            nodeDisabledState: "disabled",
             ...attrs
         }
     }
@@ -803,7 +802,7 @@ function addWidget(parent: ContainerLayout, node: ComfyWidgetNode, attrs: Partia
 
     console.debug("[layoutState] addWidget", state)
     moveItem(dragItem, parent, index)
-    runOnChangedForWidgetDefaults(dragItem)
+    // runOnChangedForWidgetDefaults(dragItem)
     return dragItem;
 }