vue-i18n-extract-plugin 1.0.50 → 1.0.51

package/lib/visitors.js

@@ -15,6 +15,84 @@ function isTFunction(node, option) {
   );
 }
 
+function isCreateTextVNodeCall(path) {
+  const node = path.node;
+  if (!path.isCallExpression()) return false;
+
+  const callee = node.callee;
+  return (
+    (t.isIdentifier(callee) && callee.name === "_createTextVNode") ||
+    (t.isMemberExpression(callee) &&
+      t.isIdentifier(callee.property) &&
+      callee.property.name === "_createTextVNode")
+  );
+}
+
+function getPropKey(propNode) {
+  if (t.isIdentifier(propNode.key)) {
+    return propNode.key.name;
+  }
+  if (t.isStringLiteral(propNode.key)) {
+    return propNode.key.value; // .replace(/-([a-z])/g, (_, char) => char.toUpperCase());
+  }
+  return null;
+}
+
+function transformDirectiveIfNeeded(path, parentPath) {
+  let hasCreateVNode = false;
+  // 属性值情况，如 title: "xxx"
+  if (parentPath.isObjectProperty()) {
+    const propKey = getPropKey(parentPath.node);
+
+    if (!propKey) return;
+
+    const vNodeCall = path.findParent(
+      p =>
+        p.isCallExpression() &&
+        (t.isIdentifier(p.node.callee, { name: "_createVNode" }) ||
+          (t.isMemberExpression(p.node.callee) &&
+            t.isIdentifier(p.node.callee.property, { name: "_createVNode" })))
+    );
+
+    if (vNodeCall) {
+      hasCreateVNode = true;
+      const args = vNodeCall.node.arguments;
+      // PatchFlag = 8 (PROPS), dynamicProps = ["title"]
+      const patchFlagIndex = 3;
+      const dynamicPropsIndex = 4;
+
+      // 确保第 3 个参数（children）存在，否则补 null
+      if (args.length === 2) {
+        args.push(t.nullLiteral()); // => 第三个参数
+      }
+
+      // 补齐参数数组长度到第 5 个参数（index: 4）
+      while (args.length <= dynamicPropsIndex) {
+        args.push(null);
+      }
+
+      // 设置 patchFlag = 8
+      args[patchFlagIndex] = t.numericLiteral(8);
+
+      // 设置 dynamicProps = ["propKey"]
+      const existingDynamicProps = args[dynamicPropsIndex];
+      if (!existingDynamicProps || !t.isArrayExpression(existingDynamicProps)) {
+        args[dynamicPropsIndex] = t.arrayExpression([t.stringLiteral(propKey)]);
+      } else {
+        const existingKeys = new Set(
+          existingDynamicProps.elements
+            .filter(el => t.isStringLiteral(el))
+            .map(el => el.value)
+        );
+        if (!existingKeys.has(propKey)) {
+          existingDynamicProps.elements.push(t.stringLiteral(propKey));
+        }
+      }
+    }
+  }
+  return hasCreateVNode;
+}
+
 function shouldTransform(path) {
   const parent = path.parentPath;
   return !(
@@ -112,10 +190,42 @@ function createI18nVisitor(option, i18nMap) {
         i18nMap[hashed] = value;
       }
 
-      const callExpression = t.callExpression(
-        t.identifier(option.translateKey),
-        [t.stringLiteral(hashed)]
-      );
+      let callExpression;
+
+      // 判断是否createTextVNode或MemberExpression（如 Vue.createTextVNode）
+      if (isCreateTextVNodeCall(parentPath)) {
+        // _ctx.$t("hashed")
+        callExpression = t.callExpression(
+          t.memberExpression(
+            t.identifier("_ctx"),
+            t.identifier(option.translateKey)
+          ),
+          [t.stringLiteral(hashed)]
+        );
+
+        // 如果 createTextVNode 参数只有一个，则补充第二个参数 1
+        const args = parentPath.node.arguments;
+        if (args.length === 1) {
+          parentPath.node.arguments = [callExpression, t.numericLiteral(1)];
+        }
+      } else {
+        const hasCreateVNode = transformDirectiveIfNeeded(path, parentPath);
+        if (hasCreateVNode) {
+          callExpression = t.callExpression(
+            t.memberExpression(
+              t.identifier("_ctx"),
+              t.identifier(option.translateKey)
+            ),
+            [t.stringLiteral(hashed)]
+          );
+        } else {
+          // $t("hashed")
+          callExpression = t.callExpression(t.identifier(option.translateKey), [
+            t.stringLiteral(hashed)
+          ]);
+        }
+      }
+
       // 如果是 JSX 属性值，需要包裹在 JSXExpressionContainer 中
       if (parentPath.isJSXAttribute()) {
         const jsxExpression = t.jsxExpressionContainer(callExpression);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vue-i18n-extract-plugin",
-  "version": "1.0.50",
+  "version": "1.0.51",
   "main": "lib/index.js",
   "types": "types/index.d.ts",
   "bin": {