feat: support export default config

plantain-00 · May 11, 2020 · e8d8d73 · e8d8d73
1 parent 61b496d
commit e8d8d73
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/demo/config.ts b/demo/config.ts
@@ -1,6 +1,6 @@
 import { generateTypescriptOfFunctionParameter, TypeDeclaration } from '../dist/core'
 
-export = (typeDeclarations: TypeDeclaration[]) => {
+export default (typeDeclarations: TypeDeclaration[]): string => {
   const result = []
   for (const declaration of typeDeclarations) {
     if (declaration.kind === 'function') {

diff --git a/src/index.ts b/src/index.ts
@@ -128,7 +128,12 @@ async function executeCommandLine() {
       if (configFilePath.endsWith('.ts')) {
         require('ts-node/register/transpile-only')
       }
-      const action = require(configFilePath) as (typeDeclarations: TypeDeclaration[], modules: typeof typescriptGenerator) => string
+      type Action = (typeDeclarations: TypeDeclaration[], modules: typeof typescriptGenerator) => string
+      // eslint-disable-next-line @typescript-eslint/no-var-requires
+      let action: Action & { default?: Action } = require(configFilePath)
+      if (action.default) {
+        action = action.default
+      }
       const customContent = action(generator.declarations, typescriptGenerator)
       fs.writeFileSync(customPath, customContent)
     }