/*
 * @Date: 2025-03-13 16:17:13
 * @LastEditors: hookehuyr hookehuyr@gmail.com
 * @LastEditTime: 2025-03-13 16:21:47
 * @FilePath: /logic-flow2/src/views/adv-menu/customEdge.js
 * @Description: 文件描述
 */
import { PolylineEdge, PolylineEdgeModel } from "@logicflow/core";

class CustomModel extends PolylineEdgeModel {
  setAttributes() {
    // 右键菜单
    this.menu = [
      {
        className: "lf-menu-delete",
        icon: true,
        callback: (edge) => {
          const comfirm = window.confirm("你确定要删除吗？");
          comfirm && this.graphModel.deleteEdgeById(edge.id);
        },
      },
    ];
  }
}

export default {
  type: "custom_edge",
  view: PolylineEdge,
  model: CustomModel,
};