@sylphx/image-reader-mcp 0.1.2 → 0.1.5

package/src/utils/palette.ts

@@ -1,36 +1,51 @@
-import sharp from 'sharp';
+/** Cheap local palette sample via optional sharp stats (not ML segmentation). */
 
-/** Cheap local palette sample via sharp stats (not ML segmentation). */
-export async function samplePalette(
-  filePath: string,
-  maxColors = 5
-): Promise<Array<{ hex: string; approx_share: number }>> {
-  const image = sharp(filePath, { failOn: 'none' }).resize(64, 64, { fit: 'inside' });
-  const stats = await image.stats();
-  const dominant = stats.dominant;
-  if (!dominant) return [];
-
-  const hex = (r: number, g: number, b: number) =>
-    `#${[r, g, b]
-      .map((v) =>
-        Math.max(0, Math.min(255, Math.round(v)))
-          .toString(16)
-          .padStart(2, '0')
-      )
-      .join('')}`;
-
-  const colors: Array<{ hex: string; approx_share: number }> = [
-    { hex: hex(dominant.r, dominant.g, dominant.b), approx_share: 0.55 },
-  ];
+type SharpLike = (
+  input: string,
+  opts?: { failOn?: string }
+) => {
+  resize: (
+    w: number,
+    h: number,
+    opts: { fit: string }
+  ) => {
+    stats: () => Promise<{
+      dominant?: { r: number; g: number; b: number };
+      channels?: Array<{ mean?: number }>;
+    }>;
+  };
+};
 
-  // Channel means as a second honest swatch (not true multi-color clustering).
-  const ch = stats.channels;
-  if (ch && ch.length >= 3) {
-    colors.push({
-      hex: hex(ch[0]?.mean ?? dominant.r, ch[1]?.mean ?? dominant.g, ch[2]?.mean ?? dominant.b),
-      approx_share: 0.45,
-    });
+export async function samplePalette(
+  filePath: string
+): Promise<Array<{ hex: string; approx_share: number }> | undefined> {
+  let sharp: SharpLike;
+  try {
+    const mod = await import('sharp');
+    const candidate = (mod as { default?: SharpLike }).default ?? (mod as unknown as SharpLike);
+    sharp = candidate;
+  } catch {
+    return undefined;
   }
 
-  return colors.slice(0, maxColors);
+  try {
+    const image = sharp(filePath, { failOn: 'none' }).resize(64, 64, { fit: 'inside' });
+    const stats = await image.stats();
+    const channels = stats.channels ?? [];
+    if (stats.dominant) {
+      const { r, g, b } = stats.dominant;
+      const hex = `#${[r, g, b].map((n) => n.toString(16).padStart(2, '0')).join('')}`;
+      return [{ hex, approx_share: 1 }];
+    }
+    if (channels.length >= 3) {
+      const r = Math.round(channels[0]?.mean ?? 0);
+      const g = Math.round(channels[1]?.mean ?? 0);
+      const b = Math.round(channels[2]?.mean ?? 0);
+      const hex = `#${[r, g, b].map((n) => n.toString(16).padStart(2, '0')).join('')}`;
+      return [{ hex, approx_share: 1 }];
+    }
+    return undefined;
+  } catch {
+    return undefined;
+  }
 }