hybard-agent 0.1.0

package/bin/cli.js

@@ -0,0 +1,326 @@
+#!/usr/bin/env node
+
+const { runAgent } = require('../lib/agent');
+const { HybardAI, startChat } = require('../lib/online');
+const { CodingAgent } = require('../lib/coding-agent');
+const { ProjectAnalyzer } = require('../lib/analyzer');
+const { getLocalIP, getServerInfo } = require('../lib/server');
+const ui = require('../lib/ui');
+const readline = require('readline');
+const path = require('path');
+const fs = require('fs');
+
+const args = process.argv.slice(2);
+const command = args[0] || '';
+const rest = args.slice(1).join(' ').trim();
+
+// Config file
+const CONFIG_DIR = path.join(require('os').homedir(), '.hybard');
+const NETWORK_CONFIG = path.join(CONFIG_DIR, 'network.json');
+
+function loadNetworkConfig() {
+  try {
+    if (fs.existsSync(NETWORK_CONFIG)) return JSON.parse(fs.readFileSync(NETWORK_CONFIG, 'utf8'));
+  } catch (e) {}
+  return { ip: '', port: 7860, autoConnect: true };
+}
+
+function saveNetworkConfig(config) {
+  if (!fs.existsSync(CONFIG_DIR)) fs.mkdirSync(CONFIG_DIR, { recursive: true });
+  fs.writeFileSync(NETWORK_CONFIG, JSON.stringify(config, null, 2));
+}
+
+function parsePort() {
+  const idx = args.indexOf('--port');
+  if (idx !== -1 && args[idx + 1]) return parseInt(args[idx + 1], 10) || 7860;
+  return loadNetworkConfig().port || 7860;
+}
+
+// ─── Interactive Chat ──────────────────────────────────
+function startChatMode() {
+  ui.printBanner();
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout, prompt: '' });
+  console.log(ui.c('  Commands: help, config, models, project, create, exit', ui.C.dim));
+  console.log('');
+
+  function prompt() {
+    rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (line) => handleChat(line, rl));
+  }
+  prompt();
+  rl.on('close', () => { console.log(ui.c('\n  Goodbye!\n', ui.C.green)); process.exit(0); });
+}
+
+async function handleChat(line, rl) {
+  const text = (line || '').trim();
+  if (!text) { rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleChat(l, rl)); return; }
+  if (text === 'exit' || text === 'quit') { console.log(ui.c('\n  Goodbye!\n', ui.C.green)); rl.close(); return; }
+  if (text === 'help') { ui.printHelp(); rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleChat(l, rl)); return; }
+
+  if (text === 'config') {
+    const ai = new HybardAI();
+    ui.printTable(['Setting', 'Value'], [
+      ['Provider', ai.config.provider],
+      ['Model', ai.config.model],
+      ['API Key', ai.getApiKey() ? '***' + ai.getApiKey().slice(-4) : 'NOT SET'],
+      ['Online Mode', ai.getApiKey() ? 'YES' : 'NO (Offline)'],
+    ]);
+    rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleChat(l, rl));
+    return;
+  }
+
+  if (text.startsWith('config provider ')) {
+    const provider = text.slice(16).trim();
+    const ai = new HybardAI();
+    try { ai.setProvider(provider); ui.printSuccess('Switched to ' + provider); }
+    catch (err) { ui.printError(err.message); }
+    rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleChat(l, rl));
+    return;
+  }
+
+  if (text === 'models') {
+    const { ModelRouter } = require('../lib/models');
+    const router = new ModelRouter();
+    for (const prov of router.listProviders()) {
+      const models = router.listModels(prov.id);
+      console.log(ui.c('\n  ' + prov.name, ui.C.brightYellow, ui.C.bold));
+      ui.printTable(['Model', 'Name', 'Size'], models.map(m => [m.modelId, m.name, m.size]));
+    }
+    rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleChat(l, rl));
+    return;
+  }
+
+  if (text === 'project') {
+    const analyzer = new ProjectAnalyzer();
+    const info = analyzer.analyze();
+    ui.printProjectFiles(analyzer.files.slice(0, 20), info.projectType);
+    rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleChat(l, rl));
+    return;
+  }
+
+  if (text.startsWith('create ')) {
+    console.log('');
+    const stop = ui.startThinking('Creating project');
+    rl.pause();
+    try { await runAgent(['create ' + text.slice(7)]); stop(); ui.printSuccess('Project created!'); }
+    catch (err) { stop(); ui.printError(err.message); }
+    rl.resume();
+    rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleChat(l, rl));
+    return;
+  }
+
+  // AI Chat
+  console.log('');
+  ui.printUser(text);
+  const stop = ui.startThinking('Generating');
+  rl.pause();
+  try {
+    const agent = new CodingAgent();
+    agent.init();
+    const response = await agent.chat(text);
+    stop();
+    ui.printAI(response);
+    // Save code to files
+    const blocks = agent.extractCodeBlocks(response);
+    if (blocks.length > 0) {
+      console.log(ui.c('\n  Files to save:', ui.C.brightGreen, ui.C.bold));
+      for (const b of blocks) console.log(ui.c('    - ' + b.filename, ui.C.green));
+    }
+  } catch (err) { stop(); ui.printError(err.message); }
+  rl.resume();
+  rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleChat(l, rl));
+}
+
+// ─── Coding Agent ──────────────────────────────────────
+async function startAgentMode(initialPrompt) {
+  ui.printBanner();
+  const agent = new CodingAgent();
+  // Skip slow project analysis - just set basic info
+  agent.projectContext = { projectType: 'unknown', languages: [], frameworks: [], summary: 'Project loaded', fileCount: 0 };
+  const info = agent.getModelInfo();
+  ui.printStatus(info.provider, info.model, info.projectSummary);
+  console.log(ui.c('\n  Type your request. Commands: /models, /switch, /info, exit', ui.C.dim));
+  console.log('');
+
+  if (initialPrompt) {
+    await processAgentPrompt(initialPrompt, agent);
+    return;
+  }
+
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout, prompt: '' });
+  function prompt() { rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleAgent(l, rl, agent)); }
+  prompt();
+  rl.on('close', () => { console.log(ui.c('\n  Goodbye!\n', ui.C.green)); process.exit(0); });
+}
+
+async function processAgentPrompt(text, agent) {
+  ui.printUser(text);
+  const stop = ui.startThinking('Generating');
+  try {
+    const response = await agent.chat(text);
+    stop();
+    ui.printAI(response);
+    const blocks = agent.extractCodeBlocks(response);
+    if (blocks.length > 0) {
+      console.log(ui.c('\n  Files generated:', ui.C.brightGreen, ui.C.bold));
+      for (const b of blocks) console.log(ui.c('    - ' + b.filename, ui.C.green));
+      // Auto-save files
+      console.log(ui.c('\n  Saving files...', ui.C.dim));
+      const written = agent.writeCodeBlocks(blocks);
+      for (const f of written) console.log(ui.c('    Saved: ' + f, ui.C.green));
+    }
+  } catch (err) { stop(); ui.printError(err.message); }
+}
+
+async function handleAgent(line, rl, agent) {
+  const text = (line || '').trim();
+  if (!text) { rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleAgent(l, rl, agent)); return; }
+  if (text === 'exit' || text === 'quit') { console.log(ui.c('\n  Goodbye!\n', ui.C.green)); rl.close(); return; }
+
+  if (text === '/models') {
+    const models = agent.getModels();
+    ui.printTable(['Model', 'Name', 'Size'], models.map(m => [m.modelId, m.name, m.size]));
+    rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleAgent(l, rl, agent));
+    return;
+  }
+
+  if (text.startsWith('/switch ')) {
+    const parts = text.slice(8).split(' ');
+    agent.setModel(parts[0], parts[1] || '');
+    ui.printSuccess('Switched to ' + parts[0]);
+    rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleAgent(l, rl, agent));
+    return;
+  }
+
+  if (text === '/info') {
+    const info = agent.getModelInfo();
+    ui.printTable(['Setting', 'Value'], [['Provider', info.provider], ['Model', info.model], ['Project', info.projectSummary]]);
+    rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleAgent(l, rl, agent));
+    return;
+  }
+
+  if (text === 'project') {
+    if (agent.analyzer) ui.printProjectFiles(agent.analyzer.files.slice(0, 20), agent.projectContext?.projectType);
+    rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleAgent(l, rl, agent));
+    return;
+  }
+
+  // Generate and save
+  console.log('');
+  ui.printUser(text);
+  const stop = ui.startThinking('Generating');
+  rl.pause();
+  try {
+    const response = await agent.chat(text);
+    stop();
+    ui.printAI(response);
+    const blocks = agent.extractCodeBlocks(response);
+    if (blocks.length > 0) {
+      console.log(ui.c('\n  Files generated:', ui.C.brightGreen, ui.C.bold));
+      for (const b of blocks) console.log(ui.c('    - ' + b.filename, ui.C.green));
+      console.log(ui.c('\n  Saving files...', ui.C.dim));
+      const written = agent.writeCodeBlocks(blocks);
+      for (const f of written) console.log(ui.c('    Saved: ' + f, ui.C.green));
+    }
+  } catch (err) { stop(); ui.printError(err.message); }
+  rl.resume();
+  rl.question(ui.c('  You > ', ui.C.blue, ui.C.bold), (l) => handleAgent(l, rl, agent));
+}
+
+// ─── Settings ──────────────────────────────────────────
+function showSettings() {
+  const config = loadNetworkConfig();
+  const detectedIP = getLocalIP();
+  const ip = config.ip || detectedIP;
+  const port = config.port || 7860;
+
+  ui.printNetworkSetup(ip, port);
+
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+  rl.question(ui.c('  Select option (1-3): ', ui.C.cyan), (answer) => {
+    if (answer === '1') {
+      saveNetworkConfig({ ...config, ip: detectedIP, autoConnect: true });
+      ui.printSuccess('Auto-detect IP: ' + detectedIP);
+      rl.close();
+    } else if (answer === '2') {
+      rl.question(ui.c('  Enter IP address: ', ui.C.cyan), (customIP) => {
+        saveNetworkConfig({ ...config, ip: customIP, autoConnect: false });
+        ui.printSuccess('Custom IP set: ' + customIP);
+        rl.close();
+      });
+    } else if (answer === '3') {
+      saveNetworkConfig({ ...config, ip: 'localhost', autoConnect: false });
+      ui.printSuccess('Using localhost only');
+      rl.close();
+    } else {
+      ui.printError('Invalid option');
+      rl.close();
+    }
+  });
+}
+
+// ─── Server ────────────────────────────────────────────
+function startServerMode() {
+  const config = loadNetworkConfig();
+  const detectedIP = getLocalIP();
+  const ip = config.autoConnect !== false ? detectedIP : (config.ip || detectedIP);
+  const port = parsePort();
+
+  ui.printBanner();
+  console.log(ui.c('  SERVER STARTED', ui.C.brightGreen, ui.C.bold));
+  console.log('');
+  ui.printTable(['Setting', 'Value'], [
+    ['Local URL', 'http://localhost:' + port],
+    ['Network URL', 'http://' + ip + ':' + port],
+    ['IP', ip],
+    ['Auto Connect', config.autoConnect !== false ? 'YES' : 'NO'],
+  ]);
+  console.log(ui.c('  Connect: http://' + ip + ':' + port, ui.C.brightCyan, ui.C.bold, ui.C.underline));
+  console.log('');
+
+  const { startHybardServer } = require('../lib/server');
+  startHybardServer({ port });
+}
+
+// ─── Main ──────────────────────────────────────────────
+async function main() {
+  if (args.length === 0) { startChatMode(); return; }
+
+  switch (command) {
+    case 'help': case '--help': case '-h': ui.printBanner(); ui.printHelp(); break;
+    case 'version': case '-v': ui.printSuccess('hybard-agent v1.0.0'); break;
+    case 'chat': await startChat(); break;
+    case 'agent': await startAgentMode(rest); break;
+    case 'server': startServerMode(); break;
+    case 'settings': showSettings(); break;
+    case 'models':
+      const { ModelRouter } = require('../lib/models');
+      const router = new ModelRouter();
+      for (const p of router.listProviders()) {
+        console.log(ui.c('\n  ' + p.name, ui.C.brightYellow, ui.C.bold));
+        ui.printTable(['Model', 'Name', 'Size'], router.listModels(p.id).map(m => [m.modelId, m.name, m.size]));
+      }
+      break;
+    case 'config':
+      if (!rest) {
+        const ai = new HybardAI();
+        ui.printTable(['Setting', 'Value'], [
+          ['Provider', ai.config.provider], ['Model', ai.config.model],
+          ['API Key', ai.getApiKey() ? '***' + ai.getApiKey().slice(-4) : 'NOT SET'],
+        ]);
+      } else if (rest.startsWith('provider ')) {
+        const ai = new HybardAI();
+        try { ai.setProvider(rest.slice(9).trim()); ui.printSuccess('Switched to ' + rest.slice(9).trim()); }
+        catch (err) { ui.printError(err.message); }
+      } else if (rest.startsWith('set-key ')) {
+        const ai = new HybardAI();
+        ai.saveConfig({ apiKey: rest.slice(8).trim() });
+        ui.printSuccess('API key saved');
+      }
+      break;
+    case 'create': await runAgent(['create ' + rest]); break;
+    case 'fix': case 'inspect': case 'install': case 'plan': await runAgent([command + ' ' + rest]); break;
+    default: await runAgent([args.join(' ')]); break;
+  }
+}
+
+main().catch((err) => { ui.printError(err.message); process.exit(1); });

package/bin/hybard.cmd

@@ -0,0 +1,2 @@
+@ECHO OFF
+node "%~dp0cli.js" %*