webstudio/packages/project/src/shared/tree-utils/insert-instance.ts

import type { Instance } from "@webstudio-is/project-build";

export type InstanceInsertionSpec = {
  parentId: Instance["id"];
  position: number | "end";
};

export const insertInstanceMutable = (
  rootInstance: Instance,
  instance: Instance,
  spec: undefined | InstanceInsertionSpec
): boolean => {
  if (spec === undefined) {
    return false;
  }
  if (spec.parentId !== rootInstance.id) {
    for (const child of rootInstance.children) {
      if (child.type === "text") {
        continue;
      }
      const hasInserted = insertInstanceMutable(child, instance, spec);
      if (hasInserted === true) {
        return true;
      }
    }
    return false;
  }

  // Inserting inside the selected/hovered instance.
  if (typeof spec.position === "number") {
    rootInstance.children.splice(spec.position, 0, instance);
    return true;
  }

  // Inserting after all children
  if (spec.position === "end") {
    rootInstance.children.push(instance);
    return true;
  }

  return true;
};