webeditor3d/src/commands/duplicate-selection-command.js

import { cloneModelInstance } from "../assets/model-instances";
import { createOpaqueId } from "../core/ids";
import { cloneEditorSelection } from "../core/selection";
import { cloneBoxBrush } from "../document/brushes";
import { cloneEntityInstance } from "../entities/entity-instances";

function duplicateBrush(brush) {
  const duplicatedBrush = cloneBoxBrush(brush);
  duplicatedBrush.id = createOpaqueId("brush");
  return duplicatedBrush;
}

function duplicateEntity(entity) {
  const duplicatedEntity = cloneEntityInstance(entity);
  duplicatedEntity.id = createOpaqueId(`entity-${duplicatedEntity.kind}`);
  return duplicatedEntity;
}

function duplicateModelInstance(modelInstance) {
  const duplicatedModelInstance = cloneModelInstance(modelInstance);
  duplicatedModelInstance.id = createOpaqueId("model-instance");
  return duplicatedModelInstance;
}

function resolveDuplicatableBrushIds(selection) {
  switch (selection.kind) {
    case "brushes":
      return selection.ids;
    case "brushFace":
    case "brushEdge":
    case "brushVertex":
      return [selection.brushId];
    default:
      return null;
  }
}

function createDuplicateSelectionResult(currentDocument, selection) {
  const duplicatableBrushIds = resolveDuplicatableBrushIds(selection);

  if (duplicatableBrushIds !== null) {
    if (duplicatableBrushIds.length === 0) {
      throw new Error("Select at least one whitebox solid to duplicate.");
    }

    const duplicatedBrushes = duplicatableBrushIds.map((brushId) => {
      const sourceBrush = currentDocument.brushes[brushId];

      if (sourceBrush === undefined) {
        throw new Error(`Box brush ${brushId} does not exist.`);
      }

      if (sourceBrush.kind !== "box") {
        throw new Error(`Brush ${brushId} is not a supported box brush.`);
      }

      return duplicateBrush(sourceBrush);
    });

    return {
      selection: {
        kind: "brushes",
        ids: duplicatedBrushes.map((brush) => brush.id)
      },
      brushes: duplicatedBrushes,
      entities: null,
      modelInstances: null
    };
  }

  if (selection.kind === "entities") {
    if (selection.ids.length === 0) {
      throw new Error("Select at least one entity to duplicate.");
    }

    const duplicatedEntities = selection.ids.map((entityId) => {
      const sourceEntity = currentDocument.entities[entityId];

      if (sourceEntity === undefined) {
        throw new Error(`Entity ${entityId} does not exist.`);
      }

      return duplicateEntity(sourceEntity);
    });

    return {
      selection: {
        kind: "entities",
        ids: duplicatedEntities.map((entity) => entity.id)
      },
      brushes: null,
      entities: duplicatedEntities,
      modelInstances: null
    };
  }

  if (selection.kind === "modelInstances") {
    if (selection.ids.length === 0) {
      throw new Error("Select at least one model instance to duplicate.");
    }

    const duplicatedModelInstances = selection.ids.map((modelInstanceId) => {
      const sourceModelInstance = currentDocument.modelInstances[modelInstanceId];

      if (sourceModelInstance === undefined) {
        throw new Error(`Model instance ${modelInstanceId} does not exist.`);
      }

      return duplicateModelInstance(sourceModelInstance);
    });

    return {
      selection: {
        kind: "modelInstances",
        ids: duplicatedModelInstances.map((modelInstance) => modelInstance.id)
      },
      brushes: null,
      entities: null,
      modelInstances: duplicatedModelInstances
    };
  }

  throw new Error("Selection must contain whitebox solids, entities, or model instances to duplicate.");
}

export function createDuplicateSelectionCommand() {
  let previousSelection = null;
  let previousToolMode = null;
  let duplicateSelectionResult = null;

  return {
    id: createOpaqueId("command"),
    label: "Duplicate selection",
    execute(context) {
      const currentDocument = context.getDocument();

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

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

      if (duplicateSelectionResult === null) {
        duplicateSelectionResult = createDuplicateSelectionResult(currentDocument, context.getSelection());
      }

      if (duplicateSelectionResult.brushes !== null) {
        context.setDocument({
          ...currentDocument,
          brushes: {
            ...currentDocument.brushes,
            ...Object.fromEntries(duplicateSelectionResult.brushes.map((brush) => [brush.id, cloneBoxBrush(brush)]))
          }
        });
      } else if (duplicateSelectionResult.entities !== null) {
        context.setDocument({
          ...currentDocument,
          entities: {
            ...currentDocument.entities,
            ...Object.fromEntries(duplicateSelectionResult.entities.map((entity) => [entity.id, cloneEntityInstance(entity)]))
          }
        });
      } else if (duplicateSelectionResult.modelInstances !== null) {
        context.setDocument({
          ...currentDocument,
          modelInstances: {
            ...currentDocument.modelInstances,
            ...Object.fromEntries(
              duplicateSelectionResult.modelInstances.map((modelInstance) => [modelInstance.id, cloneModelInstance(modelInstance)])
            )
          }
        });
      }

      context.setSelection(cloneEditorSelection(duplicateSelectionResult.selection));
      context.setToolMode("select");
    },
    undo(context) {
      if (duplicateSelectionResult === null) {
        return;
      }

      const currentDocument = context.getDocument();

      if (duplicateSelectionResult.brushes !== null) {
        const nextBrushes = {
          ...currentDocument.brushes
        };

        for (const duplicatedBrush of duplicateSelectionResult.brushes) {
          delete nextBrushes[duplicatedBrush.id];
        }

        context.setDocument({
          ...currentDocument,
          brushes: nextBrushes
        });
      } else if (duplicateSelectionResult.entities !== null) {
        const nextEntities = {
          ...currentDocument.entities
        };

        for (const duplicatedEntity of duplicateSelectionResult.entities) {
          delete nextEntities[duplicatedEntity.id];
        }

        context.setDocument({
          ...currentDocument,
          entities: nextEntities
        });
      } else if (duplicateSelectionResult.modelInstances !== null) {
        const nextModelInstances = {
          ...currentDocument.modelInstances
        };

        for (const duplicatedModelInstance of duplicateSelectionResult.modelInstances) {
          delete nextModelInstances[duplicatedModelInstance.id];
        }

        context.setDocument({
          ...currentDocument,
          modelInstances: nextModelInstances
        });
      }

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

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