Support custom page extensions (#14)

fuma-nama · Nov 5, 2024 · 3d1095d · 3d1095d
1 parent ca3bb81
commit 3d1095d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/scan.ts b/src/scan.ts
@@ -22,6 +22,7 @@ export type ScanOptions = {
 
   populate?: PopulateParams;
   meta?: Record<string, UrlMeta>;
+  extensions?: string[];
 };
 
 export type ScanResult = {
@@ -48,16 +49,19 @@ function isDirExists(dir: string): Promise<boolean> {
 }
 
 export async function scanURLs(options: ScanOptions = {}): Promise<ScanResult> {
+  const ext = options.extensions ?? ['js', 'jsx', 'tsx', 'md', 'mdx'];
+  const cwd = options.cwd ?? process.cwd();
+
   async function getFiles() {
-    const cwd = options.cwd ?? process.cwd();
+    const suffix = ext.length > 0? `.{${ext.join(",")}}` : '';
 
-    const appFiles = await fg('**/page.{js,jsx,tsx}', {
+    const appFiles = await fg(`**/page${suffix}`, {
       cwd: (await isDirExists(path.join(cwd, 'src/app')))
         ? path.join(cwd, 'src/app')
         : path.join(cwd, 'app'),
     });
 
-    const pagesFiles = await fg('**/*.{js,jsx,tsx}', {
+    const pagesFiles = await fg(`**/*${suffix}`, {
       cwd: (await isDirExists(path.join(cwd, 'src/pages')))
         ? path.join(cwd, 'src/pages')
         : path.join(cwd, 'pages'),