tldraw/state/commands/align.ts
2021-06-02 16:05:44 +01:00

102 lines
3.2 KiB
TypeScript

import Command from './command'
import history from '../history'
import { AlignType, Data } from 'types'
import { getCommonBounds, getPage, getSelectedShapes } from 'utils/utils'
import { getShapeUtils } from 'lib/shape-utils'
export default function alignCommand(data: Data, type: AlignType) {
const { currentPageId } = data
const selectedShapes = getSelectedShapes(data)
const entries = selectedShapes.map(
(shape) => [shape.id, getShapeUtils(shape).getBounds(shape)] as const
)
const boundsForShapes = Object.fromEntries(entries)
const commonBounds = getCommonBounds(...entries.map((entry) => entry[1]))
const midX = commonBounds.minX + commonBounds.width / 2
const midY = commonBounds.minY + commonBounds.height / 2
history.execute(
data,
new Command({
name: 'aligned',
category: 'canvas',
do(data) {
const { shapes } = getPage(data, currentPageId)
switch (type) {
case AlignType.Top: {
for (let id in boundsForShapes) {
const shape = shapes[id]
getShapeUtils(shape).setProperty(shape, 'point', [
shape.point[0],
commonBounds.minY,
])
}
break
}
case AlignType.CenterVertical: {
for (let id in boundsForShapes) {
const shape = shapes[id]
getShapeUtils(shape).setProperty(shape, 'point', [
shape.point[0],
midY - boundsForShapes[id].height / 2,
])
}
break
}
case AlignType.Bottom: {
for (let id in boundsForShapes) {
const shape = shapes[id]
getShapeUtils(shape).setProperty(shape, 'point', [
shape.point[0],
commonBounds.maxY - boundsForShapes[id].height,
])
}
break
}
case AlignType.Left: {
for (let id in boundsForShapes) {
const shape = shapes[id]
getShapeUtils(shape).setProperty(shape, 'point', [
commonBounds.minX,
shape.point[1],
])
}
break
}
case AlignType.CenterHorizontal: {
for (let id in boundsForShapes) {
const shape = shapes[id]
getShapeUtils(shape).setProperty(shape, 'point', [
midX - boundsForShapes[id].width / 2,
shape.point[1],
])
}
break
}
case AlignType.Right: {
for (let id in boundsForShapes) {
const shape = shapes[id]
getShapeUtils(shape).setProperty(shape, 'point', [
commonBounds.maxX - boundsForShapes[id].width,
shape.point[1],
])
}
break
}
}
},
undo(data) {
const { shapes } = getPage(data, currentPageId)
for (let id in boundsForShapes) {
const shape = shapes[id]
const initialBounds = boundsForShapes[id]
getShapeUtils(shape).setProperty(shape, 'point', [
initialBounds.minX,
initialBounds.minY,
])
}
},
})
)
}