ai-localize-codemods 1.0.1 → 2.0.1

package/src/vue-codemod.ts

@@ -1,60 +1,68 @@
 import * as fs from 'fs';
 
-import type { DetectedText } from 'ai-localize-shared';
+import type { DetectedText, CodemodConfig } from 'ai-localize-shared';
 import type { CodemodResult } from './react-codemod.js';
 
+const DEFAULT_TRANSLATION_FN = '$t';
+
 /**
- * Vue i18n codemod: wraps hardcoded text with $t() calls in Vue SFCs.
- * Handles both Options API and Composition API.
+ * Vue i18n codemod: wraps hardcoded text with $t() / t() calls in Vue SFCs.
+ * Handles both Options API (global $t) and Composition API (useI18n / useTranslation).
+ *
+ * The translation function name can be overridden via the `codemods.translationFunction`
+ * config field (default: "$t").
  */
 export class VueCodemod {
   private filePath: string;
   private content: string;
   private texts: DetectedText[];
+  private translationFn: string;
 
-  constructor(filePath: string, content: string, texts: DetectedText[]) {
+  constructor(filePath: string, content: string, texts: DetectedText[], codemodConfig?: CodemodConfig) {
     this.filePath = filePath;
     this.content = content;
-    this.texts = texts;
+ this.texts = texts;
+    this.translationFn = codemodConfig?.translationFunction ?? DEFAULT_TRANSLATION_FN;
   }
 
   transform(): CodemodResult {
     if (this.texts.length === 0) {
-    return this.unchanged();
-  }
+ return this.unchanged();
+    }
 
     // Parse <template> and <script> sections separately
     const templateMatch = this.content.match(/<template[^>]*>([\s\S]*?)<\/template>/);
     const scriptMatch = this.content.match(/<script[^>]*>([\s\S]*?)<\/script>/);
 
- if (!templateMatch && !scriptMatch) {
+    if (!templateMatch && !scriptMatch) {
       return this.unchanged();
     }
 
     const textKeyMap = new Map<string, string>();
     for (const dt of this.texts) {
-      textKeyMap.set(dt.text, dt.suggestedKey);
-    }
+  textKeyMap.set(dt.text, dt.suggestedKey);
+  }
 
     let transformedContent = this.content;
     let replacedTexts = 0;
+    const fn = this.translationFn;
 
-    // Replace in template: >text< => >{{ $t('key') }}<
+  // Replace in template: >text< => >{{ $t('key') }}<
     if (templateMatch) {
       let template = templateMatch[1];
-    for (const [text, key] of textKeyMap) {
-        const escaped = text.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
-        const re = new RegExp(`(>\\s*)${escaped}(\\s*<)`, 'g');
- template = template.replace(re, `$1{{ $t('${key}') }}$2`);
-        // Also replace in attribute values: title="text" => :title="$t('key')"
-  const attrRe = new RegExp(`((?:title|placeholder|alt|label)=)"(${escaped})"`, 'g');
-        template = template.replace(attrRe, `:$1"$t('${key}')"`);
-        replacedTexts++;
+      for (const [text, key] of textKeyMap) {
+    const escaped = text.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+  const re = new RegExp(`(>\\s*)${escaped}(\\s*<)`, 'g');
+        template = template.replace(re, `$1{{ ${fn}('${key}') }}$2`);
+   // Also replace in attribute values: title="text" => :title="$t('key')"
+        const attrRe = new RegExp(`((?:title|placeholder|alt|label)=)"(${escaped})"`, 'g');
+        template = template.replace(attrRe, `:$1"${fn}('${key}')"`);
+      replacedTexts++;
       }
       transformedContent = transformedContent.replace(templateMatch[1], template);
     }
 
-    if (replacedTexts === 0) {
+  if (replacedTexts === 0) {
       return this.unchanged();
     }
 
@@ -65,16 +73,16 @@ export class VueCodemod {
       changed: true,
       injectedImport: false, // vue-i18n is globally injected
       injectedHook: false,
-      replacedTexts,
+   replacedTexts,
     };
   }
 
   private unchanged(): CodemodResult {
     return {
-    filePath: this.filePath,
-    originalContent: this.content,
+      filePath: this.filePath,
+      originalContent: this.content,
       transformedContent: this.content,
-  changed: false,
+      changed: false,
       injectedImport: false,
       injectedHook: false,
       replacedTexts: 0,
@@ -85,10 +93,11 @@ export class VueCodemod {
 export function applyVueCodemod(
   filePath: string,
   texts: DetectedText[],
-  dryRun = false
+  dryRun = false,
+  codemodConfig?: CodemodConfig
 ): CodemodResult {
   const content = fs.readFileSync(filePath, 'utf-8');
-  const codemod = new VueCodemod(filePath, content, texts);
+  const codemod = new VueCodemod(filePath, content, texts, codemodConfig);
   const result = codemod.transform();
   if (result.changed && !dryRun) {
     fs.writeFileSync(filePath, result.transformedContent, 'utf-8');