@scoutello/i18n-magic 0.53.0 → 0.55.0

package/src/mcp-server.ts

@@ -5,7 +5,7 @@ import {
   ListToolsRequestSchema,
 } from "@modelcontextprotocol/sdk/types.js"
 import { z } from "zod"
-import { addTranslationKey, getMissingKeys, loadConfig, loadLocalesFile } from "./lib/utils.js"
+import { addTranslationKey, addTranslationKeys, getMissingKeys, loadConfig, loadLocalesFile } from "./lib/utils.js"
 import type { Configuration } from "./lib/types.js"
 import path from "path"
 import fs from "fs"
@@ -67,7 +67,17 @@ function resolveProjectRoot(): string {
 // Zod schema for the add_translation_key tool parameters
 const AddTranslationKeySchema = z.object({
   key: z.string().describe("The translation key to add (e.g., \"welcomeMessage\")"),
-  value: z.string().describe("The English text value for this translation key"),
+  value: z.string().describe("The text value for this translation key"),
+  language: z.string().optional().describe("The language code of the provided value (e.g., \"en\", \"de\", \"fr\"). Defaults to \"en\" (English) if not specified."),
+})
+
+// Zod schema for the add_translation_keys (batch) tool parameters
+const AddTranslationKeysSchema = z.object({
+  keys: z.array(z.object({
+    key: z.string().describe("The translation key to add (e.g., \"welcomeMessage\")"),
+    value: z.string().describe("The text value for this translation key"),
+    language: z.string().optional().describe("The language code of the provided value (e.g., \"en\", \"de\", \"fr\"). Defaults to \"en\" (English) if not specified."),
+  })).describe("Array of translation keys to add in batch. Use this for adding multiple keys at once for better performance."),
 })
 
 // Zod schema for the list_untranslated_keys tool parameters
@@ -94,7 +104,8 @@ const GetTranslationKeySchema = z.object({
 // Zod schema for the update_translation_key tool parameters
 const UpdateTranslationKeySchema = z.object({
   key: z.string().describe("The translation key to update (e.g., \"welcomeMessage\")"),
-  value: z.string().describe("The new English text value for this translation key"),
+  value: z.string().describe("The new text value for this translation key"),
+  language: z.string().optional().describe("The language code of the provided value (e.g., \"en\", \"de\", \"fr\"). Defaults to \"en\" (English) if not specified."),
   namespace: z
     .string()
     .optional()
@@ -182,7 +193,7 @@ class I18nMagicServer {
         tools: [
           {
             name: "add_translation_key",
-            description:  "Add a new translation key with an English value.",
+            description:  "Add a new translation key with a text value. You can optionally specify the language of the value you're providing (defaults to English). For adding multiple keys at once, use add_translation_keys instead for better performance.",
             inputSchema: {
               type: "object",
               properties: {
@@ -194,12 +205,49 @@ class I18nMagicServer {
                 value: {
                   type: "string",
                   description:
-                    "The English text value for this translation key",
+                    "The text value for this translation key",
+                },
+                language: {
+                  type: "string",
+                  description:
+                    "The language code of the provided value (e.g., \"en\" for English, \"de\" for German, \"fr\" for French). Defaults to \"en\" if not specified.",
                 },
               },
               required: ["key", "value"],
             },
           },
+          {
+            name: "add_translation_keys",
+            description: "Add multiple translation keys in batch. This is optimized for performance - when adding 2 or more keys, prefer this over multiple add_translation_key calls. It performs a single codebase scan, batches file I/O operations, and batches translations for much better performance.",
+            inputSchema: {
+              type: "object",
+              properties: {
+                keys: {
+                  type: "array",
+                  description: "Array of translation keys to add in batch",
+                  items: {
+                    type: "object",
+                    properties: {
+                      key: {
+                        type: "string",
+                        description: "The translation key to add (e.g., \"welcomeMessage\")",
+                      },
+                      value: {
+                        type: "string",
+                        description: "The text value for this translation key",
+                      },
+                      language: {
+                        type: "string",
+                        description: "The language code of the provided value (e.g., \"en\" for English, \"de\" for German, \"fr\" for French). Defaults to \"en\" if not specified.",
+                      },
+                    },
+                    required: ["key", "value"],
+                  },
+                },
+              },
+              required: ["keys"],
+            },
+          },
           {
             name: "list_untranslated_keys",
             description:
@@ -240,7 +288,7 @@ class I18nMagicServer {
           {
             name: "update_translation_key",
             description:
-              "Update an existing translation key with a new English value. This will update the key across all locales (translating automatically to other languages) and all namespaces where the key exists. Use this when you need to fix typos, improve wording, or change the text of an existing translation. If you're not sure if a key exists, use get_translation_key or search_translations first.",
+              "Update an existing translation key with a new text value. You can optionally specify the language of the value you're providing (defaults to English). This will update the key across all locales (translating automatically to other languages) and all namespaces where the key exists. Use this when you need to fix typos, improve wording, or change the text of an existing translation. If you're not sure if a key exists, use get_translation_key or search_translations first.",
             inputSchema: {
               type: "object",
               properties: {
@@ -252,7 +300,12 @@ class I18nMagicServer {
                 value: {
                   type: "string",
                   description:
-                    "The new English text value for this translation key",
+                    "The new text value for this translation key",
+                },
+                language: {
+                  type: "string",
+                  description:
+                    "The language code of the provided value (e.g., \"en\" for English, \"de\" for German, \"fr\" for French). Defaults to \"en\" if not specified.",
                 },
                 namespace: {
                   type: "string",
@@ -311,6 +364,7 @@ class I18nMagicServer {
             result = await addTranslationKey({
               key: params.key,
               value: params.value,
+              language: params.language || "en",
               config,
             })
           } finally {
@@ -325,14 +379,15 @@ class I18nMagicServer {
                 text: JSON.stringify(
                   {
                     success: true,
-                    message: `Successfully added translation key "${result.key}" to affected namespaces: ${result.namespace} (${result.locale})`,
+                    message: `Successfully added translation key "${result.key}" to affected namespaces: ${result.namespace} (locales: ${result.locale})`,
                     key: result.key,
                     value: result.value,
+                    providedLanguage: params.language || "en",
                     namespace: result.namespace,
-                    locale: result.locale,
+                    locales: result.locale,
                     nextStep: result.locale.includes(',')
-                      ? "Run 'i18n-magic sync' to translate this key to other locales"
-                      : "Key was translated to default locale. Run 'i18n-magic sync' to translate to other locales",
+                      ? "Key was automatically translated to multiple locales"
+                      : "Run 'i18n-magic sync' to translate this key to other locales",
                     diagnostics: logMessages.join('\n'),
                   },
                   null,
@@ -385,6 +440,131 @@ class I18nMagicServer {
         }
       }
 
+      if (request.params.name === "add_translation_keys") {
+        try {
+          // Validate parameters
+          const params = AddTranslationKeysSchema.parse(request.params.arguments)
+
+          if (!params.keys || params.keys.length === 0) {
+            return {
+              content: [
+                {
+                  type: "text",
+                  text: JSON.stringify(
+                    {
+                      success: false,
+                      error: "No keys provided. The 'keys' array must contain at least one key-value pair.",
+                    },
+                    null,
+                    2,
+                  ),
+                },
+              ],
+              isError: true,
+            }
+          }
+
+          // Ensure config is loaded
+          const config = await this.ensureConfig()
+
+          // Capture console.log output for diagnostics
+          const originalConsoleLog = console.log
+          const logMessages: string[] = []
+          console.log = (...args: any[]) => {
+            const message = args.map(arg => 
+              typeof arg === 'object' ? JSON.stringify(arg) : String(arg)
+            ).join(' ')
+            logMessages.push(message)
+            // Also log to stderr for debugging
+            console.error(`[i18n-magic] ${message}`)
+          }
+
+          let result
+          try {
+            // Add the translation keys in batch
+            result = await addTranslationKeys({
+              keys: params.keys.map(k => ({
+                key: k.key,
+                value: k.value,
+                language: k.language || "en",
+              })),
+              config,
+            })
+          } finally {
+            // Restore console.log
+            console.log = originalConsoleLog
+          }
+
+          return {
+            content: [
+              {
+                type: "text",
+                text: JSON.stringify(
+                  {
+                    success: true,
+                    message: `Successfully added ${result.results.length} translation key(s) in batch`,
+                    results: result.results,
+                    performance: result.performance,
+                    summary: {
+                      totalKeys: result.results.length,
+                      totalTime: `${result.performance.totalTime.toFixed(2)}ms`,
+                      averageTimePerKey: `${(result.performance.totalTime / result.results.length).toFixed(2)}ms`,
+                      scanTime: `${result.performance.scanTime.toFixed(2)}ms`,
+                      translationTime: `${result.performance.translationTime.toFixed(2)}ms`,
+                      fileIOTime: `${result.performance.fileIOTime.toFixed(2)}ms`,
+                    },
+                    diagnostics: logMessages.join('\n'),
+                  },
+                  null,
+                  2,
+                ),
+              },
+            ],
+          }
+        } catch (error) {
+          const errorMessage =
+            error instanceof Error ? error.message : "Unknown error occurred"
+          
+          // Get more detailed error information
+          let errorDetails = errorMessage
+          if (error instanceof Error) {
+            // Check if there's a cause
+            const cause = (error as any).cause
+            if (cause instanceof Error) {
+              errorDetails = `${errorMessage}\nCause: ${cause.message}\nStack: ${cause.stack}`
+            } else if (cause) {
+              errorDetails = `${errorMessage}\nCause: ${JSON.stringify(cause)}`
+            }
+            // Include stack trace
+            if (error.stack) {
+              errorDetails = `${errorDetails}\nStack: ${error.stack}`
+            }
+          }
+          
+          // Log detailed error to stderr for debugging
+          console.error(`[i18n-magic MCP] Error adding translation keys in batch:`)
+          console.error(errorDetails)
+          
+          return {
+            content: [
+              {
+                type: "text",
+                text: JSON.stringify(
+                  {
+                    success: false,
+                    error: errorMessage,
+                    details: errorDetails,
+                  },
+                  null,
+                  2,
+                ),
+              },
+            ],
+            isError: true,
+          }
+        }
+      }
+
       if (request.params.name === "list_untranslated_keys") {
         try {
           // Validate parameters
@@ -681,20 +861,21 @@ class I18nMagicServer {
             }
 
             // Build translation cache with new value
+            const inputLanguage = params.language || "en"
             const translationCache: Record<string, string> = {
-              en: params.value,
+              [inputLanguage]: params.value,
             }
 
             // Translate to all other locales
-            const nonEnglishLocales = config.locales.filter((l) => l !== "en")
-            if (nonEnglishLocales.length > 0 && config.openai) {
+            const otherLocales = config.locales.filter((l) => l !== inputLanguage)
+            if (otherLocales.length > 0 && config.openai) {
               const { translateKey } = await import("./lib/utils.js")
               
               await Promise.all(
-                nonEnglishLocales.map(async (locale) => {
+                otherLocales.map(async (locale) => {
                   const translation = await translateKey({
                     context: config.context || "",
-                    inputLanguage: "en",
+                    inputLanguage: inputLanguage,
                     outputLanguage: locale,
                     object: {
                       [params.key]: params.value,
@@ -730,6 +911,7 @@ class I18nMagicServer {
                       message: `Successfully updated translation key "${params.key}" in ${targetNamespaces.length} namespace(s) and ${config.locales.length} locale(s)`,
                       key: params.key,
                       newValue: params.value,
+                      providedLanguage: params.language || "en",
                       namespaces: targetNamespaces,
                       locales: config.locales,
                       diagnostics: logMessages.join('\n'),