@takisvc/code-canvas 0.1.0

package/dist/code-canvas.js

@@ -0,0 +1,282 @@
+#!/usr/bin/env node
+
+// client.ts
+import { privateKeyToAccount, generatePrivateKey } from "viem/accounts";
+import { formatSIWEMessage } from "@worldcoin/agentkit";
+import { existsSync, readFileSync, mkdirSync, writeFileSync } from "fs";
+import { execSync } from "child_process";
+import { homedir } from "os";
+import { join } from "path";
+import ora from "ora";
+import chalk from "chalk";
+function ensureWallet() {
+  const dir = join(homedir(), ".code-canvas");
+  const keyPath = join(dir, "key");
+  if (existsSync(keyPath)) {
+    const key2 = readFileSync(keyPath, "utf8").trim();
+    return privateKeyToAccount(key2);
+  }
+  mkdirSync(dir, { recursive: true });
+  const key = generatePrivateKey();
+  writeFileSync(keyPath, key, { mode: 384 });
+  const account2 = privateKeyToAccount(key);
+  console.log(chalk.green("Generated new agent wallet"));
+  console.log(`  Address:  ${chalk.bold(account2.address)}`);
+  console.log(`  Key file: ${keyPath}`);
+  console.log();
+  return account2;
+}
+var SPINNER_MESSAGES = [
+  "Conjuring pixels...",
+  "Asking the AI gods...",
+  "Brewing creativity...",
+  "Painting with algorithms...",
+  "Summoning the muse...",
+  "Mixing digital paint...",
+  "Consulting the pixel oracle...",
+  "Channeling pure imagination..."
+];
+function randomSpinnerMessage() {
+  return SPINNER_MESSAGES[Math.floor(Math.random() * SPINNER_MESSAGES.length)];
+}
+function supportsInlineImages() {
+  if (process.env.TERM_PROGRAM === "iTerm.app" || process.env.TERM_PROGRAM === "WezTerm") {
+    return "iterm";
+  }
+  if (process.env.KITTY_WINDOW_ID) {
+    return "kitty";
+  }
+  return false;
+}
+function displayInlineImage(base64) {
+  const protocol = supportsInlineImages();
+  if (protocol === "iterm") {
+    process.stdout.write(
+      `\x1B]1337;File=inline=1;preserveAspectRatio=1:${base64}\x07
+`
+    );
+    return true;
+  }
+  if (protocol === "kitty") {
+    const chunkSize = 4096;
+    for (let i = 0; i < base64.length; i += chunkSize) {
+      const chunk = base64.slice(i, i + chunkSize);
+      const more = i + chunkSize < base64.length ? 1 : 0;
+      if (i === 0) {
+        process.stdout.write(
+          `\x1B_Gf=100,a=T,m=${more};${chunk}\x1B\\`
+        );
+      } else {
+        process.stdout.write(`\x1B_Gm=${more};${chunk}\x1B\\`);
+      }
+    }
+    process.stdout.write("\n");
+    return true;
+  }
+  return false;
+}
+function openFile(filepath) {
+  try {
+    if (process.platform === "darwin") {
+      execSync(`open "${filepath}"`, { stdio: "ignore" });
+    } else if (process.platform === "linux") {
+      execSync(`xdg-open "${filepath}"`, { stdio: "ignore" });
+    } else if (process.platform === "win32") {
+      execSync(`start "" "${filepath}"`, { stdio: "ignore" });
+    }
+  } catch {
+  }
+}
+function saveImage(base64, prompt2) {
+  const filename = `generated-${Date.now()}.png`;
+  const buffer = Buffer.from(base64, "base64");
+  writeFileSync(filename, buffer);
+  const shown = displayInlineImage(base64);
+  const tweetText = encodeURIComponent(
+    `My AI agent just roasted my coding session \u{1F525}
+
+Prompt: "${prompt2}"
+
+Powered by @worldcoin AgentKit`
+  );
+  console.log(`  Saved:       ${chalk.bold(filename)}`);
+  console.log(
+    `  Share on X:  https://twitter.com/intent/tweet?text=${tweetText}`
+  );
+  if (!shown) {
+    openFile(filename);
+  }
+}
+function showUsage(usage) {
+  if (!usage) return;
+  const remaining = Math.max(0, usage.total - usage.used);
+  if (remaining > 0) {
+    console.log(
+      chalk.dim(`  Free generations: ${remaining} of ${usage.total} remaining`)
+    );
+  } else {
+    console.log(
+      chalk.yellow(
+        `  Free trial exhausted. Register with World ID for unlimited access or pay $1.20 USDC.`
+      )
+    );
+  }
+}
+function extractAgentkitChallenge(body) {
+  return body?.extensions?.agentkit ?? body?.accepts?.[0]?.extensions?.agentkit ?? body?.paymentRequirements?.[0]?.extensions?.agentkit ?? body?.challenge;
+}
+function runRegistration(address) {
+  console.log();
+  console.log(
+    chalk.yellow("Launching World ID registration \u2014 scan the QR code with your World App:")
+  );
+  console.log();
+  try {
+    execSync(
+      `npx @worldcoin/agentkit-cli register ${address} --network world`,
+      { stdio: "inherit" }
+    );
+    console.log();
+    console.log(chalk.green("Registration successful!"));
+    return true;
+  } catch {
+    console.log();
+    console.log(chalk.red("Registration failed or was cancelled."));
+    return false;
+  }
+}
+async function generateImage(endpoint2, prompt2, account2) {
+  const spinner = ora(randomSpinnerMessage()).start();
+  const interval = setInterval(() => {
+    spinner.text = randomSpinnerMessage();
+  }, 3e3);
+  try {
+    const firstResponse = await fetch(endpoint2, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ prompt: prompt2 })
+    });
+    if (firstResponse.status === 200) {
+      const body = await firstResponse.json();
+      spinner.succeed(chalk.green("Image generated!"));
+      saveImage(body.image, prompt2);
+      showUsage(body.usage);
+      return true;
+    }
+    if (firstResponse.status !== 402) {
+      spinner.fail("Request failed");
+      const body = await firstResponse.json().catch(() => ({}));
+      console.error(chalk.red(body.error ?? `HTTP ${firstResponse.status}`));
+      return false;
+    }
+    const paymentRequiredHeader = firstResponse.headers.get("payment-required");
+    let paymentData = null;
+    if (paymentRequiredHeader) {
+      try {
+        paymentData = JSON.parse(
+          Buffer.from(paymentRequiredHeader, "base64").toString()
+        );
+      } catch {
+      }
+    }
+    const challenge = extractAgentkitChallenge(paymentData);
+    if (!challenge?.info || !challenge?.supportedChains?.length) {
+      spinner.fail("Server did not return an agentkit challenge");
+      if (paymentData) {
+        console.error(
+          "402 payment data:",
+          JSON.stringify(paymentData, null, 2)
+        );
+      } else {
+        console.error("No payment-required header found");
+      }
+      return false;
+    }
+    spinner.text = "Signing with wallet...";
+    const { info, supportedChains } = challenge;
+    const chain = supportedChains[0];
+    const message = formatSIWEMessage(
+      { ...info, chainId: chain.chainId },
+      account2.address
+    );
+    const signature = await account2.signMessage({ message });
+    const payload = {
+      ...info,
+      address: account2.address,
+      chainId: chain.chainId,
+      type: chain.type,
+      signature
+    };
+    const encoded = Buffer.from(JSON.stringify(payload)).toString("base64");
+    spinner.text = "Verifying World ID...";
+    const secondResponse = await fetch(endpoint2, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        agentkit: encoded
+      },
+      body: JSON.stringify({ prompt: prompt2 })
+    });
+    if (secondResponse.status === 200) {
+      const body = await secondResponse.json();
+      spinner.succeed(
+        chalk.green("Image generated (free \u2014 World ID verified!)")
+      );
+      saveImage(body.image, prompt2);
+      showUsage(body.usage);
+      return true;
+    }
+    const errorBody = await secondResponse.json().catch(() => ({}));
+    if (secondResponse.status !== 402) {
+      spinner.fail(`Server error (${secondResponse.status})`);
+      console.error(chalk.red(errorBody?.error ?? "Unknown error"));
+      return false;
+    }
+    spinner.stop();
+    console.log();
+    console.log(chalk.yellow("Your agent is not registered with World ID."));
+    const registered = runRegistration(account2.address);
+    if (registered) {
+      console.log();
+      console.log(chalk.cyan("Retrying image generation..."));
+      return false;
+    }
+    console.log();
+    console.log(
+      `  ${chalk.bold("Option 1:")} Try registration again manually:`
+    );
+    console.log(
+      `    npx @worldcoin/agentkit-cli register ${account2.address} --network world`
+    );
+    console.log();
+    console.log(
+      `  ${chalk.bold("Option 2:")} Pay ${chalk.cyan("$1.20 USDC")} per image:`
+    );
+    console.log(`    Fund your wallet with USDC on World Chain and re-run.`);
+    console.log(`    Wallet: ${account2.address}`);
+    if (errorBody?.error) {
+      console.log();
+      console.log(chalk.dim(`Server: ${errorBody.error}`));
+    }
+    return false;
+  } finally {
+    clearInterval(interval);
+  }
+}
+var SERVER_URL = process.env.SERVER_URL ?? "https://takis.dev";
+var prompt = process.argv[2];
+if (!prompt) {
+  console.log(`Usage: ${chalk.bold('code-canvas "<prompt>"')}`);
+  console.log();
+  console.log("Examples:");
+  console.log('  code-canvas "a cat riding a skateboard"');
+  console.log('  code-canvas "cyberpunk city at sunset"');
+  process.exit(1);
+}
+var account = ensureWallet();
+var endpoint = `${SERVER_URL}/code-canvas/generate-image`;
+var success = await generateImage(endpoint, prompt, account);
+if (!success && !process.exitCode) {
+  success = await generateImage(endpoint, prompt, account);
+}
+process.exit(success ? 0 : 1);

package/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "@takisvc/code-canvas",
+  "version": "0.1.0",
+  "type": "module",
+  "bin": {
+    "code-canvas": "./dist/code-canvas.js"
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "dev": "tsx dev.ts",
+    "server": "tsx dev.ts",
+    "deploy": "wrangler deploy",
+    "build": "tsup",
+    "prepublishOnly": "npm run build",
+    "generate": "tsx client.ts"
+  },
+  "dependencies": {
+    "@hono/node-server": "^1.13.0",
+    "@openrouter/sdk": "^0.9.11",
+    "@worldcoin/agentkit": "^0.1.5",
+    "@x402/core": "^2.6.0",
+    "@x402/evm": "^2.6.0",
+    "@x402/hono": "^2.6.0",
+    "chalk": "^5.4.0",
+    "dotenv": "^17.3.1",
+    "hono": "^4.7.0",
+    "ora": "^8.2.0",
+    "viem": "^2.46.0"
+  },
+  "devDependencies": {
+    "tsup": "^8.4.0",
+    "tsx": "^4.7.0",
+    "typescript": "^5.3.0",
+    "wrangler": "^4.0.0"
+  }
+}