SWC中基于JavaScript/TypeScript的AST操作实现代码转换

import swc from '@swc/core';

// 待转换的源代码
let source = `
import abc from './abc.ts'
import def from "./def.mts"
import('./ghi.cts')
const foo = require('./bar.ts')
`;

async function transformImports(code: string) {
  // 1. 解析源代码为AST
  let program = await swc.parse(code, {
    syntax: 'typescript', // 指定源代码语法为TypeScript
    comments: false,     // 不保留注释
    script: true,        // 视为脚本文件
    target: 'esnext',    // 目标ES版本
  });

  // 2. 实现自定义的插件逻辑：遍历并修改AST
  // AST的根节点是Program，其body属性包含了顶层语句数组
  for (let item of program.body) {
    switch (item.type) {
      // 处理 ImportDeclaration 类型的节点（如 import abc from './abc.ts'）
      case 'ImportDeclaration':
        // 导入声明的source属性包含导入路径的字面量信息
        // raw属性是原始的字符串表示，如 "'./abc.ts'"
        if (item.source && item.source.raw) {
          item.source.raw = item.source.raw
            .replace('.ts"', '.js"')
            .replace('.mts"', '.js"')
            .replace('.cts"', '.js"');
        }
        break;
      // 处理 ExpressionStatement 类型的节点，例如动态导入 import('./ghi.cts')
      case 'ExpressionStatement':
        // 检查表达式是否为 CallExpression (函数调用)
        if (item.expression.type === 'CallExpression') {
          const callExpr = item.expression;
          // 检查调用的callee是否为Import (动态导入)
          if (callExpr.callee.type === 'Import') {
            // 动态导入的参数是数组，通常第一个参数是导入路径
            if (callExpr.arguments.length > 0 && callExpr.arguments[0].expression.type === 'StringLiteral') {
              const arg = callExpr.arguments[0].expression;
              arg.raw = arg.raw
                .replace('.ts"', '.js"')
                .replace('.mts"', '.js"')
                .replace('.cts"', '.js"');
            }
          }
        }
        break;
      // 针对 CommonJS require() 的处理（虽然SWC主要用于ESM，但也可处理）
      case 'VariableDeclaration':
        for (const declarator of item.declarations) {
          if (declarator.init && declarator.init.type === 'CallExpression' &&
              declarator.init.callee.type === 'Identifier' && declarator.init.callee.value === 'require') {
            if (declarator.init.arguments.length > 0 && declarator.init.arguments[0].expression.type === 'StringLiteral') {
              const arg = declarator.init.arguments[0].expression;
              arg.raw = arg.raw
                .replace('.ts"', '.js"')
                .replace('.mts"', '.js"')
                .replace('.cts"', '.js"');
            }
          }
        }
        break;
      default:
        // 打印其他类型的节点，以便了解AST结构
        // console.log(`Unhandled node type: ${item.type}`);
    }
  }

  // 3. 将修改后的AST转换回代码
  const transformedCode = await swc.transform(program, {
    jsc: {
      target: 'esnext', // 目标ES版本
      // minify: {
      //   // 注意：SWC的minify功能在某些版本可能存在已知问题，
      //   // 使用时请谨慎测试或避免在此阶段使用
      //   // compress: {},
      //   mangle: {},
      // },
    },
  });

  return transformedCode.code;
}

// 执行转换并打印结果
transformImports(source).then(result => {
  console.log('--- Original Code ---');
  console.log(source);
  console.log('\n--- Transformed Code ---');
  console.log(result);
}).catch(error => {
  console.error('Transformation failed:', error);
});