You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.2 KiB
TypeScript
41 lines
1.2 KiB
TypeScript
import BasicNode from '@/components/FlowEditor/node/basicNode/BasicNode';
|
|
import SwitchNode from '@/components/FlowEditor/node/switchNode/SwitchNode';
|
|
import ImageNode from '@/components/FlowEditor/node/imageNode/ImageNode';
|
|
import CodeNode from '@/components/FlowEditor/node/codeNode/CodeNode';
|
|
import RestNode from '@/components/FlowEditor/node/restNode/RestNode';
|
|
import LocalNode from '@/components/FlowEditor/node/localNode/LocalNode';
|
|
import { nodeTypeMap, registerNodeType } from '@/components/FlowEditor/node';
|
|
|
|
export const resolveNodeComponent = (nodeType: string) => {
|
|
switch (nodeType) {
|
|
case 'BASIC':
|
|
case 'SUB':
|
|
return BasicNode;
|
|
case 'SWITCH':
|
|
return SwitchNode;
|
|
case 'IMAGE':
|
|
return ImageNode;
|
|
case 'CODE':
|
|
return CodeNode;
|
|
case 'REST':
|
|
return RestNode;
|
|
default:
|
|
return LocalNode;
|
|
}
|
|
};
|
|
|
|
export const ensureNodeTypeRegistered = (
|
|
nodeType: string,
|
|
nodeName: string,
|
|
component?: any
|
|
) => {
|
|
const nodeMap = Array.from(Object.values(nodeTypeMap).map((key) => key));
|
|
if (!nodeMap.includes(nodeType)) {
|
|
registerNodeType(
|
|
nodeType,
|
|
component || resolveNodeComponent(nodeType),
|
|
nodeName
|
|
);
|
|
}
|
|
};
|