barrel-import-to-full-import.ts

import { Project, SyntaxKind } from "ts-morph";

/**
 * This script converts all "barrel" imports to fully qualified imports.
 *
 * Use 'bun' to run the script:
 *   bun run morphs/fully-qualified-gen-imports.ts
 */

const project = new Project({
  tsConfigFilePath: "tsconfig.json",
});

const barrelFile = project.getSourceFileOrThrow("src/gen/model/index.ts");

const exportedModules = barrelFile
  .getExportDeclarations()
  .map((exp) => {

    const referencedSourceFile = exp.getModuleSpecifierSourceFileOrThrow();

    const moduleSpecifier = project
      .getRootDirectories()[0]
      .getRelativePathAsModuleSpecifierTo(referencedSourceFile.getFilePath())
      // There's probably some way to resolve this using the "paths" in tsconfig but I couldn't figure it out
      .substring("./src/".length);

    const namedExports = referencedSourceFile.getExportedDeclarations();

    return Array.from(namedExports.keys()).map((name) => ({
      name,
      moduleSpecifier,
    }));
  })
  .flat();

project.getSourceFiles().forEach((sourceFile) => {
  sourceFile.getImportDeclarations().forEach((importDeclaration) => {
    if (importDeclaration.getModuleSpecifierValue() === "gen/model") {
      const namedImports = importDeclaration.getNamedImports();
      namedImports.forEach((namedImport) => {
        const correspondingExport = exportedModules.find(
          (exp) => exp.name === namedImport.getName(),
        );
        if (correspondingExport) {
          sourceFile.addImportDeclaration({
            moduleSpecifier: correspondingExport.moduleSpecifier,
            namedImports: [namedImport.getName()],
          });
        }
      });
      importDeclaration.remove();
    }
  });
  sourceFile.saveSync();
});