added: extensions

space307 · Dec 28, 2023 · 1cd2713 · 1cd2713
1 parent eb33bf5
commit 1cd2713
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 6 deletions.
diff --git a/bin/collectUsages.js b/bin/collectUsages.js
@@ -6,6 +6,7 @@ import { createStatusAPI } from './utils/index.js'
  *   config: {
  *      collectUsages: string
  *      exclude: Set<string>
+ *      extensions: Array<string>
  *      babelPlugins: Set<string>
  *      batch: {
  *        default: number

diff --git a/bin/fileTraversal/getFiles.js b/bin/fileTraversal/getFiles.js
@@ -5,10 +5,14 @@ import { fdir } from 'fdir'
 const getRepoRoot = () =>
   execSync('git rev-parse --show-toplevel', { encoding: 'utf8' }).trim()
 
+const formattedExtensions = list =>
+  list.reduce((acc, ext) => acc + (acc ? ',' : '') + ext.slice(1), '')
+
 /**
  * @param {{
  *   config: {
  *      exclude: Set<string>
+ *      extensions: Array<string>
  *   }
  *   pkg: {
  *      name: string
@@ -30,7 +34,11 @@ const getFiles = async ({ config, pkg }) => {
     .join('|')
   const excludeRegExp = new RegExp(exclude)
   const repoRoot = getRepoRoot()
-  const source = join(repoRoot, '**', '*.{ts,tsx}')
+  const source = join(
+    repoRoot,
+    '**',
+    `*.{${formattedExtensions(config.extensions)}}`
+  )
 
   const files = new fdir()
     .exclude(dirName => excludeRegExp.test(dirName))

diff --git a/bin/fileTraversal/index.js b/bin/fileTraversal/index.js
@@ -6,6 +6,7 @@ import { getFiles } from './getFiles.js'
  *   cmd: {function(_: string): void}
  *   config: {
  *      exclude: Set<string>
+ *      extensions: Array<string>
  *      babelPlugins: Set<string>
  *      batch: {
  *        default: number

diff --git a/bin/getConfig.js b/bin/getConfig.js
@@ -3,11 +3,12 @@ import { lilconfig } from 'lilconfig'
 import meow from 'meow'
 
 const BASE_CONFIG = {
-  entry: 'index.ts',
-  exclude: new Set(['node_modules']),
   babelPlugins: new Set(['typescript']),
   batch: { default: 100 },
-  collectUsages: null
+  collectUsages: null,
+  entry: 'index.ts',
+  exclude: new Set(['node_modules']),
+  extensions: ['.ts', '.tsx']
 }
 
 const cli = meow(
@@ -41,7 +42,8 @@ const getConfig = async () => {
     exclude = [],
     babelPlugins = [],
     entry = BASE_CONFIG.entry,
-    batch = {}
+    batch = {},
+    extensions = BASE_CONFIG.extensions
   } = result.config
 
   return result === null
@@ -53,7 +55,8 @@ const getConfig = async () => {
         batch: {
           default: batch.default || BASE_CONFIG.batch.defaul
         },
-        collectUsages: cli.flags.collectUsages || BASE_CONFIG.collectUsages
+        collectUsages: cli.flags.collectUsages || BASE_CONFIG.collectUsages,
+        extensions
       }
 }
 

diff --git a/bin/main.js b/bin/main.js
@@ -7,6 +7,7 @@ import { createStatusAPI, readJSON } from './utils/index.js'
  *   config: {
  *      entry: string
  *      exclude: Set<string>
+ *      extensions: Array<string>
  *      babelPlugins: Set<string>
  *      batch: {
  *        default: number