feat: add support for dirs option

chore: wip

chore: lint

Author: chrisbbreuer

README.md

@@ -29,6 +29,7 @@ import { autoImports } from 'bun-plugin-auto-imports'
 const options: AutoImportsOptions = {
   presets: ['solid-js'], // any unimport presets are valid
   imports: [{ name: 'z', from: 'zod' }],
+  dirs: ['./src'],
   dts: `./src/auto-import.d.ts`, // default is `./auto-import.d.ts`
 }
 

src/plugin.ts

@@ -1,5 +1,6 @@
 import type { Loader, PluginBuilder } from 'bun'
 import type { AutoImportsOptions, AutoImportsPlugin, ESLintOptions } from './types'
+import { Glob } from 'bun'
 import { generateESLintGlobals } from './eslint'
 
 function getLoader(path: string): string {
@@ -14,18 +15,94 @@ function getLoader(path: string): string {
 
 export const GENERATED_COMMENT = '// Generated by bun-plugin-auto-imports\n'
 
+async function scanDirExports(dir: string): Promise<string[]> {
+  const exports: Set<string> = new Set()
+
+  // Create glob for TypeScript/JavaScript files
+  const glob = new Glob('**/*.{ts,tsx,js,jsx}')
+
+  // Scan the directory for matching files
+  for await (const file of glob.scan({
+    cwd: dir,
+    absolute: true,
+    onlyFiles: true,
+    followSymlinks: false,
+  })) {
+    // Skip definition files and node_modules
+    if (file.includes('node_modules') || file.endsWith('.d.ts')) {
+      continue
+    }
+
+    const content = await Bun.file(file).text()
+
+    // Match export declarations
+    const exportMatches = [
+      // Named exports
+      ...content.matchAll(/export\s+(?:const|let|var|function|class|type|interface)\s+([a-zA-Z_$][\w$]*)/g),
+      // Export statements
+      ...content.matchAll(/export\s+\{([^}]+)\}/g),
+      // Default exports
+      ...content.matchAll(/export\s+default\s+(?:(?:function|class)\s*)?([a-zA-Z_$][\w$]*)/g),
+    ]
+
+    for (const match of exportMatches) {
+      if (match[1]) {
+        // Handle multiple exports in a single statement
+        const names = match[1].split(',').map(name =>
+          name.trim()
+            .replace(/\s+as\s+(?:\S.*)?$/, '') // Remove "as" aliases
+            .replace(/^.*(?:[\n\r\u2028\u2029]\s*|[\t\v\f \xA0\u1680\u2000-\u200A\u202F\u205F\u3000\uFEFF])as\s+/, ''), // Handle "originalName as exportName"
+        )
+        names.forEach(name => exports.add(name))
+      }
+    }
+  }
+
+  return Array.from(exports)
+}
+
 export function autoImports(options: Partial<AutoImportsOptions>): AutoImportsPlugin {
   return {
     name: 'bun-plugin-auto-imports',
 
     async setup(builder: PluginBuilder): Promise<void> {
+      // console.log('Auto-imports plugin setup', options)
       const { createUnimport } = await import('unimport')
-      const { injectImports, generateTypeDeclarations } = createUnimport({
+
+      // Scan directories for exports
+      const dirExports: Record<string, string[]> = {}
+      if (options.dirs) {
+        for (const dir of options.dirs) {
+          const dirPath = typeof dir === 'string' ? dir : ('path' in dir ? dir.path : dir)
+          // console.log('Scanning directory:', dirPath)
+          dirExports[dirPath] = await scanDirExports(dirPath)
+          // console.log('Found exports:', dirExports[dirPath])
+        }
+      }
+
+      // Convert scanned exports to unimport format
+      const scannedImports = Object.entries(dirExports).flatMap(([dir, exports]) =>
+        exports.map(name => ({
+          from: dir,
+          name,
+          as: name,
+        })),
+      )
+
+      const unimport = createUnimport({
         ...options,
+        imports: [
+          ...(options.imports || []),
+          ...scannedImports,
+        ],
         dts: undefined,
       } as AutoImportsOptions)
 
-      const dtsContent = await generateTypeDeclarations()
+      const { injectImports } = unimport
+
+      // Generate types declarations
+      const dtsContent = await unimport.generateTypeDeclarations()
+
       // Add generated comment to d.ts file
       await Bun.write(
         options.dts ?? './auto-imports.d.ts',
@@ -40,7 +117,7 @@ export function autoImports(options: Partial<AutoImportsOptions>): AutoImportsPl
         }
 
         const eslintContent = generateESLintGlobals(dtsContent, eslintOptions)
-        await Bun.write(eslintOptions.filepath ?? './auto-imports.d.ts', eslintContent)
+        await Bun.write(eslintOptions.filepath ?? './.eslint-auto-import.json', eslintContent)
       }
 
       builder.onLoad({ filter: /.*/ }, async (args) => {

src/types.ts

@@ -1,6 +1,15 @@
 import type { PluginBuilder } from 'bun'
 import type { UnimportOptions } from 'unimport'
 
+export interface ScanDir {
+  path: string
+  /**
+   * Whether to scan for type exports
+   * @default true
+   */
+  types?: boolean
+}
+
 export interface AutoImportsPlugin {
   name: string
   setup: (builder: PluginBuilder) => Promise<void>
@@ -28,4 +37,5 @@ export interface ESLintOptions {
 export type AutoImportsOptions = Partial<UnimportOptions> & {
   dts?: string
   eslint?: ESLintOptions
+  dirs?: (string | ScanDir)[]
 }

test.ts

@@ -0,0 +1,22 @@
+import type { AutoImportsOptions } from './src'
+import { plugin } from 'bun'
+import path from 'node:path'
+import { autoImports } from './src'
+
+const options: AutoImportsOptions = {
+  // presets: ['solid-js'], // any unimport presets are valid
+  dirs: [path.resolve('./src')],
+  dts: `./src/auto-import.d.ts`, // default is `./auto-import.d.ts`
+  eslint: {
+    enabled: true,
+  },
+}
+
+plugin(autoImports(options))
+
+Bun.serve({
+  fetch() {
+    return new Response('Bun!')
+  },
+  port: 3000,
+})