td-octopus 0.2.1 → 0.2.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "td-octopus",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "I18N tool",
   "author": "Anthony Li",
   "bin": {
@@ -17,6 +17,7 @@
     "colors": "^1.4.0",
     "eslint": "^8.15.0",
     "eslint-plugin-prettier": "^4.2.1",
+    "eslint-plugin-td-rules-plugin": "^1.0.1",
     "eslint-plugin-unused-imports": "^4.3.0",
     "glob": "^8.0.3",
     "google-translate": "^3.0.0",

package/src/transLang/single.js

@@ -1,4 +1,3 @@
-
 const fs = require('fs');
 const path = require('path');
 const parser = require('@babel/parser');
@@ -9,36 +8,96 @@ const { ESLint } = require('eslint');
 const prettier = require('eslint-plugin-prettier');
 const unusedImports = require('eslint-plugin-unused-imports');
 
-// ===== 工具函数 =====
-const astNodeToString = (node) => {
+// ===== 工具函数：判断 key 是否匹配指定名称 =====
+function isKeyNamed(node, name) {
+    if (t.isIdentifier(node)) {
+        return node.name === name;
+    }
     if (t.isStringLiteral(node)) {
-        return node.value;
+        return node.value === name;
     }
-    if (t.isIdentifier(node)) {
-        return node.name;
+    return false;
+}
+
+function extractCnKeyDirect(node) {
+    if (t.isStringLiteral(node)) {
+        return node.value;
     }
+
     if (t.isMemberExpression(node)) {
-        const objectStr = astNodeToString(node.object);
-        const propertyStr = astNodeToString(node.property);
-        if(objectStr === 'I18N'){
-            return propertyStr;
+        let parts = [];
+        let current = node;
+
+        while (t.isMemberExpression(current)) {
+            if (current.computed) return null;
+            const prop = current.property;
+            if (!t.isIdentifier(prop)) return null;
+            parts.unshift(prop.name);
+            current = current.object;
+        }
+
+        if (t.isIdentifier(current) && current.name === 'I18N') {
+            parts.unshift('I18N');
+        } else {
+            return null;
+        }
+
+        if (parts.length === 4) {
+            return parts.join('.');
         }
-        return `${objectStr}.${propertyStr}`;
     }
-    return generate(node, { concise: true, comments: false }).code;
+
+    return null;
 }
 
-function isKeyNamed(node, name) {
+// ===== 提取 cn 的 key（仅支持：字符串字面量 或 I18N.a.b.c 四层）=====
+function extractCnKey(node, scope) {
+    // 情况1: 字符串字面量
+    if (t.isStringLiteral(node)) {
+        return node.value;
+    }
+ 
+     // 情况2: Identifier → 查找其定义
     if (t.isIdentifier(node)) {
-        return node.name === name;
+        const binding = scope.getBinding(node.name);
+        
+        if (binding && binding.path.isVariableDeclarator()) {
+            const init = binding.path.node.init;
+           
+            if (init) {
+                // 递归解析初始化表达式（不再传 scope，避免无限递归）
+                return extractCnKeyDirect(init);
+            }
+        }
+        return null;
     }
+
+    // 情况3: 检查是否为 I18N.a.b.c（必须是四层 MemberExpression，非 computed）
+   return extractCnKeyDirect(node);
+}
+
+function extractEnValue(node, scope) {
+    // 直接是字符串字面量
     if (t.isStringLiteral(node)) {
-        return node.value === name;
+        return node.value;
     }
-    return false;
+
+    // 是变量（Identifier），尝试回溯
+    if (t.isIdentifier(node)) {
+        const binding = scope.getBinding(node.name);
+        if (binding && binding.path.isVariableDeclarator()) {
+            const init = binding.path.node.init;
+            if (init && t.isStringLiteral(init)) {
+                return init.value;
+            }
+        }
+    }
+
+    // 其他情况（表达式、模板、多变量等）不支持
+    return null;
 }
 
-// ===== 单文件转换函数（接受 code 和 filePath）=====
+// ===== 单文件转换函数 =====
 const transformToChinese = (code, globalList) => {
     let localList = {}; // 每个文件独立的 list
 
@@ -49,57 +108,64 @@ const transformToChinese = (code, globalList) => {
 
     const langVariableNames = new Set();
 
-    // 第一阶段：找 getLanguage()
+    // 第一阶段：收集 getLang / getLanguage 的变量名
     traverse(ast, {
         VariableDeclarator(path) {
             const { id, init } = path.node;
-            if (t.isIdentifier(id) && t.isCallExpression(init) && t.isIdentifier(init.callee, { name: 'getLanguage' })) {
+            if (
+                t.isIdentifier(id) &&
+                t.isCallExpression(init) &&
+                t.isIdentifier(init.callee) &&
+                ['getLanguage', 'getLang'].includes(init.callee.name)
+            ) {
                 langVariableNames.add(id.name);
             }
         }
     });
 
-    // 第二阶段：处理多语言对象
+    // 第二阶段：处理对象和成员表达式
     traverse(ast, {
         ObjectExpression(path) {
             const props = path.node.properties;
             if (props.length === 0) return;
 
             const cnProp = props.find(
-                (prop) => t.isObjectProperty(prop) && !prop.computed && isKeyNamed(prop.key,  'cn') && !prop.method
-                
+                (prop) => t.isObjectProperty(prop) && !prop.computed && isKeyNamed(prop.key, 'cn') && !prop.method
             );
-
             const enProp = props.find(
-                (prop) => t.isObjectProperty(prop) && !prop.computed && isKeyNamed(prop.key, 'en') && !prop.method &&  t.isStringLiteral(prop.value)
+                (prop) => t.isObjectProperty(prop) && !prop.computed && isKeyNamed(prop.key, 'en') && !prop.method
             );
-  
+            
+            // 仅当 en 是字符串字面量，且 cn 符合提取规则时，才记录
             if (cnProp && enProp) {
-                const cnText = astNodeToString(cnProp.value);
-                localList[cnText] = enProp.value.value
-            }
-
-            if (cnProp) {
-                path.replaceWith(cnProp.value);
+                const enStr = extractEnValue(enProp.value, path.scope);
+                const cnKey = extractCnKey(cnProp.value, path.scope);
+                
+                if (cnKey !== null && enStr !== null) {
+                    localList[cnKey] = enStr;
+                }
             }
         },
 
         MemberExpression(path) {
             const node = path.node;
-            if (t.isMemberExpression(node.object) && t.isIdentifier(node.property) && langVariableNames.has(node.property.name)) {
-                path.replaceWith(node.object);
-            }
-        },
+            if (!node.computed) return;
 
-        Identifier(path) {
-            if (langVariableNames.has(path.node.name)) {
-                const parent = path.parentPath;
-                if (parent.isExpressionStatement()) {
-                    parent.remove();
-                }
+            const isLangAccess =
+                (t.isIdentifier(node.property) && langVariableNames.has(node.property.name)) ||
+                (t.isCallExpression(node.property) &&
+                    t.isIdentifier(node.property.callee) &&
+                    ['getLang', 'getLanguage'].includes(node.property.callee.name));
+
+            if (isLangAccess) {
+                // 替换为 obj['cn']
+                path.replaceWith(
+                    t.memberExpression(node.object, t.stringLiteral('cn'), true)
+                );
             }
         },
 
+        // 清理无用的 lang 变量声明
         VariableDeclarator(path) {
             const { id } = path.node;
             if (t.isIdentifier(id) && langVariableNames.has(id.name)) {
@@ -112,16 +178,26 @@ const transformToChinese = (code, globalList) => {
                     }
                 }
             }
+        },
+
+        Identifier(path) {
+            if (langVariableNames.has(path.node.name)) {
+                const parent = path.parentPath;
+                if (parent.isExpressionStatement()) {
+                    parent.remove();
+                }
+            }
         }
     });
 
-    // 第三阶段：清理 getLanguage 导入
+    // 第三阶段：清理 getLang / getLanguage 导入
     traverse(ast, {
         ImportDeclaration(path) {
             const specifiers = path.node.specifiers;
             const newSpecifiers = specifiers.filter((spec) => {
                 if (t.isImportSpecifier(spec)) {
-                    return !t.isIdentifier(spec.imported, { name: 'getLanguage' });
+                    return !t.isIdentifier(spec.imported, { name: 'getLanguage' }) &&
+                           !t.isIdentifier(spec.imported, { name: 'getLang' });
                 }
                 return true;
             });
@@ -140,19 +216,16 @@ const transformToChinese = (code, globalList) => {
         comments: false
     }).code;
 
-    // 合并到全局 list
-    
+    // 合并到全局翻译表
     Object.assign(globalList, localList);
     return transformedCode;
-}
+};
 
-// ===== ESLint 实例（复用）=====
+// ===== ESLint 实例（用于格式化和清理未使用导入）=====
 const eslint = new ESLint({
     fix: true,
     cwd: process.cwd(),
-    useEslintrc: false,
     plugins: {
-        prettier,
         'unused-imports': unusedImports
     },
     baseConfig: {
@@ -164,15 +237,8 @@ const eslint = new ESLint({
         }
     },
     overrideConfig: {
-        env: { browser: true, es2021: true, node: true },
-        extends: ['eslint:recommended'],
-        parserOptions: { ecmaVersion: 2022, sourceType: 'module' },
+        plugins: ["td-rules-plugin"],
         rules: {
-            'no-unused-vars': ['error', { vars: 'all', args: 'none', caughtErrors: 'none', ignoreRestSiblings: true }],
-            semi: ['error', 'always'],
-            indent: ['error', 4, { SwitchCase: 1 }],
-            quotes: ['error', 'single'],
-            'comma-dangle': ['error', 'never'],
             'prettier/prettier': [
                 1,
                 {
@@ -182,14 +248,14 @@ const eslint = new ESLint({
                     singleQuote: true,
                     tabWidth: 4,
                     trailingComma: 'none',
-                    jsxBracketSameLine: true
+                    jsxBracketSameLine: true,
                 }
             ]
         }
     }
 });
 
-// ===== 递归读取所有 .js 文件 =====
+// ===== 递归读取所有 .js 文件（排除 index.js）=====
 const getAllJsFiles = (dir) => {
     let results = [];
     const files = fs.readdirSync(dir);
@@ -198,14 +264,13 @@ const getAllJsFiles = (dir) => {
         const fullPath = path.join(dir, file);
         const stat = fs.statSync(fullPath);
         if (stat.isDirectory()) {
-            results = results.concat(getAllJsFiles(fullPath)); // 递归
-        } else if (file.endsWith('.js') && !['index.js'].includes(file)) {
+            results = results.concat(getAllJsFiles(fullPath));
+        } else if (file.endsWith('.js') && file !== 'index.js') {
             results.push(fullPath);
         }
     }
-
     return results;
-}
+};
 
 // ===== 处理单个文件 =====
 const processFile = async (filePath, globalList) => {
@@ -216,9 +281,11 @@ const processFile = async (filePath, globalList) => {
     const finalCode = results[0].output || transformedCode;
 
     fs.writeFileSync(filePath, finalCode, 'utf-8');
-}
+};
 
+// ===== 导出接口 =====
 module.exports = {
     processFile,
-    getAllJsFiles
-}
+    getAllJsFiles,
+    transformToChinese // 便于单元测试
+};