89 lines
2.2 KiB
TypeScript
89 lines
2.2 KiB
TypeScript
import { createId } from '../../tools/tools'
|
|
import { NODE_DATATYPE, NODE_COMPONENT } from '../tools/index.d'
|
|
interface NodeOptions {
|
|
id?: string
|
|
position?: { x: number, y: number }
|
|
positionX?: number
|
|
positionY?: number
|
|
component?: any
|
|
data?: object
|
|
}// 不可传入type class （内部使用）
|
|
|
|
export class NodeManager {
|
|
stateManager: any
|
|
vueFlow: any
|
|
constructor(options) {
|
|
this.stateManager = options.stateManager;
|
|
this.vueFlow = options.vueFlow
|
|
}
|
|
|
|
/** 删除节点 */
|
|
deleteNode(id: string) {
|
|
this.stateManager.deleteNode(id)
|
|
}
|
|
/** 添加节点 */
|
|
addNode(node: any) {
|
|
this.stateManager.addNode(node)
|
|
}
|
|
|
|
/** 创建节点 */
|
|
createNode(options: NodeOptions) {
|
|
const superiorID = options?.data?.superiorID
|
|
const snode = superiorID ? this.stateManager.flowManager.getNodeById(superiorID) : this.stateManager.flowManager.getLastNode();
|
|
const id = options.id || createId()
|
|
const positionX = options.positionX || 0
|
|
const positionY = options.positionY || 0
|
|
const position = options.position ||
|
|
!snode ? { x: positionX, y: positionY } :
|
|
{
|
|
x: snode.position.x + snode.dimensions.width + 50 + positionX,
|
|
y: snode.position.y + positionY
|
|
}
|
|
const data = options?.data || {}
|
|
data['component'] = options.component
|
|
const options_ = {
|
|
id,
|
|
position,
|
|
data
|
|
}
|
|
this.addNode(options_)
|
|
return options_;
|
|
}
|
|
/** 创建结果节点 */
|
|
createResultNode(options?: NodeOptions) {
|
|
const options_ = {
|
|
...(options ? options : {}),
|
|
component: NODE_COMPONENT.RESULT_IMAGE,
|
|
data: {
|
|
tier: 0,
|
|
type: NODE_DATATYPE.RESULT_IMAGE,
|
|
isHeader: true,
|
|
...(options?.data || {}),
|
|
},
|
|
}
|
|
return this.createNode(options_)
|
|
}
|
|
/** 创建卡片选择节点 */
|
|
createCardsSelect(options?: NodeOptions) {
|
|
const options_ = {
|
|
...(options ? options : {}),
|
|
component: NODE_COMPONENT.CARD,
|
|
positionY: 50,
|
|
data: {
|
|
tier: 1,
|
|
type: NODE_DATATYPE.CARDS_SELECT,
|
|
...(options?.data || {}),
|
|
},
|
|
}
|
|
return this.createNode(options_)
|
|
}
|
|
/** 创建卡片节点 */
|
|
createCardNode(options?: NodeOptions) {
|
|
const options_ = {
|
|
...(options ? options : {}),
|
|
component: NODE_COMPONENT.CARD,
|
|
}
|
|
return this.createNode(options_)
|
|
}
|
|
}
|