tldraw/state/commands/stretch.ts
2021-06-24 13:34:43 +01:00

83 lines
2.5 KiB
TypeScript

import Command from './command'
import history from '../history'
import { StretchType, Data, Corner } from 'types'
import { deepClone, getCommonBounds, getPage, getSelectedShapes } from 'utils'
import { getShapeUtils } from 'state/shape-utils'
export default function stretchCommand(data: Data, type: StretchType): void {
const initialShapes = getSelectedShapes(data).map((shape) => deepClone(shape))
const snapshot = Object.fromEntries(
initialShapes.map((shape) => [
shape.id,
{
initialShape: shape,
initialBounds: getShapeUtils(shape).getBounds(shape),
},
])
)
const commonBounds = getCommonBounds(
...initialShapes.map((shape) => getShapeUtils(shape).getBounds(shape))
)
history.execute(
data,
new Command({
name: 'stretched_shapes',
category: 'canvas',
do(data) {
const { shapes } = getPage(data)
switch (type) {
case StretchType.Horizontal: {
Object.values(snapshot).forEach(
({ initialShape, initialBounds }) => {
const newBounds = { ...initialBounds }
newBounds.minX = commonBounds.minX
newBounds.width = commonBounds.width
newBounds.maxX = commonBounds.maxX
const shape = shapes[initialShape.id]
getShapeUtils(shape).transform(shape, newBounds, {
type: Corner.TopLeft,
scaleX: newBounds.width / initialBounds.width,
scaleY: 1,
initialShape,
transformOrigin: [0.5, 0.5],
})
}
)
break
}
case StretchType.Vertical: {
Object.values(snapshot).forEach(
({ initialShape, initialBounds }) => {
const newBounds = { ...initialBounds }
newBounds.minY = commonBounds.minY
newBounds.height = commonBounds.height
newBounds.maxY = commonBounds.maxY
const shape = shapes[initialShape.id]
getShapeUtils(shape).transform(shape, newBounds, {
type: Corner.TopLeft,
scaleX: 1,
scaleY: newBounds.height / initialBounds.height,
initialShape,
transformOrigin: [0.5, 0.5],
})
}
)
}
}
},
undo(data) {
const { shapes } = getPage(data)
initialShapes.forEach((shape) => (shapes[shape.id] = shape))
},
})
)
}