Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-03-03 04:00:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Properly update vue app's node defs on refresh (#493)

Browse Source
This commit is contained in:
Chenlei Hu
2024-08-17 13:23:47 -04:00
committed by GitHub
parent 5ba524fd94
commit 8c40f83b35
1 changed files with 33 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
src/scripts/app.ts
View File

@@ -1955,6 +1955,37 @@ export class ComfyApp {
this.canvas?.draw(true, true)
}
updateVueAppNodeDefs(defs: Record<string, ComfyNodeDef>) {
// Frontend only nodes registered by custom nodes.
// Example: https://github.com/rgthree/rgthree-comfy/blob/dd534e5384be8cf0c0fa35865afe2126ba75ac55/src_web/comfyui/fast_groups_bypasser.ts#L10
const rawDefs = Object.fromEntries(
Object.entries(LiteGraph.registered_node_types).map(([name, _]) => [
name,
{
name,
display_name: name,
category: '__frontend_only__',
input: { required: {}, optional: {} },
output: [],
output_name: [],
output_is_list: [],
python_module: 'custom_nodes.frontend_only',
description: `Frontend only node for ${name}`
}
])
)
const allNodeDefs = {
...rawDefs,
...defs,
...SYSTEM_NODE_DEFS
}
const nodeDefStore = useNodeDefStore()
nodeDefStore.updateNodeDefs(Object.values(allNodeDefs))
nodeDefStore.updateWidgets(this.widgets)
}
/**
* Registers nodes with the graph
*/
@@ -1964,34 +1995,7 @@ export class ComfyApp {
await this.registerNodesFromDefs(defs)
await this.#invokeExtensionsAsync('registerCustomNodes')
if (this.vueAppReady) {
// Frontend only nodes registered by custom nodes.
// Example: https://github.com/rgthree/rgthree-comfy/blob/dd534e5384be8cf0c0fa35865afe2126ba75ac55/src_web/comfyui/fast_groups_bypasser.ts#L10
const rawDefs = Object.fromEntries(
Object.entries(LiteGraph.registered_node_types).map(([name, _]) => [
name,
{
name,
display_name: name,
category: '__frontend_only__',
input: { required: {}, optional: {} },
output: [],
output_name: [],
output_is_list: [],
python_module: 'custom_nodes.frontend_only',
description: `Frontend only node for ${name}`
}
])
)
const allNodeDefs = {
...rawDefs,
...defs,
...SYSTEM_NODE_DEFS
}
const nodeDefStore = useNodeDefStore()
nodeDefStore.updateNodeDefs(Object.values(allNodeDefs))
nodeDefStore.updateWidgets(this.widgets)
this.updateVueAppNodeDefs(defs)
}
}
@@ -2899,6 +2903,7 @@ export class ComfyApp {
await this.#invokeExtensionsAsync('refreshComboInNodes', defs)
if (this.vueAppReady) {
this.updateVueAppNodeDefs(defs)
useToastStore().remove(requestToastMessage)
useToastStore().add({
severity: 'success',