From a49cbcd1ecd73cdf71919d90391626cf20393b12 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9C=B1=E7=AB=9E=E9=93=AD?= <zhu.jingming@trs.com.cn>
Date: Thu, 1 Jun 2023 10:33:22 +0800
Subject: [PATCH] feat: dirs add `exportAllByFileName` config (#388)

---
 playground/App.vue             | 13 ++++++---
 playground/utils/foo.ts        |  3 ++
 playground/utils/nested/bar.ts |  3 ++
 playground/vite.config.ts      |  5 +++-
 src/core/ctx.ts                | 50 ++++++++++++++++++++++++++++------
 src/core/unplugin.ts           |  4 +--
 src/types.ts                   |  7 ++++-
 7 files changed, 68 insertions(+), 17 deletions(-)
 create mode 100644 playground/utils/foo.ts
 create mode 100644 playground/utils/nested/bar.ts

diff --git a/playground/App.vue b/playground/App.vue
index 2bcbb3fd..d8625151 100644
--- a/playground/App.vue
+++ b/playground/App.vue
@@ -2,8 +2,11 @@
 import HelloWorld from './HelloWorld.vue'
 
 ElMessage.warning('Test')
-const foo = useFoo()
-const bar = useBar()
+const composableFoo = useFoo()
+const composableBar = useBar()
+
+const utilsFoo = foo.useFoo()
+const utilsBar = bar.useBar()
 </script>
 
 <script lang="ts">
@@ -19,8 +22,10 @@ export default defineComponent({
 
 <template>
   <HelloWorld msg="hi" />
-  <pre>{{ foo }}</pre>
-  <pre>{{ bar }}</pre>
+  <pre>{{ composableFoo }}</pre>
+  <pre>{{ composableBar }}</pre>
+  <pre>{{ utilsFoo }}</pre>
+  <pre>{{ utilsBar }}</pre>
   <pre>{{ FOOBAR }}</pre>
   <div v-loading="false">
     <ElButton>Hello</ElButton>
diff --git a/playground/utils/foo.ts b/playground/utils/foo.ts
new file mode 100644
index 00000000..134369c1
--- /dev/null
+++ b/playground/utils/foo.ts
@@ -0,0 +1,3 @@
+export function useFoo() {
+  return 'foo from ./utils/'
+}
diff --git a/playground/utils/nested/bar.ts b/playground/utils/nested/bar.ts
new file mode 100644
index 00000000..661d7294
--- /dev/null
+++ b/playground/utils/nested/bar.ts
@@ -0,0 +1,3 @@
+export function useBar() {
+  return 'bar from ./utils/nested/'
+}
diff --git a/playground/vite.config.ts b/playground/vite.config.ts
index ac78a5d9..9da77d41 100644
--- a/playground/vite.config.ts
+++ b/playground/vite.config.ts
@@ -15,9 +15,12 @@ export default defineConfig({
       ],
       dirs: [
         './composables/**',
+        {
+          dir: './utils/**',
+          exportAllByFileName: true,
+        },
       ],
       vueTemplate: true,
-      cache: true,
     }),
   ],
 })
diff --git a/src/core/ctx.ts b/src/core/ctx.ts
index c9d313ea..394e0cad 100644
--- a/src/core/ctx.ts
+++ b/src/core/ctx.ts
@@ -1,16 +1,16 @@
-import { dirname, isAbsolute, relative, resolve } from 'node:path'
+import { basename, dirname, extname, isAbsolute, relative, resolve } from 'node:path'
 import { existsSync, promises as fs } from 'node:fs'
-import { slash, throttle, toArray } from '@antfu/utils'
+import { isString, slash, throttle, toArray } from '@antfu/utils'
 import { createFilter } from '@rollup/pluginutils'
 import { isPackageExists } from 'local-pkg'
-import type { Import, InlinePreset } from 'unimport'
-import { createUnimport, resolvePreset, scanDirExports } from 'unimport'
+import type { Import, InlinePreset, ScanDirExportsOptions } from 'unimport'
+import { createUnimport, resolvePreset, scanDirExports, scanFilesFromDir } from 'unimport'
 
 // @ts-expect-error types
 import { vueTemplateAddon } from 'unimport/addons'
 import MagicString from 'magic-string'
 import { presets } from '../presets'
-import type { ESLintGlobalsPropValue, ESLintrc, ImportExtended, Options } from '../types'
+import type { DirsOptions, ESLintGlobalsPropValue, ESLintrc, ImportExtended, Options } from '../types'
 import { generateESLintConfigs } from './eslintrc'
 import { resolversAddon } from './resolvers'
 
@@ -19,7 +19,19 @@ export function createContext(options: Options = {}, root = process.cwd()) {
     dts: preferDTS = isPackageExists('typescript'),
   } = options
 
-  const dirs = options.dirs?.map(dir => resolve(root, dir))
+  const dirs: DirsOptions[] | undefined = options.dirs?.map((dir) => {
+    if (isString(dir)) {
+      return {
+        dir: resolve(root, dir),
+      }
+    }
+    else {
+      return {
+        ...dir,
+        dir: resolve(root, dir.dir),
+      }
+    }
+  })
 
   const eslintrc: ESLintrc = options.eslintrc || {}
   eslintrc.enabled = eslintrc.enabled === undefined ? false : eslintrc.enabled
@@ -164,12 +176,32 @@ ${dts}`.trim()}\n`
     return Promise.all(promises)
   }
 
+  async function scanExportAllByFileNameDir(dirs: string[], options: ScanDirExportsOptions): Promise<ImportExtended[]> {
+    const files = await scanFilesFromDir(dirs, options)
+
+    return files.map(i => ({
+      name: '*',
+      as: basename(i, extname(i)),
+      from: i,
+    }))
+  }
+
   async function scanDirs() {
     if (dirs?.length) {
       await unimport.modifyDynamicImports(async (imports) => {
-        const exports_ = await scanDirExports(dirs, {
-          filePatterns: ['*.{tsx,jsx,ts,js,mjs,cjs,mts,cts}'],
-        }) as ImportExtended[]
+        const exportDirs = dirs.filter(i => !i.exportAllByFileName).map(i => i.dir)
+        const exportAllByFileNameDirs = dirs.filter(i => i.exportAllByFileName).map(i => i.dir)
+        const exports_: ImportExtended[] = []
+        if (exportDirs.length) {
+          exports_.push(...await scanDirExports(exportDirs, {
+            filePatterns: ['*.{tsx,jsx,ts,js,mjs,cjs,mts,cts}'],
+          }))
+        }
+        if (exportAllByFileNameDirs.length) {
+          exports_.push(...await scanExportAllByFileNameDir(exportAllByFileNameDirs, {
+            filePatterns: ['*.{tsx,jsx,ts,js,mjs,cjs,mts,cts}'],
+          }))
+        }
         exports_.forEach(i => i.__source = 'dir')
         return modifyDefaultExportsAlias([
           ...imports.filter((i: ImportExtended) => i.__source !== 'dir'),
diff --git a/src/core/unplugin.ts b/src/core/unplugin.ts
index 84b7d3e3..bbebb30e 100644
--- a/src/core/unplugin.ts
+++ b/src/core/unplugin.ts
@@ -1,5 +1,5 @@
 import { minimatch } from 'minimatch'
-import { slash } from '@antfu/utils'
+import { isString, slash } from '@antfu/utils'
 import { createUnplugin } from 'unplugin'
 import type { Options } from '../types'
 import { createContext } from './ctx'
@@ -23,7 +23,7 @@ export default createUnplugin<Options>((options) => {
     },
     vite: {
       async handleHotUpdate({ file }) {
-        if (ctx.dirs?.some(glob => minimatch(slash(file), slash(glob))))
+        if (ctx.dirs?.some(glob => minimatch(slash(file), isString(glob) ? glob : glob.dir)))
           await ctx.scanDirs()
       },
       async configResolved(config) {
diff --git a/src/types.ts b/src/types.ts
index 1f0eb9b2..33821181 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -51,6 +51,11 @@ export type Resolver = ResolverFunction | ResolverResultObject
  */
 export type ImportsMap = Record<string, (string | ImportNameAlias)[]>
 
+export interface DirsOptions {
+  dir: string
+  exportAllByFileName?: boolean
+}
+
 export type ESLintGlobalsPropValue = boolean | 'readonly' | 'readable' | 'writable' | 'writeable'
 
 export interface ESLintrc {
@@ -93,7 +98,7 @@ export interface Options {
   /**
    * Path for directories to be auto imported
    */
-  dirs?: string[]
+  dirs?: (string | DirsOptions)[]
 
   /**
    * Pass a custom function to resolve the component importing path from the component name.