tldraw/state/commands/paste.ts

83 lines
2.1 KiB
TypeScript
Raw Normal View History

2021-06-18 10:14:23 +00:00
import Command from './command'
import history from '../history'
import { Data, Shape } from 'types'
import {
getChildIndexAbove,
getCommonBounds,
getCurrentCamera,
getPage,
getSelectedIds,
getSelectedShapes,
getViewport,
screenToWorld,
setSelectedIds,
setToArray,
} from 'utils/utils'
import { uniqueId } from 'utils/utils'
import { current } from 'immer'
import vec from 'utils/vec'
import { getShapeUtils } from 'lib/shape-utils'
import state from 'state/state'
export default function pasteCommand(data: Data, initialShapes: Shape[]) {
const { currentPageId } = data
const center = screenToWorld(
[window.innerWidth / 2, window.innerHeight / 2],
data
)
const bounds = getCommonBounds(
...initialShapes.map((shape) =>
getShapeUtils(shape).getRotatedBounds(shape)
)
)
const topLeft = vec.sub(center, [bounds.width / 2, bounds.height / 2])
const newIdMap = Object.fromEntries(
initialShapes.map((shape) => [shape.id, uniqueId()])
)
const oldSelectedIds = setToArray(getSelectedIds(data))
history.execute(
data,
new Command({
name: 'pasting_new_shapes',
category: 'canvas',
manualSelection: true,
do(data) {
const { shapes } = getPage(data, currentPageId)
let childIndex =
(state.values.currentShapes[state.values.currentShapes.length - 1]
?.childIndex || 0) + 1
for (const shape of initialShapes) {
const topLeftOffset = vec.sub(shape.point, [bounds.minX, bounds.minY])
const newId = newIdMap[shape.id]
shapes[newId] = {
...shape,
id: newId,
parentId: oldSelectedIds[shape.parentId] || data.currentPageId,
childIndex: childIndex++,
point: vec.add(topLeft, topLeftOffset),
}
}
setSelectedIds(data, Object.values(newIdMap))
},
undo(data) {
const { shapes } = getPage(data, currentPageId)
Object.values(newIdMap).forEach((id) => delete shapes[id])
setSelectedIds(data, oldSelectedIds)
},
})
)
}