25 lines
735 B
TypeScript
25 lines
735 B
TypeScript
import { DefaultNode, InputNode, OutputNode } from '../../components/Nodes'
|
|
import { NodeType } from '../../types'
|
|
|
|
export function createNodeTypes(nodeTypes: Record<string, NodeType>): Record<string, NodeType> {
|
|
const standardTypes: Record<string, NodeType> = {
|
|
input: nodeTypes.input || InputNode,
|
|
default: nodeTypes.default || DefaultNode,
|
|
output: nodeTypes.output || OutputNode,
|
|
}
|
|
|
|
const wrappedTypes = {} as NodeType
|
|
const specialTypes: NodeType = Object.keys(nodeTypes)
|
|
.filter((k) => !['input', 'default', 'output'].includes(k))
|
|
.reduce((res, key) => {
|
|
res[key] = nodeTypes[key] || DefaultNode
|
|
|
|
return res
|
|
}, wrappedTypes)
|
|
|
|
return {
|
|
...standardTypes,
|
|
...specialTypes,
|
|
}
|
|
}
|