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,
}