@agentscope-ai/i18n 1.0.1 → 1.0.2

package/lib/cli.js

@@ -2,10 +2,7 @@
 
 const { program } = require('commander');
 const {
-  init,
-  initMT,
-  patch,
-  patchMT,
+  runTranslation,
   check,
   reverse,
   translate,
@@ -15,11 +12,9 @@ const {
 const path = require('path');
 const fs = require('fs-extra');
 
-// 读取配置文件
 const getConfig = (options) => {
   let configPath = options.config;
 
-  // 如果没有指定配置文件路径，按优先级查找
   if (!configPath) {
     const configFiles = ['i18n.config.js', 'i18n.config.cjs', 'i18n.config.mjs'];
 
@@ -35,31 +30,25 @@ const getConfig = (options) => {
       throw new Error(`配置文件不存在，请在项目根目录创建 i18n.config.js 或 i18n.config.cjs`);
     }
   } else {
-    // 如果指定了配置文件路径，检查是否存在
     if (!fs.existsSync(configPath)) {
       throw new Error(`配置文件 ${configPath} 不存在`);
     }
   }
 
-  const config = require(configPath);
-  return config;
+  return require(configPath);
 };
 
-// 解析/覆盖 targetPath：命令行优先，其次配置文件
 const resolveTargetPath = (options, config) => {
   const fromCli = options.targetPath || options['target-path'];
-  const fromConfig = config && config.targetPath;
-  const raw = fromCli || fromConfig;
+  const raw = fromCli || (config && config.targetPath);
 
   if (!raw) {
     throw new Error('targetPath 未配置：请在配置文件中提供 targetPath，或通过 --target-path 传入');
   }
 
-  // 统一将相对路径转换为绝对路径（相对于当前工作目录）
   return path.isAbsolute(raw) ? raw : path.resolve(process.cwd(), raw);
 };
 
-// 将命令行通用选项统一解析并合并到 config 中
 const resolveCommonOptions = (options, config) => {
   const targetPath = resolveTargetPath(options, config);
   config.targetPath = targetPath;
@@ -67,187 +56,119 @@ const resolveCommonOptions = (options, config) => {
   return targetPath;
 };
 
-// 为每个命令添加通用入参：config + target-path + skip-confirm
 const withCommonOptions = (cmd) =>
   cmd
     .option('-c, --config <path>', 'Path to config file')
     .option('-p, --target-path <path>', 'Override targetPath in config file')
     .option('--skip-confirm', 'Skip all confirmation prompts, auto confirm (y/n) questions');
 
-program
-  .name('agentscope-ai-i18n')
-  .description('A tool for translating Chinese content in frontend code repositories')
-  .version('0.1.0');
-
-withCommonOptions(
-  program
-    .command('init-agent')
-    .description('Initialize translation for the project using Bailian Agent'),
-).action(async (options) => {
-  try {
-    const config = getConfig(options);
-    const targetPath = resolveCommonOptions(options, config);
-    await init(targetPath, config);
-  } catch (error) {
-    console.error('初始化失败:', error.message);
-    process.exit(1);
-  }
-});
-
-withCommonOptions(
-  program
-    .command('init-mt')
-    .description('Initialize translation using MT (Machine Translation) method'),
-).action(async (options) => {
-  try {
-    const config = getConfig(options);
-    const targetPath = resolveCommonOptions(options, config);
-    await initMT(targetPath, config);
-  } catch (error) {
-    console.error('MT初始化失败:', error.message);
-    process.exit(1);
-  }
-});
-
-withCommonOptions(
-  program
-    .command('init')
-    .description('Initialize translation using MT (Machine Translation) method'),
-).action(async (options) => {
-  try {
-    const config = getConfig(options);
-    const targetPath = resolveCommonOptions(options, config);
-    await initMT(targetPath, config);
-  } catch (error) {
-    console.error('MT初始化失败:', error.message);
-    process.exit(1);
-  }
-});
-
-withCommonOptions(
-  program
-    .command('patch-agent')
-    .description('Update translations for new content using Bailian Agent'),
-).action(async (options) => {
-  try {
-    const config = getConfig(options);
-    const targetPath = resolveCommonOptions(options, config);
-    await patch(targetPath, config);
-  } catch (error) {
-    console.error('补丁更新失败:', error.message);
-    process.exit(1);
-  }
-});
-
-withCommonOptions(
-  program
-    .command('patch-mt')
-    .description('Update translations for new content using MT (Machine Translation) method'),
-).action(async (options) => {
-  try {
-    const config = getConfig(options);
-    const targetPath = resolveCommonOptions(options, config);
-    await patchMT(targetPath, config);
-  } catch (error) {
-    console.error('MT补丁更新失败:', error.message);
-    process.exit(1);
-  }
-});
-
-withCommonOptions(
-  program
-    .command('patch')
-    .description('Update translations for new content using MT (Machine Translation) method'),
-).action(async (options) => {
-  try {
-    const config = getConfig(options);
-    const targetPath = resolveCommonOptions(options, config);
-    await patchMT(targetPath, config);
-  } catch (error) {
-    console.error('MT补丁更新失败:', error.message);
-    process.exit(1);
-  }
-});
-
-withCommonOptions(program.command('check').description('Check translation status'))
-  .option('--auto-delete-unused', 'Automatically delete unused translation keys')
-  .option(
-    '--summary-only',
-    'Only output missing key counts per language, skip deletion and file generation',
-  )
-  .action(async (options) => {
+/**
+ * 注册一个带通用选项的命令，自动处理 config 加载、targetPath 解析和错误处理。
+ */
+const registerCommand = (name, description, handler, { extraOptions = [], useCommon = true } = {}) => {
+  const cmd = program.command(name).description(description);
+  if (useCommon) withCommonOptions(cmd);
+  extraOptions.forEach((opt) => cmd.option(...opt));
+  cmd.action(async (options) => {
     try {
-      const config = getConfig(options);
-      config.autoDeleteUnused = options.autoDeleteUnused;
-      config.summaryOnly = options.summaryOnly;
-      const targetPath = resolveCommonOptions(options, config);
-      await check(targetPath, config);
+      if (useCommon) {
+        const config = getConfig(options);
+        resolveCommonOptions(options, config);
+        await handler(config, options);
+      } else {
+        await handler(options);
+      }
     } catch (error) {
-      console.error('检查失败:', error.message);
+      console.error(`❌ ${description}失败:`, error.message);
       process.exit(1);
     }
   });
+  return cmd;
+};
 
-withCommonOptions(
-  program.command('reverse').description('Reverse internationalized code back to Chinese'),
-).action(async (options) => {
-  try {
-    const config = getConfig(options);
-    const targetPath = resolveCommonOptions(options, config);
-    await reverse(targetPath, config);
-  } catch (error) {
-    console.error('还原失败:', error.message);
-    process.exit(1);
-  }
-});
-
-withCommonOptions(
-  program
-    .command('translate')
-    .description(
-      'Extract Chinese text, translate, and generate locale files + mds-payload.json (Step 1/3, no code modification)',
-    )
-    .option(
-      '-f, --file <path>',
-      'Process a single file instead of the entire targetPath directory',
-    ),
-).action(async (options) => {
-  try {
-    const config = getConfig(options);
-    const targetPath = resolveCommonOptions(options, config);
-    if (options.file) {
-      config.file = options.file;
-    }
-    await translate(targetPath, config);
-  } catch (error) {
-    console.error('translate 失败:', error.message);
-    process.exit(1);
-  }
-});
-
-withCommonOptions(
-  program
-    .command('replace')
-    .description(
-      'Replace Chinese text in code with $i18n.get calls using saved translation state (Step 3/3)',
-    )
-    .option(
-      '-f, --file <path>',
-      'Process a single file instead of the entire targetPath directory',
-    ),
-).action(async (options) => {
-  try {
-    const config = getConfig(options);
-    const targetPath = resolveCommonOptions(options, config);
-    if (options.file) {
-      config.file = options.file;
-    }
-    await replaceOnly(targetPath, config);
-  } catch (error) {
-    console.error('replace 失败:', error.message);
-    process.exit(1);
-  }
-});
+program
+  .name('i18n')
+  .description('A tool for translating Chinese content in frontend code repositories')
+  .version('1.0.1');
+
+// --- 核心翻译命令 ---
+
+registerCommand(
+  'init',
+  '初始化翻译',
+  (config, options) => runTranslation(config.targetPath, config, {
+    isUpdate: false,
+    engine: options.engine || 'mt',
+  }),
+  { extraOptions: [['--engine <engine>', 'Translation engine: mt (default) or agent', 'mt']] },
+);
+
+registerCommand(
+  'patch',
+  '增量翻译',
+  (config, options) => runTranslation(config.targetPath, config, {
+    isUpdate: true,
+    engine: options.engine || 'mt',
+  }),
+  { extraOptions: [['--engine <engine>', 'Translation engine: mt (default) or agent', 'mt']] },
+);
+
+registerCommand(
+  'check',
+  '检查翻译状态',
+  (config, options) => {
+    config.autoDeleteUnused = options.autoDeleteUnused;
+    config.summaryOnly = options.summaryOnly;
+    return check(config.targetPath, config);
+  },
+  {
+    extraOptions: [
+      ['--auto-delete-unused', 'Automatically delete unused translation keys'],
+      ['--summary-only', 'Only output missing key counts per language, skip deletion and file generation'],
+    ],
+  },
+);
+
+registerCommand('reverse', '还原国际化代码为中文', (config) =>
+  reverse(config.targetPath, config),
+);
+
+registerCommand(
+  'translate',
+  '提取中文并翻译生成locale文件 (Step 1/3, 不修改源码)',
+  (config, options) => {
+    if (options.file) config.file = options.file;
+    return translate(config.targetPath, config);
+  },
+  { extraOptions: [['-f, --file <path>', 'Process a single file instead of the entire targetPath directory']] },
+);
+
+registerCommand(
+  'replace',
+  '用翻译状态替换代码中的中文 (Step 3/3)',
+  (config, options) => {
+    if (options.file) config.file = options.file;
+    return replaceOnly(config.targetPath, config);
+  },
+  { extraOptions: [['-f, --file <path>', 'Process a single file instead of the entire targetPath directory']] },
+);
+
+// --- 向后兼容别名（隐藏，不在 help 中显示） ---
+
+const registerHiddenAlias = (name, target) => {
+  program.command(name, { hidden: true }).action(() => {
+    console.log(`⚠️  "${name}" 已弃用，请使用 "${target}" 代替`);
+    process.argv[2] = target;
+    program.parse(process.argv);
+  });
+};
+
+registerHiddenAlias('init-mt', 'init');
+registerHiddenAlias('init-agent', 'init --engine agent');
+registerHiddenAlias('patch-mt', 'patch');
+registerHiddenAlias('patch-agent', 'patch --engine agent');
+
+// --- 工具命令 ---
 
 program
   .command('init-config')
@@ -269,14 +190,12 @@ program
 
       fs.copySync(templatePath, configPath);
 
-      // 检查 .env 文件是否存在
       const envPath = path.resolve(cwd, '.env');
       if (!fs.existsSync(envPath)) {
         fs.writeFileSync(envPath, 'DASHSCOPE_APIKEY=\n', 'utf8');
         console.log('📝 已创建 .env 文件，请填写 DASHSCOPE_APIKEY');
       }
 
-      // 检查 .gitignore 中是否包含 .env
       const gitignorePath = path.resolve(cwd, '.gitignore');
       if (fs.existsSync(gitignorePath)) {
         const gitignoreContent = fs.readFileSync(gitignorePath, 'utf8');
@@ -313,7 +232,6 @@ program
     try {
       const cwd = process.cwd();
 
-      // 读取配置
       let config = {};
       try {
         config = getConfig(options);
@@ -339,13 +257,10 @@ program
         process.exit(1);
       }
       if (!appId) {
-        console.error(
-          '❌ 缺少 appId，请通过 --app-id 参数传入，或在 i18n.config.js 的 medusa.appId 中配置',
-        );
+        console.error('❌ 缺少 appId，请通过 --app-id 参数传入，或在 i18n.config.js 的 medusa.appId 中配置');
         process.exit(1);
       }
 
-      // 各 IDE/Agent 的 skills 目录映射
       const agentDirMap = {
         cursor: '.cursor/skills',
         qoder: '.qoder/skills',
@@ -354,10 +269,8 @@ program
         opencode: '.opencode/skill',
       };
 
-      // 确定目标 agents
       let targetAgents = options.agent || [];
       if (targetAgents.length === 0) {
-        // 自动检测：目录已存在的 IDE 默认安装，否则至少安装 cursor
         targetAgents = Object.entries(agentDirMap)
           .filter(([, dir]) => fs.existsSync(path.resolve(cwd, dir.split('/')[0])))
           .map(([name]) => name);
@@ -366,7 +279,6 @@ program
         }
       }
 
-      // 读取模板
       const skillSource = path.join(__dirname, '../skills/i18n-helper/SKILL.md');
       if (!fs.existsSync(skillSource)) {
         throw new Error(`SKILL.md 模板未找到: ${skillSource}`);
@@ -379,7 +291,6 @@ program
       content = content.replace(/\{\{LOCALES_FILE_PATH\}\}/g, localesFilePath);
       content = content.replace(/\{\{LANGUAGES\}\}/g, languages.join('、'));
 
-      // 写入各 IDE 目录
       const installed = [];
       for (const agent of targetAgents) {
         const baseDir = agentDirMap[agent];
@@ -428,7 +339,6 @@ program
       console.log('🚀 Medusa 国际化文件拉取工具');
       console.log('='.repeat(50));
 
-      // 尝试读取配置文件
       let configFileSettings = {};
       try {
         const configFromFile = getConfig(options);
@@ -441,50 +351,32 @@ program
 
       const appName = options.app || configFileSettings.appName;
       if (!appName) {
-        console.error(
-          '❌ 缺少 Medusa 应用名称 appName：请通过 --app 传入，或在 i18n.config.js 的 medusa.appName 中配置',
-        );
+        console.error('❌ 缺少 Medusa 应用名称 appName：请通过 --app 传入，或在 i18n.config.js 的 medusa.appName 中配置');
         process.exit(1);
       }
 
-      // tag 选填：命令行优先，其次配置文件，不设默认值
       const tag = options.tag ?? configFileSettings.tag;
 
-      // 合并配置文件和命令行参数（命令行参数优先）
       const config = {
-        medusa: {
-          appName,
-          tag,
-        },
+        medusa: { appName, tag },
         type: options.type || configFileSettings.type || 'json',
         cwd: process.cwd(),
         path: options.output || configFileSettings.outputPath || './locales',
         mergeExisting: options.merge !== false && configFileSettings.mergeExisting !== false,
       };
 
-      // 显示配置信息
       console.log('📝 配置信息:');
-      console.log(
-        `   应用名称: ${config.medusa.appName} ${options.app ? '(命令行)' : '(配置文件)'}`,
-      );
+      console.log(`   应用名称: ${config.medusa.appName} ${options.app ? '(命令行)' : '(配置文件)'}`);
       const tagSource =
         options.tag != null && options.tag !== ''
           ? '(命令行)'
           : configFileSettings.tag != null && configFileSettings.tag !== ''
             ? '(配置文件)'
             : '';
-      console.log(
-        `   标签名称: ${config.medusa.tag ?? '(未设置)'}${tagSource ? ` ${tagSource}` : ''}`,
-      );
-      console.log(
-        `   包类型: ${config.type} ${options.type ? '(命令行)' : configFileSettings.type ? '(配置文件)' : '(默认)'}`,
-      );
-      console.log(
-        `   输出目录: ${path.resolve(config.path)} ${options.output ? '(命令行)' : configFileSettings.outputPath ? '(配置文件)' : '(默认)'}`,
-      );
-      console.log(
-        `   合并现有文件: ${config.mergeExisting ? '是' : '否'} ${options.merge === false ? '(命令行)' : configFileSettings.mergeExisting !== undefined ? '(配置文件)' : '(默认)'}`,
-      );
+      console.log(`   标签名称: ${config.medusa.tag ?? '(未设置)'}${tagSource ? ` ${tagSource}` : ''}`);
+      console.log(`   包类型: ${config.type} ${options.type ? '(命令行)' : configFileSettings.type ? '(配置文件)' : '(默认)'}`);
+      console.log(`   输出目录: ${path.resolve(config.path)} ${options.output ? '(命令行)' : configFileSettings.outputPath ? '(配置文件)' : '(默认)'}`);
+      console.log(`   合并现有文件: ${config.mergeExisting ? '是' : '否'} ${options.merge === false ? '(命令行)' : configFileSettings.mergeExisting !== undefined ? '(配置文件)' : '(默认)'}`);
       console.log('');
 
       console.log('🔄 开始从 Medusa 平台拉取文件...');
@@ -495,7 +387,6 @@ program
     } catch (error) {
       console.error('❌ Medusa 拉取失败:', error.message);
 
-      // 提供更详细的错误信息
       if (error.response) {
         console.error('   HTTP 状态码:', error.response.status);
         if (error.response.status === 404) {

package/lib/core/ast-processor.js

@@ -2,30 +2,26 @@ const babylon = require('@babel/parser');
 const traverse = require('@babel/traverse').default;
 const generate = require('@babel/generator').default;
 const t = require('@babel/types');
+const { isMatchingCallee } = require('../utils/ast-utils');
 
 // 检查i18n调用中id和dm是否匹配
-const checkI18nCall = (node) => {
-  if (
-    node.type === 'CallExpression' &&
-    node.callee.type === 'MemberExpression' &&
-    node.callee.object &&
-    node.callee.object.name === '$i18n' &&
-    node.callee.property &&
-    node.callee.property.name === 'get'
-  ) {
-    const args = node.arguments;
-    if (args.length > 0 && args[0].type === 'ObjectExpression') {
-      const idProperty = args[0].properties.find((prop) => prop.key && prop.key.name === 'id');
-      const dmProperty = args[0].properties.find((prop) => prop.key && prop.key.name === 'dm');
+const checkI18nCall = (node, callExpr) => {
+  if (!isMatchingCallee(node, callExpr)) {
+    return null;
+  }
+
+  const args = node.arguments;
+  if (args.length > 0 && args[0].type === 'ObjectExpression') {
+    const idProperty = args[0].properties.find((prop) => prop.key && prop.key.name === 'id');
+    const dmProperty = args[0].properties.find((prop) => prop.key && prop.key.name === 'dm');
 
-      if (
-        idProperty?.value?.type === 'StringLiteral' &&
-        dmProperty?.value?.type === 'StringLiteral'
-      ) {
-        const id = idProperty.value.value;
-        const dm = dmProperty.value.value;
-        return { id, dm };
-      }
+    if (
+      idProperty?.value?.type === 'StringLiteral' &&
+      dmProperty?.value?.type === 'StringLiteral'
+    ) {
+      const id = idProperty.value.value;
+      const dm = dmProperty.value.value;
+      return { id, dm };
     }
   }
   return null;