@blockrun/clawrouter 0.12.25 → 0.12.26

package/dist/cli.js

@@ -3,6 +3,9 @@
 // src/proxy.ts
 import { createServer } from "http";
 import { finished } from "stream";
+import { homedir as homedir4 } from "os";
+import { join as join5 } from "path";
+import { mkdir as mkdir3, writeFile as writeFile2, readFile, stat as fsStat } from "fs/promises";
 import { createPublicClient as createPublicClient2, http as http2 } from "viem";
 import { base as base2 } from "viem/chains";
 import { privateKeyToAccount as privateKeyToAccount3 } from "viem/accounts";
@@ -5032,6 +5035,7 @@ ${lines.join("\n")}`;
 // src/proxy.ts
 var BLOCKRUN_API = "https://blockrun.ai/api";
 var BLOCKRUN_SOLANA_API = "https://sol.blockrun.ai/api";
+var IMAGE_DIR = join5(homedir4(), ".openclaw", "blockrun", "images");
 var AUTO_MODEL = "blockrun/auto";
 var ROUTING_PROFILES = /* @__PURE__ */ new Set([
   "blockrun/free",
@@ -5813,6 +5817,81 @@ async function startProxy(options) {
       res.end(JSON.stringify({ object: "list", data: models }));
       return;
     }
+    if (req.url?.startsWith("/images/") && req.method === "GET") {
+      const filename = req.url.slice("/images/".length).split("?")[0].replace(/[^a-zA-Z0-9._-]/g, "");
+      if (!filename) {
+        res.writeHead(400);
+        res.end("Bad request");
+        return;
+      }
+      const filePath = join5(IMAGE_DIR, filename);
+      try {
+        const s = await fsStat(filePath);
+        if (!s.isFile()) throw new Error("not a file");
+        const ext = filename.split(".").pop()?.toLowerCase() ?? "png";
+        const mime = { png: "image/png", jpg: "image/jpeg", jpeg: "image/jpeg", webp: "image/webp", gif: "image/gif" };
+        const data = await readFile(filePath);
+        res.writeHead(200, { "Content-Type": mime[ext] ?? "application/octet-stream", "Content-Length": data.length });
+        res.end(data);
+      } catch {
+        res.writeHead(404, { "Content-Type": "application/json" });
+        res.end(JSON.stringify({ error: "Image not found" }));
+      }
+      return;
+    }
+    if (req.url === "/v1/images/generations" && req.method === "POST") {
+      const chunks = [];
+      for await (const chunk of req) {
+        chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
+      }
+      const reqBody = Buffer.concat(chunks);
+      try {
+        const upstream = await payFetch(`${apiBase}/v1/images/generations`, {
+          method: "POST",
+          headers: { "content-type": "application/json", "user-agent": USER_AGENT },
+          body: reqBody
+        });
+        const text = await upstream.text();
+        if (!upstream.ok) {
+          res.writeHead(upstream.status, { "Content-Type": "application/json" });
+          res.end(text);
+          return;
+        }
+        let result;
+        try {
+          result = JSON.parse(text);
+        } catch {
+          res.writeHead(200, { "Content-Type": "application/json" });
+          res.end(text);
+          return;
+        }
+        if (result.data?.length) {
+          await mkdir3(IMAGE_DIR, { recursive: true });
+          const port2 = server.address()?.port ?? 8402;
+          for (const img of result.data) {
+            const m = img.url?.match(/^data:(image\/\w+);base64,(.+)$/);
+            if (m) {
+              const [, mimeType, b64] = m;
+              const ext = mimeType === "image/jpeg" ? "jpg" : mimeType.split("/")[1] ?? "png";
+              const filename = `${Date.now()}-${Math.random().toString(36).slice(2, 10)}.${ext}`;
+              await writeFile2(join5(IMAGE_DIR, filename), Buffer.from(b64, "base64"));
+              img.url = `http://localhost:${port2}/images/${filename}`;
+              console.log(`[ClawRouter] Image saved \u2192 ${img.url}`);
+            }
+          }
+        }
+        res.writeHead(200, { "Content-Type": "application/json" });
+        res.end(JSON.stringify(result));
+      } catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
+        console.error(`[ClawRouter] Image generation error: ${msg}`);
+        if (!res.headersSent) {
+          res.writeHead(502, { "Content-Type": "application/json" });
+          res.end(JSON.stringify({ error: "Image generation failed", details: msg }));
+        }
+      }
+      return;
+    }
     if (req.url?.match(/^\/v1\/(?:x|partner)\//)) {
       try {
         await proxyPartnerRequest(req, res, apiBase, payFetch);