/*
 * @Date: 2025-03-12 17:05:09
 * @LastEditors: hookehuyr hookehuyr@gmail.com
 * @LastEditTime: 2025-03-12 17:06:34
 * @FilePath: /logic-flow2/src/views/theme/customEdge.js
 * @Description: 文件描述
 */
import { PolylineEdge, PolylineEdgeModel } from '@logicflow/core';

class CustomEdgeModel extends PolylineEdgeModel {
  // 始终显示锚点
  getEdgeStyle() {
    const style = super.getEdgeStyle();
    return {
      ...style,
      hideAnchor: false  // 强制显示锚点
    };
  }
}

class CustomEdge extends PolylineEdge {
  // 可选：自定义锚点渲染方式
  getEndArrow() {
    // 保持原有箭头逻辑
    return super.getEndArrow();
  }
}

export default {
  type: 'custom-edge',
  model: CustomEdgeModel,
  view: CustomEdge,
}