fix: tree event handle

Author: zxch3n

packages/core/src/core/loroEventApply.ts

@@ -140,7 +140,7 @@ function applySingleEventToDraft(
                 target = parent && key !== undefined ? getAt(parent, key)! : [];
             }
             if (isJSONArray(target)) {
-                applyTreeDiff(target, e.diff.diff, ignoreSet);
+                applyTreeDiff(target, e.diff.diff);
                 // Invalidate cache for this roots array after structural change
                 ROOTS_TREE_INDEX_CACHE.delete(target as JSONValue[]);
             }
@@ -423,7 +423,6 @@ function applyTreeDiff(
               oldIndex: number;
           }
     >,
-    ignoreSet: Set<ContainerID>,
 ) {
     type Node = StateTreeNode;
 
@@ -443,7 +442,6 @@ function applyTreeDiff(
             };
             const idx = clampIndex(d.index, arr.length + 1);
             arr.splice(idx, 0, node);
-            ignoreSet.add(`cid:${d.target}:Map`);
         } else if (d.action === "delete") {
             const arr = getChildrenArray(d.oldParent);
             if (!arr) continue;

packages/core/src/core/mirror.ts

@@ -478,10 +478,8 @@ export class Mirror<S extends SchemaType> {
         this.syncing = true;
         try {
             // Incrementally update state using event deltas
-            this.state = applyEventBatchToState(
-                this.state,
-                event,
-                (id) => this.doc.getContainerById(id),
+            this.state = applyEventBatchToState(this.state, event, (id) =>
+                this.doc.getContainerById(id),
             );
             // Notify subscribers of the update
             this.notifySubscribers(SyncDirection.FROM_LORO);
@@ -1582,18 +1580,19 @@ export class Mirror<S extends SchemaType> {
         const shouldCheck =
             this.options.debug || this.options.checkStateConsistency;
         if (shouldCheck) {
-            this.checkStateConsistency(newState);
+            this.checkStateConsistency();
         }
 
         // Notify subscribers
         this.notifySubscribers(SyncDirection.TO_LORO, tags);
     }
 
-    checkStateConsistency(newState: InferType<S>) {
-        if (!deepEqual(newState, toNormalizedJson(this.doc))) {
+    checkStateConsistency() {
+        const state = this.state;
+        if (!deepEqual(state, toNormalizedJson(this.doc))) {
             console.error(
                 "State diverged",
-                JSON.stringify(newState, null, 2),
+                JSON.stringify(state, null, 2),
                 JSON.stringify(toNormalizedJson(this.doc), null, 2),
             );
             throw new Error("[InternalError] State diverged");

packages/core/tests/core/mirror-tree.test.ts

@@ -158,7 +158,7 @@ describe("LoroTree integration", () => {
             return roots;
         };
 
-        const runOnce = (seed: number) => {
+        const runOnce = async (seed: number) => {
             const rng = mulberry32(seed);
             const doc = new LoroDoc();
             const s = schema({
@@ -174,6 +174,12 @@ describe("LoroTree integration", () => {
                 checkStateConsistency: true,
             });
 
+            const docB = new LoroDoc();
+            const mB = new Mirror({
+                doc: docB,
+                schema: s,
+            });
+
             // 1) Create an initial random tree
             const initial = buildInitial(rng, 12);
             m.setState({ tree: initial } as any);
@@ -217,12 +223,18 @@ describe("LoroTree integration", () => {
                 );
                 // If the move is a no-op (same parent + index), skip sometimes to avoid churn
                 m.setState({ tree: nextTree } as any);
+                docB.import(
+                    doc.export({ mode: "update", from: docB.version() }),
+                );
+                await Promise.resolve();
+                expect(mB.getState()).toStrictEqual(m.getState());
+                mB.checkStateConsistency();
             }
         };
 
         // Try multiple seeds
         for (const seed of [1, 42, 2025]) {
-            runOnce(seed);
+            await runOnce(seed);
         }
     });