forgemesh-imagegen 1.0.0 → 2.0.0

package/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2026 CoinOpAI
+Copyright (c) 2026 ForgeMesh Labs
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

package/README.md

@@ -10,12 +10,13 @@ Part of the [ForgeMesh](https://github.com/forgemeshlabs/forgemesh) ecosystem
 
 | Tool | What it does | Price |
 |---|---|---|
-| `generate_image` | Text-to-image generation | $0.10 USDC |
-| `generate_clean` | Generate + background removal | $0.15 USDC |
-| `generate_hd` | Generate + 4x upscale (HD) | $0.20 USDC |
-| `generate_pro` | Generate + bg removal + 4x upscale | $0.30 USDC |
+| `generate_image` | Standard text-to-image generation | $0.25 USDC |
+| `generate_clean` | Generate + background removal | $0.35 USDC |
+| `generate_hd` | Premium generate + 4x upscale (HD) | $0.50 USDC |
+| `generate_pro` | Top-tier generate + bg removal + 4x upscale | $0.75 USDC |
 
 All tools accept `prompt` (required) and `aspect` (optional: `1:1`, `16:9`, `9:16`, `4:3`).
+`generate_image` also accepts `affiliate_id` for Pyrimid attribution, or uses `PYRIMID_AFFILIATE_ID` from the environment.
 
 ---
 
@@ -32,7 +33,8 @@ Add to `claude_desktop_config.json`:
       "command": "npx",
       "args": ["-y", "forgemesh-imagegen"],
       "env": {
-        "WALLET_PRIVATE_KEY": "0x..."
+        "WALLET_PRIVATE_KEY": "0x...",
+        "PYRIMID_AFFILIATE_ID": "af_your_id"
       }
     }
   }
@@ -55,7 +57,7 @@ export WALLET_PRIVATE_KEY=0x...
 ## Requirements
 
 - A Base mainnet wallet private key with USDC
-- $1 USDC ≈ 10 base images, 6 clean, 5 HD, or 3 pro
+- $1 USDC ≈ 4 base images, 2 clean, 2 HD, or 1 pro
 - No other API keys needed
 
 Get USDC on Base: [Coinbase](https://coinbase.com) → Bridge to Base, or buy directly on Base.
@@ -68,6 +70,9 @@ Get USDC on Base: [Coinbase](https://coinbase.com) → Bridge to Base, or buy di
 generate_image(prompt="a red panda in a spacesuit", aspect="1:1")
 → { image_url: "https://...", prompt: "...", aspect: "1:1", tier: "image" }
 
+generate_image(prompt="a red panda in a spacesuit", affiliate_id="af_your_id")
+→ routes through Pyrimid product 3 with no extra cost to the caller
+
 generate_clean(prompt="a product photo of a ceramic mug")
 → { image_url: "https://...", tier: "clean" }   // transparent PNG
 
@@ -86,14 +91,14 @@ Each tool call makes an HTTP request to the ForgeMesh imagegen service gated by
 
 - Network: Base mainnet (eip155:8453)
 - Token: USDC
-- Facilitator: Coinbase CDP
+- Protocol: [x402](https://x402.org)
 
 ---
 
 ## Links
 
-- [ForgeMesh](https://forgemesh.io) — ecosystem overview
-- [GitHub](https://github.com/forgemeshlabs/forgemesh-imagegen)
+- [ForgeMesh](https://github.com/forgemeshlabs/forgemesh) — ecosystem overview
+- [npm](https://www.npmjs.com/package/forgemesh-imagegen)
 - [x402 Protocol](https://x402.org)
 
 ---

package/index.js

@@ -8,26 +8,41 @@ const { x402Client, x402HTTPClient } = require("@x402/core/client");
 const { ExactEvmScheme } = require("@x402/evm/exact/client");
 const { toClientEvmSigner } = require("@x402/evm");
 const { privateKeyToAccount } = require("viem/accounts");
+const { createPublicClient, createWalletClient, http, parseAbi } = require("viem");
+const { base } = require("viem/chains");
 
 const BASE_URL = "https://imagegen.coinopai.com";
 const VALID_ASPECTS = ["1:1", "16:9", "9:16", "4:3"];
+const PYRIMID_ROUTER = "0xc949AEa380D7b7984806143ddbfE519B03ABd68B";
+const PYRIMID_VENDOR_ID = "0x034604e25078e293d7b181fa23b3f2f6";
+const USDC_ADDRESS = "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913";
+const ROUTER_ABI = parseAbi([
+  "function routePayment(bytes16 vendorId, uint256 productId, bytes16 affiliateId, address buyer, uint256 maxPrice) external",
+]);
+const USDC_ABI = parseAbi([
+  "function approve(address spender, uint256 amount) external returns (bool)",
+]);
+const PYRIMID_PRODUCTS = {
+  "/generate": { productId: 3n, priceUsdc: 250000n },
+};
 
 const TOOLS = [
   {
     name: "generate_image",
-    description: "Generate an AI image from a text prompt. Returns a PNG image URL. Costs $0.10 USDC on Base mainnet — paid automatically.",
+    description: "Generate a standard AI image from a text prompt. Returns a PNG image URL. Costs $0.25 USDC on Base mainnet — paid automatically.",
     inputSchema: {
       type: "object",
       properties: {
         prompt: { type: "string", description: "Text description of the image to generate" },
         aspect: { type: "string", enum: VALID_ASPECTS, description: "Aspect ratio — 1:1 (default), 16:9, 9:16, 4:3" },
+        affiliate_id: { type: "string", description: "Optional Pyrimid affiliate ID. Affiliate earns a commission from within the listed price — no extra cost to you." },
       },
       required: ["prompt"],
     },
   },
   {
     name: "generate_clean",
-    description: "Generate an AI image with the background removed. Returns a transparent PNG URL. Costs $0.15 USDC on Base mainnet.",
+    description: "Generate an AI image with background removed. Returns a transparent PNG URL. Costs $0.35 USDC on Base mainnet.",
     inputSchema: {
       type: "object",
       properties: {
@@ -39,7 +54,7 @@ const TOOLS = [
   },
   {
     name: "generate_hd",
-    description: "Generate an AI image upscaled to 4x resolution. Returns a high-resolution image URL. Costs $0.20 USDC on Base mainnet.",
+    description: "Generate a premium AI image upscaled 4x HD. Returns a high-resolution image URL. Costs $0.50 USDC on Base mainnet.",
     inputSchema: {
       type: "object",
       properties: {
@@ -51,7 +66,7 @@ const TOOLS = [
   },
   {
     name: "generate_pro",
-    description: "Generate an AI image with background removed AND upscaled 4x HD. The full pipeline. Costs $0.30 USDC on Base mainnet.",
+    description: "Generate a top-tier AI image with background removal and 4x HD upscale. Costs $0.75 USDC on Base mainnet.",
     inputSchema: {
       type: "object",
       properties: {
@@ -64,10 +79,10 @@ const TOOLS = [
 ];
 
 const TOOL_ENDPOINTS = {
-  generate_image: { path: "/generate", price: "$0.10" },
-  generate_clean: { path: "/generate/clean", price: "$0.15" },
-  generate_hd:    { path: "/generate/hd",    price: "$0.20" },
-  generate_pro:   { path: "/generate/pro",   price: "$0.30" },
+  generate_image: { path: "/generate", price: "$0.25" },
+  generate_clean: { path: "/generate/clean", price: "$0.35" },
+  generate_hd:    { path: "/generate/hd",    price: "$0.50" },
+  generate_pro:   { path: "/generate/pro",   price: "$0.75" },
 };
 
 function buildHttpClient() {
@@ -75,21 +90,70 @@ function buildHttpClient() {
   if (!key) {
     throw new Error(
       "WALLET_PRIVATE_KEY required — set a Base wallet private key funded with USDC.\n" +
-      "  generate_image = $0.10 | generate_clean = $0.15 | generate_hd = $0.20 | generate_pro = $0.30"
+      "  generate_image = $0.25 | generate_clean = $0.35 | generate_hd = $0.50 | generate_pro = $0.75"
     );
   }
   const pk = key.startsWith("0x") ? key : "0x" + key;
   const account = privateKeyToAccount(pk);
   const signer = toClientEvmSigner(account);
   const coreClient = new x402Client().register("eip155:*", new ExactEvmScheme(signer));
-  return new x402HTTPClient(coreClient);
+  return { httpClient: new x402HTTPClient(coreClient), account };
 }
 
-async function callPaid(httpClient, path, params) {
+const sleep = (ms) => new Promise(r => setTimeout(r, ms));
+
+async function callPyrimid(account, url, product, affiliateId) {
+  const transport = http();
+  const publicClient = createPublicClient({ chain: base, transport });
+  const walletClient = createWalletClient({ account, chain: base, transport });
+
+  const approveHash = await walletClient.writeContract({
+    address: USDC_ADDRESS,
+    abi: USDC_ABI,
+    functionName: "approve",
+    args: [PYRIMID_ROUTER, product.priceUsdc],
+  });
+  await publicClient.waitForTransactionReceipt({ hash: approveHash });
+  await sleep(3000);
+
+  const routeHash = await walletClient.writeContract({
+    address: PYRIMID_ROUTER,
+    abi: ROUTER_ABI,
+    functionName: "routePayment",
+    args: [PYRIMID_VENDOR_ID, product.productId, "0x00000000000000000000000000000000", account.address, product.priceUsdc],
+  });
+  await publicClient.waitForTransactionReceipt({ hash: routeHash });
+  await sleep(3000);
+
+  const paidRes = await fetch(url.toString(), {
+    headers: { "X-Affiliate-ID": affiliateId, "X-Payment": routeHash },
+  });
+  if (!paidRes.ok) {
+    const errBody = await paidRes.text().catch(() => paidRes.statusText);
+    throw new Error(`Pyrimid payment failed — HTTP ${paidRes.status}: ${errBody.slice(0, 200)}`);
+  }
+  return paidRes.json();
+}
+
+async function callPaid(ctx, path, params, affiliateId) {
+  const { httpClient, account } = ctx;
   const url = new URL(BASE_URL + path);
   if (params) Object.entries(params).forEach(([k, v]) => url.searchParams.set(k, v));
 
-  const res = await fetch(url.toString());
+  const product = PYRIMID_PRODUCTS[path];
+  if (!product) affiliateId = null;
+  const extraHeaders = affiliateId ? { "X-Affiliate-ID": affiliateId } : {};
+
+  if (affiliateId && product) {
+    try {
+      return await callPyrimid(account, url, product, affiliateId);
+    } catch (_) {
+      affiliateId = null;
+    }
+  }
+
+  const directHeaders = affiliateId ? extraHeaders : {};
+  const res = await fetch(url.toString(), { headers: directHeaders });
 
   if (res.status === 402) {
     let body;
@@ -100,7 +164,7 @@ async function callPaid(httpClient, path, params) {
     );
     const paymentPayload = await httpClient.createPaymentPayload(paymentRequired);
     const paidRes = await fetch(url.toString(), {
-      headers: httpClient.encodePaymentSignatureHeader(paymentPayload),
+      headers: { ...httpClient.encodePaymentSignatureHeader(paymentPayload), ...directHeaders },
     });
     if (!paidRes.ok) {
       const errBody = await paidRes.text().catch(() => paidRes.statusText);
@@ -117,16 +181,16 @@ async function callPaid(httpClient, path, params) {
 }
 
 async function main() {
-  let httpClient;
+  let ctx;
   try {
-    httpClient = buildHttpClient();
+    ctx = buildHttpClient();
   } catch (e) {
     process.stderr.write("[forgemesh-imagegen] " + e.message + "\n");
     process.exit(1);
   }
 
   const server = new Server(
-    { name: "forgemesh-imagegen", version: "1.0.0" },
+    { name: "forgemesh-imagegen", version: "1.0.2" },
     { capabilities: { tools: {} } }
   );
 
@@ -147,10 +211,11 @@ async function main() {
         throw new Error(`Invalid aspect ratio '${aspect}'. Valid: ${VALID_ASPECTS.join(", ")}`);
       }
 
-      const data = await callPaid(httpClient, endpoint.path, {
+      const affiliateId = args.affiliate_id || process.env.PYRIMID_AFFILIATE_ID || null;
+      const data = await callPaid(ctx, endpoint.path, {
         prompt: args.prompt.trim(),
         aspect,
-      });
+      }, affiliateId);
 
       return {
         content: [{

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "forgemesh-imagegen",
-  "version": "1.0.0",
+  "version": "2.0.0",
   "mcpName": "io.github.forgemeshlabs/imagegen-mcp",
   "description": "MCP server for AI image generation — generate, clean (bg removal), HD (4x upscale), Pro. Pay per image in USDC via x402 on Base. No API key needed.",
   "main": "index.js",
@@ -15,7 +15,7 @@
     "start": "node index.js"
   },
   "engines": {
-    "node": ">=18"
+    "node": ">=20"
   },
   "keywords": [
     "mcp",
@@ -41,14 +41,14 @@
   "license": "MIT",
   "repository": {
     "type": "git",
-    "url": "https://github.com/forgemeshlabs/forgemesh-imagegen.git"
+    "url": "https://github.com/forgemeshlabs/imagegen-mcp.git"
   },
-  "homepage": "https://github.com/forgemeshlabs/forgemesh-imagegen",
+  "homepage": "https://github.com/forgemeshlabs/imagegen-mcp",
   "bugs": {
-    "url": "https://github.com/forgemeshlabs/forgemesh-imagegen/issues"
+    "url": "https://github.com/forgemeshlabs/imagegen-mcp/issues"
   },
   "dependencies": {
-    "@modelcontextprotocol/sdk": "^1.10.1",
+    "@modelcontextprotocol/sdk": "^1.30.0",
     "@x402/core": "^2.11.0",
     "@x402/evm": "^2.11.0",
     "viem": "^2.0.0"