@takisvc/code-canvas 0.1.2 → 0.1.3

package/dist/code-canvas.js

@@ -63,9 +63,7 @@ function displayInlineImage(base64) {
       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\\`
-        );
+        process.stdout.write(`\x1B_Gf=100,a=T,m=${more};${chunk}\x1B\\`);
       } else {
         process.stdout.write(`\x1B_Gm=${more};${chunk}\x1B\\`);
       }
@@ -87,7 +85,7 @@ function openFile(filepath) {
   } catch {
   }
 }
-function saveImage(base64, prompt2) {
+function saveImage(base64, prompt) {
   const filename = `generated-${Date.now()}.png`;
   const buffer = Buffer.from(base64, "base64");
   writeFileSync(filename, buffer);
@@ -95,7 +93,7 @@ function saveImage(base64, prompt2) {
   const tweetText = encodeURIComponent(
     `My AI agent just roasted my coding session \u{1F525}
 
-Prompt: "${prompt2}"
+Prompt: "${prompt}"
 
 Powered by @worldcoin AgentKit`
   );
@@ -125,7 +123,7 @@ function showUsage(usage) {
 function extractAgentkitChallenge(body) {
   return body?.extensions?.agentkit ?? body?.accepts?.[0]?.extensions?.agentkit ?? body?.paymentRequirements?.[0]?.extensions?.agentkit ?? body?.challenge;
 }
-async function generateImage(endpoint2, prompt2, account2) {
+async function generateImage(endpoint2, prompt, account2) {
   const spinner = ora(randomSpinnerMessage()).start();
   const interval = setInterval(() => {
     spinner.text = randomSpinnerMessage();
@@ -134,12 +132,12 @@ async function generateImage(endpoint2, prompt2, account2) {
     const firstResponse = await fetch(endpoint2, {
       method: "POST",
       headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({ prompt: prompt2 })
+      body: JSON.stringify({ prompt })
     });
     if (firstResponse.status === 200) {
       const body = await firstResponse.json();
       spinner.succeed(chalk.green("Image generated!"));
-      saveImage(body.image, prompt2);
+      saveImage(body.image, prompt);
       showUsage(body.usage);
       return true;
     }
@@ -195,14 +193,14 @@ async function generateImage(endpoint2, prompt2, account2) {
         "Content-Type": "application/json",
         agentkit: encoded
       },
-      body: JSON.stringify({ prompt: prompt2 })
+      body: JSON.stringify({ prompt })
     });
     if (secondResponse.status === 200) {
       const body = await secondResponse.json();
       spinner.succeed(
         chalk.green("Image generated (free \u2014 World ID verified!)")
       );
-      saveImage(body.image, prompt2);
+      saveImage(body.image, prompt);
       showUsage(body.usage);
       return true;
     }
@@ -216,13 +214,19 @@ async function generateImage(endpoint2, prompt2, account2) {
     console.log();
     console.log(chalk.yellow("Your agent is not registered with World ID."));
     console.log();
-    console.log(`  ${chalk.bold("Register:")} Run this in your terminal and scan the QR code with World App:`);
+    console.log(
+      `  ${chalk.bold("Register:")} Run this in your terminal and scan the QR code with World App:`
+    );
     console.log();
-    console.log(`    npx @worldcoin/agentkit-cli register ${account2.address} --network world`);
+    console.log(
+      `    npx @worldcoin/agentkit-cli register ${account2.address} --network world`
+    );
     console.log();
     console.log(`  Then re-run the image generation command.`);
     console.log();
-    console.log(`  ${chalk.bold("Or pay:")} Fund wallet ${account2.address} with ${chalk.cyan("$1.20 USDC")} on World Chain.`);
+    console.log(
+      `  ${chalk.bold("Or pay:")} Fund wallet ${account2.address} with ${chalk.cyan("$1.20 USDC")} on World Chain.`
+    );
     if (errorBody?.error) {
       console.log();
       console.log(chalk.dim(`Server: ${errorBody.error}`));
@@ -232,10 +236,39 @@ async function generateImage(endpoint2, prompt2, account2) {
     clearInterval(interval);
   }
 }
+function runSetup(account2) {
+  console.log(chalk.bold("Code Canvas Setup"));
+  console.log();
+  console.log(`  Wallet: ${chalk.cyan(account2.address)}`);
+  console.log();
+  console.log(
+    "Registering with World ID \u2014 scan the QR code with your World App:"
+  );
+  console.log();
+  try {
+    execSync(
+      `npx @worldcoin/agentkit-cli register ${account2.address} --network world`,
+      { stdio: "inherit" }
+    );
+    console.log();
+    console.log(chalk.green("Setup complete! You can now generate images:"));
+    return true;
+  } catch {
+    console.log();
+    console.log(chalk.red("Registration failed or was cancelled."));
+    return false;
+  }
+}
 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>"')}`);
+var command = process.argv[2];
+if (!command) {
+  console.log(`Usage:`);
+  console.log(
+    `  ${chalk.bold("code-canvas setup")}              Register with World ID`
+  );
+  console.log(
+    `  ${chalk.bold('code-canvas "<prompt>"')}          Generate an image`
+  );
   console.log();
   console.log("Examples:");
   console.log('  code-canvas "a cat riding a skateboard"');
@@ -243,6 +276,10 @@ if (!prompt) {
   process.exit(1);
 }
 var account = ensureWallet();
+if (command === "setup") {
+  const ok = runSetup(account);
+  process.exit(ok ? 0 : 1);
+}
 var endpoint = `${SERVER_URL}/code-canvas/generate-image`;
-var result = await generateImage(endpoint, prompt, account);
+var result = await generateImage(endpoint, command, account);
 process.exit(result === true ? 0 : result === "needs-registration" ? 0 : 1);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@takisvc/code-canvas",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "type": "module",
   "bin": {
     "code-canvas": "./dist/code-canvas.js"