feat: use ScanDirExportsOptions from unimport (#570)

Author: ilyaliao

README.md

@@ -276,6 +276,8 @@ AutoImport({
 
   // Options for scanning directories for auto import
   dirsScanOptions: {
+    filePatterns: ['*.ts'], // Glob patterns for matching files
+    fileFilter: file => file.endsWith('.ts'), // Filter files
     types: true // Enable auto import the types under the directories
   },
 

src/types.ts

@@ -1,5 +1,5 @@
 import type { Arrayable, Awaitable } from '@antfu/utils'
-import type { AddonVueDirectivesOptions, Import, InlinePreset, PackagePreset, UnimportOptions } from 'unimport'
+import type { AddonVueDirectivesOptions, Import, InlinePreset, PackagePreset, ScanDirExportsOptions, UnimportOptions } from 'unimport'
 import type { FilterPattern } from 'unplugin-utils'
 import type { PresetName } from './presets'
 
@@ -51,15 +51,6 @@ export type Resolver = ResolverFunction | ResolverResultObject
  */
 export type ImportsMap = Record<string, (string | ImportNameAlias)[]>
 
-export interface ScanDirExportsOptions {
-  /**
-   * Register type exports
-   *
-   * @default true
-   */
-  types?: boolean
-}
-
 /**
  * Directory to search for import
  */
@@ -140,7 +131,7 @@ export interface Options {
   /**
    * Options for scanning directories for auto import
    */
-  dirsScanOptions?: ScanDirExportsOptions
+  dirsScanOptions?: Omit<ScanDirExportsOptions, 'cwd'>
 
   /**
    * Path for directories to be auto imported

test/search.test.ts

@@ -35,7 +35,7 @@ describe('search', () => {
   })
 })
 
-describe('import the types from the dirs', () => {
+describe('dirsScanOptions', () => {
   it('should top level types enable work', async () => {
     const ctx = createContext({
       dts: false,
@@ -88,4 +88,114 @@ describe('import the types from the dirs', () => {
     expect(data).not.toContain('TypeB')
     expect(data).toContain('SpecialType')
   })
+
+  it('should filePatterns work', async () => {
+    const ctx = createContext({
+      dts: false,
+      dirsScanOptions: {
+        filePatterns: ['*.{ts,tsx}'],
+      },
+      dirs: [
+        'src/views',
+        'src/types',
+      ],
+    }, root)
+
+    await ctx.scanDirs()
+    const data = await ctx.generateDTS('')
+    expect(data).toContain('PageA')
+    expect(data).toContain('PageB')
+    expect(data).toContain('TypeA')
+    expect(data).toContain('TypeB')
+    expect(data).toContain('SpecialType')
+  })
+
+  it('should specific filePatterns work', async () => {
+    const ctx = createContext({
+      dts: false,
+      dirsScanOptions: {
+        types: true,
+        filePatterns: ['*.ts'],
+      },
+      dirs: [
+        'src/views',
+        'src/types',
+      ],
+    }, root)
+
+    await ctx.scanDirs()
+    const data = await ctx.generateDTS('')
+    expect(data).not.toContain('PageA')
+    expect(data).not.toContain('PageB')
+    expect(data).not.toContain('TypeA')
+    expect(data).not.toContain('TypeB')
+    expect(data).toContain('SpecialType')
+  })
+
+  it('should fileFilter work', async () => {
+    const ctx = createContext({
+      dts: false,
+      dirsScanOptions: {
+        types: true,
+        fileFilter: (file: string) => file.includes('TypeA') || file.includes('PageA'),
+      },
+      dirs: [
+        'src/views',
+        'src/types',
+      ],
+    }, root)
+
+    await ctx.scanDirs()
+    const data = await ctx.generateDTS('')
+    expect(data).toContain('TypeA')
+    expect(data).toContain('PageA')
+    expect(data).not.toContain('TypeB')
+    expect(data).not.toContain('PageB')
+    expect(data).not.toContain('SpecialType')
+  })
+
+  it('should fileFilter work when excluding all', async () => {
+    const ctx = createContext({
+      dts: false,
+      dirsScanOptions: {
+        types: true,
+        fileFilter: (_file: string) => false,
+      },
+      dirs: [
+        'src/views',
+        'src/types',
+      ],
+    }, root)
+
+    await ctx.scanDirs()
+    const data = await ctx.generateDTS('')
+    expect(data).not.toContain('TypeA')
+    expect(data).not.toContain('PageA')
+    expect(data).not.toContain('TypeB')
+    expect(data).not.toContain('PageB')
+    expect(data).not.toContain('SpecialType')
+  })
+
+  it('should filePatterns and fileFilter work together', async () => {
+    const ctx = createContext({
+      dts: false,
+      dirsScanOptions: {
+        types: true,
+        filePatterns: ['*.{ts,tsx}'],
+        fileFilter: (file: string) => file.includes('PageA'),
+      },
+      dirs: [
+        'src/views',
+        'src/types',
+      ],
+    }, root)
+
+    await ctx.scanDirs()
+    const data = await ctx.generateDTS('')
+    expect(data).toContain('PageA')
+    expect(data).toContain('TypeA')
+    expect(data).not.toContain('SpecialType')
+    expect(data).not.toContain('PageB')
+    expect(data).not.toContain('TypeB')
+  })
 })