tldraw/state/sessions/direction-session.ts

import { Data, LineShape, RayShape } from 'types'
import vec from 'utils/vec'
import BaseSession from './base-session'
import commands from 'state/commands'
import { current } from 'immer'
import { getPage, getSelectedIds } from 'utils/utils'

export default class DirectionSession extends BaseSession {
  delta = [0, 0]
  origin: number[]
  snapshot: DirectionSnapshot

  constructor(data: Data, point: number[]) {
    super(data)
    this.origin = point
    this.snapshot = getDirectionSnapshot(data)
  }

  update(data: Data, point: number[]) {
    const { shapes } = this.snapshot

    const page = getPage(data)

    for (let { id } of shapes) {
      const shape = page.shapes[id] as RayShape | LineShape

      shape.direction = vec.uni(vec.vec(shape.point, point))
    }
  }

  cancel(data: Data) {
    const page = getPage(data, this.snapshot.currentPageId)

    for (let { id, direction } of this.snapshot.shapes) {
      const shape = page.shapes[id] as RayShape | LineShape
      shape.direction = direction
    }
  }

  complete(data: Data) {
    commands.direct(data, this.snapshot, getDirectionSnapshot(data))
  }
}

export function getDirectionSnapshot(data: Data) {
  const { shapes } = getPage(current(data))

  let snapshapes: { id: string; direction: number[] }[] = []

  getSelectedIds(data).forEach((id) => {
    const shape = shapes[id]
    if ('direction' in shape) {
      snapshapes.push({ id: shape.id, direction: shape.direction })
    }
  })

  return {
    currentPageId: data.currentPageId,
    shapes: snapshapes,
  }
}

export type DirectionSnapshot = ReturnType<typeof getDirectionSnapshot>