From 0d2caed4b952cc4a1365f17549db45d1135841f3 Mon Sep 17 00:00:00 2001
From: Ammar Ahmed <ammarahmed6506@gmail.com>
Date: Fri, 21 Aug 2026 12:35:40 +0500
Subject: [PATCH] editor: only page notes past a size threshold

---
 .../editor/src/extensions/virtualization/index.ts  | 14 ++++++++++++--
 .../src/extensions/virtualization/node-views.ts    |  9 +++++++--
 2 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/packages/editor/src/extensions/virtualization/index.ts b/packages/editor/src/extensions/virtualization/index.ts
index 4ebb81552..0468fc1ee 100644
--- a/packages/editor/src/extensions/virtualization/index.ts
+++ b/packages/editor/src/extensions/virtualization/index.ts
@@ -22,12 +22,17 @@ import { HeightMap } from "./height-map.js";
 import { withVirtualization } from "./node-views.js";
 import { virtualizationPlugin } from "./viewport-plugin.js";
 
+/** Paging only engages for notes larger than this many top-level blocks. */
+const DEFAULT_THRESHOLD_BLOCKS = 300;
+
 export type VirtualizationOptions = {
   enabled: boolean;
+  thresholdBlocks: number;
 };
 
 export type VirtualizationStorage = {
   enabled: boolean;
+  thresholdBlocks: number;
   heightMap: HeightMap;
 };
 
@@ -44,12 +49,13 @@ export const Virtualization = Extension.create<VirtualizationOptions>({
   name: "virtualization",
 
   addOptions() {
-    return { enabled: false };
+    return { enabled: false, thresholdBlocks: DEFAULT_THRESHOLD_BLOCKS };
   },
 
   addStorage(): VirtualizationStorage {
     return {
       enabled: this.options.enabled,
+      thresholdBlocks: this.options.thresholdBlocks,
       heightMap: new HeightMap()
     };
   },
@@ -87,7 +93,11 @@ export function installVirtualization(editor: Editor): void {
   Object.defineProperty(editor.extensionManager, "nodeViews", {
     configurable: true,
     get() {
-      return withVirtualization(originalGetter.call(this), storage.heightMap);
+      return withVirtualization(
+        originalGetter.call(this),
+        storage.heightMap,
+        storage.thresholdBlocks
+      );
     }
   });
 }
diff --git a/packages/editor/src/extensions/virtualization/node-views.ts b/packages/editor/src/extensions/virtualization/node-views.ts
index 0ae7d1ced..e9b2bc584 100644
--- a/packages/editor/src/extensions/virtualization/node-views.ts
+++ b/packages/editor/src/extensions/virtualization/node-views.ts
@@ -172,7 +172,8 @@ function wrapCustom(
 
 export function withVirtualization(
   nodeViews: Record<string, NodeViewConstructor>,
-  heightMap: HeightMap
+  heightMap: HeightMap,
+  thresholdBlocks: number
 ): Record<string, NodeViewConstructor> {
   const wrapped: Record<string, NodeViewConstructor> = { ...nodeViews };
 
@@ -182,9 +183,13 @@ export function withVirtualization(
       const topLevel = isTopLevel(view, getPos as () => number | undefined);
       const materialize = isMaterialized(decorations);
 
+      // Only page documents past the size threshold. Smaller notes — the
+      // overwhelming majority — render fully and are unaffected by any of this.
+      const belowThreshold = view.state.doc.childCount <= thresholdBlocks;
+
       // Nested instances (inside callouts, tables, list items) are never
       // virtualized — only the outermost blocks are paged.
-      if (!topLevel) {
+      if (!topLevel || belowThreshold) {
         return inner
           ? inner(node, view, getPos, decorations, innerDecorations)
           : createMaterializedDefault(node, heightMap);