vite-plugin-localization-ai 1.0.2 → 1.0.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vite-plugin-localization-ai",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Vite plugin to generate localization jsons using AI",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -18,7 +18,8 @@
     }
   },
   "files": [
-    "dist"
+    "dist",
+    "src"
   ],
   "scripts": {
     "build": "tsc",

package/src/index.ts

@@ -0,0 +1,47 @@
+import fs from "fs";
+import path from "path";
+import { parse as parseJsonc } from "jsonc-parser";
+import { translate } from "./providers/gemini";
+
+interface LocalizationAIOptions {
+  source: string;
+  targets: string[];
+  outputDir?: string;
+  apiKey?: string;
+  extraPromt?: string;
+}
+
+export default function localizationAI(options: LocalizationAIOptions) {
+  const {
+    source,
+    targets,
+    extraPromt,
+    outputDir = path.dirname(source),
+    apiKey = process.env.LOCALIZATION_AI_API_KEY,
+  } = options;
+
+  return {
+    name: "vite-plugin-localization-ai",
+    apply: "build",
+    buildStart: async () => {
+      console.log("[localization-ai] starting localization...");
+
+      const sourceContent = parseJsonc(fs.readFileSync(source, "utf-8"));
+
+      for (const lang of targets) {
+        console.log(`[localization-ai] Translating to ${lang}...`);
+        const translated = await translate(sourceContent, lang, apiKey);
+
+        const outputPath = path.join(outputDir, `${lang}.json`);
+        fs.writeFileSync(
+          outputPath,
+          JSON.stringify(translated, null, 2),
+          "utf-8"
+        );
+        console.log(`[localization-ai] ${lang}.json saved!`);
+      }
+
+      console.log("[localization-ai] done ✅");
+    },
+  };
+}

package/src/providers/gemini.ts

@@ -0,0 +1,35 @@
+import { GoogleGenAI } from "@google/genai";
+
+export async function translate(
+  sourceJson: Record<string, any>,
+  targetLang: string,
+  apiKey?: string,
+  extraPromt?: string
+) {
+  const client = new GoogleGenAI({
+    apiKey: apiKey,
+  });
+
+  const text = JSON.stringify(sourceJson, null, 2);
+
+  const response = await client.models.generateContent({
+    model: "gemini-2.5-flash",
+    contents: `You are a professional application interface translator.
+Translate the JSON.
+JSON:
+${text}
+
+Target language is "${targetLang}". ${extraPromt || ""}
+`,
+    config: {
+      systemInstruction:
+        "Translate JSON values exactly, keep keys and structure. The response must be plain text — no Markdown",
+      temperature: 0,
+      seed: 42,
+      responseMimeType: "application/json",
+    },
+  });
+
+  const translatedText = response.text?.trim() || "{}";
+  return JSON.parse(translatedText || "{}");
+}