whale-code 6.5.6 → 6.5.8

package/dist/server/handlers/image-gen-providers.js

@@ -0,0 +1,161 @@
+// server/handlers/image-gen-providers.ts — Image generation provider implementations
+// Extracted from image-gen.ts for single-responsibility.
+
+import OpenAI from "openai";
+import { GoogleGenAI } from "@google/genai";
+
+// ============================================================================
+// TYPES
+// ============================================================================
+
+// ============================================================================
+// CREDENTIAL RESOLUTION
+// ============================================================================
+
+export async function getImageCredentials(sb, storeId) {
+  const creds = {};
+  const keys = ["OPENAI_API_KEY", "GOOGLE_AI_API_KEY"];
+  const results = await Promise.all(keys.map(async k => {
+    try {
+      const r = await sb.rpc("decrypt_secret", {
+        p_name: k,
+        p_store_id: storeId
+      });
+      return r.data;
+    } catch {
+      return null;
+    }
+  }));
+  if (results[0]) creds.openai = results[0];
+  if (results[1]) creds.google = results[1];
+  return creds;
+}
+
+// ============================================================================
+// PROVIDER IMPLEMENTATIONS
+// ============================================================================
+
+export async function generateWithDalle(apiKey, prompt, size, quality, style) {
+  const client = new OpenAI({
+    apiKey
+  });
+  const validSizes = ["1024x1024", "1024x1792", "1792x1024"];
+  const selectedSize = validSizes.includes(size) ? size : "1024x1024";
+  const resp = await client.images.generate({
+    model: "dall-e-3",
+    prompt,
+    n: 1,
+    size: selectedSize,
+    quality: quality === "hd" ? "hd" : "standard",
+    style: style === "natural" ? "natural" : "vivid",
+    response_format: "b64_json"
+  });
+  const first = resp.data?.[0];
+  if (!first?.b64_json) throw new Error("DALL-E returned no image data");
+  return {
+    base64: first.b64_json,
+    revisedPrompt: first.revised_prompt || prompt
+  };
+}
+export async function generateWithImagen(apiKey, prompt, aspectRatio) {
+  const client = new GoogleGenAI({
+    apiKey
+  });
+  const validRatios = ["1:1", "3:4", "4:3", "9:16", "16:9"];
+  const selectedRatio = validRatios.includes(aspectRatio) ? aspectRatio : "1:1";
+  const response = await client.models.generateImages({
+    model: "imagen-4.0-generate-001",
+    prompt,
+    config: {
+      numberOfImages: 1,
+      aspectRatio: selectedRatio
+    }
+  });
+  const image = response.generatedImages?.[0];
+  if (!image?.image?.imageBytes) {
+    throw new Error("Imagen returned no image data");
+  }
+  return {
+    base64: image.image.imageBytes
+  };
+}
+export async function generateWithGemini(apiKey, prompt, model = "gemini-2.5-flash-image") {
+  const client = new GoogleGenAI({
+    apiKey
+  });
+  const response = await client.models.generateContent({
+    model,
+    contents: prompt,
+    config: {
+      responseModalities: ["TEXT", "IMAGE"]
+    }
+  });
+  const parts = response.candidates?.[0]?.content?.parts;
+  if (!parts) throw new Error("Gemini returned no content parts");
+  for (const part of parts) {
+    if (part.inlineData?.data) {
+      return {
+        base64: part.inlineData.data,
+        mimeType: part.inlineData.mimeType || "image/png"
+      };
+    }
+  }
+  throw new Error("Gemini response contained no image data");
+}
+
+// ============================================================================
+// STORAGE + MEDIA HELPERS
+// ============================================================================
+
+export async function uploadAndRecord(sb, storeId, base64Data, prompt, provider, model, mimeType = "image/png") {
+  const id = crypto.randomUUID();
+  const ext = mimeType.includes("jpeg") || mimeType.includes("jpg") ? "jpg" : "png";
+  const fileName = `${id}.${ext}`;
+  const storeIdUpper = storeId.toUpperCase();
+  const storagePath = `ai-generated/${storeIdUpper}/standalone/${fileName}`;
+  const buffer = Buffer.from(base64Data, "base64");
+  const {
+    error: uploadErr
+  } = await sb.storage.from("product-images").upload(storagePath, buffer, {
+    contentType: mimeType,
+    upsert: true
+  });
+  if (uploadErr) {
+    throw new Error(`Storage upload failed: ${uploadErr.message}`);
+  }
+  const {
+    data: urlData
+  } = sb.storage.from("product-images").getPublicUrl(storagePath);
+  const fileUrl = urlData.publicUrl;
+  const aiTags = [provider === "openai" ? "DALL-E 3" : provider === "imagen" ? "Imagen 3" : "Gemini", "AI Generated"];
+  const {
+    data: mediaRow,
+    error: mediaErr
+  } = await sb.from("store_media").insert({
+    store_id: storeId,
+    file_name: fileName,
+    file_path: storagePath,
+    file_url: fileUrl,
+    file_size: buffer.length,
+    file_type: mimeType,
+    category: "ai_generated",
+    ai_tags: aiTags,
+    ai_description: prompt.substring(0, 500),
+    source: `mcp-${provider}`,
+    folder: "ai-generated"
+  }).select("id").single();
+  if (mediaErr) {
+    console.error("[image-gen] store_media insert error:", mediaErr.message);
+  }
+  return {
+    id,
+    provider,
+    model,
+    prompt,
+    file_url: fileUrl,
+    file_name: fileName,
+    file_size: buffer.length,
+    media_id: mediaRow?.id || id
+  };
+}
+//# sourceMappingURL=image-gen-providers.js.map

package/dist/server/handlers/image-gen-providers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image-gen-providers.js","names":["OpenAI","GoogleGenAI","getImageCredentials","sb","storeId","creds","keys","results","Promise","all","map","k","r","rpc","p_name","p_store_id","data","openai","google","generateWithDalle","apiKey","prompt","size","quality","style","client","validSizes","selectedSize","includes","resp","images","generate","model","n","response_format","first","b64_json","Error","base64","revisedPrompt","revised_prompt","generateWithImagen","aspectRatio","validRatios","selectedRatio","response","models","generateImages","config","numberOfImages","image","generatedImages","imageBytes","generateWithGemini","generateContent","contents","responseModalities","parts","candidates","content","part","inlineData","mimeType","uploadAndRecord","base64Data","provider","id","crypto","randomUUID","ext","fileName","storeIdUpper","toUpperCase","storagePath","buffer","Buffer","from","error","uploadErr","storage","upload","contentType","upsert","message","urlData","getPublicUrl","fileUrl","publicUrl","aiTags","mediaRow","mediaErr","insert","store_id","file_name","file_path","file_url","file_size","length","file_type","category","ai_tags","ai_description","substring","source","folder","select","single","console","media_id"],"sources":["../../../src/server/handlers/image-gen-providers.ts"],"sourcesContent":["// server/handlers/image-gen-providers.ts — Image generation provider implementations\n// Extracted from image-gen.ts for single-responsibility.\n\nimport type { SupabaseClient } from \"@supabase/supabase-js\";\nimport OpenAI from \"openai\";\nimport { GoogleGenAI } from \"@google/genai\";\n\n// ============================================================================\n// TYPES\n// ============================================================================\n\nexport type ImageProvider = \"openai\" | \"imagen\" | \"gemini\" | \"gemini3\";\n\nexport interface GeneratedImage {\n  id: string;\n  provider: ImageProvider;\n  model: string;\n  prompt: string;\n  file_url: string;\n  file_name: string;\n  file_size: number;\n  media_id: string;\n}\n\nexport interface ImageCredentials {\n  openai?: string;\n  google?: string;\n}\n\n// ============================================================================\n// CREDENTIAL RESOLUTION\n// ============================================================================\n\nexport async function getImageCredentials(sb: SupabaseClient, storeId: string): Promise<ImageCredentials> {\n  const creds: ImageCredentials = {};\n  const keys = [\"OPENAI_API_KEY\", \"GOOGLE_AI_API_KEY\"] as const;\n  const results = await Promise.all(\n    keys.map(async (k) => {\n      try {\n        const r = await sb.rpc(\"decrypt_secret\", { p_name: k, p_store_id: storeId });\n        return r.data as string | null;\n      } catch { return null; }\n    }),\n  );\n  if (results[0]) creds.openai = results[0];\n  if (results[1]) creds.google = results[1];\n  return creds;\n}\n\n// ============================================================================\n// PROVIDER IMPLEMENTATIONS\n// ============================================================================\n\nexport async function generateWithDalle(\n  apiKey: string,\n  prompt: string,\n  size: string,\n  quality: string,\n  style: string,\n): Promise<{ base64: string; revisedPrompt: string }> {\n  const client = new OpenAI({ apiKey });\n  const validSizes = [\"1024x1024\", \"1024x1792\", \"1792x1024\"];\n  const selectedSize = validSizes.includes(size) ? size as \"1024x1024\" | \"1024x1792\" | \"1792x1024\" : \"1024x1024\";\n\n  const resp = await client.images.generate({\n    model: \"dall-e-3\",\n    prompt,\n    n: 1,\n    size: selectedSize,\n    quality: quality === \"hd\" ? \"hd\" : \"standard\",\n    style: style === \"natural\" ? \"natural\" : \"vivid\",\n    response_format: \"b64_json\",\n  });\n\n  const first = resp.data?.[0];\n  if (!first?.b64_json) throw new Error(\"DALL-E returned no image data\");\n\n  return {\n    base64: first.b64_json,\n    revisedPrompt: first.revised_prompt || prompt,\n  };\n}\n\nexport async function generateWithImagen(\n  apiKey: string,\n  prompt: string,\n  aspectRatio: string,\n): Promise<{ base64: string }> {\n  const client = new GoogleGenAI({ apiKey });\n  const validRatios = [\"1:1\", \"3:4\", \"4:3\", \"9:16\", \"16:9\"];\n  const selectedRatio = validRatios.includes(aspectRatio) ? aspectRatio : \"1:1\";\n\n  const response = await client.models.generateImages({\n    model: \"imagen-4.0-generate-001\",\n    prompt,\n    config: {\n      numberOfImages: 1,\n      aspectRatio: selectedRatio,\n    },\n  });\n\n  const image = response.generatedImages?.[0];\n  if (!image?.image?.imageBytes) {\n    throw new Error(\"Imagen returned no image data\");\n  }\n\n  return { base64: image.image.imageBytes };\n}\n\nexport async function generateWithGemini(\n  apiKey: string,\n  prompt: string,\n  model: string = \"gemini-2.5-flash-image\",\n): Promise<{ base64: string; mimeType: string }> {\n  const client = new GoogleGenAI({ apiKey });\n\n  const response = await client.models.generateContent({\n    model,\n    contents: prompt,\n    config: {\n      responseModalities: [\"TEXT\", \"IMAGE\"],\n    },\n  });\n\n  const parts = response.candidates?.[0]?.content?.parts;\n  if (!parts) throw new Error(\"Gemini returned no content parts\");\n\n  for (const part of parts) {\n    if (part.inlineData?.data) {\n      return {\n        base64: part.inlineData.data,\n        mimeType: part.inlineData.mimeType || \"image/png\",\n      };\n    }\n  }\n\n  throw new Error(\"Gemini response contained no image data\");\n}\n\n// ============================================================================\n// STORAGE + MEDIA HELPERS\n// ============================================================================\n\nexport async function uploadAndRecord(\n  sb: SupabaseClient,\n  storeId: string,\n  base64Data: string,\n  prompt: string,\n  provider: ImageProvider,\n  model: string,\n  mimeType: string = \"image/png\",\n): Promise<GeneratedImage> {\n  const id = crypto.randomUUID();\n  const ext = mimeType.includes(\"jpeg\") || mimeType.includes(\"jpg\") ? \"jpg\" : \"png\";\n  const fileName = `${id}.${ext}`;\n  const storeIdUpper = storeId.toUpperCase();\n  const storagePath = `ai-generated/${storeIdUpper}/standalone/${fileName}`;\n\n  const buffer = Buffer.from(base64Data, \"base64\");\n\n  const { error: uploadErr } = await sb.storage\n    .from(\"product-images\")\n    .upload(storagePath, buffer, { contentType: mimeType, upsert: true });\n\n  if (uploadErr) {\n    throw new Error(`Storage upload failed: ${uploadErr.message}`);\n  }\n\n  const { data: urlData } = sb.storage.from(\"product-images\").getPublicUrl(storagePath);\n  const fileUrl = urlData.publicUrl;\n\n  const aiTags = [\n    provider === \"openai\" ? \"DALL-E 3\" : provider === \"imagen\" ? \"Imagen 3\" : \"Gemini\",\n    \"AI Generated\",\n  ];\n\n  const { data: mediaRow, error: mediaErr } = await sb\n    .from(\"store_media\")\n    .insert({\n      store_id: storeId,\n      file_name: fileName,\n      file_path: storagePath,\n      file_url: fileUrl,\n      file_size: buffer.length,\n      file_type: mimeType,\n      category: \"ai_generated\",\n      ai_tags: aiTags,\n      ai_description: prompt.substring(0, 500),\n      source: `mcp-${provider}`,\n      folder: \"ai-generated\",\n    })\n    .select(\"id\")\n    .single();\n\n  if (mediaErr) {\n    console.error(\"[image-gen] store_media insert error:\", mediaErr.message);\n  }\n\n  return {\n    id,\n    provider,\n    model,\n    prompt,\n    file_url: fileUrl,\n    file_name: fileName,\n    file_size: buffer.length,\n    media_id: mediaRow?.id || id,\n  };\n}\n"],"mappings":"AAAA;AACA;;AAGA,OAAOA,MAAM,MAAM,QAAQ;AAC3B,SAASC,WAAW,QAAQ,eAAe;;AAE3C;AACA;AACA;;AAoBA;AACA;AACA;;AAEA,OAAO,eAAeC,mBAAmBA,CAACC,EAAkB,EAAEC,OAAe,EAA6B;EACxG,MAAMC,KAAuB,GAAG,CAAC,CAAC;EAClC,MAAMC,IAAI,GAAG,CAAC,gBAAgB,EAAE,mBAAmB,CAAU;EAC7D,MAAMC,OAAO,GAAG,MAAMC,OAAO,CAACC,GAAG,CAC/BH,IAAI,CAACI,GAAG,CAAC,MAAOC,CAAC,IAAK;IACpB,IAAI;MACF,MAAMC,CAAC,GAAG,MAAMT,EAAE,CAACU,GAAG,CAAC,gBAAgB,EAAE;QAAEC,MAAM,EAAEH,CAAC;QAAEI,UAAU,EAAEX;MAAQ,CAAC,CAAC;MAC5E,OAAOQ,CAAC,CAACI,IAAI;IACf,CAAC,CAAC,MAAM;MAAE,OAAO,IAAI;IAAE;EACzB,CAAC,CACH,CAAC;EACD,IAAIT,OAAO,CAAC,CAAC,CAAC,EAAEF,KAAK,CAACY,MAAM,GAAGV,OAAO,CAAC,CAAC,CAAC;EACzC,IAAIA,OAAO,CAAC,CAAC,CAAC,EAAEF,KAAK,CAACa,MAAM,GAAGX,OAAO,CAAC,CAAC,CAAC;EACzC,OAAOF,KAAK;AACd;;AAEA;AACA;AACA;;AAEA,OAAO,eAAec,iBAAiBA,CACrCC,MAAc,EACdC,MAAc,EACdC,IAAY,EACZC,OAAe,EACfC,KAAa,EACuC;EACpD,MAAMC,MAAM,GAAG,IAAIzB,MAAM,CAAC;IAAEoB;EAAO,CAAC,CAAC;EACrC,MAAMM,UAAU,GAAG,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC;EAC1D,MAAMC,YAAY,GAAGD,UAAU,CAACE,QAAQ,CAACN,IAAI,CAAC,GAAGA,IAAI,GAA8C,WAAW;EAE9G,MAAMO,IAAI,GAAG,MAAMJ,MAAM,CAACK,MAAM,CAACC,QAAQ,CAAC;IACxCC,KAAK,EAAE,UAAU;IACjBX,MAAM;IACNY,CAAC,EAAE,CAAC;IACJX,IAAI,EAAEK,YAAY;IAClBJ,OAAO,EAAEA,OAAO,KAAK,IAAI,GAAG,IAAI,GAAG,UAAU;IAC7CC,KAAK,EAAEA,KAAK,KAAK,SAAS,GAAG,SAAS,GAAG,OAAO;IAChDU,eAAe,EAAE;EACnB,CAAC,CAAC;EAEF,MAAMC,KAAK,GAAGN,IAAI,CAACb,IAAI,GAAG,CAAC,CAAC;EAC5B,IAAI,CAACmB,KAAK,EAAEC,QAAQ,EAAE,MAAM,IAAIC,KAAK,CAAC,+BAA+B,CAAC;EAEtE,OAAO;IACLC,MAAM,EAAEH,KAAK,CAACC,QAAQ;IACtBG,aAAa,EAAEJ,KAAK,CAACK,cAAc,IAAInB;EACzC,CAAC;AACH;AAEA,OAAO,eAAeoB,kBAAkBA,CACtCrB,MAAc,EACdC,MAAc,EACdqB,WAAmB,EACU;EAC7B,MAAMjB,MAAM,GAAG,IAAIxB,WAAW,CAAC;IAAEmB;EAAO,CAAC,CAAC;EAC1C,MAAMuB,WAAW,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;EACzD,MAAMC,aAAa,GAAGD,WAAW,CAACf,QAAQ,CAACc,WAAW,CAAC,GAAGA,WAAW,GAAG,KAAK;EAE7E,MAAMG,QAAQ,GAAG,MAAMpB,MAAM,CAACqB,MAAM,CAACC,cAAc,CAAC;IAClDf,KAAK,EAAE,yBAAyB;IAChCX,MAAM;IACN2B,MAAM,EAAE;MACNC,cAAc,EAAE,CAAC;MACjBP,WAAW,EAAEE;IACf;EACF,CAAC,CAAC;EAEF,MAAMM,KAAK,GAAGL,QAAQ,CAACM,eAAe,GAAG,CAAC,CAAC;EAC3C,IAAI,CAACD,KAAK,EAAEA,KAAK,EAAEE,UAAU,EAAE;IAC7B,MAAM,IAAIf,KAAK,CAAC,+BAA+B,CAAC;EAClD;EAEA,OAAO;IAAEC,MAAM,EAAEY,KAAK,CAACA,KAAK,CAACE;EAAW,CAAC;AAC3C;AAEA,OAAO,eAAeC,kBAAkBA,CACtCjC,MAAc,EACdC,MAAc,EACdW,KAAa,GAAG,wBAAwB,EACO;EAC/C,MAAMP,MAAM,GAAG,IAAIxB,WAAW,CAAC;IAAEmB;EAAO,CAAC,CAAC;EAE1C,MAAMyB,QAAQ,GAAG,MAAMpB,MAAM,CAACqB,MAAM,CAACQ,eAAe,CAAC;IACnDtB,KAAK;IACLuB,QAAQ,EAAElC,MAAM;IAChB2B,MAAM,EAAE;MACNQ,kBAAkB,EAAE,CAAC,MAAM,EAAE,OAAO;IACtC;EACF,CAAC,CAAC;EAEF,MAAMC,KAAK,GAAGZ,QAAQ,CAACa,UAAU,GAAG,CAAC,CAAC,EAAEC,OAAO,EAAEF,KAAK;EACtD,IAAI,CAACA,KAAK,EAAE,MAAM,IAAIpB,KAAK,CAAC,kCAAkC,CAAC;EAE/D,KAAK,MAAMuB,IAAI,IAAIH,KAAK,EAAE;IACxB,IAAIG,IAAI,CAACC,UAAU,EAAE7C,IAAI,EAAE;MACzB,OAAO;QACLsB,MAAM,EAAEsB,IAAI,CAACC,UAAU,CAAC7C,IAAI;QAC5B8C,QAAQ,EAAEF,IAAI,CAACC,UAAU,CAACC,QAAQ,IAAI;MACxC,CAAC;IACH;EACF;EAEA,MAAM,IAAIzB,KAAK,CAAC,yCAAyC,CAAC;AAC5D;;AAEA;AACA;AACA;;AAEA,OAAO,eAAe0B,eAAeA,CACnC5D,EAAkB,EAClBC,OAAe,EACf4D,UAAkB,EAClB3C,MAAc,EACd4C,QAAuB,EACvBjC,KAAa,EACb8B,QAAgB,GAAG,WAAW,EACL;EACzB,MAAMI,EAAE,GAAGC,MAAM,CAACC,UAAU,CAAC,CAAC;EAC9B,MAAMC,GAAG,GAAGP,QAAQ,CAAClC,QAAQ,CAAC,MAAM,CAAC,IAAIkC,QAAQ,CAAClC,QAAQ,CAAC,KAAK,CAAC,GAAG,KAAK,GAAG,KAAK;EACjF,MAAM0C,QAAQ,GAAG,GAAGJ,EAAE,IAAIG,GAAG,EAAE;EAC/B,MAAME,YAAY,GAAGnE,OAAO,CAACoE,WAAW,CAAC,CAAC;EAC1C,MAAMC,WAAW,GAAG,gBAAgBF,YAAY,eAAeD,QAAQ,EAAE;EAEzE,MAAMI,MAAM,GAAGC,MAAM,CAACC,IAAI,CAACZ,UAAU,EAAE,QAAQ,CAAC;EAEhD,MAAM;IAAEa,KAAK,EAAEC;EAAU,CAAC,GAAG,MAAM3E,EAAE,CAAC4E,OAAO,CAC1CH,IAAI,CAAC,gBAAgB,CAAC,CACtBI,MAAM,CAACP,WAAW,EAAEC,MAAM,EAAE;IAAEO,WAAW,EAAEnB,QAAQ;IAAEoB,MAAM,EAAE;EAAK,CAAC,CAAC;EAEvE,IAAIJ,SAAS,EAAE;IACb,MAAM,IAAIzC,KAAK,CAAC,0BAA0ByC,SAAS,CAACK,OAAO,EAAE,CAAC;EAChE;EAEA,MAAM;IAAEnE,IAAI,EAAEoE;EAAQ,CAAC,GAAGjF,EAAE,CAAC4E,OAAO,CAACH,IAAI,CAAC,gBAAgB,CAAC,CAACS,YAAY,CAACZ,WAAW,CAAC;EACrF,MAAMa,OAAO,GAAGF,OAAO,CAACG,SAAS;EAEjC,MAAMC,MAAM,GAAG,CACbvB,QAAQ,KAAK,QAAQ,GAAG,UAAU,GAAGA,QAAQ,KAAK,QAAQ,GAAG,UAAU,GAAG,QAAQ,EAClF,cAAc,CACf;EAED,MAAM;IAAEjD,IAAI,EAAEyE,QAAQ;IAAEZ,KAAK,EAAEa;EAAS,CAAC,GAAG,MAAMvF,EAAE,CACjDyE,IAAI,CAAC,aAAa,CAAC,CACnBe,MAAM,CAAC;IACNC,QAAQ,EAAExF,OAAO;IACjByF,SAAS,EAAEvB,QAAQ;IACnBwB,SAAS,EAAErB,WAAW;IACtBsB,QAAQ,EAAET,OAAO;IACjBU,SAAS,EAAEtB,MAAM,CAACuB,MAAM;IACxBC,SAAS,EAAEpC,QAAQ;IACnBqC,QAAQ,EAAE,cAAc;IACxBC,OAAO,EAAEZ,MAAM;IACfa,cAAc,EAAEhF,MAAM,CAACiF,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;IACxCC,MAAM,EAAE,OAAOtC,QAAQ,EAAE;IACzBuC,MAAM,EAAE;EACV,CAAC,CAAC,CACDC,MAAM,CAAC,IAAI,CAAC,CACZC,MAAM,CAAC,CAAC;EAEX,IAAIhB,QAAQ,EAAE;IACZiB,OAAO,CAAC9B,KAAK,CAAC,uCAAuC,EAAEa,QAAQ,CAACP,OAAO,CAAC;EAC1E;EAEA,OAAO;IACLjB,EAAE;IACFD,QAAQ;IACRjC,KAAK;IACLX,MAAM;IACN0E,QAAQ,EAAET,OAAO;IACjBO,SAAS,EAAEvB,QAAQ;IACnB0B,SAAS,EAAEtB,MAAM,CAACuB,MAAM;IACxBW,QAAQ,EAAEnB,QAAQ,EAAEvB,EAAE,IAAIA;EAC5B,CAAC;AACH","ignoreList":[]}

package/dist/server/handlers/image-gen.test.js

@@ -0,0 +1,205 @@
+// image-gen.test.ts — Tests for handleImageGen handler
+
+import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
+import { HAS_TEST_DB, getTestClient, seedTracked, cleanup, TEST_STORE, RUN_PREFIX } from "./__test-utils__/test-db.js";
+import { randomUUID } from "node:crypto";
+import { handleImageGen } from "./image-gen.js";
+const sb = getTestClient();
+
+/** Helper: spy on sb.rpc to intercept decrypt_secret calls */
+function stubDecryptSecret(returnValue) {
+  const originalRpc = sb.rpc.bind(sb);
+  vi.spyOn(sb, "rpc").mockImplementation((fn, params) => {
+    if (fn === "decrypt_secret") {
+      return Promise.resolve({
+        data: returnValue,
+        error: null
+      });
+    }
+    return originalRpc(fn, params);
+  });
+}
+describe.skipIf(!HAS_TEST_DB)("handleImageGen", () => {
+  beforeEach(() => {
+    vi.restoreAllMocks();
+  });
+  afterEach(async () => {
+    vi.restoreAllMocks();
+    await cleanup();
+  });
+  it("returns error when store_id is missing", async () => {
+    const result = await handleImageGen(sb, {
+      action: "generate"
+    });
+    expect(result.success).toBe(false);
+    expect(result.error).toBe("store_id is required");
+  });
+  it("returns error when prompt is missing for generate", async () => {
+    stubDecryptSecret("fake-key");
+    const result = await handleImageGen(sb, {
+      action: "generate"
+    }, TEST_STORE);
+    expect(result.success).toBe(false);
+    expect(result.error).toBe("prompt is required");
+  });
+  it("returns error when OpenAI key is not configured for openai provider", async () => {
+    stubDecryptSecret(null);
+    const result = await handleImageGen(sb, {
+      action: "generate",
+      prompt: "A whale",
+      provider: "openai"
+    }, TEST_STORE);
+    expect(result.success).toBe(false);
+    expect(result.error).toContain("OpenAI API key not configured");
+  });
+  it("returns error when Google key is not configured for imagen provider", async () => {
+    stubDecryptSecret(null);
+    const result = await handleImageGen(sb, {
+      action: "generate",
+      prompt: "A whale",
+      provider: "imagen"
+    }, TEST_STORE);
+    expect(result.success).toBe(false);
+    expect(result.error).toContain("Google AI API key not configured");
+  });
+  it("returns error when Google key is not configured for gemini provider", async () => {
+    stubDecryptSecret(null);
+    const result = await handleImageGen(sb, {
+      action: "generate",
+      prompt: "A whale",
+      provider: "gemini"
+    }, TEST_STORE);
+    expect(result.success).toBe(false);
+    expect(result.error).toContain("Google AI API key not configured");
+  });
+  it("returns error for unknown provider", async () => {
+    stubDecryptSecret("fake-key");
+    const result = await handleImageGen(sb, {
+      action: "generate",
+      prompt: "Test",
+      provider: "midjourney"
+    }, TEST_STORE);
+    expect(result.success).toBe(false);
+    expect(result.error).toContain("Unknown provider");
+  });
+  it("returns error for batch without items", async () => {
+    const result = await handleImageGen(sb, {
+      action: "batch"
+    }, TEST_STORE);
+    expect(result.success).toBe(false);
+    expect(result.error).toContain("items array is required");
+  });
+  it("returns error for batch exceeding 10 items", async () => {
+    const items = Array.from({
+      length: 11
+    }, (_, i) => ({
+      prompt: `Image ${i}`
+    }));
+    const result = await handleImageGen(sb, {
+      action: "batch",
+      items
+    }, TEST_STORE);
+    expect(result.success).toBe(false);
+    expect(result.error).toBe("Maximum 10 images per batch");
+  });
+  it("lists images from store_media", async () => {
+    await seedTracked("store_media", [{
+      id: randomUUID(),
+      store_id: TEST_STORE,
+      file_name: `${RUN_PREFIX}_image1.png`,
+      file_path: `stores/${TEST_STORE}/media/${RUN_PREFIX}_image1.png`,
+      file_url: "https://example.com/image1.png",
+      file_type: "image/png",
+      file_size: 1024,
+      category: "ai_generated",
+      status: "active"
+    }, {
+      id: randomUUID(),
+      store_id: TEST_STORE,
+      file_name: `${RUN_PREFIX}_image2.png`,
+      file_path: `stores/${TEST_STORE}/media/${RUN_PREFIX}_image2.png`,
+      file_url: "https://example.com/image2.png",
+      file_type: "image/png",
+      file_size: 2048,
+      category: "ai_generated",
+      status: "active"
+    }]);
+    const result = await handleImageGen(sb, {
+      action: "list"
+    }, TEST_STORE);
+    expect(result.success).toBe(true);
+    const data = result.data;
+    expect(data.count).toBeGreaterThanOrEqual(2);
+  });
+  it("gets a specific image by media_id", async () => {
+    const mediaId = randomUUID();
+    await seedTracked("store_media", [{
+      id: mediaId,
+      store_id: TEST_STORE,
+      file_name: `${RUN_PREFIX}_image1.png`,
+      file_path: `stores/${TEST_STORE}/media/${RUN_PREFIX}_image1.png`,
+      file_url: "https://example.com/image1.png",
+      file_type: "image/png",
+      file_size: 1024,
+      category: "ai_generated",
+      status: "active"
+    }]);
+    const result = await handleImageGen(sb, {
+      action: "get",
+      media_id: mediaId
+    }, TEST_STORE);
+    expect(result.success).toBe(true);
+    expect(result.data.file_name).toBe(`${RUN_PREFIX}_image1.png`);
+  });
+  it("returns error for get without media_id", async () => {
+    const result = await handleImageGen(sb, {
+      action: "get"
+    }, TEST_STORE);
+    expect(result.success).toBe(false);
+    expect(result.error).toBe("media_id is required");
+  });
+  it("soft-deletes an image", async () => {
+    const mediaId = randomUUID();
+    await seedTracked("store_media", [{
+      id: mediaId,
+      store_id: TEST_STORE,
+      file_name: `${RUN_PREFIX}_to-delete.png`,
+      file_path: `stores/${TEST_STORE}/media/${RUN_PREFIX}_to-delete.png`,
+      file_url: "https://example.com/to-delete.png",
+      file_type: "image/png",
+      file_size: 512,
+      category: "ai_generated",
+      status: "active"
+    }]);
+    const result = await handleImageGen(sb, {
+      action: "delete",
+      media_id: mediaId
+    }, TEST_STORE);
+    expect(result.success).toBe(true);
+    expect(result.data.deleted).toBe(mediaId);
+  });
+  it("returns error for delete without media_id", async () => {
+    const result = await handleImageGen(sb, {
+      action: "delete"
+    }, TEST_STORE);
+    expect(result.success).toBe(false);
+    expect(result.error).toBe("media_id is required");
+  });
+  it("returns provider configuration", async () => {
+    stubDecryptSecret(null);
+    const result = await handleImageGen(sb, {
+      action: "providers"
+    }, TEST_STORE);
+    expect(result.success).toBe(true);
+    const data = result.data;
+    const providers = data.providers;
+    expect(providers.length).toBe(4);
+  });
+  it("returns error for unknown action", async () => {
+    const result = await handleImageGen(sb, {
+      action: "invalid_action"
+    }, TEST_STORE);
+    expect(result.success).toBe(false);
+    expect(result.error).toContain("Unknown action");
+  });
+});

package/dist/server/handlers/inventory-audit.d.ts

@@ -0,0 +1,10 @@
+import type { SupabaseClient } from "@supabase/supabase-js";
+export declare function handleInventoryAudit(sb: SupabaseClient, args: Record<string, unknown>, storeId?: string): Promise<{
+    success: boolean;
+    error: string;
+    data?: undefined;
+} | {
+    success: boolean;
+    data: any;
+    error?: undefined;
+}>;

package/dist/server/handlers/inventory-audit.js

@@ -0,0 +1,90 @@
+// server/handlers/inventory-audit.ts — Inventory audit handlers
+
+export async function handleInventoryAudit(sb, args, storeId) {
+  const sid = storeId;
+  switch (args.action) {
+    case "start":
+      {
+        const {
+          data,
+          error
+        } = await sb.from("inventory_audits").insert({
+          store_id: sid,
+          location_id: args.location_id,
+          audit_type: args.audit_type || "full",
+          status: "in_progress",
+          started_at: new Date().toISOString()
+        }).select().single();
+        return error ? {
+          success: false,
+          error: error.message
+        } : {
+          success: true,
+          data
+        };
+      }
+    case "count":
+      {
+        // Verify the parent audit belongs to this store before updating items
+        const {
+          data: audit
+        } = await sb.from("inventory_audits").select("id").eq("id", args.audit_id).eq("store_id", sid).single();
+        if (!audit) return {
+          success: false,
+          error: "Audit not found"
+        };
+        const {
+          data,
+          error
+        } = await sb.from("inventory_audit_items").update({
+          counted_quantity: args.counted
+        }).eq("audit_id", args.audit_id).eq("product_id", args.product_id).select().single();
+        return error ? {
+          success: false,
+          error: error.message
+        } : {
+          success: true,
+          data
+        };
+      }
+    case "complete":
+      {
+        const {
+          data,
+          error
+        } = await sb.from("inventory_audits").update({
+          status: "completed",
+          completed_at: new Date().toISOString()
+        }).eq("id", args.audit_id).eq("store_id", sid).select().single();
+        return error ? {
+          success: false,
+          error: error.message
+        } : {
+          success: true,
+          data
+        };
+      }
+    case "summary":
+      {
+        const {
+          data,
+          error
+        } = await sb.from("inventory_audits").select("*, items:inventory_audit_items(*)").eq("store_id", sid).order("created_at", {
+          ascending: false
+        }).limit(args.limit || 5);
+        return error ? {
+          success: false,
+          error: error.message
+        } : {
+          success: true,
+          data
+        };
+      }
+    default:
+      return {
+        success: false,
+        error: `Unknown inventory_audit action: ${args.action}`
+      };
+  }
+}
+//# sourceMappingURL=inventory-audit.js.map

package/dist/server/handlers/inventory-audit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"inventory-audit.js","names":["handleInventoryAudit","sb","args","storeId","sid","action","data","error","from","insert","store_id","location_id","audit_type","status","started_at","Date","toISOString","select","single","success","message","audit","eq","audit_id","update","counted_quantity","counted","product_id","completed_at","order","ascending","limit"],"sources":["../../../src/server/handlers/inventory-audit.ts"],"sourcesContent":["// server/handlers/inventory-audit.ts — Inventory audit handlers\n\nimport type { SupabaseClient } from \"@supabase/supabase-js\";\n\nexport async function handleInventoryAudit(sb: SupabaseClient, args: Record<string, unknown>, storeId?: string) {\n  const sid = storeId as string;\n  switch (args.action) {\n    case \"start\": {\n      const { data, error } = await sb.from(\"inventory_audits\")\n        .insert({ store_id: sid, location_id: args.location_id, audit_type: args.audit_type || \"full\", status: \"in_progress\", started_at: new Date().toISOString() })\n        .select().single();\n      return error ? { success: false, error: error.message } : { success: true, data };\n    }\n    case \"count\": {\n      // Verify the parent audit belongs to this store before updating items\n      const { data: audit } = await sb.from(\"inventory_audits\").select(\"id\").eq(\"id\", args.audit_id as string).eq(\"store_id\", sid).single();\n      if (!audit) return { success: false, error: \"Audit not found\" };\n      const { data, error } = await sb.from(\"inventory_audit_items\")\n        .update({ counted_quantity: args.counted })\n        .eq(\"audit_id\", args.audit_id).eq(\"product_id\", args.product_id).select().single();\n      return error ? { success: false, error: error.message } : { success: true, data };\n    }\n    case \"complete\": {\n      const { data, error } = await sb.from(\"inventory_audits\")\n        .update({ status: \"completed\", completed_at: new Date().toISOString() })\n        .eq(\"id\", args.audit_id as string).eq(\"store_id\", sid).select().single();\n      return error ? { success: false, error: error.message } : { success: true, data };\n    }\n    case \"summary\": {\n      const { data, error } = await sb.from(\"inventory_audits\")\n        .select(\"*, items:inventory_audit_items(*)\").eq(\"store_id\", sid)\n        .order(\"created_at\", { ascending: false }).limit(args.limit as number || 5);\n      return error ? { success: false, error: error.message } : { success: true, data };\n    }\n    default:\n      return { success: false, error: `Unknown inventory_audit action: ${args.action}` };\n  }\n}\n"],"mappings":"AAAA;;AAIA,OAAO,eAAeA,oBAAoBA,CAACC,EAAkB,EAAEC,IAA6B,EAAEC,OAAgB,EAAE;EAC9G,MAAMC,GAAG,GAAGD,OAAiB;EAC7B,QAAQD,IAAI,CAACG,MAAM;IACjB,KAAK,OAAO;MAAE;QACZ,MAAM;UAAEC,IAAI;UAAEC;QAAM,CAAC,GAAG,MAAMN,EAAE,CAACO,IAAI,CAAC,kBAAkB,CAAC,CACtDC,MAAM,CAAC;UAAEC,QAAQ,EAAEN,GAAG;UAAEO,WAAW,EAAET,IAAI,CAACS,WAAW;UAAEC,UAAU,EAAEV,IAAI,CAACU,UAAU,IAAI,MAAM;UAAEC,MAAM,EAAE,aAAa;UAAEC,UAAU,EAAE,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC;QAAE,CAAC,CAAC,CAC5JC,MAAM,CAAC,CAAC,CAACC,MAAM,CAAC,CAAC;QACpB,OAAOX,KAAK,GAAG;UAAEY,OAAO,EAAE,KAAK;UAAEZ,KAAK,EAAEA,KAAK,CAACa;QAAQ,CAAC,GAAG;UAAED,OAAO,EAAE,IAAI;UAAEb;QAAK,CAAC;MACnF;IACA,KAAK,OAAO;MAAE;QACZ;QACA,MAAM;UAAEA,IAAI,EAAEe;QAAM,CAAC,GAAG,MAAMpB,EAAE,CAACO,IAAI,CAAC,kBAAkB,CAAC,CAACS,MAAM,CAAC,IAAI,CAAC,CAACK,EAAE,CAAC,IAAI,EAAEpB,IAAI,CAACqB,QAAkB,CAAC,CAACD,EAAE,CAAC,UAAU,EAAElB,GAAG,CAAC,CAACc,MAAM,CAAC,CAAC;QACrI,IAAI,CAACG,KAAK,EAAE,OAAO;UAAEF,OAAO,EAAE,KAAK;UAAEZ,KAAK,EAAE;QAAkB,CAAC;QAC/D,MAAM;UAAED,IAAI;UAAEC;QAAM,CAAC,GAAG,MAAMN,EAAE,CAACO,IAAI,CAAC,uBAAuB,CAAC,CAC3DgB,MAAM,CAAC;UAAEC,gBAAgB,EAAEvB,IAAI,CAACwB;QAAQ,CAAC,CAAC,CAC1CJ,EAAE,CAAC,UAAU,EAAEpB,IAAI,CAACqB,QAAQ,CAAC,CAACD,EAAE,CAAC,YAAY,EAAEpB,IAAI,CAACyB,UAAU,CAAC,CAACV,MAAM,CAAC,CAAC,CAACC,MAAM,CAAC,CAAC;QACpF,OAAOX,KAAK,GAAG;UAAEY,OAAO,EAAE,KAAK;UAAEZ,KAAK,EAAEA,KAAK,CAACa;QAAQ,CAAC,GAAG;UAAED,OAAO,EAAE,IAAI;UAAEb;QAAK,CAAC;MACnF;IACA,KAAK,UAAU;MAAE;QACf,MAAM;UAAEA,IAAI;UAAEC;QAAM,CAAC,GAAG,MAAMN,EAAE,CAACO,IAAI,CAAC,kBAAkB,CAAC,CACtDgB,MAAM,CAAC;UAAEX,MAAM,EAAE,WAAW;UAAEe,YAAY,EAAE,IAAIb,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC;QAAE,CAAC,CAAC,CACvEM,EAAE,CAAC,IAAI,EAAEpB,IAAI,CAACqB,QAAkB,CAAC,CAACD,EAAE,CAAC,UAAU,EAAElB,GAAG,CAAC,CAACa,MAAM,CAAC,CAAC,CAACC,MAAM,CAAC,CAAC;QAC1E,OAAOX,KAAK,GAAG;UAAEY,OAAO,EAAE,KAAK;UAAEZ,KAAK,EAAEA,KAAK,CAACa;QAAQ,CAAC,GAAG;UAAED,OAAO,EAAE,IAAI;UAAEb;QAAK,CAAC;MACnF;IACA,KAAK,SAAS;MAAE;QACd,MAAM;UAAEA,IAAI;UAAEC;QAAM,CAAC,GAAG,MAAMN,EAAE,CAACO,IAAI,CAAC,kBAAkB,CAAC,CACtDS,MAAM,CAAC,mCAAmC,CAAC,CAACK,EAAE,CAAC,UAAU,EAAElB,GAAG,CAAC,CAC/DyB,KAAK,CAAC,YAAY,EAAE;UAAEC,SAAS,EAAE;QAAM,CAAC,CAAC,CAACC,KAAK,CAAC7B,IAAI,CAAC6B,KAAK,IAAc,CAAC,CAAC;QAC7E,OAAOxB,KAAK,GAAG;UAAEY,OAAO,EAAE,KAAK;UAAEZ,KAAK,EAAEA,KAAK,CAACa;QAAQ,CAAC,GAAG;UAAED,OAAO,EAAE,IAAI;UAAEb;QAAK,CAAC;MACnF;IACA;MACE,OAAO;QAAEa,OAAO,EAAE,KAAK;QAAEZ,KAAK,EAAE,mCAAmCL,IAAI,CAACG,MAAM;MAAG,CAAC;EACtF;AACF","ignoreList":[]}

package/dist/server/handlers/inventory-helpers.d.ts

@@ -0,0 +1,8 @@
+import type { SupabaseClient } from "@supabase/supabase-js";
+/**
+ * Resolve location name/slug to UUID — accepts UUID passthrough, name, or slug.
+ */
+export declare function resolveLocationId(sb: SupabaseClient, sid: string, input: string): Promise<{
+    id: string;
+    name: string;
+} | null>;

package/dist/server/handlers/inventory-helpers.js

@@ -0,0 +1,19 @@
+// server/handlers/inventory-helpers.ts — Shared inventory utilities
+
+/**
+ * Resolve location name/slug to UUID — accepts UUID passthrough, name, or slug.
+ */
+export async function resolveLocationId(sb, sid, input) {
+  if (/^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i.test(input)) {
+    const {
+      data
+    } = await sb.from("locations").select("id, name").eq("id", input).eq("store_id", sid).single();
+    return data;
+  }
+  const clean = input.replace(/-/g, " ");
+  const {
+    data
+  } = await sb.from("locations").select("id, name").eq("store_id", sid).or(`name.ilike.%${clean}%,slug.ilike.%${input}%`).limit(1);
+  return data?.[0] || null;
+}
+//# sourceMappingURL=inventory-helpers.js.map

package/dist/server/handlers/inventory-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"inventory-helpers.js","names":["resolveLocationId","sb","sid","input","test","data","from","select","eq","single","clean","replace","or","limit"],"sources":["../../../src/server/handlers/inventory-helpers.ts"],"sourcesContent":["// server/handlers/inventory-helpers.ts — Shared inventory utilities\n\nimport type { SupabaseClient } from \"@supabase/supabase-js\";\n\n/**\n * Resolve location name/slug to UUID — accepts UUID passthrough, name, or slug.\n */\nexport async function resolveLocationId(\n  sb: SupabaseClient,\n  sid: string,\n  input: string\n): Promise<{ id: string; name: string } | null> {\n  if (/^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i.test(input)) {\n    const { data } = await sb.from(\"locations\").select(\"id, name\").eq(\"id\", input).eq(\"store_id\", sid).single();\n    return data;\n  }\n  const clean = input.replace(/-/g, \" \");\n  const { data } = await sb.from(\"locations\").select(\"id, name\").eq(\"store_id\", sid)\n    .or(`name.ilike.%${clean}%,slug.ilike.%${input}%`).limit(1);\n  return data?.[0] || null;\n}\n"],"mappings":"AAAA;;AAIA;AACA;AACA;AACA,OAAO,eAAeA,iBAAiBA,CACrCC,EAAkB,EAClBC,GAAW,EACXC,KAAa,EACiC;EAC9C,IAAI,iEAAiE,CAACC,IAAI,CAACD,KAAK,CAAC,EAAE;IACjF,MAAM;MAAEE;IAAK,CAAC,GAAG,MAAMJ,EAAE,CAACK,IAAI,CAAC,WAAW,CAAC,CAACC,MAAM,CAAC,UAAU,CAAC,CAACC,EAAE,CAAC,IAAI,EAAEL,KAAK,CAAC,CAACK,EAAE,CAAC,UAAU,EAAEN,GAAG,CAAC,CAACO,MAAM,CAAC,CAAC;IAC3G,OAAOJ,IAAI;EACb;EACA,MAAMK,KAAK,GAAGP,KAAK,CAACQ,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC;EACtC,MAAM;IAAEN;EAAK,CAAC,GAAG,MAAMJ,EAAE,CAACK,IAAI,CAAC,WAAW,CAAC,CAACC,MAAM,CAAC,UAAU,CAAC,CAACC,EAAE,CAAC,UAAU,EAAEN,GAAG,CAAC,CAC/EU,EAAE,CAAC,eAAeF,KAAK,iBAAiBP,KAAK,GAAG,CAAC,CAACU,KAAK,CAAC,CAAC,CAAC;EAC7D,OAAOR,IAAI,GAAG,CAAC,CAAC,IAAI,IAAI;AAC1B","ignoreList":[]}

package/dist/server/handlers/inventory-query.d.ts

@@ -0,0 +1,67 @@
+import type { SupabaseClient } from "@supabase/supabase-js";
+export declare function handleInventoryQuery(sb: SupabaseClient, args: Record<string, unknown>, storeId?: string): Promise<{
+    success: boolean;
+    data: string;
+    error?: undefined;
+} | {
+    success: boolean;
+    count: number;
+    total_products: number;
+    data: {
+        product_id: any;
+        name: any;
+        sku: any;
+        category: any;
+        total_stock: number;
+    }[];
+    error?: undefined;
+} | {
+    success: boolean;
+    error: string;
+    data?: undefined;
+    count?: undefined;
+} | {
+    success: boolean;
+    data: {
+        total_items: number;
+        by_location: {
+            location_id: string;
+            location_name: string;
+            items: number;
+            total_qty: number;
+        }[];
+        days?: undefined;
+        filters?: undefined;
+        products?: undefined;
+    };
+    error?: undefined;
+    count?: undefined;
+} | {
+    success: boolean;
+    data: {
+        days: number;
+        filters: {
+            categoryId: string;
+            locationId: string;
+            productId: string;
+        };
+        products: any;
+        total_items?: undefined;
+        by_location?: undefined;
+    };
+    error?: undefined;
+    count?: undefined;
+} | {
+    success: boolean;
+    count: number;
+    data: {
+        product_id: any;
+        product_name: any;
+        product_sku: any;
+        category: any;
+        location_id: any;
+        location_name: any;
+        quantity: any;
+    }[];
+    error?: undefined;
+}>;