fix: make changes to the schema migration

Author: nperez0111

packages/core/src/extensions/Collaboration/schemaMigration/SchemaMigrationPlugin.ts

@@ -1,5 +1,4 @@
 import { Plugin, PluginKey } from "@tiptap/pm/state";
-import { ySyncPluginKey } from "y-prosemirror";
 import * as Y from "yjs";
 
 import { BlockNoteExtension } from "../../../editor/BlockNoteExtension.js";
@@ -31,8 +30,12 @@ export class SchemaMigrationPlugin extends BlockNoteExtension {
           }
 
           if (
-            transactions.length !== 1 ||
-            !transactions[0].getMeta(ySyncPluginKey)
+            // If any of the transactions are not due to a yjs sync, we don't need to run the migration
+            !transactions.some((tr) => tr.getMeta("y-sync$")) ||
+            // If none of the transactions result in a document change, we don't need to run the migration
+            transactions.every((tr) => !tr.docChanged) ||
+            // If the fragment is still empty, we can't run the migration (since it has not yet been applied to the Y.Doc)
+            !fragment.firstChild
           ) {
             return undefined;
           }
@@ -44,6 +47,10 @@ export class SchemaMigrationPlugin extends BlockNoteExtension {
 
           this.migrationDone = true;
 
+          if (!tr.docChanged) {
+            return undefined;
+          }
+
           return tr;
         },
       }),

packages/core/src/extensions/Collaboration/schemaMigration/migrationRules/moveColorAttributes.ts

@@ -23,14 +23,13 @@ const traverseElement = (
 export const moveColorAttributes: MigrationRule = (fragment, tr) => {
   // Stores necessary info for all `blockContainer` nodes which still have
   // `textColor` or `backgroundColor` attributes that need to be moved.
-  const targetBlockContainers: Record<
+  const targetBlockContainers: Map<
     string,
     {
-      textColor?: string;
-      backgroundColor?: string;
+      textColor: string | undefined;
+      backgroundColor: string | undefined;
     }
-  > = {};
-
+  > = new Map();
   // Finds all elements which still have `textColor` or `backgroundColor`
   // attributes in the current Yjs fragment.
   fragment.forEach((element) => {
@@ -40,39 +39,53 @@ export const moveColorAttributes: MigrationRule = (fragment, tr) => {
           element.nodeName === "blockContainer" &&
           element.hasAttribute("id")
         ) {
+          const textColor = element.getAttribute("textColor");
+          const backgroundColor = element.getAttribute("backgroundColor");
+
           const colors = {
-            textColor: element.getAttribute("textColor"),
-            backgroundColor: element.getAttribute("backgroundColor"),
+            textColor:
+              textColor === defaultProps.textColor.default
+                ? undefined
+                : textColor,
+            backgroundColor:
+              backgroundColor === defaultProps.backgroundColor.default
+                ? undefined
+                : backgroundColor,
           };
 
-          if (colors.textColor === defaultProps.textColor.default) {
-            colors.textColor = undefined;
-          }
-          if (colors.backgroundColor === defaultProps.backgroundColor.default) {
-            colors.backgroundColor = undefined;
-          }
-
           if (colors.textColor || colors.backgroundColor) {
-            targetBlockContainers[element.getAttribute("id")!] = colors;
+            targetBlockContainers.set(element.getAttribute("id")!, colors);
           }
         }
       });
     }
   });
 
+  if (targetBlockContainers.size === 0) {
+    return false;
+  }
+
   // Appends transactions to add the `textColor` and `backgroundColor`
   // attributes found on each `blockContainer` node to move them to the child
   // `blockContent` node.
   tr.doc.descendants((node, pos) => {
     if (
       node.type.name === "blockContainer" &&
-      targetBlockContainers[node.attrs.id]
+      targetBlockContainers.has(node.attrs.id)
     ) {
-      tr = tr.setNodeMarkup(
-        pos + 1,
-        undefined,
-        targetBlockContainers[node.attrs.id],
-      );
+      const el = tr.doc.nodeAt(pos + 1);
+      if (!el) {
+        throw new Error("No element found");
+      }
+
+      tr.setNodeMarkup(pos + 1, undefined, {
+        // preserve existing attributes
+        ...el.attrs,
+        // add the textColor and backgroundColor attributes
+        ...targetBlockContainers.get(node.attrs.id),
+      });
     }
   });
+
+  return true;
 };