[Refactor] useComboWidget composable (#2517)

src/composables/widgets/useComboWidget.ts

@@ -0,0 +1,61 @@
+import type { LGraphNode } from '@comfyorg/litegraph'
+import type { IComboWidget } from '@comfyorg/litegraph/dist/types/widgets'
+
+import { addValueControlWidgets } from '@/scripts/widgets'
+import type { ComfyWidgetConstructor } from '@/scripts/widgets'
+import { useWidgetStore } from '@/stores/widgetStore'
+import type { InputSpec } from '@/types/apiTypes'
+
+import { useRemoteWidget } from './useRemoteWidget'
+
+export const useComboWidget = () => {
+  const widgetConstructor: ComfyWidgetConstructor = (
+    node: LGraphNode,
+    inputName: string,
+    inputData: InputSpec
+  ) => {
+    const widgetStore = useWidgetStore()
+    const { remote, options } = inputData[1]
+    const defaultValue = widgetStore.getDefaultValue(inputData)
+
+    const res = {
+      widget: node.addWidget('combo', inputName, defaultValue, () => {}, {
+        values: options ?? inputData[0]
+      }) as IComboWidget
+    }
+
+    if (remote) {
+      const remoteWidget = useRemoteWidget({
+        inputData,
+        defaultValue,
+        node,
+        widget: res.widget
+      })
+      if (remote.refresh_button) remoteWidget.addRefreshButton()
+
+      const origOptions = res.widget.options
+      res.widget.options = new Proxy(
+        origOptions as Record<string | symbol, any>,
+        {
+          get(target, prop: string | symbol) {
+            if (prop !== 'values') return target[prop]
+            return remoteWidget.getValue()
+          }
+        }
+      )
+    }
+
+    if (inputData[1]?.control_after_generate) {
+      res.widget.linkedWidgets = addValueControlWidgets(
+        node,
+        res.widget,
+        undefined,
+        undefined,
+        inputData
+      )
+    }
+    return res
+  }
+
+  return widgetConstructor
+}

src/scripts/widgets.ts

@@ -1,18 +1,16 @@
 // @ts-strict-ignore
 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 { useComboWidget } from '@/composables/widgets/useComboWidget'
 import { useFloatWidget } from '@/composables/widgets/useFloatWidget'
 import { useImageUploadWidget } from '@/composables/widgets/useImageUploadWidget'
 import { useIntWidget } from '@/composables/widgets/useIntWidget'
 import { useMarkdownWidget } from '@/composables/widgets/useMarkdownWidget'
-import { useRemoteWidget } from '@/composables/widgets/useRemoteWidget'
 import { useSeedWidget } from '@/composables/widgets/useSeedWidget'
 import { useStringWidget } from '@/composables/widgets/useStringWidget'
 import { useSettingStore } from '@/stores/settingStore'
-import { useWidgetStore } from '@/stores/widgetStore'
 import type { InputSpec } from '@/types/apiTypes'
 
 import type { ComfyApp } from './app'
@@ -255,49 +253,6 @@ export const ComfyWidgets: Record<string, ComfyWidgetConstructor> = {
   BOOLEAN: useBooleanWidget(),
   STRING: useStringWidget(),
   MARKDOWN: useMarkdownWidget(),
-  COMBO(node, inputName, inputData: InputSpec) {
-    const widgetStore = useWidgetStore()
-    const { remote, options } = inputData[1]
-    const defaultValue = widgetStore.getDefaultValue(inputData)
-
-    const res = {
-      widget: node.addWidget('combo', inputName, defaultValue, () => {}, {
-        values: options ?? inputData[0]
-      }) as IComboWidget
-    }
-
-    if (remote) {
-      const remoteWidget = useRemoteWidget({
-        inputData,
-        defaultValue,
-        node,
-        widget: res.widget
-      })
-      if (remote.refresh_button) remoteWidget.addRefreshButton()
-
-      const origOptions = res.widget.options
-      res.widget.options = new Proxy(
-        origOptions as Record<string | symbol, any>,
-        {
-          get(target, prop: string | symbol) {
-            if (prop !== 'values') return target[prop]
-            return remoteWidget.getValue()
-          }
-        }
-      )
-    }
-
-    if (inputData[1]?.control_after_generate) {
-      // TODO make combo handle a widget node type?
-      res.widget.linkedWidgets = addValueControlWidgets(
-        node,
-        res.widget,
-        undefined,
-        undefined,
-        inputData
-      )
-    }
-    return res
-  },
+  COMBO: useComboWidget(),
   IMAGEUPLOAD: useImageUploadWidget()
 }