pabal-resource-mcp 1.5.3 → 1.5.4

package/dist/bin/mcp-server.js

@@ -3523,7 +3523,7 @@ function calculateAspectRatio(width, height) {
   if (Math.abs(ratio - 4 / 3) < 0.1) return "4:3";
   return ratio < 1 ? "9:16" : "16:9";
 }
-async function translateImage(sourcePath, sourceLocale, targetLocale, outputPath) {
+async function translateImage(sourcePath, sourceLocale, targetLocale, outputPath, preserveWords) {
   try {
     const client = getGeminiClient();
     const sourceLanguage = getLanguageName(sourceLocale);
@@ -3531,6 +3531,8 @@ async function translateImage(sourcePath, sourceLocale, targetLocale, outputPath
     const { width, height } = await getImageDimensions(sourcePath);
     const aspectRatio = calculateAspectRatio(width, height);
     const { data: imageData, mimeType } = readImageAsBase64(sourcePath);
+    const preserveInstruction = preserveWords && preserveWords.length > 0 ? `
+- Do NOT translate these words, keep them exactly as-is: ${preserveWords.join(", ")}` : "";
     const prompt = `This is an app screenshot with text in ${sourceLanguage}.
 Please translate ONLY the text/words in this image to ${targetLanguage}.
 
@@ -3540,7 +3542,7 @@ IMPORTANT INSTRUCTIONS:
 - Maintain the same font style and text positioning as much as possible
 - Do NOT add any new elements or remove existing design elements
 - The output should look identical except the text language is ${targetLanguage}
-- Preserve all icons, images, and graphical elements exactly as they are`;
+- Preserve all icons, images, and graphical elements exactly as they are${preserveInstruction}`;
     const chat = client.chats.create({
       model: "gemini-3-pro-image-preview",
       config: {
@@ -3604,7 +3606,7 @@ IMPORTANT INSTRUCTIONS:
     };
   }
 }
-async function translateImagesWithProgress(translations, onProgress) {
+async function translateImagesWithProgress(translations, onProgress, preserveWords) {
   let successful = 0;
   let failed = 0;
   const errors = [];
@@ -3626,7 +3628,8 @@ async function translateImagesWithProgress(translations, onProgress) {
       translation.sourcePath,
       translation.sourceLocale,
       translation.targetLocale,
-      translation.outputPath
+      translation.outputPath,
+      preserveWords
     );
     if (result.success) {
       successful++;
@@ -3659,7 +3662,7 @@ async function getImageDimensions2(imagePath) {
     height: metadata.height
   };
 }
-async function detectEdgeColor(imagePath) {
+async function detectCornerColor(imagePath) {
   const image = sharp2(imagePath);
   const metadata = await image.metadata();
   const width = metadata.width || 100;
@@ -3667,14 +3670,17 @@ async function detectEdgeColor(imagePath) {
   const { data, info } = await image.raw().toBuffer({ resolveWithObject: true });
   const channels = info.channels;
   const colorCounts = /* @__PURE__ */ new Map();
-  const sampleEdgePixel = (x, y) => {
+  const cornerWidth = Math.min(100, Math.max(10, Math.floor(width * 0.05)));
+  const cornerHeight = Math.min(100, Math.max(10, Math.floor(height * 0.05)));
+  const samplePixel = (x, y) => {
+    if (x < 0 || x >= width || y < 0 || y >= height) return;
     const idx = (y * width + x) * channels;
     const r = data[idx];
     const g = data[idx + 1];
     const b = data[idx + 2];
-    const qr = Math.round(r / 16) * 16;
-    const qg = Math.round(g / 16) * 16;
-    const qb = Math.round(b / 16) * 16;
+    const qr = Math.round(r / 8) * 8;
+    const qg = Math.round(g / 8) * 8;
+    const qb = Math.round(b / 8) * 8;
     const key = `${qr},${qg},${qb}`;
     const existing = colorCounts.get(key);
     if (existing) {
@@ -3683,13 +3689,22 @@ async function detectEdgeColor(imagePath) {
       colorCounts.set(key, { count: 1, color: { r: qr, g: qg, b: qb } });
     }
   };
-  for (let x = 0; x < width; x += 2) {
-    sampleEdgePixel(x, 0);
-    sampleEdgePixel(x, height - 1);
-  }
-  for (let y = 0; y < height; y += 2) {
-    sampleEdgePixel(0, y);
-    sampleEdgePixel(width - 1, y);
+  const corners = [
+    { startX: 0, startY: 0 },
+    // Top-left
+    { startX: width - cornerWidth, startY: 0 },
+    // Top-right
+    { startX: 0, startY: height - cornerHeight },
+    // Bottom-left
+    { startX: width - cornerWidth, startY: height - cornerHeight }
+    // Bottom-right
+  ];
+  for (const corner of corners) {
+    for (let y = corner.startY; y < corner.startY + cornerHeight; y += 2) {
+      for (let x = corner.startX; x < corner.startX + cornerWidth; x += 2) {
+        samplePixel(x, y);
+      }
+    }
   }
   let maxCount = 0;
   let dominantColor = { r: 255, g: 255, b: 255 };
@@ -3702,7 +3717,7 @@ async function detectEdgeColor(imagePath) {
   return dominantColor;
 }
 async function resizeImage(inputPath, outputPath, targetDimensions) {
-  const bgColor = await detectEdgeColor(inputPath);
+  const bgColor = await detectCornerColor(inputPath);
   await sharp2(inputPath).resize(targetDimensions.width, targetDimensions.height, {
     fit: "contain",
     // Preserve aspect ratio
@@ -3763,8 +3778,17 @@ var localizeScreenshotsInputSchema = z7.object({
   deviceTypes: z7.array(z7.enum(["phone", "tablet"])).optional().default(["phone", "tablet"]).describe("Device types to process (default: both phone and tablet)"),
   dryRun: z7.boolean().optional().default(false).describe("Preview mode - shows what would be translated without actually translating"),
   skipExisting: z7.boolean().optional().default(true).describe("Skip translation if target file already exists (default: true)"),
-  screenshotNumbers: z7.array(z7.number().int().positive()).optional().describe(
-    "Specific screenshot numbers to process (e.g., [1, 3, 5]). If not provided, all screenshots will be processed."
+  screenshotNumbers: z7.union([
+    z7.array(z7.number().int().positive()),
+    z7.object({
+      phone: z7.array(z7.number().int().positive()).optional(),
+      tablet: z7.array(z7.number().int().positive()).optional()
+    })
+  ]).optional().describe(
+    "Specific screenshot numbers to process. Can be:\n- Array for all devices: [1, 3, 5]\n- Object for per-device: { phone: [1, 2], tablet: [1, 3, 5] }\nIf not provided, all screenshots will be processed."
+  ),
+  preserveWords: z7.array(z7.string()).optional().describe(
+    'Words to keep untranslated (e.g., brand names, product names). Example: ["Pabal", "Pro", "AI"]'
   )
 });
 var jsonSchema7 = zodToJsonSchema7(localizeScreenshotsInputSchema, {
@@ -3881,7 +3905,8 @@ async function handleLocalizeScreenshots(input) {
     deviceTypes = ["phone", "tablet"],
     dryRun = false,
     skipExisting = true,
-    screenshotNumbers
+    screenshotNumbers,
+    preserveWords
   } = input;
   const results = [];
   let appInfo;
@@ -3936,16 +3961,34 @@ async function handleLocalizeScreenshots(input) {
   let filteredScreenshots = sourceScreenshots.filter(
     (s) => deviceTypes.includes(s.type)
   );
-  if (screenshotNumbers && screenshotNumbers.length > 0) {
+  if (screenshotNumbers) {
+    const isArray = Array.isArray(screenshotNumbers);
+    const phoneNumbers = isArray ? screenshotNumbers : screenshotNumbers.phone;
+    const tabletNumbers = isArray ? screenshotNumbers : screenshotNumbers.tablet;
     filteredScreenshots = filteredScreenshots.filter((s) => {
       const match = s.filename.match(/^(\d+)\./);
-      if (match) {
-        const num = parseInt(match[1], 10);
-        return screenshotNumbers.includes(num);
+      if (!match) return false;
+      const num = parseInt(match[1], 10);
+      const numbersForDevice = s.type === "phone" ? phoneNumbers : tabletNumbers;
+      if (!numbersForDevice || numbersForDevice.length === 0) {
+        return true;
       }
-      return false;
+      return numbersForDevice.includes(num);
     });
-    results.push(`\u{1F522} Filtering screenshots: ${screenshotNumbers.join(", ")}`);
+    const filterParts = [];
+    if (isArray) {
+      filterParts.push(`all: ${screenshotNumbers.join(", ")}`);
+    } else {
+      if (phoneNumbers && phoneNumbers.length > 0) {
+        filterParts.push(`phone: ${phoneNumbers.join(", ")}`);
+      }
+      if (tabletNumbers && tabletNumbers.length > 0) {
+        filterParts.push(`tablet: ${tabletNumbers.join(", ")}`);
+      }
+    }
+    if (filterParts.length > 0) {
+      results.push(`\u{1F522} Filtering screenshots: ${filterParts.join(" | ")}`);
+    }
   }
   if (filteredScreenshots.length === 0) {
     const screenshotsDir2 = getScreenshotsDir(appInfo.slug);
@@ -4015,6 +4058,9 @@ ${screenshotsDir2}/${primaryLocale}/tablet/1.png, 2.png, ...`
   }
   results.push(`
 \u{1F680} Starting translations...`);
+  if (preserveWords && preserveWords.length > 0) {
+    results.push(`\u{1F512} Preserving words: ${preserveWords.join(", ")}`);
+  }
   const translationResult = await translateImagesWithProgress(
     tasks,
     (progress) => {
@@ -4032,7 +4078,8 @@ ${screenshotsDir2}/${primaryLocale}/tablet/1.png, 2.png, ...`
           `\u274C ${progressPrefix} ${progress.targetLocale}/${progress.deviceType}/${progress.filename}: ${progress.error}`
         );
       }
-    }
+    },
+    preserveWords
   );
   results.push(`
 \u{1F4CA} Translation Results:`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pabal-resource-mcp",
-  "version": "1.5.3",
+  "version": "1.5.4",
   "type": "module",
   "description": "MCP server for ASO data management with shared types and utilities",
   "author": "skyu",