@logicflow/core/src/model/edge/LineEdgeModel.ts

LogicFlow, help you quickly create flowcharts

import { cloneDeep } from 'lodash-es'
import LogicFlow from '../../LogicFlow'
import BaseEdgeModel from './BaseEdgeModel'
import { ModelType } from '../../constant'

import Point = LogicFlow.Point

export class LineEdgeModel extends BaseEdgeModel {
  modelType = ModelType.LINE_EDGE
  getEdgeStyle() {
    const { line } = this.graphModel.theme
    const style = super.getEdgeStyle()
    const { style: customStyle = {} } = this.properties
    return {
      ...style,
      ...cloneDeep(line),
      ...cloneDeep(customStyle),
    }
  }
  initEdgeData(data: LogicFlow.EdgeConfig): void {
    super.initEdgeData(data)
    this.points = this.getPath([this.startPoint, this.endPoint])
  }
  getPath(points: Point[]): string {
    const [start, end] = points
    return `${start.x},${start.y} ${end.x},${end.y}`
  }
  getTextPosition(): Point {
    return {
      x: (this.startPoint.x + this.endPoint.x) / 2,
      y: (this.startPoint.y + this.endPoint.y) / 2,
    }
  }
}

export default LineEdgeModel