fix(core): allow node to be remove before nested root element is unmo…

…unted
yf-yang · Apr 28, 2024 · 49b419e · 49b419e
1 parent 5d4cf3b
commit 49b419e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/packages/lexical/src/LexicalEditor.ts b/packages/lexical/src/LexicalEditor.ts
@@ -1030,13 +1030,14 @@ export class LexicalEditor {
     key: NodeKey,
     nextNestedRootElement: null | HTMLElement,
   ): void {
-    invariant(
-      $isNestedRootNode(this._editorState._nodeMap.get(key)),
-      'Cannot set nested root element for non-nested root node.',
-    );
     const prevNestedRootElement = this._keyToDOMMap.get(key) || null;
     if (nextNestedRootElement !== prevNestedRootElement) {
       if (nextNestedRootElement !== null) {
+        invariant(
+          $isNestedRootNode(this._editorState._nodeMap.get(key)),
+          'Cannot set nested root element for non-nested root node.',
+        );
+
         const style = nextNestedRootElement.style;
         style.userSelect = 'text';
         style.whiteSpace = 'pre-wrap';