@antv/x6/src/registry/node-anchor/util.ts

JavaScript diagramming library that uses SVG and HTML for rendering.

import { NumberExt } from '../../util'
import { Point } from '../../geometry'
import { EdgeView } from '../../view'

export interface ResolveOptions {
  fixedAt?: number | string
}

// eslint-disable-next-line
export function resolve<S extends Function, T>(fn: S): T {
  return function (
    this: EdgeView,
    view: EdgeView,
    magnet: SVGElement,
    ref: any,
    options: ResolveOptions,
  ) {
    if (ref instanceof Element) {
      const refView = this.graph.renderer.findViewByElem(ref)
      let refPoint
      if (refView) {
        if (refView.isEdgeElement(ref)) {
          const distance = options.fixedAt != null ? options.fixedAt : '50%'
          refPoint = getPointAtEdge(refView as EdgeView, distance)
        } else {
          refPoint = refView.getBBoxOfElement(ref).getCenter()
        }
      } else {
        refPoint = new Point()
      }
      return fn.call(this, view, magnet, refPoint, options)
    }
    return fn.apply(this, arguments) // eslint-disable-line
  } as any as T
}

export function getPointAtEdge(edgeView: EdgeView, value: string | number) {
  const isPercentage = NumberExt.isPercentage(value)
  const num = typeof value === 'string' ? parseFloat(value) : value
  if (isPercentage) {
    return edgeView.getPointAtRatio(num / 100)
  }
  return edgeView.getPointAtLength(num)
}