100 lines
3.3 KiB
TypeScript
100 lines
3.3 KiB
TypeScript
import { parseWhateverIntoImageMetadata, type ComfyBoxImageMetadata, type ComfyUploadImageType } from "$lib/utils";
|
|
import { BuiltInSlotType, LiteGraph, type IComboWidget, type ITextWidget, type PropertyLayout, type SlotLayout } from "@litegraph-ts/core";
|
|
import { get } from "svelte/store";
|
|
|
|
import GalleryWidget from "$lib/widgets/GalleryWidget.svelte";
|
|
import type { ComfyWidgetProperties } from "./ComfyWidgetNode";
|
|
import ComfyWidgetNode from "./ComfyWidgetNode";
|
|
|
|
export interface ComfyGalleryProperties extends ComfyWidgetProperties {
|
|
index: number,
|
|
updateMode: "replace" | "append",
|
|
}
|
|
|
|
export default class ComfyGalleryNode extends ComfyWidgetNode<ComfyBoxImageMetadata[]> {
|
|
override properties: ComfyGalleryProperties = {
|
|
tags: [],
|
|
defaultValue: [],
|
|
index: 0,
|
|
updateMode: "replace",
|
|
}
|
|
|
|
static slotLayout: SlotLayout = {
|
|
inputs: [
|
|
{ name: "images", type: "OUTPUT" },
|
|
{ name: "store", type: BuiltInSlotType.ACTION, options: { color_off: "rebeccapurple", color_on: "rebeccapurple" } }
|
|
],
|
|
outputs: [
|
|
{ name: "images", type: "COMFYBOX_IMAGES" },
|
|
{ name: "selected_index", type: "number" },
|
|
]
|
|
}
|
|
|
|
static propertyLayout: PropertyLayout = [
|
|
{ name: "updateMode", defaultValue: "replace", type: "enum", options: { values: ["replace", "append"] } }
|
|
]
|
|
|
|
override svelteComponentType = GalleryWidget
|
|
override defaultValue = []
|
|
override saveUserState = false;
|
|
override outputSlotName = null;
|
|
override changedEventName = null;
|
|
|
|
selectedFilename: string | null = null;
|
|
|
|
selectedIndexWidget: ITextWidget;
|
|
modeWidget: IComboWidget;
|
|
|
|
constructor(name?: string) {
|
|
super(name, [])
|
|
this.selectedIndexWidget = this.addWidget("text", "Selected", String(this.properties.index), "index")
|
|
this.selectedIndexWidget.disabled = true;
|
|
this.modeWidget = this.addWidget("combo", "Mode", this.properties.updateMode, null, { property: "updateMode", values: ["replace", "append"] })
|
|
}
|
|
|
|
override onPropertyChanged(property: any, value: any) {
|
|
if (property === "updateMode") {
|
|
this.modeWidget.value = value;
|
|
}
|
|
}
|
|
|
|
override onExecute() {
|
|
this.setOutputData(0, get(this.value))
|
|
this.setOutputData(1, this.properties.index)
|
|
}
|
|
|
|
override onAction(action: any, param: any, options: { action_call?: string }) {
|
|
super.onAction(action, param, options)
|
|
}
|
|
|
|
override formatValue(value: ComfyBoxImageMetadata[] | null): string {
|
|
return `Images: ${value?.length || 0}`
|
|
}
|
|
|
|
override parseValue(param: any): ComfyBoxImageMetadata[] {
|
|
const meta = parseWhateverIntoImageMetadata(param) || [];
|
|
|
|
console.debug("[ComfyGalleryNode] Received output!", param)
|
|
|
|
if (this.properties.updateMode === "append") {
|
|
const currentValue = get(this.value)
|
|
return currentValue.concat(meta)
|
|
}
|
|
else {
|
|
return meta;
|
|
}
|
|
}
|
|
|
|
override setValue(value: any, noChangedEvent: boolean = false) {
|
|
super.setValue(value, noChangedEvent)
|
|
this.setProperty("index", null)
|
|
}
|
|
}
|
|
|
|
LiteGraph.registerNodeType({
|
|
class: ComfyGalleryNode,
|
|
title: "UI.Gallery",
|
|
desc: "Gallery that shows most recent outputs",
|
|
type: "ui/gallery"
|
|
})
|