FiDA_Front/src/components/Canvas/FlowCanvas/manager/NodeManager.ts

import { createId } from '../../tools/tools'
import card from '../components/cards/index.vue'
import resultImage from '../components/result/result-image.vue'
import { NODE_DATATYPE } 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 lastNode = this.stateManager.flowManager.getLastNode();
		const id = options.id || createId()
		const positionX = options.positionX || 0
		const positionY = options.positionY || 0
		const position = options.position ||
			!lastNode ? { x: positionX, y: positionY } :
			{ x: lastNode.position.x + lastNode.dimensions.width + 50 + positionX, y: lastNode.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: resultImage,
			data: {
				tier: 0,
				type: NODE_DATATYPE.RESULT_IMAGE,
				isHeader: true,
				...(options?.data || {}),
			},
		}
		return this.createNode(options_)
	}
	/** 创建卡片选择节点 */
	createCardsSelect(options?: NodeOptions) {
		const options_ = {
			...(options ? options : {}),
			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: card,
		}
		return this.createNode(options_)
	}
}