neuro_llc/ComfyBox
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Auto add new widgets for backend nodes

Browse Source
This commit is contained in:
space-nuko
2023-05-04 11:42:21 -05:00
parent 93cf2ed98a
commit 71a3abf518
13 changed files with 150 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/lib/GraphSync.ts
View File

@@ -2,6 +2,7 @@ import type { LGraph, LGraphNode } from "@litegraph-ts/core";
import type ComfyApp from "./components/ComfyApp";
import type { Unsubscriber, Writable } from "svelte/store";
import type { ComfyWidgetNode } from "./nodes";
import type ComfyGraph from "./ComfyGraph";
type WidgetSubStore = {
store: WidgetUIStateStore,
@@ -28,13 +29,11 @@ export default class GraphSync {
// nodeId -> widgetSubStore
private stores: Record<string, WidgetSubStore> = {}
constructor(app: ComfyApp) {
this.graph = app.lGraph;
app.eventBus.on("nodeAdded", this.onNodeAdded.bind(this));
app.eventBus.on("nodeRemoved", this.onNodeRemoved.bind(this));
constructor(graph: ComfyGraph) {
this.graph = graph;
}
private onNodeAdded(node: LGraphNode) {
onNodeAdded(node: LGraphNode) {
// TODO assumes only a single graph's widget state.
if ("svelteComponentType" in node) {
@@ -44,7 +43,7 @@ export default class GraphSync {
this.graph.setDirtyCanvas(true, true);
}
private onNodeRemoved(node: LGraphNode) {
onNodeRemoved(node: LGraphNode) {
if ("svelteComponentType" in node) {
this.removeStore(node as ComfyWidgetNode);
}