windmill/frontend/src/lib/components/graph/model.ts

import type { FlowStatusModule } from '$lib/gen'
import type { UserNodeType } from './svelvet/types'

export type ModuleHost = 'workspace' | 'inline' | 'hub'

export type Node = UserNodeType & {
	parentIds: string[]
	edgeLabel?: string
	host?: ModuleHost
	type: 'node'
	loopDepth: number
}

export type Loop = {
	type: 'loop'
	items: NestedNodes
}

export type Branch = {
	node: Node
	nodeEnd: Node
	type: 'branch'
	items: NestedNodes[]
}

export type GraphItem = Node | Loop | Branch

export type GraphModuleState = {
	type: FlowStatusModule.type
	args: any
	logs?: string
	result?: any
	scheduled_for?: Date
	job_id?: string
	parent_module?: string
	iteration_total?: number
	retries?: number
}

export type NestedNodes = GraphItem[]

export function isNode(item: GraphItem | NestedNodes | undefined): item is Node {
	return item?.['type'] === 'node'
}

export function isLoop(item: GraphItem | NestedNodes | undefined): item is Loop {
	return item?.['type'] === 'loop'
}

export function isBranch(item: GraphItem | NestedNodes | undefined): item is Branch {
	return item?.['type'] === 'branch'
}