Refresh combo boxes button

src/lib/nodes/ComfyActionNodes.ts

@@ -91,3 +91,37 @@ LiteGraph.registerNodeType({
     desc: "Copies its input to its output when an event is received",
     type: "actions/copy"
 })
+
+export interface ComfySwapActionProperties extends Record<any, any> {
+}
+
+export class ComfySwapAction extends ComfyGraphNode {
+    override properties: ComfySwapActionProperties = {
+    }
+
+    static slotLayout: SlotLayout = {
+        inputs: [
+            { name: "A", type: "*" },
+            { name: "B", type: "*" },
+            { name: "swap", type: BuiltInSlotType.ACTION }
+        ],
+        outputs: [
+            { name: "B", type: "*" },
+            { name: "A", type: "*" }
+        ],
+    }
+
+    override onAction(action: any, param: any) {
+        const a = this.getInputData(0)
+        const b = this.getInputData(1)
+        this.setOutputData(0, a)
+        this.setOutputData(1, b)
+    };
+}
+
+LiteGraph.registerNodeType({
+    class: ComfySwapAction,
+    title: "Comfy.SwapAction",
+    desc: "Swaps two inputs when triggered",
+    type: "actions/swap"
+})

src/lib/nodes/ComfyGraphNode.ts

@@ -1,9 +1,10 @@
 import type { ComfyInputConfig } from "$lib/IComfyInputSlot";
 import type { SerializedPrompt } from "$lib/components/ComfyApp";
 import type ComfyWidget from "$lib/components/widgets/ComfyWidget";
-import { LGraph, LGraphNode } from "@litegraph-ts/core";
+import { LGraph, LGraphNode, LiteGraph, type SerializedLGraphNode } from "@litegraph-ts/core";
 import type { SvelteComponentDev } from "svelte/internal";
 import type { ComfyWidgetNode } from "./ComfyWidgetNodes";
+import type IComfyInputSlot from "$lib/IComfyInputSlot";
 
 export type DefaultWidgetSpec = {
     defaultWidgetNode: new (name?: string) => ComfyWidgetNode,
@@ -21,4 +22,34 @@ export default class ComfyGraphNode extends LGraphNode {
     onExecuted?(output: any): void;
 
     defaultWidgets?: DefaultWidgetLayout
+
+    override onSerialize(o: SerializedLGraphNode) {
+        for (let index = 0; index < this.inputs.length; index++) {
+            const input = this.inputs[index]
+            const serInput = o.inputs[index]
+            if ("defaultWidgetNode" in input) {
+                const comfyInput = input as IComfyInputSlot
+                const widgetNode = comfyInput.defaultWidgetNode
+                const ty = Object.values(LiteGraph.registered_node_types)
+                    .find(v => v.class === widgetNode)
+                if (ty)
+                    (serInput as any).widgetNodeType = ty.type
+            }
+        }
+    }
+
+    override onConfigure(o: SerializedLGraphNode) {
+        for (let index = 0; index < this.inputs.length; index++) {
+            const input = this.inputs[index]
+            const serInput = o.inputs[index]
+            if ("widgetNodeType" in serInput) {
+                const comfyInput = input as IComfyInputSlot
+                const ty: string = serInput.widgetNodeType as any
+                const widgetNode = Object.values(LiteGraph.registered_node_types)
+                    .find(v => v.type === ty)
+                if (widgetNode)
+                    comfyInput.defaultWidgetNode = widgetNode.class as any
+            }
+        }
+    }
 }

src/lib/nodes/ComfyWidgetNodes.ts

@@ -27,7 +27,7 @@ export abstract class ComfyWidgetNode<T = any> extends ComfyGraphNode {
     abstract properties: ComfyWidgetProperties;
 
     value: Writable<T>
-    propsChanged: Writable<boolean> = writable(true) // dummy to indicate if props changed
+    propsChanged: Writable<number> = writable(0) // dummy to indicate if props changed
     unsubscribe: Unsubscriber;
 
     /** Svelte class for the frontend logic */
@@ -119,6 +119,7 @@ export abstract class ComfyWidgetNode<T = any> extends ComfyGraphNode {
             if (this.inputs.length >= this.inputIndex) {
                 const data = this.getInputData(this.inputIndex)
                 if (data) { // TODO can "null" be a legitimate value here?
+                    console.log(data)
                     this.setValue(data)
                     const input = this.getInputLink(this.inputIndex)
                     input.data = null;
@@ -145,30 +146,33 @@ export abstract class ComfyWidgetNode<T = any> extends ComfyGraphNode {
         inputNode: LGraphNode,
         inputIndex: number
     ): boolean {
-        if (this.autoConfig) {
-            // Copy properties from default config in input slot
-            if ("config" in input) {
-                const comfyInput = input as IComfyInputSlot;
-                for (const key in comfyInput.config)
-                    this.setProperty(key, comfyInput.config[key])
-
-                if ("defaultValue" in this.properties)
-                    this.setValue(this.properties.defaultValue)
-
-                const widget = layoutState.findLayoutForNode(this.id)
-                if (widget && input.name !== "") {
-                    widget.attrs.title = input.name;
-                }
-
-                console.debug("Property copy", input, this.properties)
-
-                this.setValue(get(this.value))
-            }
+        if (this.autoConfig && "config" in input) {
+            this.doAutoConfig(input as IComfyInputSlot)
         }
 
         return true;
     }
 
+    doAutoConfig(input: IComfyInputSlot) {
+        // Copy properties from default config in input slot
+        const comfyInput = input as IComfyInputSlot;
+        for (const key in comfyInput.config)
+            this.setProperty(key, comfyInput.config[key])
+
+        if ("defaultValue" in this.properties)
+            this.setValue(this.properties.defaultValue)
+
+        const widget = layoutState.findLayoutForNode(this.id)
+        if (widget && input.name !== "") {
+            widget.attrs.title = input.name;
+        }
+
+        console.debug("Property copy", input, this.properties)
+
+        this.setValue(get(this.value))
+        this.propsChanged.set(get(this.propsChanged) + 1)
+    }
+
     onConnectionsChange(
         type: LConnectionKind,
         slotIndex: number,
@@ -195,12 +199,13 @@ export abstract class ComfyWidgetNode<T = any> extends ComfyGraphNode {
         }
 
         // Force reactivity change so the frontend can be updated with the new props
-        this.propsChanged.set(!get(this.propsChanged))
+        this.propsChanged.set(get(this.propsChanged) + 1)
     }
 
     clampOneConfig(input: IComfyInputSlot) { }
 
     override onSerialize(o: SerializedLGraphNode) {
+        super.onSerialize(o);
         (o as any).comfyValue = get(this.value);
         (o as any).shownOutputProperties = this.shownOutputProperties
     }

src/lib/nodes/index.ts

@@ -1,5 +1,5 @@
 export { default as ComfyReroute } from "./ComfyReroute"
 export { ComfyWidgetNode, ComfySliderNode, ComfyComboNode, ComfyTextNode } from "./ComfyWidgetNodes"
-export { ComfyCopyAction } from "./ComfyActionNodes"
+export { ComfyCopyAction, ComfySwapAction } from "./ComfyActionNodes"
 export { default as ComfyValueControl } from "./ComfyValueControl"
 export { default as ComfySelector } from "./ComfySelector"