feat: use locations for insertBlocks operations

Author: nperez0111

packages/core/src/api/blockManipulation/commands/insertBlocks/insertBlocks.ts

@@ -2,15 +2,15 @@ import { Fragment, Slice } from "prosemirror-model";
 import type { Transaction } from "prosemirror-state";
 import { ReplaceStep } from "prosemirror-transform";
 import { Block, PartialBlock } from "../../../../blocks/defaultBlocks.js";
+import { resolveLocation } from "../../../../locations/location.js";
+import { Location } from "../../../../locations/types.js";
 import {
-  BlockIdentifier,
   BlockSchema,
   InlineContentSchema,
   StyleSchema,
 } from "../../../../schema/index.js";
 import { blockToNode } from "../../../nodeConversions/blockToNode.js";
 import { nodeToBlock } from "../../../nodeConversions/nodeToBlock.js";
-import { getNodeById } from "../../../nodeUtil.js";
 import { getPmSchema } from "../../../pmUtil.js";
 
 export function insertBlocks<
@@ -20,25 +20,16 @@ export function insertBlocks<
 >(
   tr: Transaction,
   blocksToInsert: PartialBlock<BSchema, I, S>[],
-  referenceBlock: BlockIdentifier,
+  referenceBlock: Location,
   placement: "before" | "after" = "before",
 ): Block<BSchema, I, S>[] {
-  const id =
-    typeof referenceBlock === "string" ? referenceBlock : referenceBlock.id;
   const pmSchema = getPmSchema(tr);
   const nodesToInsert = blocksToInsert.map((block) =>
     blockToNode(block, pmSchema),
   );
 
-  const posInfo = getNodeById(id, tr.doc);
-  if (!posInfo) {
-    throw new Error(`Block with ID ${id} not found`);
-  }
-
-  let pos = posInfo.posBeforeNode;
-  if (placement === "after") {
-    pos += posInfo.node.nodeSize;
-  }
+  const resolved = resolveLocation(tr.doc, referenceBlock);
+  const pos = placement === "after" ? resolved.head + 1 : resolved.anchor;
 
   tr.step(
     new ReplaceStep(pos, pos, new Slice(Fragment.from(nodesToInsert), 0, 0)),