[Refactor] useBooleanWidget composable (#2516)

This commit is contained in:
Chenlei Hu
2025-02-11 11:05:58 -05:00
committed by GitHub
parent ab305059bc
commit cfa46ebacb
2 changed files with 27 additions and 12 deletions

View File

@@ -3,6 +3,7 @@ import type { LGraphNode } from '@comfyorg/litegraph'
import type { IWidget } from '@comfyorg/litegraph'
import type { IComboWidget } from '@comfyorg/litegraph/dist/types/widgets'
import { useBooleanWidget } from '@/composables/widgets/useBooleanWidget'
import { useFloatWidget } from '@/composables/widgets/useFloatWidget'
import { useImageUploadWidget } from '@/composables/widgets/useImageUploadWidget'
import { useIntWidget } from '@/composables/widgets/useIntWidget'
@@ -251,18 +252,7 @@ export const ComfyWidgets: Record<string, ComfyWidgetConstructor> = {
'INT:noise_seed': SeedWidget,
INT: useIntWidget(),
FLOAT: useFloatWidget(),
BOOLEAN(node, inputName, inputData) {
let defaultVal = false
let options = {}
if (inputData[1]) {
if (inputData[1].default) defaultVal = inputData[1].default
if (inputData[1].label_on) options['on'] = inputData[1].label_on
if (inputData[1].label_off) options['off'] = inputData[1].label_off
}
return {
widget: node.addWidget('toggle', inputName, defaultVal, () => {}, options)
}
},
BOOLEAN: useBooleanWidget(),
STRING: useStringWidget(),
MARKDOWN: useMarkdownWidget(),
COMBO(node, inputName, inputData: InputSpec) {