trufflesuite/truffle

import debugModule from "debug";
const debug = debugModule("profiler:getImports");

import path from "path";

//HACK: do *not* import ResolvedSource from @truffle/resolver because
//that would create a circular dependency!
//don't import ResolverSource for the same reasons
export interface ResolvedSource {
  filePath: string;
  body: string;
  source: {
    resolveDependencyPath(importPath: string, dependencyPath: string): string;
  };
}

export interface GetImportsOptions {
  source: ResolvedSource;
  parseImports(body: string): Promise<string[]>;
  shouldIncludePath(filePath: string): boolean;
}

export async function getImports({
  source: { filePath, body, source },
  shouldIncludePath,
  parseImports
}: GetImportsOptions): Promise<string[]> {
  if (!shouldIncludePath(filePath) || !parseImports) return [];

  debug("filePath: %s", filePath);

  const imports = await parseImports(body);

  debug("imports: %O", imports);

  // Convert relative dependencies of modules back into module paths.
  // note: the check for what's a relative dependency has been removed from
  // here, that's now the responsibility of the individual resolverSource to check
  return (await Promise.all(imports.map(
    dependencyPath => source.resolveDependencyPath(filePath, dependencyPath)
  ))).filter(sourcePath => sourcePath) //filter out Vyper failures
  .map(sourcePath => sourcePath.replace(/\//g, path.sep)); //make sure to use
  //backslash on Windows (for same reason as in requiredSources.ts)
}