63 lines
1.6 KiB
TypeScript
63 lines
1.6 KiB
TypeScript
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>
|