import { Curve, Path } from '@antv/x6-geometry'
import { Connector } from './index'

export interface SmoothConnectorOptions extends Connector.BaseOptions {
  direction?: 'H' | 'V'
}

export const smooth: Connector.Definition<SmoothConnectorOptions> = function (
  sourcePoint,
  targetPoint,
  routePoints,
  options = {},
) {
  let path
  let direction = options.direction

  if (routePoints && routePoints.length !== 0) {
    const points = [sourcePoint, ...routePoints, targetPoint]
    const curves = Curve.throughPoints(points)
    path = new Path(curves)
  } else {
    // If we have no route, use a default cubic bezier curve, cubic bezier
    // requires two control points, the control points have `x` midway
    // between source and target. This produces an S-like curve.

    path = new Path()
    path.appendSegment(Path.createSegment('M', sourcePoint))

    if (!direction) {
      direction =
        Math.abs(sourcePoint.x - targetPoint.x) >=
        Math.abs(sourcePoint.y - targetPoint.y)
          ? 'H'
          : 'V'
    }

    if (direction === 'H') {
      const controlPointX = (sourcePoint.x + targetPoint.x) / 2
      path.appendSegment(
        Path.createSegment(
          'C',
          controlPointX,
          sourcePoint.y,
          controlPointX,
          targetPoint.y,
          targetPoint.x,
          targetPoint.y,
        ),
      )
    } else {
      const controlPointY = (sourcePoint.y + targetPoint.y) / 2
      path.appendSegment(
        Path.createSegment(
          'C',
          sourcePoint.x,
          controlPointY,
          targetPoint.x,
          controlPointY,
          targetPoint.x,
          targetPoint.y,
        ),
      )
    }
  }

  return options.raw ? path : path.serialize()
}
