import { BaseEdge } from '@antv/g6';
import { Line } from '@antv/g';

import type { BaseEdgeStyleProps } from '@antv/g6';

class ExtendBaseEdge extends BaseEdge {
  // 重写 getKeyStyle 方法
  protected getKeyStyle(attributes: Required<BaseEdgeStyleProps>) {
    return { ...super.getKeyStyle(attributes), lineWidth: 2, stroke: '#000' };
  }

  // 实现 getKeyPath 方法
  protected getKeyPath(attributes) {
    const { sourceNode, targetNode } = this;
    const [x1, y1] = sourceNode.getPosition();
    const [x2, y2] = targetNode.getPosition();

    return [
      ['M', x1, y1],
      ['L', x2, y2],
    ];
  }
}