td-octopus 0.1.14 → 0.1.16

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "td-octopus",
-  "version": "0.1.14",
+  "version": "0.1.16",
   "description": "I18N tool",
   "author": "Anthony Li",
   "bin": {

package/src/check/index.js

@@ -3,7 +3,7 @@
  * @Author: 郑泳健
  * @Date: 2024-12-12 15:00:24
  * @LastEditors: 郑泳健
- * @LastEditTime: 2025-09-09 17:09:27
+ * @LastEditTime: 2025-09-11 09:56:54
  */
 const path = require('path')
 const fs = require('fs')
@@ -12,7 +12,7 @@ const { getSpecifiedFiles, isFile } = require('../utils/file')
 const syncLang = require('../utils/syncLang')
 const { flatObject, rewriteFiles, getFileKeyValueList, getAdjustLangObjAndAddList } = require('../utils/translate');
 const { failInfo, highlightText } = require('../utils/colors');
-const { getProjectConfig } = require('../utils/index')
+const { getProjectConfig, autoImportJSFiles } = require('../utils/index')
 const ora = require('ora');
 
 const CONFIG = getProjectConfig();
@@ -102,6 +102,11 @@ function main() {
 
         rewriteFiles(getFileKeyValueList(zhCNResult), 'zh-CN');
         rewriteFiles(getFileKeyValueList(enCNResult), 'en-US');
+        const cnJSON = fs.readFileSync(path.resolve(process.cwd(), '.octopus/zh-CN/index.js'), 'utf-8');
+        const enJSON = fs.readFileSync(path.resolve(process.cwd(), '.octopus/en-US/index.js'), 'utf-8');
+
+        autoImportJSFiles(path.resolve(process.cwd(), '.octopus/zh-CN'), cnJSON)
+        autoImportJSFiles(path.resolve(process.cwd(), '.octopus/en-US'), enJSON)
         spinner.succeed(`同步完成`);
     })()
 }

package/src/utils/index.js

@@ -270,6 +270,55 @@ function prettierFile(fileContent, proType) {
     }
   }
 
+  function autoImportJSFiles(__dirname, indexContent) {
+    // 1. 提取已有 import 的模块名
+    const importRegex = /import\s+([a-zA-Z0-9_$]+)\s+from\s+'\.\/([^']+)'/g;
+    const existingImports = new Map();
+    let match;
+
+    while ((match = importRegex.exec(indexContent)) !== null) {
+        existingImports.set(match[2], match[1]);
+    }
+
+    // 2. 获取目录下所有 .js 文件（排除 index.js）
+    const allFiles = fs.readdirSync(__dirname)
+    .filter(file => file.endsWith('.js') && file !== 'index.js')
+    .map(file => path.basename(file, '.js'));
+
+    // 3. 找出缺少的
+    const missing = allFiles.filter(file => !existingImports.has(file));
+
+    if (missing.length === 0) {
+    process.exit(0);
+    }
+
+    // 4. 构造 import 和 export 内容
+    let newImports = '';
+    let newExports = '';
+
+    missing.forEach((file) => {
+    newImports += `import ${file} from './${file}';\n`;
+    newExports += `    ${file},\n`;
+    });
+
+    // 5. 插入到 index.js
+    // 找到最后一个 import 的位置
+    const lastImportIndex = indexContent.lastIndexOf("import ");
+    const firstExportIndex = indexContent.indexOf("export default");
+
+    const updatedContent =
+    indexContent.slice(0, lastImportIndex) +
+    indexContent.slice(lastImportIndex, firstExportIndex) + newImports + '\n' +
+    indexContent.slice(firstExportIndex).replace(
+        /(export default Object\.assign\(\{\},\s*\{[\s\S]*?)(\}\);)/,
+        `$1\n${newExports}$2`
+    );
+
+    // 6. 写回 index.js
+    fs.writeFileSync(path.resolve(__dirname, 'index.js'), updatedContent, 'utf-8');
+
+}
+
 module.exports = {
   getOtpDir,
   getLangDir,
@@ -285,5 +334,6 @@ module.exports = {
   lookForFiles,
   translateKeyText,
   spining,
-  prettierFile
+  prettierFile,
+  autoImportJSFiles
 };