atbb / packages / lexicon / scripts / fix-imports.ts

#!/usr/bin/env tsx
/**
 * Post-processing script to add .js extensions to relative imports in compiled JavaScript files.
 * Required because @atproto/lex-cli generates TypeScript without file extensions,
 * but Node ESM requires explicit extensions for relative imports.
 */
import { readdir, readFile, writeFile } from "node:fs/promises";
import { join } from "node:path";

async function* walkDir(dir: string): AsyncGenerator<string> {
  const entries = await readdir(dir, { withFileTypes: true });
  for (const entry of entries) {
    const path = join(dir, entry.name);
    if (entry.isDirectory()) {
      yield* walkDir(path);
    } else if (entry.isFile() && path.endsWith(".js")) {
      yield path;
    }
  }
}

async function fixImports(filePath: string): Promise<void> {
  const content = await readFile(filePath, "utf-8");

  // Match relative imports without .js extension
  // Examples: from '../../../lexicons' -> from '../../../lexicons.js'
  //          from './util' -> from './util.js'
  const fixed = content.replace(
    /from\s+(['"])(\.\.?\/.+?)\1/g,
    (match, quote, importPath) => {
      // Skip if already has extension
      if (importPath.endsWith(".js") || importPath.endsWith(".json")) {
        return match;
      }
      return `from ${quote}${importPath}.js${quote}`;
    }
  );

  if (fixed !== content) {
    await writeFile(filePath, fixed, "utf-8");
    console.log(`Fixed imports in: ${filePath}`);
  }
}

async function main() {
  const distDir = join(process.cwd(), "dist/types");
  console.log(`Fixing relative imports in ${distDir}...`);

  let count = 0;
  for await (const file of walkDir(distDir)) {
    await fixImports(file);
    count++;
  }

  console.log(`Processed ${count} JavaScript files.`);
}

main().catch((error) => {
  console.error("Failed to fix imports:", error);
  process.exit(1);
});