Subgraph ergonomic improvements & node def types

src/lib/ComfyNodeDef.ts

@@ -0,0 +1,52 @@
+import { range } from "./utils"
+import ComfyWidgets from "./widgets"
+
+export type ComfyNodeDef = {
+    name: string
+    display_name?: string
+    category: string
+    input: ComfyNodeDefInputs
+    /** Output type like "LATENT" or "IMAGE" */
+    output: string[]
+    output_name: string[]
+    output_is_list: boolean[]
+}
+
+export type ComfyNodeDefInputs = {
+    required: Record<string, ComfyNodeDefInput>,
+    optional?: Record<string, ComfyNodeDefInput>
+}
+export type ComfyNodeDefInput = [ComfyNodeDefInputType, ComfyNodeDefInputOptions | null]
+export type ComfyNodeDefInputType = string[] | keyof typeof ComfyWidgets | string
+export type ComfyNodeDefInputOptions = {
+    forceInput?: boolean
+}
+
+// TODO when comfy refactors
+export type ComfyNodeDefOutput = {
+    type: string,
+    name: string,
+    is_list?: boolean
+}
+
+export function isBackendNodeDefInputType(inputName: string, type: ComfyNodeDefInputType): type is string {
+    return !Array.isArray(type) && !(type in ComfyWidgets) && !(`${type}:${inputName}` in ComfyWidgets);
+}
+
+export function iterateNodeDefInputs(def: ComfyNodeDef): Iterable<[string, ComfyNodeDefInput]> {
+    var inputs = def.input.required
+    if (def.input.optional != null) {
+        inputs = Object.assign({}, def.input.required, def.input.optional)
+    }
+    return Object.entries(inputs);
+}
+
+export function iterateNodeDefOutputs(def: ComfyNodeDef): Iterable<ComfyNodeDefOutput> {
+    return range(def.output.length).map(i => {
+        return {
+            type: def.output[i],
+            name: def.output_name[i] || def.output[i],
+            is_list: def.output_is_list[i],
+        }
+    })
+}