tldraw/state/code/index.ts

import { Mutable, Shape, ShapeUtility } from 'types'
import { createShape, getShapeUtils } from 'state/shape-utils'
import vec from 'utils/vec'

export const codeShapes = new Set<CodeShape<Shape>>([])

/**
 * A base class for code shapes. Note that creating a shape adds it to the
 * shape map, while deleting it removes it from the collected shapes set
 */
export default class CodeShape<T extends Shape> {
  private _shape: Mutable<T>
  private utils: ShapeUtility<T>

  constructor(props: T) {
    this._shape = createShape(props.type, props) as Mutable<T>
    this.utils = getShapeUtils<T>(this._shape)
    codeShapes.add(this)
  }

  export(): Mutable<T> {
    return { ...this._shape }
  }

  destroy(): void {
    codeShapes.delete(this)
  }

  moveTo(point: number[]): CodeShape<T> {
    this.utils.setProperty(this._shape, 'point', point)
    return this
  }

  translate(delta: number[]): CodeShape<T> {
    this.utils.setProperty(
      this._shape,
      'point',
      vec.add(this._shape.point, delta)
    )
    return this
  }

  rotate(rotation: number): CodeShape<T> {
    this.utils.setProperty(this._shape, 'rotation', rotation)
    return this
  }

  getBounds(): CodeShape<T> {
    this.utils.getBounds(this.shape)
    return this
  }

  hitTest(point: number[]): CodeShape<T> {
    this.utils.hitTest(this.shape, point)
    return this
  }

  get shape(): T {
    return this._shape
  }

  get point(): number[] {
    return [...this.shape.point]
  }

  get rotation(): number {
    return this.shape.rotation
  }
}