mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-21 15:24:09 +00:00
a1c54ad7aa
Fixes #8882 Adds explicit type annotations to all extension callback parameters (`nodeCreated`, `beforeRegisterNodeDef`, `addCustomNodeDefs`) across 14 core extension files. While the types were already inferred from the `ComfyExtension` interface, explicit annotations improve readability and make the code self-documenting. ## Changes - Annotate `node` parameter as `LGraphNode` in all `nodeCreated` callbacks - Annotate `nodeType` as `typeof LGraphNode` and `nodeData` as `ComfyNodeDef` in all `beforeRegisterNodeDef` callbacks - Annotate `defs` as `Record<string, ComfyNodeDef>` in `addCustomNodeDefs` - Add necessary type imports where missing ┆Issue is synchronized with this [Notion page](https://www.notion.so/PR-8966-fix-add-explicit-type-annotations-to-extension-callback-parameters-30b6d73d36508125b074f509aa38145f) by [Unito](https://www.unito.io)
32 lines
1.2 KiB
TypeScript
32 lines
1.2 KiB
TypeScript
import type { LGraphNode } from '@/lib/litegraph/src/LGraphNode'
|
|
import { useExtensionService } from '@/services/extensionService'
|
|
import { processDynamicPrompt } from '@/utils/formatUtil'
|
|
|
|
// Allows for simple dynamic prompt replacement
|
|
// Inputs in the format {a|b} will have a random value of a or b chosen when the prompt is queued.
|
|
|
|
useExtensionService().registerExtension({
|
|
name: 'Comfy.DynamicPrompts',
|
|
nodeCreated(node: LGraphNode) {
|
|
if (node.widgets) {
|
|
// Locate dynamic prompt text widgets
|
|
// Include any widgets with dynamicPrompts set to true, and customtext
|
|
const widgets = node.widgets.filter((w) => w.dynamicPrompts)
|
|
for (const widget of widgets) {
|
|
// Override the serialization of the value to resolve dynamic prompts for all widgets supporting it in this node
|
|
widget.serializeValue = (workflowNode, widgetIndex) => {
|
|
if (typeof widget.value !== 'string') return widget.value
|
|
|
|
const prompt = processDynamicPrompt(widget.value)
|
|
|
|
// Overwrite the value in the serialized workflow pnginfo
|
|
if (workflowNode?.widgets_values)
|
|
workflowNode.widgets_values[widgetIndex] = prompt
|
|
|
|
return prompt
|
|
}
|
|
}
|
|
}
|
|
}
|
|
})
|