Function visitor

scripts/build.ts:23–87

(node)

  context,
) => {
  const visitor: ts.Visitor = (node) => {
    // 1. Handle Static Imports/Exports (import ... from "./x.ts")
    if (ts.isImportDeclaration(node) || ts.isExportDeclaration(node)) {
      if (node.moduleSpecifier && ts.isStringLiteral(node.moduleSpecifier)) {
        const text = node.moduleSpecifier.text;
        if (text.endsWith(".ts")) {
          const newSpecifier = ts.factory.createStringLiteral(
            text.replace(/\.ts$/, ".js"),
          );

          if (ts.isImportDeclaration(node)) {
            return ts.factory.updateImportDeclaration(
              node,
              node.modifiers,
              node.importClause,
              newSpecifier,
              node.attributes,
            );
          } else {
            return ts.factory.updateExportDeclaration(
              node,
              node.modifiers,
              node.isTypeOnly,
              node.exportClause,
              newSpecifier,
              node.attributes,
            );
          }
        }
      }
    }

    // 2. Handle Dynamic Worker URL (new URL("./worker.ts", ...))
    if (ts.isNewExpression(node)) {
      // Check if the expression being new'ed is "URL"
      if (ts.isIdentifier(node.expression) && node.expression.text === "URL") {
        if (node.arguments && node.arguments.length > 0) {
          const firstArg = node.arguments[0]!;

          // Check if first arg is a string literal ending in .ts
          if (ts.isStringLiteral(firstArg) && firstArg.text.endsWith(".ts")) {
            console.log(
              `[Transform] Rewriting new URL("${firstArg.text}") -> .js`,
            );

            const newArg = ts.factory.createStringLiteral(
              firstArg.text.replace(/\.ts$/, ".js"),
            );

            // Create a new arguments array, preserving the second argument (import.meta.url)
            const newArguments = [newArg, ...node.arguments.slice(1)];

            return ts.factory.updateNewExpression(
              node,
              node.expression,
              node.typeArguments,
              newArguments,
            );