Add rotate box brush command

src/commands/rotate-box-brush-command.ts

@@ -0,0 +1,79 @@
+import type { ToolMode } from "../core/tool-mode";
+
+import { createOpaqueId } from "../core/ids";
+import type { EditorSelection } from "../core/selection";
+import type { Vec3 } from "../core/vector";
+
+import { cloneSelectionForCommand, getBoxBrushOrThrow, replaceBrush, setSingleBrushSelection } from "./brush-command-helpers";
+import type { EditorCommand } from "./command";
+
+interface RotateBoxBrushCommandOptions {
+  brushId: string;
+  rotationDegrees: Vec3;
+  label?: string;
+}
+
+export function createRotateBoxBrushCommand(options: RotateBoxBrushCommandOptions): EditorCommand {
+  let previousRotationDegrees: Vec3 | null = null;
+  let previousSelection: EditorSelection | null = null;
+  let previousToolMode: ToolMode | null = null;
+
+  return {
+    id: createOpaqueId("command"),
+    label: options.label ?? "Rotate box brush",
+    execute(context) {
+      const currentDocument = context.getDocument();
+      const brush = getBoxBrushOrThrow(currentDocument, options.brushId);
+
+      if (previousRotationDegrees === null) {
+        previousRotationDegrees = {
+          ...brush.rotationDegrees
+        };
+      }
+
+      if (previousSelection === null) {
+        previousSelection = cloneSelectionForCommand(context.getSelection());
+      }
+
+      if (previousToolMode === null) {
+        previousToolMode = context.getToolMode();
+      }
+
+      context.setDocument(
+        replaceBrush(currentDocument, {
+          ...brush,
+          rotationDegrees: {
+            ...options.rotationDegrees
+          }
+        })
+      );
+      context.setSelection(setSingleBrushSelection(options.brushId));
+      context.setToolMode("select");
+    },
+    undo(context) {
+      if (previousRotationDegrees === null) {
+        return;
+      }
+
+      const currentDocument = context.getDocument();
+      const brush = getBoxBrushOrThrow(currentDocument, options.brushId);
+
+      context.setDocument(
+        replaceBrush(currentDocument, {
+          ...brush,
+          rotationDegrees: {
+            ...previousRotationDegrees
+          }
+        })
+      );
+
+      if (previousSelection !== null) {
+        context.setSelection(previousSelection);
+      }
+
+      if (previousToolMode !== null) {
+        context.setToolMode(previousToolMode);
+      }
+    }
+  };
+}