Function collectModuleExports

packages/mocker/src/node/parsers.ts:33–124

(
  filename: string,
  code: string,
  format: 'module' | 'commonjs',
  exports: string[] = [],
)

const cachedFileExports = new Map<string, string[]>()

export function collectModuleExports(
  filename: string,
  code: string,
  format: 'module' | 'commonjs',
  exports: string[] = [],
): string[] {
  if (format === 'module') {
    const [imports_, exports_] = parseModuleSyntax(code, filename)
    const fileExports = [...exports_.map(p => p.n)]
    imports_.forEach(({ ss: start, se: end, n: name }) => {
      const substring = code.substring(start, end).replace(/ +/g, ' ')
      if (name && substring.startsWith('export *') && !substring.startsWith('export * as')) {
        fileExports.push(...tryParseModule(name))
      }
    })
    cachedFileExports.set(filename, fileExports)
    exports.push(...fileExports)
  }
  else {
    const { exports: exports_, reexports } = parseCjsSyntax(code, filename)
    const fileExports = [...exports_]
    reexports.forEach((name) => {
      fileExports.push(...tryParseModule(name))
    })
    cachedFileExports.set(filename, fileExports)
    exports.push(...fileExports)
  }

  function tryParseModule(name: string): string[] {
    try {
      return parseModule(name)
    }
    catch (error) {
      console.warn(`[module mocking] Failed to parse '${name}' imported from ${filename}:`, error)
      return []
    }
  }

  let __require: NodeJS.Require | undefined
  function getModuleRequire() {
    return (__require ??= createRequire(filename))
  }

  function parseModule(name: string): string[] {
    if (isBuiltin(name)) {
      if (cachedFileExports.has(name)) {
        const cachedExports = cachedFileExports.get(name)!
        return cachedExports
      }

      const builtinModule = getBuiltinModule(name)
      const builtinExports = Object.keys(builtinModule)
      cachedFileExports.set(name, builtinExports)
      return builtinExports
    }

    const resolvedModuleUrl = format === 'module'
      ? import.meta.resolve(name, pathToFileURL(filename).toString())