mlogclub
42 lines
1.1 KiB
TypeScript
42 lines
1.1 KiB
TypeScript
import type { WorkflowNodeRegistry } from "@flowgram.ai/free-layout-editor"
|
|
|
|
import type { AIWorkflowNodeSpec } from "@/lib/api/admin"
|
|
|
|
export function buildFlowgramNodeRegistries(nodeSpecs: AIWorkflowNodeSpec[]): WorkflowNodeRegistry[] {
|
|
const seen = new Set<string>()
|
|
const specs = nodeSpecs.length > 0
|
|
? nodeSpecs
|
|
: [
|
|
{ type: "start", title: "开始" },
|
|
{ type: "end", title: "结束" },
|
|
]
|
|
|
|
return specs
|
|
.filter((spec) => {
|
|
if (!spec.type || seen.has(spec.type)) {
|
|
return false
|
|
}
|
|
seen.add(spec.type)
|
|
return true
|
|
})
|
|
.map((spec) => ({
|
|
type: spec.type,
|
|
meta: {
|
|
defaultExpanded: true,
|
|
deleteDisable: spec.type === "start" || spec.type === "end",
|
|
copyDisable: spec.type === "start" || spec.type === "end",
|
|
defaultPorts: defaultPortsForNodeType(spec.type),
|
|
},
|
|
}))
|
|
}
|
|
|
|
function defaultPortsForNodeType(type: string) {
|
|
if (type === "start") {
|
|
return [{ type: "output" as const }]
|
|
}
|
|
if (type === "end") {
|
|
return [{ type: "input" as const }]
|
|
}
|
|
return [{ type: "input" as const }, { type: "output" as const }]
|
|
}
|