webeditor3d/src/commands/resize-box-brush-command.ts

import type { ToolMode } from "../core/tool-mode";
import { DEFAULT_GRID_SIZE, snapPositiveSizeToGrid } from "../geometry/grid-snapping";
import { cloneBoxBrushGeometry, scaleBoxBrushGeometryToSize } from "../document/brushes";

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 ResizeBoxBrushCommandOptions {
  brushId: string;
  size: Vec3;
  snapToGrid?: boolean;
  gridSize?: number;
  label?: string;
}

export function createResizeBoxBrushCommand(options: ResizeBoxBrushCommandOptions): EditorCommand {
  const resolvedSize =
    options.snapToGrid === false ? options.size : snapPositiveSizeToGrid(options.size, options.gridSize ?? DEFAULT_GRID_SIZE);

  let previousSize: Vec3 | null = null;
  let previousGeometry: ReturnType<typeof cloneBoxBrushGeometry> | null = null;
  let previousSelection: EditorSelection | null = null;
  let previousToolMode: ToolMode | null = null;

  return {
    id: createOpaqueId("command"),
    label: options.label ?? "Resize box brush",
    execute(context) {
      const currentDocument = context.getDocument();
      const brush = getBoxBrushOrThrow(currentDocument, options.brushId);

      if (previousSize === null) {
        previousSize = {
          ...brush.size
        };
        previousGeometry = cloneBoxBrushGeometry(brush.geometry);
      }

      if (previousSelection === null) {
        previousSelection = cloneSelectionForCommand(context.getSelection());
      }

      if (previousToolMode === null) {
        previousToolMode = context.getToolMode();
      }

      const nextGeometry = scaleBoxBrushGeometryToSize(brush.geometry, resolvedSize);

      context.setDocument(
        replaceBrush(currentDocument, {
          ...brush,
          size: {
            ...resolvedSize
          },
          geometry: nextGeometry
        })
      );
      context.setSelection(setSingleBrushSelection(options.brushId));
      context.setToolMode("select");
    },
    undo(context) {
      if (previousSize === null || previousGeometry === null) {
        return;
      }

      const currentDocument = context.getDocument();
      const brush = getBoxBrushOrThrow(currentDocument, options.brushId);

      context.setDocument(
        replaceBrush(currentDocument, {
          ...brush,
          size: {
            ...previousSize
          },
          geometry: cloneBoxBrushGeometry(previousGeometry)
        })
      );

      if (previousSelection !== null) {
        context.setSelection(previousSelection);
      }

      if (previousToolMode !== null) {
        context.setToolMode(previousToolMode);
      }
    }
  };
}