mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-09 01:20:09 +00:00
Add group nodes to search and node library (#861)
* Register group nodes in nodeDefStore * Add playwright tests * Update test expectations [skip ci] * Mock nodeDefStore in group node unit test --------- Co-authored-by: github-actions <github-actions@github.com>
This commit is contained in:
@@ -4,6 +4,7 @@ import { mergeIfValid } from './widgetInputs'
|
||||
import { ManageGroupDialog } from './groupNodeManage'
|
||||
import type { LGraphNode } from '@comfyorg/litegraph'
|
||||
import { LGraphCanvas, LiteGraph } from '@comfyorg/litegraph'
|
||||
import { useNodeDefStore } from '@/stores/nodeDefStore'
|
||||
|
||||
const GROUP = Symbol()
|
||||
|
||||
@@ -194,6 +195,10 @@ export class GroupNodeConfig {
|
||||
display_name: this.name,
|
||||
category: 'group nodes' + ('/' + source),
|
||||
input: { required: {} },
|
||||
description: `Group node combining ${this.nodeData.nodes
|
||||
.map((n) => n.type)
|
||||
.join(', ')}`,
|
||||
python_module: 'custom_nodes.' + this.name,
|
||||
|
||||
[GROUP]: this
|
||||
}
|
||||
@@ -212,6 +217,7 @@ export class GroupNodeConfig {
|
||||
}
|
||||
this.#convertedToProcess = null
|
||||
await app.registerNodeDef('workflow/' + this.name, this.nodeDef)
|
||||
useNodeDefStore().addNodeDef(this.nodeDef)
|
||||
}
|
||||
|
||||
getLinks() {
|
||||
|
||||
@@ -314,6 +314,11 @@ export const useNodeDefStore = defineStore('nodeDef', {
|
||||
this.nodeDefsByName = newNodeDefsByName
|
||||
this.nodeDefsByDisplayName = nodeDefsByDisplayName
|
||||
},
|
||||
addNodeDef(nodeDef: ComfyNodeDef) {
|
||||
const nodeDefImpl = plainToClass(ComfyNodeDefImpl, nodeDef)
|
||||
this.nodeDefsByName[nodeDef.name] = nodeDefImpl
|
||||
this.nodeDefsByDisplayName[nodeDef.display_name] = nodeDefImpl
|
||||
},
|
||||
updateWidgets(widgets: Record<string, ComfyWidgetConstructor>) {
|
||||
this.widgets = widgets
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user