53 lines
1.6 KiB
TypeScript
53 lines
1.6 KiB
TypeScript
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],
|
|
}
|
|
})
|
|
}
|