Use the actual node count rather than the dirty node count (#3569)

packages/lexical-react/src/LexicalTreeView.tsx

@@ -98,20 +98,23 @@ export function TreeView({
   );
 
   useEffect(() => {
-    setContent(
-      generateContent(
-        editor.getEditorState(),
-        editor._config,
-        editor._compositionKey,
-        editor._editable,
-      ),
-    );
-  }, [editor]);
+    const editorState = editor.getEditorState();
+    if (!showLimited && editorState._nodeMap.size > 1000) {
+      setContent(
+        generateContent(
+          editorState,
+          editor._config,
+          editor._compositionKey,
+          editor._editable,
+        ),
+      );
+    }
+  }, [editor, showLimited]);
 
   useEffect(() => {
     return mergeRegister(
-      editor.registerUpdateListener(({editorState, dirtyLeaves}) => {
-        if (!showLimited && dirtyLeaves.size > 1000) {
+      editor.registerUpdateListener(({editorState}) => {
+        if (!showLimited && editorState._nodeMap.size > 1000) {
           lastEditorStateRef.current = editorState;
           setIsLimited(true);
           if (!showLimited) {

packages/lexical/src/LexicalSelection.ts

@@ -2743,7 +2743,7 @@ export function updateDOMSelection(
   domSelection: Selection,
   tags: Set<string>,
   rootElement: HTMLElement,
-  dirtyLeavesCount: number,
+  nodeCount: number,
 ): void {
   const anchorDOMNode = domSelection.anchorNode;
   const focusDOMNode = domSelection.focusNode;
@@ -2852,7 +2852,7 @@ export function updateDOMSelection(
       // The downside is that is makes the computation within Lexical more
       // complex, as now, we've sync update the DOM, but selection no longer
       // matches.
-      if (IS_CHROME && dirtyLeavesCount > 1000) {
+      if (IS_CHROME && nodeCount > 1000) {
         window.requestAnimationFrame(() =>
           domSelection.setBaseAndExtent(
             nextAnchorNode as Node,

packages/lexical/src/LexicalUpdates.ts

@@ -521,7 +521,7 @@ export function commitPendingUpdates(editor: LexicalEditor): void {
   const normalizedNodes = editor._normalizedNodes;
   const tags = editor._updateTags;
   const deferred = editor._deferred;
-  const dirtyLeavesCount = dirtyLeaves.size;
+  const nodeCount = pendingEditorState._nodeMap.size;
 
   if (needsUpdate) {
     editor._dirtyType = NO_DIRTY_NODES;
@@ -569,7 +569,7 @@ export function commitPendingUpdates(editor: LexicalEditor): void {
           domSelection,
           tags,
           rootElement as HTMLElement,
-          dirtyLeavesCount,
+          nodeCount,
         );
       }
       updateDOMBlockCursorElement(