feat: handle layers

rezozero · Feb 23, 2024 · 044ffac · 044ffac
1 parent 499ffcf
commit 044ffac
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/playground/nuxt.config.ts b/playground/nuxt.config.ts
@@ -1,5 +1,6 @@
 export default defineNuxtConfig({
     devtools: { enabled: true },
-    modules: ['../src/module'],
+    // modules: ['../src/module'],
     css: ['~/assets/css/main.css'],
+    extends: ['github:rezozero/nuxt-layer'],
 })
diff --git a/src/module.ts b/src/module.ts
@@ -84,14 +84,24 @@ export default defineNuxtModule<NuxtStoriesOptions>({
 
         // PAGES
         extendPages(async (pages) => {
+            // scan project files
             const files = await resolveFiles(nuxt.options.rootDir, pattern)
 
+            // scan layers files
+            const layersFiles = await Promise.all(
+                nuxt.options._layers.map((layer) => resolveFiles(layer.config.rootDir, pattern)),
+            )
+
+            files.concat(layersFiles.flat())
+
+            // generate child routes
             files.forEach((file) => {
                 const fileRoute = getFileRoute(file)
 
                 route.children!.push(fileRoute)
             })
 
+            // add route
             pages.push(route)
         })