@blockrun/clawrouter 0.11.2 → 0.11.4

package/dist/index.js

@@ -4934,6 +4934,27 @@ async function proxyPartnerRequest(req, res, apiBase, payFetch) {
   }).catch(() => {
   });
 }
+async function uploadDataUriToHost(dataUri) {
+  const match = dataUri.match(/^data:(image\/\w+);base64,(.+)$/);
+  if (!match) throw new Error("Invalid data URI format");
+  const [, mimeType, b64Data] = match;
+  const ext = mimeType === "image/jpeg" ? "jpg" : mimeType.split("/")[1] ?? "png";
+  const buffer = Buffer.from(b64Data, "base64");
+  const blob = new Blob([buffer], { type: mimeType });
+  const form = new FormData();
+  form.append("reqtype", "fileupload");
+  form.append("fileToUpload", blob, `image.${ext}`);
+  const resp = await fetch("https://catbox.moe/user/api.php", {
+    method: "POST",
+    body: form
+  });
+  if (!resp.ok) throw new Error(`catbox.moe upload failed: HTTP ${resp.status}`);
+  const result = await resp.text();
+  if (result.startsWith("https://")) {
+    return result.trim();
+  }
+  throw new Error(`catbox.moe upload failed: ${result}`);
+}
 async function startProxy(options) {
   const apiBase = options.apiBase ?? BLOCKRUN_API;
   const listenPort = options.port ?? getProxyPort();
@@ -5551,7 +5572,19 @@ async function proxyRequest(req, res, apiBase, payFetch, options, routerOpts, de
             } else {
               const lines = [];
               for (const img of images) {
-                if (img.url) lines.push(img.url);
+                if (img.url) {
+                  if (img.url.startsWith("data:")) {
+                    try {
+                      const hostedUrl = await uploadDataUriToHost(img.url);
+                      lines.push(hostedUrl);
+                    } catch (uploadErr) {
+                      console.error(`[ClawRouter] /imagegen: failed to upload data URI: ${uploadErr instanceof Error ? uploadErr.message : String(uploadErr)}`);
+                      lines.push("Image generated but upload failed. Try again or use --model dall-e-3.");
+                    }
+                  } else {
+                    lines.push(img.url);
+                  }
+                }
                 if (img.revised_prompt) lines.push(`Revised prompt: ${img.revised_prompt}`);
               }
               lines.push("", `Model: ${imageModel} | Size: ${imageSize}`);