snapshot of https://github.com/shadcn-ui/ui.git (main, 2026-09-15; blobs > 131072B omitted)

apps/v4/lib/page-tree.ts

@@ -0,0 +1,64 @@
+import type { source } from "@/lib/source"
+
+export type PageTreeNode = (typeof source.pageTree)["children"][number]
+export type PageTreeFolder = Extract<PageTreeNode, { type: "folder" }>
+export type PageTreePage = Extract<PageTreeNode, { type: "page" }>
+
+// Recursively find all pages in a folder tree.
+export function getAllPagesFromFolder(folder: PageTreeFolder): PageTreePage[] {
+  const pages: PageTreePage[] = []
+
+  for (const child of folder.children) {
+    if (child.type === "page") {
+      pages.push(child)
+    } else if (child.type === "folder") {
+      pages.push(...getAllPagesFromFolder(child))
+    }
+  }
+
+  return pages
+}
+
+// Get the pages from a folder, handling nested base folders (radix/base).
+export function getPagesFromFolder(
+  folder: PageTreeFolder,
+  currentBase: string
+): PageTreePage[] {
+  // For the components folder, find the base subfolder.
+  if (folder.$id === "components" || folder.name === "Components") {
+    for (const child of folder.children) {
+      if (child.type === "folder") {
+        // Match by $id or by name.
+        const isRadix = child.$id === "radix" || child.name === "Radix UI"
+        const isBase = child.$id === "base" || child.name === "Base UI"
+        const isReactAria = child.$id === "aria" || child.name === "React Aria"
+
+        if (
+          (currentBase === "radix" && isRadix) ||
+          (currentBase === "base" && isBase) ||
+          (currentBase === "aria" && isReactAria)
+        ) {
+          return child.children.filter(
+            (c): c is PageTreePage => c.type === "page"
+          )
+        }
+      }
+    }
+
+    // Fallback: return all pages from nested folders.
+    return getAllPagesFromFolder(folder).filter(
+      (page) => !page.url.endsWith("/components")
+    )
+  }
+
+  // For other folders, return direct page children.
+  return folder.children.filter(
+    (child): child is PageTreePage => child.type === "page"
+  )
+}
+
+// Get current base (radix, base, or aria) from pathname.
+export function getCurrentBase(pathname: string): string {
+  const baseMatch = pathname.match(/\/docs\/components\/(radix|base|aria)\//)
+  return baseMatch ? baseMatch[1] : "base" // Default to base.
+}