65 lines
2.1 KiB
TypeScript
65 lines
2.1 KiB
TypeScript
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.
|
|
}
|