create-polyclaw-agent 0.1.0

package/src/index.ts

@@ -0,0 +1,211 @@
+#!/usr/bin/env node
+
+/**
+ * create-polyclaw-agent
+ *
+ * Interactive CLI to set up a Polyclaw AI trading agent.
+ * Generates a sea creature name, registers the agent, and scaffolds a bot project.
+ */
+
+import chalk from 'chalk';
+import prompts from 'prompts';
+import { generateName, registerAgent, getLinkUrl } from './api.js';
+import { scaffoldProject, nameToDir } from './scaffold.js';
+
+const LOBSTER = '\u{1F99E}';
+
+async function main() {
+  console.log();
+  console.log(chalk.bold('  Welcome to Polyclaw!'));
+  console.log();
+
+  // Step 1: Generate and pick a name
+  let agentName: string;
+  try {
+    agentName = await pickName();
+  } catch (err) {
+    console.log();
+    console.log(chalk.red('  Failed to connect to Polyclaw API.'));
+    console.log(chalk.dim('  Make sure you have an internet connection and try again.'));
+    console.log();
+    process.exit(1);
+  }
+
+  if (!agentName) {
+    console.log();
+    console.log(chalk.dim('  Cancelled.'));
+    process.exit(0);
+  }
+
+  // Step 2: Ask for owner name (optional)
+  console.log();
+  const ownerResponse = await prompts({
+    type: 'text',
+    name: 'ownerName',
+    message: 'Your name (shown on leaderboard, press Enter to skip)',
+  });
+
+  const ownerName = ownerResponse.ownerName?.trim() || undefined;
+
+  // Step 3: Register the agent
+  console.log();
+  process.stdout.write(chalk.dim(`  Registering ${chalk.white(agentName)}... `));
+
+  let credentials;
+  try {
+    credentials = await registerAgent(agentName, { ownerName });
+  } catch (err) {
+    console.log(chalk.red('failed'));
+    console.log();
+    console.log(chalk.red(`  Error: ${(err as Error).message}`));
+    process.exit(1);
+  }
+
+  console.log(chalk.green('done'));
+
+  // Step 3: Choose language
+  console.log();
+  const langResponse = await prompts({
+    type: 'select',
+    name: 'language',
+    message: 'Choose your language',
+    choices: [
+      { title: 'Python', value: 'python' },
+      { title: 'TypeScript', value: 'typescript' },
+    ],
+  });
+
+  if (!langResponse.language) {
+    console.log(chalk.dim('  Cancelled.'));
+    process.exit(0);
+  }
+
+  const language = langResponse.language as 'python' | 'typescript';
+
+  // Step 4: Ask for AI provider
+  console.log();
+  const aiResponse = await prompts({
+    type: 'select',
+    name: 'aiProvider',
+    message: 'Which AI powers your agent?',
+    choices: [
+      { title: 'Anthropic (Claude)', value: 'anthropic' },
+      { title: 'OpenAI (GPT)', value: 'openai' },
+      { title: 'None / I\'ll add my own later', value: 'none' },
+    ],
+  });
+
+  if (!aiResponse.aiProvider) {
+    console.log(chalk.dim('  Cancelled.'));
+    process.exit(0);
+  }
+
+  const aiProvider = aiResponse.aiProvider as 'anthropic' | 'openai' | 'none';
+
+  let aiApiKey: string | undefined;
+  if (aiProvider !== 'none') {
+    const providerName = aiProvider === 'anthropic' ? 'Anthropic' : 'OpenAI';
+    const keyResponse = await prompts({
+      type: 'text',
+      name: 'apiKey',
+      message: `${providerName} API key`,
+    });
+    aiApiKey = keyResponse.apiKey?.trim() || undefined;
+
+    if (!aiApiKey) {
+      console.log(chalk.dim(`  No key provided. You can add it to .env later.`));
+    }
+  }
+
+  // Step 5: Scaffold the project
+  const dirName = nameToDir(agentName);
+  console.log();
+  console.log(chalk.dim(`  Creating ${chalk.white(dirName)}/`));
+
+  const projectDir = scaffoldProject({
+    name: agentName,
+    dirName,
+    apiKey: credentials.apiKey,
+    apiSecret: credentials.apiSecret,
+    language,
+    aiProvider,
+    aiApiKey,
+  });
+
+  // Show created files
+  if (language === 'python') {
+    console.log(chalk.dim('    .env               API credentials saved'));
+    console.log(chalk.dim('    bot.py             Trading bot template'));
+    console.log(chalk.dim('    requirements.txt   Dependencies'));
+  } else {
+    console.log(chalk.dim('    .env               API credentials saved'));
+    console.log(chalk.dim('    bot.ts             Trading bot template'));
+    console.log(chalk.dim('    package.json       Dependencies'));
+  }
+
+  // Step 5: Success!
+  const linkUrl = getLinkUrl(credentials.linkToken);
+
+  const installCmd = language === 'python'
+    ? 'python3 -m venv venv && source venv/bin/activate && pip install -r requirements.txt'
+    : 'npm install';
+
+  const runCmd = language === 'python'
+    ? 'python bot.py'
+    : 'npm start';
+
+  console.log();
+  console.log(chalk.bold.green(`  ${LOBSTER} ${agentName} is ready!`));
+  console.log();
+  console.log(chalk.bold('  Step 1: Link your wallet'));
+  console.log();
+  console.log(`  Open this link in your browser:`);
+  console.log();
+  console.log(`  ${chalk.cyan(linkUrl)}`);
+  console.log();
+  console.log(chalk.bold('  Step 2: Start your bot'));
+  console.log();
+  console.log(`  ${chalk.white(`cd ${dirName}`)}`);
+  console.log(`  ${chalk.white(installCmd)}`);
+  console.log(`  ${chalk.white(runCmd)}`);
+  console.log();
+}
+
+async function pickName(): Promise<string> {
+  let name = await generateName();
+
+  while (true) {
+    console.log(`  Your agent: ${chalk.bold.white(name)}`);
+    console.log();
+
+    const response = await prompts({
+      type: 'select',
+      name: 'action',
+      message: 'Choose an action',
+      choices: [
+        { title: `Accept "${name}"`, value: 'accept' },
+        { title: 'Re-roll (get a new name)', value: 'reroll' },
+        { title: 'Cancel', value: 'cancel' },
+      ],
+    });
+
+    if (!response.action || response.action === 'cancel') {
+      return '';
+    }
+
+    if (response.action === 'accept') {
+      return name;
+    }
+
+    // Re-roll
+    process.stdout.write(chalk.dim('  Generating new name... '));
+    name = await generateName();
+    console.log(chalk.green('done'));
+    console.log();
+  }
+}
+
+main().catch((err) => {
+  console.error(chalk.red(`Unexpected error: ${err.message}`));
+  process.exit(1);
+});