Fixed node size error for multi-step transaction.

Author: VV-EE

src/__tests__/withSuggestChanges.test.ts

@@ -1,4 +1,3 @@
- 
 /* eslint-disable @typescript-eslint/no-non-null-assertion */
 import { EditorState } from "prosemirror-state";
 import { Fragment, Slice } from "prosemirror-model";
@@ -38,26 +37,30 @@ describe("withSuggestChanges", () => {
     originalTransaction.step(step1);
 
     // Step 2: Replace first "test" with "WORK" in second paragraph
-    // Use the updated document from the transaction
+    // Create step with original positions, then map it
     const step2 = replaceStep(
-      originalTransaction.doc,
+      doc,
       doc.tag["c"]!,
       doc.tag["d"],
       new Slice(Fragment.from(testBuilders.schema.text("WORK")), 0, 0),
     ) as ReplaceStep | null;
     assert(step2, "Could not create step2");
-    originalTransaction.step(step2);
+    const mappedStep2 = step2.map(originalTransaction.mapping);
+    assert(mappedStep2, "Could not map step2");
+    originalTransaction.step(mappedStep2);
 
     // Step 3: Replace second "test" with "WORK" in second paragraph
-    // Use the updated document from the transaction
+    // Create step with original positions, then map it
     const step3 = replaceStep(
-      originalTransaction.doc,
+      doc,
       doc.tag["e"]!,
       doc.tag["f"],
       new Slice(Fragment.from(testBuilders.schema.text("WORK")), 0, 0),
     ) as ReplaceStep | null;
     assert(step3, "Could not create step3");
-    originalTransaction.step(step3);
+    const mappedStep3 = step3.map(originalTransaction.mapping);
+    assert(mappedStep3, "Could not map step3");
+    originalTransaction.step(mappedStep3);
 
     // Verify the transaction produces expected result
     const originalTrState = editorState.apply(originalTransaction);
@@ -72,6 +75,85 @@ describe("withSuggestChanges", () => {
     );
     const newState = editorState.apply(suggestedTr);
 
+    // Should pass without error - document should be modified
+    expect(newState.doc.toJSON()).not.toEqual(doc.toJSON());
+  });
+  it("shouldn't throw nodeSize errors on multiline edits", () => {
+    // Create document with all tags for the replacements
+    const doc = testBuilders.doc(
+      testBuilders.paragraph({ id: null }, "<a>test<b> <c>test<d>"),
+      testBuilders.paragraph({ id: null }, "<e>test<f> <g>test<h>"),
+    ) as TaggedNode;
+
+    const editorState = EditorState.create({
+      doc,
+      plugins: [suggestChanges()],
+    });
+
+    const originalTransaction = editorState.tr;
+
+    // Step 1: Replace first "test" with "WORKKKKKKK"
+    const step1 = replaceStep(
+      doc,
+      doc.tag["a"]!,
+      doc.tag["b"],
+      new Slice(Fragment.from(testBuilders.schema.text("WORKKKKKKK")), 0, 0),
+    ) as ReplaceStep | null;
+    assert(step1, "Could not create step1");
+    originalTransaction.step(step1);
+
+    // Step 2: Replace second "test" with "WORKKKKKKK"
+    // Create step with original positions, then map it
+    const step2 = replaceStep(
+      doc,
+      doc.tag["c"]!,
+      doc.tag["d"],
+      new Slice(Fragment.from(testBuilders.schema.text("WORKKKKKKK")), 0, 0),
+    ) as ReplaceStep | null;
+    assert(step2, "Could not create step2");
+    const mappedStep2 = step2.map(originalTransaction.mapping);
+    assert(mappedStep2, "Could not map step2");
+    originalTransaction.step(mappedStep2);
+
+    // Step 3: Replace third "test" with "WORKKKKKKK"
+    // Create step with original positions, then map it
+    const step3 = replaceStep(
+      doc,
+      doc.tag["e"]!,
+      doc.tag["f"],
+      new Slice(Fragment.from(testBuilders.schema.text("WORKKKKKKK")), 0, 0),
+    ) as ReplaceStep | null;
+    assert(step3, "Could not create step3");
+    const mappedStep3 = step3.map(originalTransaction.mapping);
+    assert(mappedStep3, "Could not map step3");
+    originalTransaction.step(mappedStep3);
+
+    // Step 4: Replace fourth "test" with "WORKKKKKKK"
+    // Create step with original positions, then map it
+    const step4 = replaceStep(
+      doc,
+      doc.tag["g"]!,
+      doc.tag["h"],
+      new Slice(Fragment.from(testBuilders.schema.text("WORKKKKKKK")), 0, 0),
+    ) as ReplaceStep | null;
+    assert(step4, "Could not create step4");
+    const mappedStep4 = step4.map(originalTransaction.mapping);
+    assert(mappedStep4, "Could not map step4");
+    originalTransaction.step(mappedStep4);
+
+    // Verify the transaction produces expected result
+    const originalTrState = editorState.apply(originalTransaction);
+    expect(originalTrState.doc.textContent).toEqual(
+      "WORKKKKKKK WORKKKKKKKWORKKKKKKK WORKKKKKKK",
+    );
+
+    // Transform to suggestion transaction
+    const suggestedTr = transformToSuggestionTransaction(
+      originalTransaction,
+      editorState,
+    );
+    const newState = editorState.apply(suggestedTr);
+
     // Should pass without error - document should be modified
     expect(newState.doc.toJSON()).not.toEqual(doc.toJSON());
   });

src/rebasePos.ts

@@ -9,6 +9,7 @@ import { type Step } from "prosemirror-transform";
  */
 export function rebasePos(pos: number, back: Step[], forth: Step[]) {
   const reset = back
+    .slice()
     .reverse()
     .reduce((acc, step) => step.getMap().invert().map(acc), pos);
   return forth.reduce((acc, step) => step.getMap().map(acc), reset);

src/withSuggestChanges.ts

@@ -66,6 +66,7 @@ function getStepHandler<S extends Step>(step: S): StepHandler<S> {
     prevSteps: Step[],
   ) => {
     const reset = prevSteps
+      .slice()
       .reverse()
       .reduce<Step | null>(
         (acc, step) => acc?.map(step.getMap().invert()) ?? null,
@@ -224,6 +225,7 @@ export function withSuggestChanges(
       isSuggestChangesEnabled(this.state) &&
       !tr.getMeta("history$") &&
       !tr.getMeta("collab$") &&
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
       !ySyncMeta.isUndoRedoOperation &&
       !ySyncMeta.isChangeOrigin &&
       !("skip" in (tr.getMeta(suggestChangesKey) ?? {}))