feat!(nuxt3): extend auto imports on `generateApp` hook (#3480)

packages/nuxt3/src/auto-imports/composables.ts

@@ -6,39 +6,45 @@ import { AutoImport } from '@nuxt/schema'
 import { resolveFiles } from '@nuxt/kit'
 import { filterInPlace } from './utils'
 
-export async function scanForComposables (dir: string, autoImports: AutoImport[]) {
-  if (!existsSync(dir)) { return }
+export async function scanForComposables (dir: string | string[], autoImports: AutoImport[]) {
+  const performScan = async (entry: string) => {
+    const files = await resolveFiles(entry, [
+      '*.{ts,js,mjs,cjs,mts,cts}',
+      '*/index.{ts,js,mjs,cjs,mts,cts}'
+    ])
 
-  const files = await resolveFiles(dir, [
-    '*.{ts,js,mjs,cjs,mts,cts}',
-    '*/index.{ts,js,mjs,cjs,mts,cts}'
-  ])
+    await Promise.all(
+      files.map(async (path) => {
+        // Remove original entries from the same import (for build watcher)
+        filterInPlace(autoImports, i => i.from !== path)
 
-  await Promise.all(
-    files.map(async (path) => {
-      // Remove original entries from the same import (for build watcher)
-      filterInPlace(autoImports, i => i.from !== path)
+        const code = await fsp.readFile(path, 'utf-8')
+        const exports = findExports(code)
+        const defaultExport = exports.find(i => i.type === 'default')
 
-      const code = await fsp.readFile(path, 'utf-8')
-      const exports = findExports(code)
-      const defaultExport = exports.find(i => i.type === 'default')
-
-      if (defaultExport) {
-        let name = parsePath(path).name
-        if (name === 'index') {
-          name = parsePath(path.split('/').slice(0, -1).join('/')).name
-        }
-        autoImports.push({ name: 'default', as: camelCase(name), from: path })
-      }
-      for (const exp of exports) {
-        if (exp.type === 'named') {
-          for (const name of exp.names) {
-            autoImports.push({ name, as: name, from: path })
+        if (defaultExport) {
+          let name = parsePath(path).name
+          if (name === 'index') {
+            name = parsePath(path.split('/').slice(0, -1).join('/')).name
           }
-        } else if (exp.type === 'declaration') {
-          autoImports.push({ name: exp.name, as: exp.name, from: path })
+          autoImports.push({ name: 'default', as: camelCase(name), from: path })
         }
-      }
-    })
-  )
+        for (const exp of exports) {
+          if (exp.type === 'named') {
+            for (const name of exp.names) {
+              autoImports.push({ name, as: name, from: path })
+            }
+          } else if (exp.type === 'declaration') {
+            autoImports.push({ name: exp.name, as: exp.name, from: path })
+          }
+        }
+      })
+    )
+  }
+
+  for (const entry of Array.isArray(dir) ? dir : [dir]) {
+    if (!existsSync(entry)) { continue }
+
+    await performScan(entry)
+  }
 }

packages/nuxt3/src/auto-imports/module.ts

@@ -72,9 +72,7 @@ export default defineNuxtModule<AutoImportsOptions>({
           : { name: importName.name, as: importName.as || importName.name, from: source.from }
       ))
       // Scan composables/
-      for (const composablesDir of composablesDirs) {
-        await scanForComposables(composablesDir, ctx.autoImports)
-      }
+      await scanForComposables(composablesDirs, ctx.autoImports)
       // Allow modules extending
       await nuxt.callHook('autoImports:extend', ctx.autoImports)
       // Update context
@@ -96,10 +94,13 @@ export default defineNuxtModule<AutoImportsOptions>({
     nuxt.hook('builder:watch', async (_, path) => {
       const _resolved = resolve(nuxt.options.srcDir, path)
       if (composablesDirs.find(dir => _resolved.startsWith(dir))) {
-        await regenerateAutoImports()
         await nuxt.callHook('builder:generateApp')
       }
     })
+
+    nuxt.hook('builder:generateApp', async () => {
+      await regenerateAutoImports()
+    })
   }
 })