@scoutello/i18n-magic 0.54.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"
@@ -71,6 +71,15 @@ const AddTranslationKeySchema = z.object({
   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
 const ListUntranslatedKeysSchema = z.object({
   namespace: z
@@ -184,7 +193,7 @@ class I18nMagicServer {
         tools: [
           {
             name: "add_translation_key",
-            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).",
+            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: {
@@ -207,6 +216,38 @@ class I18nMagicServer {
               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:
@@ -399,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