migrate.ts

import fs from "fs";
import path from "path";

// Also need to find and replace `declare module ../../registry` and stand alone imports like `import "../../registry";`

const rootDir = (): string => path.join(__dirname);

updateImportsInWorkspaces();

function updateImportsInWorkspaces() {
  const dir = rootDir();
  const rootPackageJsonPath = path.join(dir, "package.json");
  const rootPackageJson = require(rootPackageJsonPath);
  rootPackageJson.workspaces.forEach((workspaceDir: string) => {
    const workspacePackageDir = path.join(dir, workspaceDir);
    console.log("Updating imports in", workspacePackageDir);
    updateTsFilesInDirectory(workspacePackageDir);
  });
}

function updateTsFilesInDirectory(workspaceDir: string) {
  findTsFiles(workspaceDir).forEach((file) => {
    const content = fs.readFileSync(file, "utf8");
    const updatedContent = updateImportPaths(file, content);
    fs.writeFileSync(file, updatedContent, "utf8");
  });
}

function findTsFiles(dir: string): string[] {
  const contents = fs.readdirSync(dir).map((file) => path.join(dir, file));
  const dirs = contents.filter((filePath) => fs.statSync(filePath).isDirectory());

  // Recurse first, then look for imports in the current directory
  const nestedFiles = dirs.flatMap((dir) => findTsFiles(dir));

  const currentFiles = contents.filter(
    (filePath) => fs.statSync(filePath).isFile() && path.basename(filePath).endsWith(".ts"),
  )!;

  return [...nestedFiles, ...currentFiles];
}

// Function to modify the import/export statements
function updateImportPaths(filePath: string, fileContent: string) {
  return fileContent.replace(/from\s+['"]([^'"]*)['"]/g, (match, p1) => {
    // Check if the path is relative
    if (!p1.startsWith(".") && !p1.startsWith("/")) {
      return match; // Skip if the import is not a relative path
    }

    // TODO: clobber flag?
    // Check if the path already has a file extension
    if (path.extname(p1)) {
      return match; // Skip if the import statement has an extension
    }

    const maybeDir = path.join(path.dirname(filePath), p1);
    try {
      if (fs.statSync(maybeDir).isDirectory()) {
        // path.join strips ./ from the beginning of the path
        p1 = "./" + path.join(p1, "index");
      }
    } catch (e) {
      console.error("failed to stat", maybeDir);
    }

    if (p1 === ".") p1 = "./";
    if (p1.endsWith("/")) p1 += "index";

    return `from '${p1}.js'`; // Append .mjs to the import path
  });
}