Potentially fixed undo/redo error

Author: PatrickSachs

src/commands/AddTreeElementCommand.ts

@@ -2,23 +2,24 @@ import TreeElement from "./../gameobjects/IBranchContainer";
 import { ITreeElementDetails } from "@/gameobjects/IBranchContainer";
 import TreeGameObject from "@/gameobjects/TreeGameObject";
 import ICommand from "@/commands/ICommand";
+import SlowTreeError from "@/errors/SlowTreeError";
 
 export default class AddTreeElementCommand implements ICommand {
-    private element: TreeElement | null = null;
-
     constructor(private tree: TreeGameObject, private parent: string, private details: ITreeElementDetails) {
     }
 
     do(): void {
         const owner = this.tree.find(this.parent);
         if (owner) {
-            this.element = owner.addTreeElement(this.details);
+            owner.addTreeElement(this.details).id;
         }
     }
 
     undo(): void {
-        if (this.element) {
-            this.element.destroy();
+        const element = this.tree.find(this.details.id);
+        if (!element) {
+            throw new SlowTreeError("Could not find element with ID " + this.details.id);
         }
+        element.destroy();
     }
 }