Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-04-29 18:52:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Type serialized NodeSlot (#605)

Browse Source
This commit is contained in:
Chenlei Hu
2025-02-25 19:38:14 -05:00
committed by GitHub
parent dbe016b934
commit 953ae434ea
3 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/LGraphNode.ts
View File

@@ -707,8 +707,8 @@ export class LGraphNode implements Positionable, IPinnable, IColorable {
if (this.constructor === LGraphNode && this.last_serialization) if (this.constructor === LGraphNode && this.last_serialization)
return this.last_serialization return this.last_serialization
if (this.inputs) o.inputs = this.inputs.map(serializeSlot) if (this.inputs) o.inputs = this.inputs.map(input => serializeSlot(input))
if (this.outputs) o.outputs = this.outputs.map(serializeSlot) if (this.outputs) o.outputs = this.outputs.map(output => serializeSlot(output))
if (this.title && this.title != this.constructor.title) o.title = this.title if (this.title && this.title != this.constructor.title) o.title = this.title

6
src/NodeSlot.ts
View File

@@ -4,6 +4,8 @@ import type { LinkId } from "./LLink"
import { LinkDirection, RenderShape } from "./types/globalEnums" import { LinkDirection, RenderShape } from "./types/globalEnums"
import { LabelPosition, SlotShape, SlotType } from "./draw" import { LabelPosition, SlotShape, SlotType } from "./draw"
import { LiteGraph } from "./litegraph" import { LiteGraph } from "./litegraph"
import { ISerialisedNodeOutputSlot } from "./types/serialisation"
import { ISerialisedNodeInputSlot } from "./types/serialisation"
export interface ConnectionColorContext { export interface ConnectionColorContext {
default_connection_color: { default_connection_color: {
@@ -27,7 +29,9 @@ interface IDrawOptions {
highlight?: boolean highlight?: boolean
} }
export function serializeSlot<T extends INodeSlot>(slot: T): T { export function serializeSlot(slot: INodeInputSlot): ISerialisedNodeInputSlot
export function serializeSlot(slot: INodeOutputSlot): ISerialisedNodeOutputSlot
export function serializeSlot(slot: INodeInputSlot | INodeOutputSlot): ISerialisedNodeInputSlot | ISerialisedNodeOutputSlot {
const serialized = { ...slot } const serialized = { ...slot }
delete serialized._layoutElement delete serialized._layoutElement
if ("_data" in serialized) { if ("_data" in serialized) {

9
src/types/serialisation.ts
View File

@@ -40,6 +40,11 @@ export interface SerialisableGraph {
extra?: Record<any, any> extra?: Record<any, any>
} }
export type ISerialisedNodeInputSlot = Omit<INodeInputSlot, "_layoutElement"> & {
widget?: { name?: string }
}
export type ISerialisedNodeOutputSlot = Omit<INodeOutputSlot, "_layoutElement" | "_data">
/** Serialised LGraphNode */ /** Serialised LGraphNode */
export interface ISerialisedNode { export interface ISerialisedNode {
title?: string title?: string
@@ -50,8 +55,8 @@ export interface ISerialisedNode {
flags?: INodeFlags flags?: INodeFlags
order?: number order?: number
mode?: number mode?: number
outputs?: INodeOutputSlot[] outputs?: ISerialisedNodeOutputSlot[]
inputs?: INodeInputSlot[] inputs?: ISerialisedNodeInputSlot[]
properties?: Dictionary<unknown> properties?: Dictionary<unknown>
shape?: RenderShape shape?: RenderShape
boxcolor?: string boxcolor?: string