64 lines
1.8 KiB
TypeScript
64 lines
1.8 KiB
TypeScript
import LoopNode from './index.vue'
|
|
import { t } from '@/locales'
|
|
import { AppNode, AppNodeModel } from '@/workflow/common/app-node'
|
|
class LoopBodyNodeView extends AppNode {
|
|
constructor(props: any) {
|
|
super(props, LoopNode)
|
|
}
|
|
getNodeName() {
|
|
return t('views.applicationWorkflow.nodes.loopBodyNode.label')
|
|
}
|
|
get_up_node_field_list(contain_self: boolean, use_cache: boolean) {
|
|
const loop_node_id = this.props.model.properties.loop_node_id
|
|
if (typeof loop_node_id === 'string') {
|
|
const loop_node = this.props.graphModel.getNodeModelById(loop_node_id)
|
|
if (loop_node && typeof loop_node.get_up_node_field_list === 'function') {
|
|
return loop_node.get_up_node_field_list(contain_self, use_cache)
|
|
}
|
|
}
|
|
return []
|
|
}
|
|
}
|
|
class LoopBodyModel extends AppNodeModel {
|
|
refreshBranch() {
|
|
// 更新节点连接边的path
|
|
this.incoming.edges.forEach((edge: any) => {
|
|
// 调用自定义的更新方案
|
|
edge.updatePathByAnchor()
|
|
})
|
|
this.outgoing.edges.forEach((edge: any) => {
|
|
edge.updatePathByAnchor()
|
|
})
|
|
}
|
|
getDefaultAnchor() {
|
|
const { id, x, y, width, height } = this
|
|
const showNode = this.properties.showNode === undefined ? true : this.properties.showNode
|
|
const anchors: any = []
|
|
anchors.push({
|
|
edgeAddable: false,
|
|
x: x,
|
|
y: y - height / 2 + 10,
|
|
id: `${id}_children`,
|
|
type: 'children',
|
|
})
|
|
|
|
return anchors
|
|
}
|
|
setHeight(height: number) {
|
|
this.properties['height'] = height
|
|
this.outgoing.edges.forEach((edge: any) => {
|
|
// 调用自定义的更新方案
|
|
edge.updatePathByAnchor()
|
|
})
|
|
this.incoming.edges.forEach((edge: any) => {
|
|
// 调用自定义的更新方案
|
|
edge.updatePathByAnchor()
|
|
})
|
|
}
|
|
}
|
|
export default {
|
|
type: 'loop-body-node',
|
|
model: LoopBodyModel,
|
|
view: LoopBodyNodeView,
|
|
}
|