From 00fc3dd60df803c8b2c5a171630c64edbfbbd39c Mon Sep 17 00:00:00 2001
From: Daniel Roe <daniel@roe.dev>
Date: Fri, 21 Feb 2025 22:28:50 +0000
Subject: [PATCH] feat: support scanning works from nuxt layers

resolves https://github.com/danielroe/nuxt-workers/issues/92
---
 playground/app.vue                             |  3 +++
 playground/layers/test/nuxt.config.ts          |  1 +
 playground/layers/test/workers/layer-worker.ts |  3 +++
 src/module.ts                                  | 12 +++++++++---
 test/e2e.spec.ts                               |  4 +++-
 5 files changed, 19 insertions(+), 4 deletions(-)
 create mode 100644 playground/layers/test/nuxt.config.ts
 create mode 100644 playground/layers/test/workers/layer-worker.ts

diff --git a/playground/app.vue b/playground/app.vue
index d858d31..99fc410 100644
--- a/playground/app.vue
+++ b/playground/app.vue
@@ -1,5 +1,6 @@
 <script setup lang="ts">
 const message = await hi()
+const layerMessage = await fromLayer()
 
 const clientSideMessage = ref('')
 async function loadClientSideMessage() {
@@ -11,6 +12,8 @@ async function loadClientSideMessage() {
   <div>
     {{ message }}
     <br>
+    {{ layerMessage }}
+    <br>
     <div>
       {{ clientSideMessage }}
     </div>
diff --git a/playground/layers/test/nuxt.config.ts b/playground/layers/test/nuxt.config.ts
new file mode 100644
index 0000000..268da7f
--- /dev/null
+++ b/playground/layers/test/nuxt.config.ts
@@ -0,0 +1 @@
+export default defineNuxtConfig({})
diff --git a/playground/layers/test/workers/layer-worker.ts b/playground/layers/test/workers/layer-worker.ts
new file mode 100644
index 0000000..2c8ee6d
--- /dev/null
+++ b/playground/layers/test/workers/layer-worker.ts
@@ -0,0 +1,3 @@
+export function fromLayer() {
+  return 'Hello from layer worker!'
+}
diff --git a/src/module.ts b/src/module.ts
index 10b370b..3a763df 100644
--- a/src/module.ts
+++ b/src/module.ts
@@ -21,16 +21,22 @@ export default defineNuxtModule<ModuleOptions>({
     name: 'nuxt-workers',
   },
   defaults: {
-    dir: '~/workers',
+    dir: 'workers',
   },
   async setup(options, nuxt) {
     const scanPattern = nuxt.options.extensions.map(e => `*${e}`)
 
-    const dirs: string[] = []
+    const _dirs = new Set<string>()
     for (const dir of Array.isArray(options.dir) ? options.dir : [options.dir]) {
-      dirs.push(resolve(nuxt.options.srcDir, resolveAlias(dir, nuxt.options.alias)))
+      for (const layer of nuxt.options._layers) {
+        _dirs.add(resolve(layer.config.srcDir || layer.cwd, resolveAlias(dir, nuxt.options.alias)))
+      }
     }
 
+    const dirs = [..._dirs]
+
+    console.log({ dirs })
+
     const context = {
       workerExports: Object.create(null) as Record<string, string>,
       reverseMap: Object.create(null) as Record<string, string[]>,
diff --git a/test/e2e.spec.ts b/test/e2e.spec.ts
index e957e21..6686f11 100644
--- a/test/e2e.spec.ts
+++ b/test/e2e.spec.ts
@@ -10,7 +10,9 @@ await setup({
 
 describe('nuxt-workers', () => {
   it('should work on the server', async () => {
-    expect(await $fetch('/')).toContain('Hello from worker!')
+    const html = await $fetch('/')
+    expect(html).toContain('Hello from worker!')
+    expect(html).toContain('Hello from layer worker!')
   })
 
   it('should work on the client', async () => {
-- 
2.51.2