mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-04 07:00:23 +00:00
6f9d2bfa17
* Revert "Revert "Add support for LiteGraph to convert to classes (#334)" (#386)"
This reverts commit d607f6c7f7.
* Format
53 lines
1.3 KiB
TypeScript
53 lines
1.3 KiB
TypeScript
import { LiteGraph, LGraphCanvas } from '@comfyorg/litegraph'
|
|
import { app } from '../../scripts/app'
|
|
import { ComfyWidgets } from '../../scripts/widgets'
|
|
import { LGraphNode } from '@comfyorg/litegraph'
|
|
// Node that add notes to your project
|
|
|
|
app.registerExtension({
|
|
name: 'Comfy.NoteNode',
|
|
registerCustomNodes() {
|
|
class NoteNode extends LGraphNode {
|
|
static category: string
|
|
|
|
color = LGraphCanvas.node_colors.yellow.color
|
|
bgcolor = LGraphCanvas.node_colors.yellow.bgcolor
|
|
groupcolor = LGraphCanvas.node_colors.yellow.groupcolor
|
|
isVirtualNode: boolean
|
|
collapsable: boolean
|
|
title_mode: number
|
|
|
|
constructor(title?: string) {
|
|
super(title)
|
|
if (!this.properties) {
|
|
this.properties = { text: '' }
|
|
}
|
|
ComfyWidgets.STRING(
|
|
// Should we extends LGraphNode? Yesss
|
|
this,
|
|
'',
|
|
// @ts-expect-error
|
|
['', { default: this.properties.text, multiline: true }],
|
|
app
|
|
)
|
|
|
|
this.serialize_widgets = true
|
|
this.isVirtualNode = true
|
|
}
|
|
}
|
|
|
|
// Load default visibility
|
|
|
|
LiteGraph.registerNodeType(
|
|
'Note',
|
|
Object.assign(NoteNode, {
|
|
title_mode: LiteGraph.NORMAL_TITLE,
|
|
title: 'Note',
|
|
collapsable: true
|
|
})
|
|
)
|
|
|
|
NoteNode.category = 'utils'
|
|
}
|
|
})
|