pabal-resource-mcp 1.5.3 → 1.5.5

package/dist/bin/mcp-server.js

@@ -3432,6 +3432,87 @@ import { GoogleGenAI } from "@google/genai";
 import fs10 from "fs";
 import path10 from "path";
 import sharp from "sharp";
+var GEMINI_ASPECT_RATIOS = {
+  "1:1": { ratio: 1 / 1, width: 2048, height: 2048 },
+  "2:3": { ratio: 2 / 3, width: 1696, height: 2528 },
+  "3:2": { ratio: 3 / 2, width: 2528, height: 1696 },
+  "3:4": { ratio: 3 / 4, width: 1792, height: 2400 },
+  "4:3": { ratio: 4 / 3, width: 2400, height: 1792 },
+  "4:5": { ratio: 4 / 5, width: 1856, height: 2304 },
+  "5:4": { ratio: 5 / 4, width: 2304, height: 1856 },
+  "9:16": { ratio: 9 / 16, width: 1536, height: 2752 },
+  "16:9": { ratio: 16 / 9, width: 2752, height: 1536 },
+  "21:9": { ratio: 21 / 9, width: 1584, height: 672 }
+};
+var DEVICE_ASPECT_RATIOS = {
+  phone: "9:16",
+  tablet: "3:4"
+};
+var GEMINI_SUPPORTED_LOCALES = {
+  // English variants
+  "en": "EN",
+  "en-US": "EN",
+  "en-GB": "EN",
+  "en-AU": "EN",
+  "en-CA": "EN",
+  // Arabic
+  "ar": "ar-EG",
+  "ar-EG": "ar-EG",
+  "ar-SA": "ar-EG",
+  // German
+  "de": "de-DE",
+  "de-DE": "de-DE",
+  // Spanish
+  "es": "es-MX",
+  "es-MX": "es-MX",
+  "es-ES": "es-MX",
+  "es-419": "es-MX",
+  // French
+  "fr": "fr-FR",
+  "fr-FR": "fr-FR",
+  "fr-CA": "fr-FR",
+  // Hindi
+  "hi": "hi-IN",
+  "hi-IN": "hi-IN",
+  // Indonesian
+  "id": "id-ID",
+  "id-ID": "id-ID",
+  // Italian
+  "it": "it-IT",
+  "it-IT": "it-IT",
+  // Japanese
+  "ja": "ja-JP",
+  "ja-JP": "ja-JP",
+  // Korean
+  "ko": "ko-KR",
+  "ko-KR": "ko-KR",
+  // Portuguese
+  "pt": "pt-BR",
+  "pt-BR": "pt-BR",
+  "pt-PT": "pt-BR",
+  // Russian
+  "ru": "ru-RU",
+  "ru-RU": "ru-RU",
+  // Ukrainian
+  "uk": "ua-UA",
+  "uk-UA": "ua-UA",
+  "ua-UA": "ua-UA",
+  // Vietnamese
+  "vi": "vi-VN",
+  "vi-VN": "vi-VN",
+  // Chinese
+  "zh": "zh-CN",
+  "zh-CN": "zh-CN",
+  "zh-Hans": "zh-CN",
+  "zh-TW": "zh-CN",
+  "zh-Hant": "zh-CN"
+};
+function isGeminiSupportedLocale(locale) {
+  return locale in GEMINI_SUPPORTED_LOCALES;
+}
+function getUnsupportedLocales(locales) {
+  return locales.filter((locale) => !isGeminiSupportedLocale(locale));
+}
 var LANGUAGE_NAMES = {
   "en-US": "English (US)",
   "en-GB": "English (UK)",
@@ -3507,30 +3588,18 @@ function readImageAsBase64(imagePath) {
   }
   return { data: base64, mimeType };
 }
-async function getImageDimensions(imagePath) {
-  const metadata = await sharp(imagePath).metadata();
-  return {
-    width: metadata.width || 1080,
-    height: metadata.height || 1920
-  };
-}
-function calculateAspectRatio(width, height) {
-  const ratio = width / height;
-  if (Math.abs(ratio - 1) < 0.1) return "1:1";
-  if (Math.abs(ratio - 9 / 16) < 0.1) return "9:16";
-  if (Math.abs(ratio - 16 / 9) < 0.1) return "16:9";
-  if (Math.abs(ratio - 3 / 4) < 0.1) return "3:4";
-  if (Math.abs(ratio - 4 / 3) < 0.1) return "4:3";
-  return ratio < 1 ? "9:16" : "16:9";
+function getAspectRatioForDevice(deviceType) {
+  return DEVICE_ASPECT_RATIOS[deviceType];
 }
-async function translateImage(sourcePath, sourceLocale, targetLocale, outputPath) {
+async function translateImage(sourcePath, sourceLocale, targetLocale, outputPath, deviceType, preserveWords) {
   try {
     const client = getGeminiClient();
     const sourceLanguage = getLanguageName(sourceLocale);
     const targetLanguage = getLanguageName(targetLocale);
-    const { width, height } = await getImageDimensions(sourcePath);
-    const aspectRatio = calculateAspectRatio(width, height);
+    const aspectRatio = getAspectRatioForDevice(deviceType);
     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 +3609,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 +3673,7 @@ IMPORTANT INSTRUCTIONS:
     };
   }
 }
-async function translateImagesWithProgress(translations, onProgress) {
+async function translateImagesWithProgress(translations, onProgress, preserveWords) {
   let successful = 0;
   let failed = 0;
   const errors = [];
@@ -3626,7 +3695,9 @@ async function translateImagesWithProgress(translations, onProgress) {
       translation.sourcePath,
       translation.sourceLocale,
       translation.targetLocale,
-      translation.outputPath
+      translation.outputPath,
+      translation.deviceType,
+      preserveWords
     );
     if (result.success) {
       successful++;
@@ -3649,7 +3720,7 @@ async function translateImagesWithProgress(translations, onProgress) {
 // src/tools/aso/utils/localize-screenshots/image-resizer.util.ts
 import sharp2 from "sharp";
 import fs11 from "fs";
-async function getImageDimensions2(imagePath) {
+async function getImageDimensions(imagePath) {
   const metadata = await sharp2(imagePath).metadata();
   if (!metadata.width || !metadata.height) {
     throw new Error(`Unable to read dimensions from ${imagePath}`);
@@ -3659,7 +3730,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 +3738,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 +3757,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,20 +3785,22 @@ 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
     withoutEnlargement: false,
     // Allow enlargement if needed
-    background: bgColor
+    background: bgColor,
     // Use detected edge color
+    kernel: "lanczos3"
+    // High-quality downscaling algorithm
   }).flatten({ background: bgColor }).png().toFile(outputPath + ".tmp");
   fs11.renameSync(outputPath + ".tmp", outputPath);
 }
 async function validateAndResizeImage(sourcePath, translatedPath) {
-  const sourceDimensions = await getImageDimensions2(sourcePath);
-  const translatedDimensions = await getImageDimensions2(translatedPath);
+  const sourceDimensions = await getImageDimensions(sourcePath);
+  const translatedDimensions = await getImageDimensions(translatedPath);
   const needsResize = sourceDimensions.width !== translatedDimensions.width || sourceDimensions.height !== translatedDimensions.height;
   if (needsResize) {
     await resizeImage(translatedPath, translatedPath, sourceDimensions);
@@ -3763,8 +3848,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, {
@@ -3841,12 +3935,14 @@ function getTargetLocales(allLocales, primaryLocale, requestedTargets) {
     );
     if (invalidTargets.length > 0) {
       console.warn(
-        `Warning: Some requested locales are not supported: ${invalidTargets.join(", ")}`
+        `Warning: Some requested locales are not in product: ${invalidTargets.join(", ")}`
       );
     }
     targets = validTargets.filter((t) => t !== primaryLocale);
   }
-  return targets;
+  const skippedLocales = getUnsupportedLocales(targets);
+  const supportedTargets = targets.filter((t) => isGeminiSupportedLocale(t));
+  return { targets: supportedTargets, skippedLocales };
 }
 function buildTranslationTasks(slug, screenshots, primaryLocale, targetLocales, skipExisting) {
   const tasks = [];
@@ -3881,7 +3977,8 @@ async function handleLocalizeScreenshots(input) {
     deviceTypes = ["phone", "tablet"],
     dryRun = false,
     skipExisting = true,
-    screenshotNumbers
+    screenshotNumbers,
+    preserveWords
   } = input;
   const results = [];
   let appInfo;
@@ -3916,36 +4013,58 @@ async function handleLocalizeScreenshots(input) {
       ]
     };
   }
-  const targetLocales = getTargetLocales(
+  const { targets: targetLocales, skippedLocales } = getTargetLocales(
     allLocales,
     primaryLocale,
     requestedTargetLocales
   );
   if (targetLocales.length === 0) {
+    const skippedMsg = skippedLocales.length > 0 ? ` (Skipped due to Gemini limitation: ${skippedLocales.join(", ")})` : "";
     return {
       content: [
         {
           type: "text",
-          text: `\u274C No target locales to translate to. Primary locale: ${primaryLocale}, Available: ${allLocales.join(", ")}`
+          text: `\u274C No target locales to translate to. Primary locale: ${primaryLocale}, Available: ${allLocales.join(", ")}${skippedMsg}`
         }
       ]
     };
   }
   results.push(`\u{1F3AF} Target locales: ${targetLocales.join(", ")}`);
+  if (skippedLocales.length > 0) {
+    results.push(`\u26A0\uFE0F Skipped locales (not supported by Gemini): ${skippedLocales.join(", ")}`);
+  }
   const sourceScreenshots = scanLocaleScreenshots(appInfo.slug, primaryLocale);
   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 +4134,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 +4154,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.5",
   "type": "module",
   "description": "MCP server for ASO data management with shared types and utilities",
   "author": "skyu",