pixelate-mcp 0.1.0 → 0.3.0

package/index.js

@@ -1,6 +1,8 @@
 #!/usr/bin/env node
 
 import { readFile, writeFile } from "node:fs/promises";
+import os from "node:os";
+import crypto from "node:crypto";
 import sharp from "sharp";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
@@ -13,18 +15,36 @@ if (!API_URL || !API_KEY) {
   process.exit(1);
 }
 
+// Stable per-machine id. A key binds to the first machine that uses it, so a
+// shared key won't work on someone else's device. Derived from host/user/home
+// (deterministic, no extra files); override with PIXELATE_MACHINE_ID if needed.
+let _uname = "";
+try { _uname = os.userInfo().username; } catch { /* ignore */ }
+const MACHINE_ID =
+  process.env.PIXELATE_MACHINE_ID ||
+  crypto.createHash("sha256").update([os.hostname(), _uname, os.homedir(), os.platform()].join("|")).digest("hex").slice(0, 24);
+
 const stripDataUrl = (s) => {
   const i = s.indexOf(",");
   return s.startsWith("data:") && i !== -1 ? s.slice(i + 1) : s;
 };
 const errText = (text) => ({ content: [{ type: "text", text }], isError: true });
+const usd = (cents) => `$${(cents / 100).toFixed(2)}`;
+// One-line billing summary from a /job done response (credits_cents may be null
+// for unlimited subscription keys).
+function billingLine(res) {
+  const price = res.price_cents != null ? `${usd(res.price_cents)}/image` : null;
+  if (res.credits_cents == null) return price ? ` Plan: unlimited (${price}).` : "";
+  const imgs = res.price_cents ? Math.floor(res.credits_cents / res.price_cents) : null;
+  return ` Charged ${price}; ${usd(res.credits_cents)} credit left${imgs != null ? ` (~${imgs} images)` : ""}.`;
+}
 
 async function jobReq(query, body) {
+  const headers = { "x-api-key": API_KEY, "x-machine-id": MACHINE_ID };
+  if (body) headers["content-type"] = "application/octet-stream";
   const res = await fetch(`${API_URL}/job?${query}`, {
     method: "POST",
-    headers: body
-      ? { "content-type": "application/octet-stream", "x-api-key": API_KEY }
-      : { "x-api-key": API_KEY },
+    headers,
     body: body ?? undefined,
   });
   const j = await res.json().catch(() => ({}));
@@ -33,7 +53,27 @@ async function jobReq(query, body) {
   return j;
 }
 
-const server = new McpServer({ name: "pixelate-mcp", version: "0.1.0" });
+const server = new McpServer({ name: "pixelate-mcp", version: "0.3.0" });
+
+server.tool(
+  "pixelate_account",
+  "Show your Pixelate Cloud credit balance (how much you have left) and the cost per image.",
+  {},
+  async () => {
+    try {
+      const res = await fetch(`${API_URL}/account`, { headers: { "x-api-key": API_KEY, "x-machine-id": MACHINE_ID } });
+      const j = await res.json().catch(() => ({}));
+      if (!res.ok || !j.success) return errText(`account error (${res.status}): ${j.error || "unknown"}`);
+      const text =
+        j.plan === "subscription"
+          ? `Plan: unlimited subscription. Cost per image: ${usd(j.price_per_image_cents)} (not charged on your plan).`
+          : `Balance: $${j.credits_usd} (~${j.images_remaining} images left). Cost per image: $${j.price_per_image_usd}.`;
+      return { content: [{ type: "text", text }] };
+    } catch (e) {
+      return errText(`pixelate-mcp error: ${e?.message || e}`);
+    }
+  },
+);
 
 server.tool(
   "pixelate_image",
@@ -80,6 +120,7 @@ server.tool(
         await writeFile(a.output_path, Buffer.from(res.image_base64, "base64"));
         summary += ` Saved to ${a.output_path}.`;
       }
+      summary += billingLine(res);
       return {
         content: [
           { type: "text", text: summary },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pixelate-mcp",
-  "version": "0.1.0",
+  "version": "0.3.0",
   "description": "MCP server: turn any image into grid-perfect pixel art via the Pixelate Cloud API. Works in Codex, Claude Desktop, Cursor.",
   "type": "module",
   "bin": {