From 38764d45abdf69f2c7f3ce93d3df429ee6898317 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Germ=C3=A1n=20Jablo=C3=B1ski?=
 <43938777+GermanJablo@users.noreply.github.com>
Date: Mon, 9 Dec 2024 15:25:32 -0300
Subject: [PATCH] fix updateDOM

---
 packages/lexical/src/LexicalReconciler.ts | 10 +++++++++-
 packages/lexical/src/LexicalUtils.ts      |  8 ++++++--
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/packages/lexical/src/LexicalReconciler.ts b/packages/lexical/src/LexicalReconciler.ts
index 855fd44c373..216db70ea48 100644
--- a/packages/lexical/src/LexicalReconciler.ts
+++ b/packages/lexical/src/LexicalReconciler.ts
@@ -641,8 +641,16 @@ function $reconcileNode(
     );
   }
 
+  const nextClasses = nextNode.__classes || {};
+  const prevClasses = prevNode.__classes || {};
+  const classesChanged = !(
+    Object.keys(nextClasses).length === Object.keys(prevClasses).length &&
+    Object.keys(nextClasses).every(
+      (_key) => nextClasses[_key] === prevClasses[_key],
+    )
+  );
   // Update node. If it returns true, we need to unmount and re-create the node
-  if (nextNode.updateDOM(prevNode, dom, activeEditorConfig)) {
+  if (nextNode.updateDOM(prevNode, dom, activeEditorConfig) || classesChanged) {
     const replacementDOM = $createNode(key, null);
 
     if (parentDOM === null) {
diff --git a/packages/lexical/src/LexicalUtils.ts b/packages/lexical/src/LexicalUtils.ts
index 67a8d13d24b..eb0a5011b6d 100644
--- a/packages/lexical/src/LexicalUtils.ts
+++ b/packages/lexical/src/LexicalUtils.ts
@@ -1436,7 +1436,9 @@ export function $isRootOrShadowRoot(
  */
 export function $copyNode<T extends LexicalNode>(node: T): T {
   const copy = node.constructor.clone(node) as T;
-  copy.__classes = {...node.__classes};
+  if (node.__classes) {
+    copy.__classes = {...node.__classes};
+  }
   $setNodeKey(copy, null);
   return copy;
 }
@@ -1842,7 +1844,9 @@ export function getCachedTypeToNodeMap(
 export function $cloneWithProperties<T extends LexicalNode>(latestNode: T): T {
   const constructor = latestNode.constructor;
   const mutableNode = constructor.clone(latestNode) as T;
-  mutableNode.__classes = {...latestNode.__classes};
+  if (latestNode.__classes) {
+    mutableNode.__classes = {...latestNode.__classes};
+  }
   mutableNode.afterCloneFrom(latestNode);
   if (__DEV__) {
     invariant(