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 } 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: "translate_shapes",
|
|
category: "canvas",
|
|
do(data) {
|
|
const { shapes } = getPage(data)
|
|
|
|
data.selectedIds.clear()
|
|
|
|
// 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
|
|
}
|
|
},
|
|
})
|
|
)
|
|
}
|