feat: add maxDepth param to getAllPagesInSpace

Author: transitive-bullshit

packages/notion-utils/src/get-all-pages-in-space.ts

@@ -12,7 +12,6 @@ import { parsePageId } from './parse-page-id'
  * If `rootSpaceId` is not defined, the space ID of the root page will be used
  * to scope traversal.
  *
- *
  * @param rootPageId - Page ID to start from.
  * @param rootSpaceId - Space ID to scope traversal.
  * @param getPage - Function used to fetch a single page.
@@ -25,18 +24,24 @@ export async function getAllPagesInSpace(
   {
     concurrency = 4,
     traverseCollections = true,
-    targetPageId
+    targetPageId,
+    maxDepth = Number.POSITIVE_INFINITY
   }: {
     concurrency?: number
     traverseCollections?: boolean
     targetPageId?: string
+    maxDepth?: number
   } = {}
 ): Promise<PageMap> {
   const pages: PageMap = {}
   const pendingPageIds = new Set<string>()
   const queue = new PQueue({ concurrency })
 
-  async function processPage(pageId: string) {
+  async function processPage(pageId: string, depth = 0) {
+    if (depth > maxDepth) {
+      return
+    }
+
     if (targetPageId && pendingPageIds.has(targetPageId)) {
       return
     }
@@ -94,7 +99,7 @@ export async function getAllPagesInSpace(
 
             return true
           })) {
-            void processPage(subPageId)
+            void processPage(subPageId, depth + 1)
           }
 
           // traverse collection item pages as they may contain subpages as well
@@ -107,7 +112,7 @@ export async function getAllPagesInSpace(
 
                 if (blockIds) {
                   for (const collectionItemId of blockIds) {
-                    void processPage(collectionItemId)
+                    void processPage(collectionItemId, depth + 1)
                   }
                 }
               }