@agentscope-ai/i18n 0.1.2

package/lib/cli.js

@@ -0,0 +1,224 @@
+#!/usr/bin/env node
+
+const { program } = require('commander');
+const { init, initMT, patch, patchMT, check, reverse, medusaRun } = require('./index');
+const path = require('path');
+const fs = require('fs-extra');
+
+// 读取配置文件
+const getConfig = (options) => {
+  const configPath = options.config || path.resolve(process.cwd(), 'i18n.config.js');
+  if (!fs.existsSync(configPath)) {
+    throw new Error(`配置文件 ${configPath} 不存在`);
+  }
+  const config = require(configPath);
+  return config;
+};
+
+program
+  .name('agentscope-ai-i18n')
+  .description('A tool for translating Chinese content in frontend code repositories')
+  .version('0.1.0');
+
+program
+  .command('init-agent')
+  .description('Initialize translation for the project using Bailian Agent')
+  .option('-c, --config <path>', 'Path to config file')
+  .action(async (options) => {
+    try {
+      const config = getConfig(options);
+      await init(config.targetPath, config);
+    } catch (error) {
+      console.error('初始化失败:', error.message);
+      process.exit(1);
+    }
+  });
+
+program
+  .command('init-mt')
+  .description('Initialize translation using MT (Machine Translation) method')
+  .option('-c, --config <path>', 'Path to config file')
+  .action(async (options) => {
+    try {
+      const config = getConfig(options);
+      await initMT(config.targetPath, config);
+    } catch (error) {
+      console.error('MT初始化失败:', error.message);
+      process.exit(1);
+    }
+  });
+
+program
+  .command('init')
+  .description('Initialize translation using MT (Machine Translation) method')
+  .option('-c, --config <path>', 'Path to config file')
+  .action(async (options) => {
+    try {
+      const config = getConfig(options);
+      await initMT(config.targetPath, config);
+    } catch (error) {
+      console.error('MT初始化失败:', error.message);
+      process.exit(1);
+    }
+  });
+
+program
+  .command('patch-agent')
+  .description('Update translations for new content using Bailian Agent')
+  .option('-c, --config <path>', 'Path to config file')
+  .action(async (options) => {
+    try {
+      const config = getConfig(options);
+      await patch(config.targetPath, config);
+    } catch (error) {
+      console.error('补丁更新失败:', error.message);
+      process.exit(1);
+    }
+  });
+
+program
+  .command('patch-mt')
+  .description('Update translations for new content using MT (Machine Translation) method')
+  .option('-c, --config <path>', 'Path to config file')
+  .action(async (options) => {
+    try {
+      const config = getConfig(options);
+      await patchMT(config.targetPath, config);
+    } catch (error) {
+      console.error('MT补丁更新失败:', error.message);
+      process.exit(1);
+    }
+  });
+
+program
+  .command('patch')
+  .description('Update translations for new content using MT (Machine Translation) method')
+  .option('-c, --config <path>', 'Path to config file')
+  .action(async (options) => {
+    try {
+      const config = getConfig(options);
+      await patchMT(config.targetPath, config);
+    } catch (error) {
+      console.error('MT补丁更新失败:', error.message);
+      process.exit(1);
+    }
+  });
+
+program
+  .command('check')
+  .description('Check translation status')
+  .option('-c, --config <path>', 'Path to config file')
+  .option('--auto-delete-unused', 'Automatically delete unused translation keys')
+  .action(async (options) => {
+    try {
+      const config = getConfig(options);
+      // 传递auto-delete-unused参数到check方法
+      config.autoDeleteUnused = options.autoDeleteUnused;
+      await check(config.targetPath, config);
+    } catch (error) {
+      console.error('检查失败:', error.message);
+      process.exit(1);
+    }
+  });
+
+program
+  .command('reverse')
+  .description('Reverse internationalized code back to Chinese')
+  .option('-c, --config <path>', 'Path to config file')
+  .action(async (options) => {
+    try {
+      const config = getConfig(options);
+      await reverse(config.targetPath, config);
+    } catch (error) {
+      console.error('还原失败:', error.message);
+      process.exit(1);
+    }
+  });
+
+program
+  .command('medusa')
+  .description('Pull translation files from Medusa platform')
+  .option('-c, --config <path>', 'Path to config file')
+  .option('-o, --output <dir>', 'Output directory (overrides config)')
+  .option('-a, --app <name>', 'Medusa app name (overrides config)')
+  .option('-t, --tag <name>', 'Medusa tag name (overrides config)')
+  .option('--type <type>', 'Package type (json/android/ios) (overrides config)')
+  .option('--no-merge', 'Do not merge with existing files, overwrite directly')
+  .action(async (options) => {
+    try {
+      console.log('🚀 Medusa 国际化文件拉取工具');
+      console.log('='.repeat(50));
+
+      // 尝试读取配置文件
+      let configFileSettings = {};
+      try {
+        const configFromFile = getConfig(options);
+        configFileSettings = configFromFile.medusa || {};
+      } catch (error) {
+        console.log('⚠️  未找到配置文件，使用默认配置');
+        console.log(`   ${error.message}`);
+        console.log('');
+      }
+
+      // 合并配置文件和命令行参数（命令行参数优先）
+      const config = {
+        medusa: {
+          appName: options.app || configFileSettings.appName || 'broadscope-bailian-frontend',
+          tag: options.tag || configFileSettings.tag || 'bailian-fe',
+        },
+        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 ? '(命令行)' : configFileSettings.appName ? '(配置文件)' : '(默认)'}`,
+      );
+      console.log(
+        `   标签名称: ${config.medusa.tag} ${options.tag ? '(命令行)' : configFileSettings.tag ? '(配置文件)' : '(默认)'}`,
+      );
+      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 平台拉取文件...');
+      await medusaRun(config);
+      console.log('');
+      console.log('✅ 拉取成功！');
+      console.log(`📁 文件已保存到: ${path.resolve(config.path)}`);
+    } catch (error) {
+      console.error('❌ Medusa 拉取失败:', error.message);
+
+      // 提供更详细的错误信息
+      if (error.response) {
+        console.error('   HTTP 状态码:', error.response.status);
+        if (error.response.status === 404) {
+          console.error('   可能的原因:');
+          console.error('   - 应用名称不存在');
+          console.error('   - 标签名称不存在');
+          console.error('   - API 端点已更改');
+        }
+      }
+
+      if (error.code === 'ENOTFOUND' || error.code === 'ECONNREFUSED') {
+        console.error('   网络连接错误，请检查:');
+        console.error('   - 网络连接是否正常');
+        console.error('   - 是否需要连接内网或VPN');
+        console.error('   - 防火墙设置');
+      }
+
+      process.exit(1);
+    }
+  });
+
+program.parse();

package/lib/core/ast-processor.js

@@ -0,0 +1,75 @@
+const babylon = require('@babel/parser');
+const traverse = require('@babel/traverse').default;
+const generate = require('@babel/generator').default;
+const t = require('@babel/types');
+
+// 检查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');
+
+      if (
+        idProperty?.value?.type === 'StringLiteral' &&
+        dmProperty?.value?.type === 'StringLiteral'
+      ) {
+        const id = idProperty.value.value;
+        const dm = dmProperty.value.value;
+        return { id, dm };
+      }
+    }
+  }
+  return null;
+};
+
+const parseAST = (code) => {
+  return babylon.parse(code, {
+    sourceType: 'module',
+    allowImportExportEverywhere: true,
+    plugins: [
+      'jsx',
+      'objectRestSpread',
+      'classProperties',
+      'functionBind',
+      'typescript',
+      'decorators-legacy',
+      'dynamicImport',
+      'classPrivateProperties',
+      'classPrivateMethods',
+      'optionalChaining',
+      'nullishCoalescingOperator',
+    ].filter(Boolean),
+    tokens: true,
+    errorRecovery: true,
+  });
+};
+
+const generateCode = (ast, originalCode) => {
+  return generate(
+    ast,
+    {
+      jsescOption: {
+        minimal: true,
+        quotes: 'single',
+      },
+      retainLines: true,
+      compact: false,
+    },
+    originalCode,
+  ).code;
+};
+
+module.exports = {
+  checkI18nCall,
+  parseAST,
+  generateCode,
+};

package/lib/core/file-processor.js

@@ -0,0 +1,397 @@
+const find = require('find');
+const fs = require('fs-extra');
+const traverse = require('@babel/traverse').default;
+const t = require('@babel/types');
+const { parseAST, generateCode } = require('./ast-processor');
+const nodePath = require('path');
+const { injectImportStatement } = require('../utils/file-utils');
+const {
+  batchKeyTranslate,
+  batchTranslate,
+  singleKeyTranslate,
+  singleTranslate,
+} = require('../utils/translation-utils');
+const { extractStringValue } = require('../utils/ast-utils');
+const { askQuestion } = require('../utils/cli-utils');
+const { generateMedusaExcel } = require('../utils/excel-utils');
+const parseJSX = require('../parse-jsx');
+
+// 提取共用的文案处理函数
+const processFiles = async (files, params) => {
+  console.log(`    [开始] 提取和替换文件中的中文内容...`);
+
+  const { keyPrefix, functionImportPath } = params;
+  const fileCodeMap = new Map(); // 存储文件路径和替换后的代码的映射
+  const needImportFiles = new Set(); // 存储需要添加导入语句的文件
+
+  const promises = files.map((file) => {
+    if (/(\.js|\.ts|\.tsx|\.jsx)$/i.test(file)) {
+      return parseJSX(file, { ...params, keyPrefix }).then(({ i18n, newCode }) => {
+        if (Object.keys(i18n).length > 0 && functionImportPath) {
+          needImportFiles.add(file);
+        }
+        if (newCode) {
+          fileCodeMap.set(file, newCode);
+        }
+        return i18n;
+      });
+    }
+    return Promise.resolve({});
+  });
+
+  const results = await Promise.all(promises);
+  const i18n = {};
+  results.forEach((langObj) => {
+    Object.keys(langObj).forEach((lang) => {
+      i18n[lang] = langObj[lang];
+    });
+  });
+
+  return { i18n, fileCodeMap, needImportFiles };
+};
+
+// 提取共用的翻译和文件更新函数
+const translateAndUpdateFiles = async (
+  i18n,
+  path,
+  localesFilePath,
+  fileType,
+  fileCodeMap,
+  needImportFiles,
+  functionImportPath,
+  isUpdate = false,
+  dashScope,
+  medusa,
+) => {
+  if (Object.keys(i18n).length === 0) {
+    console.log('    [提示] 没有找到需要提取的文案');
+    return;
+  }
+
+  // 生成临时的key.json文件
+  const keyJsonPath = `${localesFilePath}/key.json`;
+  fs.ensureDirSync(localesFilePath);
+  fs.writeJsonSync(keyJsonPath, i18n, { spaces: 2 });
+  console.log(`    [进度] 生成临时key.json文件，共${Object.keys(i18n).length}条文案`);
+
+  // 调用key翻译
+  console.log(`    [进度] AI正在翻译key...请稍后`);
+  const { keyMap, zhCN } = await batchKeyTranslate(i18n, dashScope);
+
+  // 更新zh-cn.json
+  const zhCNPath = `${localesFilePath}/zh-cn.json`;
+  let existingZhCN = {};
+  if (isUpdate) {
+    try {
+      existingZhCN = fs.readJsonSync(zhCNPath);
+    } catch (error) {
+      console.log('    [提示] 未找到现有的zh-cn.json文件，将创建新文件');
+    }
+  }
+  const finalZhCN = isUpdate ? { ...existingZhCN, ...zhCN } : zhCN;
+  fs.writeJsonSync(zhCNPath, finalZhCN, { spaces: 2 });
+  console.log(`    [完成] ${isUpdate ? '更新' : '生成'}zh-cn.json文件成功`);
+
+  // 调用value翻译生成多语言文件
+  console.log(`    [进度] AI正在翻译value...请稍后`);
+  const translatedResults = await batchTranslate(zhCN, dashScope);
+
+  // 为每种语言生成对应的翻译文件
+  const generatedFiles = [];
+  for (const [language, translatedI18n] of Object.entries(translatedResults)) {
+    const languagePath = `${localesFilePath}/${language}.json`;
+    let existingLanguage = {};
+
+    if (isUpdate) {
+      try {
+        existingLanguage = fs.readJsonSync(languagePath);
+      } catch (error) {
+        console.log(`    [提示] 未找到现有的${language}.json文件，将创建新文件`);
+      }
+    }
+
+    const finalLanguage = isUpdate ? { ...existingLanguage, ...translatedI18n } : translatedI18n;
+    fs.writeJsonSync(languagePath, finalLanguage, { spaces: 2 });
+    console.log(`    [完成] ${isUpdate ? '更新' : '生成'}${language}.json文件成功`);
+    generatedFiles.push(languagePath);
+  }
+
+  // 提示用户检查翻译文件
+  console.log('\n=== 翻译文件已生成 ===');
+  console.log('请检查以下文件：');
+  console.log(`1. ${zhCNPath}`);
+  generatedFiles.forEach((file, index) => {
+    console.log(`${index + 2}. ${file}`);
+  });
+
+  const answer = await askQuestion('翻译文件检查无误后，是否继续替换代码中的文案？(y/n): ');
+  if (answer === 'y' || answer === 'yes') {
+    // 替换文件中的key
+    console.log(`    [进度] 开始替换文件中的key...`);
+    await replaceKeysInFiles(path, keyMap, fileType, fileCodeMap);
+
+    // 添加导入语句
+    if (needImportFiles.size > 0) {
+      console.log(`    [进度] 开始添加导入语句...`);
+      needImportFiles.forEach((file) => {
+        injectImportStatement(file, functionImportPath);
+      });
+      console.log(`    [完成] 已为${needImportFiles.size}个文件添加导入语句`);
+    }
+
+    console.log('    [完成] 代码替换完成！');
+  } else {
+    console.log('    [取消] 已取消代码替换操作');
+  }
+
+  // 询问是否生成medusa excel文件
+  if (medusa && medusa.appName && medusa.group && medusa.keyMap) {
+    const excelAnswer = await askQuestion('是否需要生成medusa excel文件？(y/n): ');
+    if (excelAnswer === 'y' || excelAnswer === 'yes') {
+      try {
+        const newKeys = Object.keys(zhCN); // 新增的key
+        await generateMedusaExcel({
+          localesFilePath,
+          medusa,
+          newKeys,
+        });
+      } catch (error) {
+        console.error('    [错误] 生成medusa excel文件失败:', error.message);
+      }
+    } else {
+      console.log('    [取消] 已取消生成medusa excel文件');
+    }
+  }
+
+  fs.removeSync(keyJsonPath);
+  console.log(`    [提示] key.json文件已保留用于调试`);
+
+  return Object.keys(zhCN).length;
+};
+
+// 单条翻译和文件更新函数（使用MT方式）
+const singleTranslateAndUpdateFiles = async (
+  i18n,
+  path,
+  localesFilePath,
+  fileType,
+  fileCodeMap,
+  needImportFiles,
+  functionImportPath,
+  isUpdate = false,
+  dashScope,
+  medusa,
+) => {
+  if (Object.keys(i18n).length === 0) {
+    console.log('    [提示] 没有找到需要提取的文案');
+    return;
+  }
+
+  // 生成临时的key.json文件
+  const keyJsonPath = `${localesFilePath}/key.json`;
+  fs.ensureDirSync(localesFilePath);
+  fs.writeJsonSync(keyJsonPath, i18n, { spaces: 2 });
+  console.log(`    [进度] 生成临时key.json文件，共${Object.keys(i18n).length}条文案`);
+
+  // 调用单条key翻译
+  console.log(`    [进度] 使用MT方式逐条翻译key...请稍后`);
+  const { keyMap, zhCN } = await singleKeyTranslate(i18n, dashScope);
+
+  // 更新zh-cn.json
+  const zhCNPath = `${localesFilePath}/zh-cn.json`;
+  let existingZhCN = {};
+  if (isUpdate) {
+    try {
+      existingZhCN = fs.readJsonSync(zhCNPath);
+    } catch (error) {
+      console.log('    [提示] 未找到现有的zh-cn.json文件，将创建新文件');
+    }
+  }
+  const finalZhCN = isUpdate ? { ...existingZhCN, ...zhCN } : zhCN;
+  fs.writeJsonSync(zhCNPath, finalZhCN, { spaces: 2 });
+  console.log(`    [完成] ${isUpdate ? '更新' : '生成'}zh-cn.json文件成功`);
+
+  // 调用单条value翻译生成多语言文件
+  console.log(`    [进度] 使用MT方式逐条翻译value...请稍后`);
+  const translatedResults = await singleTranslate(zhCN, dashScope);
+
+  // 为每种语言生成对应的翻译文件
+  const generatedFiles = [];
+  for (const [language, translatedI18n] of Object.entries(translatedResults)) {
+    const languagePath = `${localesFilePath}/${language}.json`;
+    let existingLanguage = {};
+
+    if (isUpdate) {
+      try {
+        existingLanguage = fs.readJsonSync(languagePath);
+      } catch (error) {
+        console.log(`    [提示] 未找到现有的${language}.json文件，将创建新文件`);
+      }
+    }
+
+    const finalLanguage = isUpdate ? { ...existingLanguage, ...translatedI18n } : translatedI18n;
+    fs.writeJsonSync(languagePath, finalLanguage, { spaces: 2 });
+    console.log(`    [完成] ${isUpdate ? '更新' : '生成'}${language}.json文件成功`);
+    generatedFiles.push(languagePath);
+  }
+
+  // 提示用户检查翻译文件
+  console.log('\n=== 翻译文件已生成（MT方式） ===');
+  console.log('请检查以下文件：');
+  console.log(`1. ${zhCNPath}`);
+  generatedFiles.forEach((file, index) => {
+    console.log(`${index + 2}. ${file}`);
+  });
+
+  const answer = await askQuestion('翻译文件检查无误后，是否继续替换代码中的文案？(y/n): ');
+  if (answer === 'y' || answer === 'yes') {
+    // 替换文件中的key
+    console.log(`    [进度] 开始替换文件中的key...`);
+    await replaceKeysInFiles(path, keyMap, fileType, fileCodeMap);
+
+    // 添加导入语句
+    if (needImportFiles.size > 0) {
+      console.log(`    [进度] 开始添加导入语句...`);
+      needImportFiles.forEach((file) => {
+        injectImportStatement(file, functionImportPath);
+      });
+      console.log(`    [完成] 已为${needImportFiles.size}个文件添加导入语句`);
+    }
+
+    console.log('    [完成] 代码替换完成！');
+  } else {
+    console.log('    [取消] 已取消代码替换操作');
+  }
+
+  // 询问是否生成medusa excel文件
+  if (medusa && medusa.appName && medusa.group && medusa.keyMap) {
+    const excelAnswer = await askQuestion('是否需要生成medusa excel文件？(y/n): ');
+    if (excelAnswer === 'y' || excelAnswer === 'yes') {
+      try {
+        const newKeys = Object.keys(zhCN); // 新增的key
+        await generateMedusaExcel({
+          localesFilePath,
+          medusa,
+          newKeys,
+        });
+      } catch (error) {
+        console.error('    [错误] 生成medusa excel文件失败:', error.message);
+      }
+    } else {
+      console.log('    [取消] 已取消生成medusa excel文件');
+    }
+  }
+
+  fs.removeSync(keyJsonPath);
+  console.log(`    [提示] key.json文件已删除`);
+
+  return Object.keys(zhCN).length;
+};
+
+// 修改replaceKeysInFiles函数以支持fileCodeMap
+const replaceKeysInFiles = async (path, keyMap, fileType, fileCodeMap) => {
+  const fileTypes = fileType.split(',').map((s) => s.trim());
+  const re = new RegExp(`(${fileTypes.map((ft) => ft.replace('.', '\\.')).join('|')})$`);
+  const files = find.fileSync(re, path);
+
+  console.log(`    [进度] 开始替换${files.length}个文件中的key...`);
+
+  // 仅当目标项目根目录存在 .prettierrc 时才尝试格式化
+  const hasPrettierRc = fs.existsSync(nodePath.join(path, '.prettierrc'));
+
+  // 动态导入 Prettier 并应用项目配置（如果可用）
+  const formatWithPrettier = async (text, filePath) => {
+    if (!hasPrettierRc) return text;
+    try {
+      const prettierModule = await import('prettier');
+      const prettier = prettierModule.default || prettierModule;
+      const resolved = (await prettier.resolveConfig(filePath)) || {};
+      return prettier.format(text, {
+        ...resolved,
+        // 传入 filepath 以便 Prettier 自动推断 parser（ts/tsx/js/jsx）
+        filepath: filePath,
+      });
+    } catch (_error) {
+      // Prettier 不可用或解析失败则跳过格式化
+      return text;
+    }
+  };
+
+  for (const file of files) {
+    try {
+      // 使用fileCodeMap中的代码，如果没有则读取文件
+      const code = fileCodeMap.get(file) || fs.readFileSync(file, 'utf8');
+
+      // 解析文件
+      const ast = parseAST(code);
+
+      let hasChanges = false;
+
+      // 遍历并替换key
+      traverse(ast, {
+        CallExpression(path) {
+          // 检查是否是 $i18n.get 调用
+          if (
+            path.node.callee.type === 'MemberExpression' &&
+            path.node.callee.object &&
+            path.node.callee.object.name === '$i18n' &&
+            path.node.callee.property &&
+            path.node.callee.property.name === 'get'
+          ) {
+            const args = path.node.arguments;
+            // 检查第一个参数是否是对象表达式
+            if (args.length > 0 && args[0].type === 'ObjectExpression') {
+              // 查找 id 属性
+              const idProperty = args[0].properties.find(
+                (prop) => prop.key && prop.key.name === 'id',
+              );
+
+              if (idProperty && idProperty.value) {
+                // 尝试提取 key 值（支持多种格式）
+                const oldKey = extractStringValue(idProperty.value);
+
+                if (oldKey) {
+                  const newKey = keyMap[oldKey];
+
+                  // 如果在 keyMap 中找到对应的新 key，进行替换
+                  if (newKey && newKey !== oldKey) {
+                    // 只替换字符串字面量，其他类型保持不变
+                    if (idProperty.value.type === 'StringLiteral') {
+                      idProperty.value = t.stringLiteral(newKey);
+                      hasChanges = true;
+                    } else {
+                      console.log(
+                        `    [警告] 无法替换非字符串字面量的 key: ${oldKey} (${idProperty.value.type})`,
+                      );
+                    }
+                  }
+                }
+              }
+            }
+          }
+        },
+      });
+
+      // 如果有改动，写回文件
+      if (hasChanges) {
+        const newCode = generateCode(ast, code);
+        const formatted = await formatWithPrettier(newCode, file);
+        fs.writeFileSync(file, formatted);
+      }
+    } catch (error) {
+      console.error(`    [错误] 更新文件失败: ${file}`, error);
+      if (error.loc) {
+        console.error(`    [错误] 错误位置: 第${error.loc.line}行, 第${error.loc.column}列`);
+      }
+    }
+  }
+
+  console.log('    [完成] 完成所有key替换');
+};
+
+module.exports = {
+  processFiles,
+  translateAndUpdateFiles,
+  singleTranslateAndUpdateFiles,
+  replaceKeysInFiles,
+};