1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

import fs from 'fs'
import path from 'path'
import resolve from 'resolve'
import detective from 'detective-typescript'
import normalizePath from 'normalize-path'

function createModule(file: string): { file: string; requires: string[] } {
  let source = fs.readFileSync(file, 'utf-8')
  return { file, requires: detective(source, { mixedImports: true }) }
}

function* _getModuleDependencies(entryFile: string): Generator<string> {
  yield entryFile

  let mod = createModule(entryFile)

  let ext = path.extname(entryFile)
  let isTypeScript = ext === '.ts' || ext === '.cts' || ext === '.mts'
  let extensions = [...(isTypeScript ? ['.ts', '.cts', '.mts'] : []), '.js', '.cjs', '.mjs']

  // Iterate over the modules, even when new
  // ones are being added
  for (let dep of mod.requires) {
    // Only track local modules, not node_modules
    if (!dep.startsWith('./') && !dep.startsWith('../')) {
      continue
    }

    try {
      let basedir = path.dirname(mod.file)
      let depPath = resolve.sync(dep, { basedir, extensions })
      yield* _getModuleDependencies(depPath)
    } catch {}
  }
}

export function getModuleDependencies(entryFile: string): string[] {
  return Array.from(_getModuleDependencies(entryFile))
    .filter((file) => file !== entryFile)
    .map((file) => normalizePath(file))
}