ComfyUI_frontend/src/litegraph.ts

import type { Point, ConnectingLink } from "./interfaces"
import type { INodeSlot, INodeInputSlot, INodeOutputSlot, CanvasColour, Direction, IBoundaryNodes, IContextMenuOptions, IContextMenuValue, IFoundSlot, IInputOrOutput, INodeFlags, IOptionalInputsData, ISlotType, KeysOfType, MethodNames, PickByType, Rect, Rect32, Size } from "./interfaces"
import type { SlotShape, LabelPosition, SlotDirection, SlotType } from "./draw"
import type { IWidget } from "./types/widgets"
import type { TitleMode } from "./types/globalEnums"
import { LiteGraphGlobal } from "./LiteGraphGlobal"
import { loadPolyfills } from "./polyfills"

import { LGraph } from "./LGraph"
import { LGraphCanvas } from "./LGraphCanvas"
import { DragAndScale } from "./DragAndScale"
import { LGraphNode } from "./LGraphNode"
import { LGraphGroup } from "./LGraphGroup"
import { LLink } from "./LLink"
import { ContextMenu } from "./ContextMenu"
import { CurveEditor } from "./CurveEditor"
import { LGraphBadge, BadgePosition } from "./LGraphBadge"

export const LiteGraph = new LiteGraphGlobal()
export { LGraph, LGraphCanvas, DragAndScale, LGraphNode, LGraphGroup, LLink, ContextMenu, CurveEditor }
export { INodeSlot, INodeInputSlot, INodeOutputSlot, ConnectingLink, CanvasColour, Direction, IBoundaryNodes, IContextMenuOptions, IContextMenuValue, IFoundSlot, IInputOrOutput, INodeFlags, IOptionalInputsData, ISlotType, KeysOfType, MethodNames, PickByType, Rect, Rect32, Size }
export { IWidget }
export { LGraphBadge, BadgePosition }
export { SlotShape, LabelPosition, SlotDirection, SlotType }

// TODO: Remove legacy accessors
LiteGraph.LGraph = LGraph
LiteGraph.LLink = LLink
LiteGraph.LGraphNode = LGraphNode
LiteGraph.LGraphGroup = LGraphGroup
LiteGraph.DragAndScale = DragAndScale
LiteGraph.LGraphCanvas = LGraphCanvas
LiteGraph.ContextMenu = ContextMenu
LiteGraph.CurveEditor = CurveEditor

export function clamp(v: number, a: number, b: number): number {
    return a > v ? a : b < v ? b : v
};

// Load legacy polyfills
loadPolyfills()

// Backwards compat

// Type definitions for litegraph.js 0.7.0
// Project: litegraph.js
// Definitions by: NateScarlet <https://github.com/NateScarlet>
export type Vector2 = Point
export type Vector4 = [number, number, number, number]

export interface IContextMenuItem {
    content: string
    callback?: ContextMenuEventListener
    /** Used as innerHTML for extra child element */
    title?: string
    disabled?: boolean
    has_submenu?: boolean
    submenu?: {
        options: IContextMenuItem[]
    } & IContextMenuOptions
    className?: string
}

export type ContextMenuEventListener = (
    value: IContextMenuItem,
    options: IContextMenuOptions,
    event: MouseEvent,
    parentMenu: ContextMenu | undefined,
    node: LGraphNode
) => boolean | void

export interface LinkReleaseContext {
    node_to?: LGraphNode
    node_from?: LGraphNode
    slot_from: INodeSlot
    type_filter_in?: string
    type_filter_out?: string
}

export interface LinkReleaseContextExtended {
    links: ConnectingLink[]
}

export type LiteGraphCanvasEventType = "empty-release" | "empty-double-click" | "group-double-click"

export interface LiteGraphCanvasEvent extends CustomEvent<{
    subType: string
    originalEvent: Event
    linkReleaseContext?: LinkReleaseContextExtended
    group?: LGraphGroup
}> { }

export interface LiteGraphCanvasGroupEvent extends CustomEvent<{
    subType: "group-double-click"
    originalEvent: MouseEvent
    group: LGraphGroup
}> { }

/** https://github.com/jagenjo/litegraph.js/blob/master/guides/README.md#lgraphnode */

export interface LGraphNodeConstructor<T extends LGraphNode = LGraphNode> {
    title?: string
    type?: string
    size?: Size
    min_height?: number
    slot_start_y?: number
    widgets_info?: any
    collapsable?: boolean
    nodeData: any
    new(): T
}

// End backwards compat