@openagents-org/agent-connector 0.1.0

package/src/cli.js

@@ -0,0 +1,526 @@
+'use strict';
+
+const { AgentConnector, Daemon } = require('./index');
+
+// ---------------------------------------------------------------------------
+// Arg parsing
+// ---------------------------------------------------------------------------
+
+function parseArgs(argv) {
+  const args = argv.slice(2);
+  const flags = {};
+  const allPositional = [];
+
+  for (let i = 0; i < args.length; i++) {
+    const a = args[i];
+    if (a.startsWith('--')) {
+      const eq = a.indexOf('=');
+      if (eq > 0) {
+        flags[a.slice(2, eq)] = a.slice(eq + 1);
+      } else if (i + 1 < args.length && !args[i + 1].startsWith('--')) {
+        flags[a.slice(2)] = args[i + 1];
+        i++;
+      } else {
+        flags[a.slice(2)] = true;
+      }
+    } else {
+      allPositional.push(a);
+    }
+  }
+
+  const cmd = allPositional[0] || 'status';
+  const positional = allPositional.slice(1);
+
+  return { cmd, flags, positional };
+}
+
+// ---------------------------------------------------------------------------
+// Helpers
+// ---------------------------------------------------------------------------
+
+function getConnector(flags) {
+  const opts = {};
+  if (flags.config) opts.configDir = flags.config;
+  return new AgentConnector(opts);
+}
+
+function print(msg) { process.stdout.write(msg + '\n'); }
+
+function table(rows, headers) {
+  if (rows.length === 0) return;
+  const widths = headers.map((h, i) =>
+    Math.max(h.length, ...rows.map((r) => String(r[i] || '').length))
+  );
+  print(headers.map((h, i) => h.padEnd(widths[i])).join('  '));
+  print(widths.map((w) => '-'.repeat(w)).join('  '));
+  for (const row of rows) {
+    print(row.map((c, i) => String(c || '').padEnd(widths[i])).join('  '));
+  }
+}
+
+// ---------------------------------------------------------------------------
+// Commands
+// ---------------------------------------------------------------------------
+
+async function cmdUp(connector, flags) {
+  const pid = connector.getDaemonPid();
+  if (pid) {
+    print(`Daemon already running (PID ${pid})`);
+    return;
+  }
+
+  if (flags.foreground) {
+    // Run in foreground (used by daemonize child)
+    const daemon = connector.createDaemon();
+    await daemon.start();
+  } else {
+    // Daemonize
+    const foregroundArgs = [process.argv[1], 'up', '--foreground'];
+    if (flags.config) foregroundArgs.push('--config', flags.config);
+    connector.startDaemon(foregroundArgs);
+  }
+}
+
+async function cmdDown(connector) {
+  const stopped = connector.stopDaemon();
+  if (stopped) {
+    print('Daemon stopped');
+  } else {
+    print('Daemon is not running');
+  }
+}
+
+async function cmdStatus(connector) {
+  const pid = connector.getDaemonPid();
+  if (!pid) {
+    print('Daemon is not running');
+  } else {
+    print(`Daemon running (PID ${pid})`);
+  }
+
+  const agents = connector.listAgents();
+  if (agents.length === 0) {
+    print('\nNo agents configured. Run: agent-connector create <name> --type <type>');
+    return;
+  }
+
+  const status = connector.getDaemonStatus();
+  const rows = agents.map((a) => {
+    const s = status[a.name] || {};
+    const state = s.state || (pid ? 'stopped' : '-');
+    const restarts = s.restarts || 0;
+    return [a.name, a.type, state, a.network || '(local)', restarts > 0 ? `${restarts}` : ''];
+  });
+  print('');
+  table(rows, ['NAME', 'TYPE', 'STATE', 'NETWORK', 'RESTARTS']);
+}
+
+async function cmdCreate(connector, flags, positional) {
+  const name = positional[0];
+  if (!name) { print('Usage: agent-connector create <name> [--type <type>]'); return; }
+  const type = flags.type || 'openclaw';
+  const role = flags.role || 'worker';
+
+  try {
+    connector.addAgent({ name, type, role, path: flags.path });
+    print(`Agent '${name}' created (type: ${type})`);
+
+    // Auto-install if not installed
+    if (!connector.isInstalled(type)) {
+      print(`Installing ${type}...`);
+      try {
+        await connector.install(type);
+        print(`${type} installed`);
+      } catch (e) {
+        print(`Warning: install failed: ${e.message}`);
+      }
+    }
+  } catch (e) {
+    print(`Error: ${e.message}`);
+    process.exitCode = 1;
+  }
+}
+
+async function cmdRemove(connector, _flags, positional) {
+  const name = positional[0];
+  if (!name) { print('Usage: agent-connector remove <name>'); return; }
+  connector.removeAgent(name);
+  print(`Agent '${name}' removed`);
+}
+
+async function cmdStart(connector, _flags, positional) {
+  const name = positional[0];
+  if (!name) { print('Usage: agent-connector start <name>'); return; }
+  connector.sendDaemonCommand(`restart:${name}`);
+  print(`Sent start command for '${name}'`);
+}
+
+async function cmdStop(connector, _flags, positional) {
+  const name = positional[0];
+  if (!name) { print('Usage: agent-connector stop <name>'); return; }
+  connector.sendDaemonCommand(`stop:${name}`);
+  print(`Sent stop command for '${name}'`);
+}
+
+async function cmdInstall(connector, _flags, positional) {
+  const type = positional[0];
+  if (!type) { print('Usage: agent-connector install <type>'); return; }
+
+  if (connector.isInstalled(type)) {
+    print(`${type} is already installed`);
+    return;
+  }
+
+  print(`Installing ${type}...`);
+  try {
+    const result = await connector.install(type);
+    print(`${type} installed successfully`);
+    if (result.output) print(result.output);
+  } catch (e) {
+    print(`Error: ${e.message}`);
+    process.exitCode = 1;
+  }
+}
+
+async function cmdUninstall(connector, _flags, positional) {
+  const type = positional[0];
+  if (!type) { print('Usage: agent-connector uninstall <type>'); return; }
+
+  print(`Uninstalling ${type}...`);
+  try {
+    const result = await connector.uninstall(type);
+    print(`${type} uninstalled`);
+    if (result.output) print(result.output);
+  } catch (e) {
+    print(`Error: ${e.message}`);
+    process.exitCode = 1;
+  }
+}
+
+async function cmdSearch(connector, flags, positional) {
+  const query = positional[0] || '';
+  let catalog;
+  try {
+    catalog = await connector.getCatalog();
+  } catch {
+    catalog = connector.registry.getCatalogSync().map((e) => ({
+      ...e,
+      installed: connector.isInstalled(e.name),
+    }));
+  }
+
+  if (query) {
+    const q = query.toLowerCase();
+    catalog = catalog.filter((e) =>
+      e.name.includes(q) || (e.label || '').toLowerCase().includes(q) ||
+      (e.description || '').toLowerCase().includes(q) ||
+      (e.tags || []).some((t) => t.includes(q))
+    );
+  }
+
+  if (catalog.length === 0) {
+    print(query ? `No agents matching '${query}'` : 'No agents in catalog');
+    return;
+  }
+
+  const rows = catalog.map((e) => [
+    e.name,
+    e.label || e.name,
+    e.installed ? 'installed' : '',
+    (e.description || '').slice(0, 50),
+  ]);
+  table(rows, ['NAME', 'LABEL', 'STATUS', 'DESCRIPTION']);
+}
+
+async function cmdList(connector) {
+  const agents = connector.listAgents();
+  if (agents.length === 0) {
+    print('No agents configured');
+    return;
+  }
+  const rows = agents.map((a) => [
+    a.name, a.type, a.role, a.network || '(local)',
+  ]);
+  table(rows, ['NAME', 'TYPE', 'ROLE', 'NETWORK']);
+}
+
+async function cmdRuntimes(connector) {
+  let catalog;
+  try {
+    catalog = await connector.getCatalog();
+  } catch {
+    catalog = connector.registry.getCatalogSync().map((e) => ({
+      ...e, installed: connector.isInstalled(e.name),
+    }));
+  }
+
+  const installed = catalog.filter((e) => e.installed);
+  if (installed.length === 0) {
+    print('No agent runtimes installed');
+    return;
+  }
+
+  const rows = installed.map((e) => {
+    const binary = connector.installer.which(e.name) || '-';
+    return [e.name, e.label || e.name, binary];
+  });
+  table(rows, ['NAME', 'LABEL', 'PATH']);
+}
+
+async function cmdConnect(connector, flags, positional) {
+  const name = positional[0];
+  const token = positional[1] || flags.token;
+  if (!name || !token) {
+    print('Usage: agent-connector connect <agent-name> <token>');
+    return;
+  }
+
+  print(`Resolving workspace token...`);
+  try {
+    const info = await connector.resolveToken(token);
+    const slug = info.slug || info.workspace_id;
+    const wsName = info.name || slug;
+
+    // Save network
+    connector.config.addNetwork({
+      id: info.workspace_id,
+      slug,
+      name: wsName,
+      endpoint: info.endpoint || connector.workspace.endpoint,
+      token,
+    });
+
+    // Connect agent
+    connector.connectWorkspace(name, slug);
+    print(`'${name}' connected to workspace '${wsName}'`);
+
+    // Signal daemon reload
+    const pid = connector.getDaemonPid();
+    if (pid) {
+      connector.sendDaemonCommand(`restart:${name}`);
+      print('Daemon notified');
+    }
+  } catch (e) {
+    print(`Error: ${e.message}`);
+    process.exitCode = 1;
+  }
+}
+
+async function cmdDisconnect(connector, _flags, positional) {
+  const name = positional[0];
+  if (!name) { print('Usage: agent-connector disconnect <agent-name>'); return; }
+  connector.disconnectWorkspace(name);
+  print(`'${name}' disconnected from workspace`);
+
+  const pid = connector.getDaemonPid();
+  if (pid) {
+    connector.sendDaemonCommand(`restart:${name}`);
+  }
+}
+
+async function cmdLogs(connector, flags, positional) {
+  const agent = positional[0] || flags.agent;
+  const lines = parseInt(flags.lines || flags.n || '50', 10);
+  const logLines = connector.getLogs(agent, lines);
+  for (const line of logLines) {
+    if (line) print(line);
+  }
+}
+
+async function cmdWorkspace(connector, flags, positional) {
+  const sub = positional[0] || 'list';
+  const subArgs = positional.slice(1);
+
+  switch (sub) {
+    case 'create': {
+      const name = subArgs[0] || flags.name || 'My Workspace';
+      print(`Creating workspace '${name}'...`);
+      try {
+        const result = await connector.createWorkspace({ name });
+        print(`Workspace created: ${result.name}`);
+        print(`  Slug:  ${result.slug}`);
+        print(`  Token: ${result.token}`);
+        print(`  URL:   ${result.url}`);
+      } catch (e) {
+        print(`Error: ${e.message}`);
+        process.exitCode = 1;
+      }
+      break;
+    }
+
+    case 'join': {
+      const token = subArgs[0] || flags.token;
+      if (!token) { print('Usage: agent-connector workspace join <token>'); return; }
+      try {
+        const info = await connector.resolveToken(token);
+        connector.config.addNetwork({
+          id: info.workspace_id,
+          slug: info.slug || info.workspace_id,
+          name: info.name || info.slug,
+          endpoint: info.endpoint || connector.workspace.endpoint,
+          token,
+        });
+        print(`Joined workspace '${info.name || info.slug}'`);
+      } catch (e) {
+        print(`Error: ${e.message}`);
+        process.exitCode = 1;
+      }
+      break;
+    }
+
+    case 'list':
+    default: {
+      const workspaces = connector.listWorkspaces();
+      if (workspaces.length === 0) {
+        print('No workspaces configured');
+        return;
+      }
+      const rows = workspaces.map((w) => [w.slug, w.name, w.endpoint || '-']);
+      table(rows, ['SLUG', 'NAME', 'ENDPOINT']);
+      break;
+    }
+  }
+}
+
+async function cmdEnv(connector, flags, positional) {
+  const type = positional[0];
+  if (!type) { print('Usage: agent-connector env <type> [--set KEY=VALUE]'); return; }
+
+  const setVal = flags.set;
+  if (setVal) {
+    const eq = setVal.indexOf('=');
+    if (eq < 1) { print('Usage: --set KEY=VALUE'); return; }
+    const key = setVal.slice(0, eq);
+    const val = setVal.slice(eq + 1);
+    connector.saveAgentEnv(type, { [key]: val });
+    print(`Saved ${key} for ${type}`);
+    return;
+  }
+
+  // Show current env
+  const env = connector.getAgentEnv(type);
+  const fields = connector.getEnvFields(type);
+
+  if (fields.length > 0) {
+    for (const field of fields) {
+      const val = env[field.name];
+      const display = field.password && val ? '***' : (val || '(not set)');
+      print(`  ${field.name}: ${display}  ${field.required ? '(required)' : ''}`);
+    }
+  } else {
+    const entries = Object.entries(env);
+    if (entries.length === 0) {
+      print(`No env vars configured for ${type}`);
+    } else {
+      for (const [k, v] of entries) {
+        print(`  ${k}: ${v}`);
+      }
+    }
+  }
+}
+
+async function cmdTestLLM(connector, _flags, positional) {
+  const type = positional[0];
+  if (!type) { print('Usage: agent-connector test-llm <type>'); return; }
+
+  const env = connector.getAgentEnv(type);
+  const resolved = connector.resolveAgentEnv(type, env);
+  const effective = { ...env, ...resolved };
+
+  print(`Testing LLM connection for ${type}...`);
+  const result = await connector.testLLM(effective);
+  if (result.success) {
+    print(`Success! Model: ${result.model}, Response: ${result.response}`);
+  } else {
+    print(`Failed: ${result.error}`);
+    process.exitCode = 1;
+  }
+}
+
+async function cmdVersion() {
+  const pkg = require('../package.json');
+  print(`${pkg.name} v${pkg.version}`);
+}
+
+async function cmdHelp() {
+  print(`Usage: agent-connector <command> [options]
+
+Commands:
+  up [--foreground]           Start daemon (background by default)
+  down                        Stop daemon
+  status                      Show agent status
+  list                        List configured agents
+  create <name> [--type T]    Create a new agent
+  remove <name>               Remove an agent
+  start <name>                Start a single agent
+  stop <name>                 Stop a single agent
+  install <type>              Install an agent runtime
+  uninstall <type>            Uninstall an agent runtime
+  search [query]              Browse agent catalog
+  runtimes                    List installed runtimes
+  connect <agent> <token>     Connect agent to workspace
+  disconnect <agent>          Disconnect agent from workspace
+  env <type> [--set K=V]      View/set env vars for agent type
+  test-llm <type>             Test LLM connection
+  logs [agent] [--lines N]    View daemon logs
+  workspace create [name]     Create a new workspace
+  workspace join <token>      Join workspace with token
+  workspace list              List configured workspaces
+  version                     Show version
+  help                        Show this help
+
+Options:
+  --config <dir>              Config directory (default: ~/.openagents)
+`);
+}
+
+// ---------------------------------------------------------------------------
+// Main
+// ---------------------------------------------------------------------------
+
+async function main() {
+  const { cmd, flags, positional } = parseArgs(process.argv);
+
+  if (cmd === 'help' || flags.help) { await cmdHelp(); return; }
+  if (cmd === 'version' || flags.version) { await cmdVersion(); return; }
+
+  const connector = getConnector(flags);
+
+  const commands = {
+    up: () => cmdUp(connector, flags),
+    down: () => cmdDown(connector),
+    status: () => cmdStatus(connector),
+    list: () => cmdList(connector),
+    create: () => cmdCreate(connector, flags, positional),
+    remove: () => cmdRemove(connector, flags, positional),
+    start: () => cmdStart(connector, flags, positional),
+    stop: () => cmdStop(connector, flags, positional),
+    install: () => cmdInstall(connector, flags, positional),
+    uninstall: () => cmdUninstall(connector, flags, positional),
+    search: () => cmdSearch(connector, flags, positional),
+    runtimes: () => cmdRuntimes(connector),
+    connect: () => cmdConnect(connector, flags, positional),
+    disconnect: () => cmdDisconnect(connector, flags, positional),
+    logs: () => cmdLogs(connector, flags, positional),
+    workspace: () => cmdWorkspace(connector, flags, positional),
+    env: () => cmdEnv(connector, flags, positional),
+    'test-llm': () => cmdTestLLM(connector, flags, positional),
+  };
+
+  const handler = commands[cmd];
+  if (!handler) {
+    print(`Unknown command: ${cmd}`);
+    print('Run: agent-connector help');
+    process.exitCode = 1;
+    return;
+  }
+
+  try {
+    await handler();
+  } catch (e) {
+    print(`Error: ${e.message}`);
+    process.exitCode = 1;
+  }
+}
+
+main();