@xyz-credit/agent-cli 1.1.2 → 1.2.1

package/src/commands/setup.js

@@ -0,0 +1,295 @@
+/**
+ * Setup Wizard — Automated NPX onboarding
+ *
+ * Chains inquirer prompts for:
+ *   1. Platform URL
+ *   2. LLM provider selection + API key validation
+ *   3. Custom MCP server (optional)
+ *   4. Auto-generate .env + config.json
+ *   5. Trigger first boot-up
+ */
+const inquirer = require('inquirer');
+const chalk = require('chalk');
+const ora = require('ora');
+const fetch = require('node-fetch');
+const fs = require('fs');
+const path = require('path');
+const boxen = require('boxen');
+const { config, saveCredentials, isAuthenticated, getCredentials } = require('../config');
+
+const LLM_PROVIDERS = [
+  { name: 'OpenAI (GPT-4o, GPT-5.2)', value: 'openai' },
+  { name: 'Anthropic (Claude Sonnet 4.5)', value: 'anthropic' },
+  { name: 'Google Gemini', value: 'gemini' },
+  { name: 'Mistral (Local via Ollama)', value: 'ollama' },
+  { name: 'Groq', value: 'groq' },
+];
+
+async function validateLLMKey(provider, apiKey, platformUrl) {
+  if (provider === 'ollama') return { valid: true };
+
+  const spinner = ora(`Validating ${provider} API key...`).start();
+  try {
+    const res = await fetch(`${platformUrl}/api/cli/validate-llm`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ provider, api_key: apiKey }),
+      timeout: 15000,
+    });
+    if (res.ok) {
+      const data = await res.json();
+      if (data.valid) {
+        spinner.succeed(`${provider} API key validated`);
+        return { valid: true };
+      }
+      spinner.fail(`${provider} key invalid: ${data.error || 'authentication failed'}`);
+      return { valid: false, error: data.error };
+    }
+    spinner.fail('Validation request failed');
+    return { valid: false };
+  } catch (e) {
+    // Fallback: try direct validation
+    try {
+      let valid = false;
+      if (provider === 'openai') {
+        const r = await fetch('https://api.openai.com/v1/models', {
+          headers: { Authorization: `Bearer ${apiKey}` }, timeout: 8000,
+        });
+        valid = r.status === 200;
+      } else if (provider === 'anthropic') {
+        const r = await fetch('https://api.anthropic.com/v1/messages', {
+          method: 'POST',
+          headers: { 'x-api-key': apiKey, 'anthropic-version': '2023-06-01', 'content-type': 'application/json' },
+          body: JSON.stringify({ model: 'claude-3-haiku-20240307', max_tokens: 1, messages: [{ role: 'user', content: 'hi' }] }),
+          timeout: 8000,
+        });
+        valid = r.status === 200 || r.status === 201;
+      } else if (provider === 'gemini') {
+        const r = await fetch(`https://generativelanguage.googleapis.com/v1/models?key=${apiKey}`, { timeout: 8000 });
+        valid = r.status === 200;
+      } else if (provider === 'groq') {
+        const r = await fetch('https://api.groq.com/openai/v1/models', {
+          headers: { Authorization: `Bearer ${apiKey}` }, timeout: 8000,
+        });
+        valid = r.status === 200;
+      }
+      if (valid) {
+        spinner.succeed(`${provider} API key validated (direct)`);
+        return { valid: true };
+      }
+      spinner.fail(`${provider} key invalid`);
+      return { valid: false };
+    } catch {
+      spinner.warn(`Could not validate (network issue) — saving key anyway`);
+      return { valid: true, warning: 'offline_validation' };
+    }
+  }
+}
+
+async function setupCommand() {
+  console.log('');
+  console.log(boxen(
+    chalk.bold.cyan('  xyz.credit Agent Setup Wizard\n\n') +
+    chalk.dim('  Automated onboarding — configure identity, LLM, and MCP\n') +
+    chalk.dim('  in one seamless flow. No manual file editing needed.'),
+    { padding: 1, margin: 1, borderColor: 'cyan', borderStyle: 'round', title: 'Setup', titleAlignment: 'center' }
+  ));
+
+  // Check existing setup
+  if (isAuthenticated()) {
+    const creds = getCredentials();
+    console.log(chalk.yellow(`  Existing agent found: ${creds.agentName} (${creds.agentId.slice(0, 12)}...)`));
+    const { reconfigure } = await inquirer.prompt([{
+      type: 'confirm', name: 'reconfigure',
+      message: 'Reconfigure from scratch?', default: false,
+    }]);
+    if (!reconfigure) {
+      console.log(chalk.green('  Using existing configuration.\n'));
+      return;
+    }
+  }
+
+  // ── Step 1: Platform URL ──
+  console.log(chalk.bold('\n  Step 1/4: Platform Connection\n'));
+  const { platformUrl } = await inquirer.prompt([{
+    type: 'input', name: 'platformUrl',
+    message: 'Platform URL:',
+    default: config.get('platformUrl') || 'https://agent-messaging.preview.emergentagent.com',
+    validate: (v) => v.startsWith('http') ? true : 'Must start with http(s)://',
+  }]);
+
+  const pingSpinner = ora('Testing platform connection...').start();
+  try {
+    const r = await fetch(`${platformUrl}/api/health`, { timeout: 8000 });
+    if (r.ok) pingSpinner.succeed('Platform connected');
+    else pingSpinner.warn('Platform responded but may have issues');
+  } catch {
+    pingSpinner.warn('Cannot reach platform — continuing offline');
+  }
+
+  config.set('platformUrl', platformUrl);
+
+  // ── Step 2: LLM Provider ──
+  console.log(chalk.bold('\n  Step 2/4: LLM Configuration\n'));
+  const { llmProvider } = await inquirer.prompt([{
+    type: 'list', name: 'llmProvider',
+    message: 'Select your LLM provider:',
+    choices: LLM_PROVIDERS,
+  }]);
+
+  let llmApiKey = '';
+  let llmValidated = false;
+
+  if (llmProvider === 'ollama') {
+    console.log(chalk.dim('  Ollama runs locally — no API key needed.'));
+    const { ollamaUrl } = await inquirer.prompt([{
+      type: 'input', name: 'ollamaUrl',
+      message: 'Ollama server URL:',
+      default: 'http://localhost:11434',
+    }]);
+    config.set('llmProvider', 'ollama');
+    config.set('llmEndpoint', ollamaUrl);
+    llmValidated = true;
+  } else {
+    let attempts = 0;
+    while (attempts < 3 && !llmValidated) {
+      const { apiKey } = await inquirer.prompt([{
+        type: 'password', name: 'apiKey', mask: '*',
+        message: `${llmProvider} API Key:`,
+        validate: (v) => v.length > 10 ? true : 'API key seems too short',
+      }]);
+
+      const result = await validateLLMKey(llmProvider, apiKey, platformUrl);
+      if (result.valid) {
+        llmApiKey = apiKey;
+        llmValidated = true;
+      } else {
+        attempts++;
+        if (attempts < 3) console.log(chalk.yellow(`  Try again (${3 - attempts} attempts left)`));
+      }
+    }
+
+    if (!llmValidated) {
+      console.log(chalk.red('  LLM validation failed. You can update the key later with `xyz-agent config`.'));
+    }
+
+    config.set('llmProvider', llmProvider);
+    config.set('llmApiKey', llmApiKey);
+  }
+
+  // ── Step 3: Custom MCP Server ──
+  console.log(chalk.bold('\n  Step 3/4: MCP Server (Optional)\n'));
+  const { hasMcp } = await inquirer.prompt([{
+    type: 'confirm', name: 'hasMcp',
+    message: 'Do you have a custom MCP server for your own products?',
+    default: false,
+  }]);
+
+  let mcpUrl = '';
+  if (hasMcp) {
+    const answer = await inquirer.prompt([{
+      type: 'input', name: 'mcpUrl',
+      message: 'MCP server URL:',
+      default: 'http://localhost:3001/mcp',
+      validate: (v) => v.startsWith('http') ? true : 'Must be a valid URL',
+    }]);
+    mcpUrl = answer.mcpUrl;
+    config.set('localMcpUrl', mcpUrl);
+
+    // Try connecting
+    const mcpSpinner = ora('Testing MCP connection...').start();
+    try {
+      const r = await fetch(mcpUrl.replace('/sse', ''), {
+        method: 'POST', timeout: 5000,
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ jsonrpc: '2.0', id: 1, method: 'initialize', params: { protocolVersion: '2025-03-26', capabilities: {}, clientInfo: { name: 'xyz-agent-cli', version: '2.0.0' } } }),
+      });
+      if (r.ok) mcpSpinner.succeed('MCP server connected');
+      else mcpSpinner.warn('MCP server responded with non-200');
+    } catch {
+      mcpSpinner.warn('MCP not reachable now — will retry on start');
+    }
+  }
+
+  // ── Step 4: Finalization ──
+  console.log(chalk.bold('\n  Step 4/4: Generating Configuration\n'));
+
+  const outputDir = process.cwd();
+  const envPath = path.join(outputDir, '.env');
+  const configJsonPath = path.join(outputDir, 'config.json');
+
+  // Generate .env
+  const envContent = [
+    `# xyz.credit Agent Configuration`,
+    `# Generated by setup wizard at ${new Date().toISOString()}`,
+    ``,
+    `PLATFORM_URL=${platformUrl}`,
+    `LLM_PROVIDER=${llmProvider}`,
+    llmApiKey ? `LLM_API_KEY=${llmApiKey}` : `# LLM_API_KEY=`,
+    llmProvider === 'ollama' ? `OLLAMA_URL=${config.get('llmEndpoint') || 'http://localhost:11434'}` : '',
+    mcpUrl ? `LOCAL_MCP_SOURCE=${mcpUrl}` : `# LOCAL_MCP_SOURCE=`,
+    ``,
+    `# Agent identity (populated after auth)`,
+    `# AGENT_ID=`,
+    `# API_KEY=`,
+    ``,
+    `# Heartbeat interval (seconds)`,
+    `HEARTBEAT_INTERVAL=1800`,
+  ].filter(Boolean).join('\n') + '\n';
+
+  // Generate config.json
+  const configJson = {
+    platform_url: platformUrl,
+    llm: {
+      provider: llmProvider,
+      validated: llmValidated,
+      ...(llmProvider === 'ollama' ? { endpoint: config.get('llmEndpoint') } : {}),
+    },
+    mcp: mcpUrl ? { url: mcpUrl, connected: false } : null,
+    heartbeat: { interval_seconds: 1800, enabled: true },
+    agent: { id: config.get('agentId') || null, name: config.get('agentName') || null },
+    created_at: new Date().toISOString(),
+  };
+
+  const genSpinner = ora('Writing configuration files...').start();
+
+  fs.writeFileSync(envPath, envContent);
+  fs.writeFileSync(configJsonPath, JSON.stringify(configJson, null, 2) + '\n');
+
+  genSpinner.succeed('Configuration files generated');
+
+  console.log(chalk.dim(`  .env       → ${envPath}`));
+  console.log(chalk.dim(`  config.json→ ${configJsonPath}`));
+
+  // ── Auto boot-up ──
+  console.log('');
+  console.log(boxen(
+    chalk.green.bold('  Setup Complete!\n\n') +
+    chalk.white(`  Platform:  ${chalk.cyan(platformUrl)}\n`) +
+    chalk.white(`  LLM:       ${chalk.cyan(llmProvider)} ${llmValidated ? chalk.green('(validated)') : chalk.yellow('(unverified)')}\n`) +
+    chalk.white(`  MCP:       ${mcpUrl ? chalk.cyan(mcpUrl) : chalk.dim('Not configured')}\n\n`) +
+    chalk.dim('  Next steps:\n') +
+    chalk.white(`  1. ${chalk.cyan('xyz-agent auth')}              Authenticate your agent\n`) +
+    chalk.white(`  2. ${chalk.cyan('xyz-agent connect')}           Bridge your MCP tools\n`) +
+    chalk.white(`  3. ${chalk.cyan('xyz-agent register-service')}  Publish to marketplace\n`) +
+    chalk.white(`  4. ${chalk.cyan('xyz-agent start')}             Start the agent with dashboard`),
+    { padding: 1, margin: 1, borderColor: 'green', borderStyle: 'round', title: 'Ready', titleAlignment: 'center' }
+  ));
+
+  // If already authenticated, offer to start immediately
+  if (isAuthenticated()) {
+    const { autoStart } = await inquirer.prompt([{
+      type: 'confirm', name: 'autoStart',
+      message: 'Agent is already authenticated. Start now?',
+      default: true,
+    }]);
+    if (autoStart) {
+      console.log(chalk.dim('\n  Launching agent...\n'));
+      const { startCommand } = require('./start');
+      await startCommand({});
+    }
+  }
+  console.log('');
+}
+
+module.exports = { setupCommand };