72 lines
1.6 KiB
TypeScript
72 lines
1.6 KiB
TypeScript
import Command from './command'
|
|
import history from '../history'
|
|
import { CodeControl, Data, Shape } from 'types'
|
|
import { current } from 'immer'
|
|
import { getPage, getSelectedIds, setSelectedIds } from 'utils/utils'
|
|
|
|
export default function generateCommand(
|
|
data: Data,
|
|
currentPageId: string,
|
|
generatedShapes: Shape[]
|
|
) {
|
|
const cData = current(data)
|
|
const page = getPage(cData)
|
|
|
|
const currentShapes = page.shapes
|
|
|
|
const prevGeneratedShapes = Object.values(currentShapes).filter(
|
|
(shape) => shape.isGenerated
|
|
)
|
|
|
|
// Remove previous generated shapes
|
|
for (let id in currentShapes) {
|
|
if (currentShapes[id].isGenerated) {
|
|
delete currentShapes[id]
|
|
}
|
|
}
|
|
|
|
// Add new ones
|
|
for (let shape of generatedShapes) {
|
|
currentShapes[shape.id] = shape
|
|
}
|
|
|
|
history.execute(
|
|
data,
|
|
new Command({
|
|
name: 'generate_shapes',
|
|
category: 'canvas',
|
|
do(data) {
|
|
const { shapes } = getPage(data)
|
|
|
|
setSelectedIds(data, [])
|
|
|
|
// Remove previous generated shapes
|
|
for (let id in shapes) {
|
|
if (shapes[id].isGenerated) {
|
|
delete shapes[id]
|
|
}
|
|
}
|
|
|
|
// Add new generated shapes
|
|
for (let shape of generatedShapes) {
|
|
shapes[shape.id] = shape
|
|
}
|
|
},
|
|
undo(data) {
|
|
const { shapes } = getPage(data)
|
|
|
|
// Remove generated shapes
|
|
for (let id in shapes) {
|
|
if (shapes[id].isGenerated) {
|
|
delete shapes[id]
|
|
}
|
|
}
|
|
|
|
// Restore previous generated shapes
|
|
for (let shape of prevGeneratedShapes) {
|
|
shapes[shape.id] = shape
|
|
}
|
|
},
|
|
})
|
|
)
|
|
}
|