pabal-resource-mcp 1.5.2 → 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,13 +3662,70 @@ async function getImageDimensions2(imagePath) {
     height: metadata.height
   };
 }
+async function detectCornerColor(imagePath) {
+  const image = sharp2(imagePath);
+  const metadata = await image.metadata();
+  const width = metadata.width || 100;
+  const height = metadata.height || 100;
+  const { data, info } = await image.raw().toBuffer({ resolveWithObject: true });
+  const channels = info.channels;
+  const colorCounts = /* @__PURE__ */ new Map();
+  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 / 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) {
+      existing.count++;
+    } else {
+      colorCounts.set(key, { count: 1, color: { r: qr, g: qg, b: qb } });
+    }
+  };
+  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 };
+  for (const { count, color } of colorCounts.values()) {
+    if (count > maxCount) {
+      maxCount = count;
+      dominantColor = color;
+    }
+  }
+  return dominantColor;
+}
 async function resizeImage(inputPath, outputPath, targetDimensions) {
+  const bgColor = await detectCornerColor(inputPath);
   await sharp2(inputPath).resize(targetDimensions.width, targetDimensions.height, {
-    fit: "fill",
-    // Exact resize to target dimensions
-    withoutEnlargement: false
+    fit: "contain",
+    // Preserve aspect ratio
+    withoutEnlargement: false,
     // Allow enlargement if needed
-  }).toFile(outputPath + ".tmp");
+    background: bgColor
+    // Use detected edge color
+  }).flatten({ background: bgColor }).png().toFile(outputPath + ".tmp");
   fs11.renameSync(outputPath + ".tmp", outputPath);
 }
 async function validateAndResizeImage(sourcePath, translatedPath) {
@@ -3717,7 +3777,19 @@ 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)")
+  skipExisting: z7.boolean().optional().default(true).describe("Skip translation if target file already exists (default: true)"),
+  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, {
   name: "LocalizeScreenshotsInput",
@@ -3832,7 +3904,9 @@ async function handleLocalizeScreenshots(input) {
     targetLocales: requestedTargetLocales,
     deviceTypes = ["phone", "tablet"],
     dryRun = false,
-    skipExisting = true
+    skipExisting = true,
+    screenshotNumbers,
+    preserveWords
   } = input;
   const results = [];
   let appInfo;
@@ -3884,9 +3958,38 @@ async function handleLocalizeScreenshots(input) {
   }
   results.push(`\u{1F3AF} Target locales: ${targetLocales.join(", ")}`);
   const sourceScreenshots = scanLocaleScreenshots(appInfo.slug, primaryLocale);
-  const filteredScreenshots = sourceScreenshots.filter(
+  let filteredScreenshots = sourceScreenshots.filter(
     (s) => deviceTypes.includes(s.type)
   );
+  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) return false;
+      const num = parseInt(match[1], 10);
+      const numbersForDevice = s.type === "phone" ? phoneNumbers : tabletNumbers;
+      if (!numbersForDevice || numbersForDevice.length === 0) {
+        return true;
+      }
+      return numbersForDevice.includes(num);
+    });
+    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);
     return {
@@ -3955,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) => {
@@ -3972,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.2",
+  "version": "1.5.4",
   "type": "module",
   "description": "MCP server for ASO data management with shared types and utilities",
   "author": "skyu",