habi-agent 0.1.3 → 0.1.5

package/src/cli.ts

@@ -1,27 +1,47 @@
 #!/usr/bin/env node
 /**
- * HABI Agent CLI
+ * HABI Agent CLI v0.1.5
+ *
  * Usage:
- *   habi-agent start     — start the local identity daemon
- *   habi-agent status    — check if agent is running
- *   habi-agent identity  — print hardware identity
- *   habi-agent init      — generate a Project Context File
- *   habi-agent verify    — verify current session
+ *   npx habi-agent@latest start --key <key> --name <name>
+ *     Installs HABI as a persistent background service, registers the device,
+ *     injects Claude Code MCP config, and exits. Terminal can be closed.
+ *
+ *   habi-agent stop         Stop the background service
+ *   habi-agent restart      Restart the background service
+ *   habi-agent uninstall    Remove the background service
+ *   habi-agent status       Show service + agent health
+ *   habi-agent identity     Print hardware fingerprint
+ *   habi-agent logs         Tail the agent log file
  */
 
-import { HabiLocalServer } from './server';
-import { readHardwareIdentity, generateSessionFingerprint, sha256 } from './hardware';
-import { findContextFile, loadContext } from './context';
-import { registerDevice, sendHeartbeat } from './cloud';
-import * as fs from 'fs';
-import * as path from 'path';
-import * as http from 'http';
-import * as os from 'os';
+import * as fs      from 'fs';
+import * as path    from 'path';
+import * as http    from 'http';
+import * as os      from 'os';
+import * as readline from 'readline';
+import { execSync, spawnSync } from 'child_process';
 
-const AGENT_PORT = 7432;
-const command = process.argv[2] || 'start';
+import { readHardwareIdentity } from './hardware';
+import { registerDevice }       from './cloud';
+import {
+  saveConfig,
+  configExists,
+  installService,
+  uninstallService,
+  stopService,
+  restartService,
+  isServiceInstalled,
+} from './service';
+
+const AGENT_PORT  = 7432;
+const CONFIG_DIR  = path.join(os.homedir(), '.habi');
+const LOG_PATH    = path.join(CONFIG_DIR, 'habi.log');
+const HABI_VERSION = '0.1.5';
+
+// ── Arg parsing ─────────────────────────────────────────────────────────────
+const command = process.argv[2] || 'help';
 
-// Parse --key and --name arguments (works on all platforms)
 const keyArgIdx = process.argv.indexOf('--key');
 if (keyArgIdx !== -1 && process.argv[keyArgIdx + 1]) {
   process.env.HABI_KEY = process.argv[keyArgIdx + 1];
@@ -31,8 +51,8 @@ if (nameArgIdx !== -1 && process.argv[nameArgIdx + 1]) {
   process.env.HABI_DEVICE_NAME = process.argv[nameArgIdx + 1];
 }
 
-// ── Utility: ping the running agent ────────────────────────────────────────
-function pingAgent(p: string = '/health'): Promise<Record<string, unknown>> {
+// ── Utility ──────────────────────────────────────────────────────────────────
+function pingAgent(p = '/health'): Promise<Record<string, unknown>> {
   return new Promise((resolve, reject) => {
     const req = http.get(
       { host: '127.0.0.1', port: AGENT_PORT, path: p },
@@ -41,7 +61,7 @@ function pingAgent(p: string = '/health'): Promise<Record<string, unknown>> {
         res.on('data', (chunk: Buffer) => body += chunk.toString());
         res.on('end', () => {
           try { resolve(JSON.parse(body)); }
-          catch { reject(new Error('Invalid JSON from agent')); }
+          catch { reject(new Error('Invalid JSON')); }
         });
       }
     );
@@ -50,225 +70,310 @@ function pingAgent(p: string = '/health'): Promise<Record<string, unknown>> {
   });
 }
 
-async function cmdStart() {
-  console.log('\n┌─────────────────────────────────────────────┐');
-  console.log('│  HABI Local Identity Agent v0.1.0           │');
-  console.log('│  On Spot Solutions LLC — hardware-first AI  │');
-  console.log('└─────────────────────────────────────────────┘\n');
+function ask(question: string): Promise<string> {
+  return new Promise((resolve) => {
+    const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+    rl.question(question, (ans) => { rl.close(); resolve(ans.trim()); });
+  });
+}
+
+function banner() {
+  console.log(`
+┌─────────────────────────────────────────────────┐
+│  HABI Local Identity Agent v${HABI_VERSION}             │
+│  On Spot Solutions LLC — hardware-first AI       │
+│  imhabi.com                                      │
+└─────────────────────────────────────────────────┘`);
+}
+
+// ── Claude Code detection & MCP injection ───────────────────────────────────
+function claudeCodePaths(): string[] {
+  const home = os.homedir();
+  return [
+    path.join(home, '.claude', 'settings.json'),
+    path.join(home, 'AppData', 'Roaming', 'Claude', 'settings.json'),   // Windows
+    path.join(home, 'Library', 'Application Support', 'Claude', 'settings.json'), // macOS alt
+  ];
+}
+
+function findClaudeSettings(): string | null {
+  return claudeCodePaths().find(p => fs.existsSync(p)) || null;
+}
 
-  // Check if already running
+function isClaudeCodeInstalled(): boolean {
   try {
-    await pingAgent();
-    console.log('⚠  HABI agent is already running on port 7432.');
-    console.log('   Run: habi-agent status  to check it.');
-    process.exit(0);
-  } catch {
-    // Not running — proceed
+    const r = spawnSync(os.platform() === 'win32' ? 'where' : 'which', ['claude'], { encoding: 'utf8' });
+    return r.status === 0 && r.stdout.trim().length > 0;
+  } catch { return false; }
+}
+
+function injectMcpConfig(habiKey: string): { injected: boolean; path: string } {
+  const settingsPath = findClaudeSettings() || claudeCodePaths()[0];
+  const settingsDir  = path.dirname(settingsPath);
+
+  if (!fs.existsSync(settingsDir)) fs.mkdirSync(settingsDir, { recursive: true });
+
+  let existing: Record<string, unknown> = {};
+  if (fs.existsSync(settingsPath)) {
+    try { existing = JSON.parse(fs.readFileSync(settingsPath, 'utf8')); }
+    catch { existing = {}; }
   }
 
-  const server = new HabiLocalServer();
+  const mcpServers = (existing.mcpServers || {}) as Record<string, unknown>;
 
-  // Graceful shutdown
-  const shutdown = () => {
-    console.log('\n[HABI] Shutting down agent...');
-    server.stop();
-    process.exit(0);
+  // Inject or update HABI entry
+  mcpServers['habi'] = {
+    type: 'url',
+    url:  'https://dokcixjitwihtbdlkbdb.supabase.co/functions/v1/habi-mcp',
+    headers: { 'x-habi-key': habiKey },
   };
-  process.on('SIGINT', shutdown);
-  process.on('SIGTERM', shutdown);
 
+  existing.mcpServers = mcpServers;
+  fs.writeFileSync(settingsPath, JSON.stringify(existing, null, 2));
+  return { injected: true, path: settingsPath };
+}
+
+// ── COMMANDS ─────────────────────────────────────────────────────────────────
+
+async function cmdStart() {
+  banner();
+
+  const habiKey    = process.env.HABI_KEY || '';
+  const deviceName = process.env.HABI_DEVICE_NAME || os.hostname();
+
+  if (!habiKey) {
+    console.error('\n✗ --key is required. Get your key from imhabi.com\n');
+    process.exit(1);
+  }
+
+  console.log('\n[HABI] Reading hardware identity...');
+  const hw = readHardwareIdentity();
+  console.log(`[HABI] Hardware: ${hw.fingerprint} (${hw.type}) on ${hw.iface}`);
+
+  // ── 1. Register device ───────────────────────────────────────────────────
+  console.log(`[HABI] Registering "${deviceName}" with imhabi.com...`);
+  process.env.HABI_KEY         = habiKey;
+  process.env.HABI_DEVICE_NAME = deviceName;
+
+  const cloudState = await registerDevice({
+    friendlyName:        deviceName,
+    hardwareFingerprint: hw.fingerprint,
+    fingerprintType:     hw.type,
+    agentVersion:        HABI_VERSION,
+    osPlatform:          os.platform(),
+  });
+
+  if (cloudState) {
+    console.log(`[HABI] ✓ "${deviceName}" registered — visible on imhabi.com`);
+  } else {
+    console.error('[HABI] ✗ Registration failed — check your key and internet connection');
+    process.exit(1);
+  }
+
+  // ── 2. Save config for daemon ────────────────────────────────────────────
+  saveConfig(habiKey, deviceName);
+  console.log(`[HABI] ✓ Config saved to ~/.habi/config.json`);
+
+  // ── 3. Install background service ───────────────────────────────────────
+  console.log(`[HABI] Installing background service...`);
   try {
-    await server.start();
+    await installService();
+    console.log(`[HABI] ✓ Background service installed — runs at every login`);
+    console.log(`[HABI] ✓ You can close this terminal. HABI keeps running.`);
+  } catch (err) {
+    console.error(`[HABI] ✗ Service install failed: ${err}`);
+    console.log(`[HABI]   You can still run manually: npx habi-agent@latest daemon`);
+  }
 
-    // ── Cloud registration ──────────────────────────────────────────────
-    const hw = readHardwareIdentity();
-    const habiKey = process.env.HABI_KEY || process.env.HABI_OWNER_EMAIL || "";
-    const friendlyName = process.env.HABI_DEVICE_NAME || os.hostname();
+  // ── 4. Claude Code detection ─────────────────────────────────────────────
+  console.log('\n[HABI] Checking for Claude Code...');
+  const claudeInstalled = isClaudeCodeInstalled();
 
-    if (!habiKey) {
-      console.log('[HABI Cloud] HABI_OWNER_EMAIL not set — running in local-only mode.');
-      console.log('[HABI Cloud] Set export HABI_OWNER_EMAIL=you@example.com to enable cloud registration.\n');
-    } else {
-      console.log(`[HABI Cloud] Registering "${friendlyName}" via HABI key...`);
-      const cloudState = await registerDevice({
-        friendlyName,
-        hardwareFingerprint: hw.fingerprint,
-        fingerprintType: hw.type,
-        agentVersion: '0.1.0',
-        osPlatform: os.platform(),
-      });
-
-      if (cloudState) {
-        console.log(`[HABI Cloud] ✓ Device registered — ID: ${cloudState.deviceId.slice(0, 8)}...`);
-        console.log(`[HABI Cloud] ✓ Session token active until ${new Date(cloudState.sessionTokenExpiresAt).toLocaleTimeString()}\n`);
-
-        // ── Heartbeat loop every 30s ──────────────────────────────────
-        let currentToken = cloudState.sessionToken;
-        setInterval(async () => {
-          const hb = await sendHeartbeat({
-            deviceId: cloudState.deviceId,
-            sessionToken: currentToken,
-            hardwareFingerprint: hw.fingerprint,
-          });
-          if (hb) {
-            currentToken = hb.sessionToken;
-            if (hb.pendingJobs && hb.pendingJobs.length > 0) {
-              console.log(`[HABI Cloud] ${hb.pendingJobs.length} pending job(s) received from dispatch queue.`);
-              // Job execution handled in Phase 2 — log for now
-              for (const job of hb.pendingJobs) {
-                console.log(`[HABI Cloud]   Job ${job.id.slice(0, 8)}: ${JSON.stringify(job.payload).slice(0, 80)}`);
-              }
-            }
-          }
-        }, 30000);
+  if (!claudeInstalled) {
+    console.log('[HABI] Claude Code not detected.');
+    const answer = await ask('[HABI] Install Claude Code now to enable MCP enforcement? (Y/n) ');
+    if (answer.toLowerCase() !== 'n') {
+      console.log('[HABI] Installing Claude Code (npm install -g @anthropic-ai/claude-code)...');
+      try {
+        execSync('npm install -g @anthropic-ai/claude-code', { stdio: 'inherit' });
+        console.log('[HABI] ✓ Claude Code installed');
+        const mcp = injectMcpConfig(habiKey);
+        console.log(`[HABI] ✓ HABI MCP server configured → ${mcp.path}`);
+      } catch (err) {
+        console.error(`[HABI] ✗ Claude Code install failed: ${err}`);
+        console.log('[HABI]   Install manually: npm install -g @anthropic-ai/claude-code');
       }
+    } else {
+      console.log('[HABI] Skipping Claude Code install. MCP enforcement disabled.');
     }
-
-    // Keep alive
-    await new Promise(() => {});
-  } catch (err) {
-    console.error(`\n✗ HABI agent failed to start:\n  ${err}\n`);
-    process.exit(1);
+  } else {
+    console.log('[HABI] ✓ Claude Code detected');
+    const mcp = injectMcpConfig(habiKey);
+    console.log(`[HABI] ✓ HABI MCP server configured → ${mcp.path}`);
   }
+
+  // ── 5. Done ──────────────────────────────────────────────────────────────
+  console.log(`
+┌─────────────────────────────────────────────────────────────┐
+│  ✓ HABI is running as a background service                  │
+│                                                             │
+│  Machine:    ${(deviceName + ' ').padEnd(44)} │
+│  Hardware:   ${(hw.fingerprint + ' ').padEnd(44)} │
+│  Dashboard:  https://imhabi.com                             │
+│  Logs:       ~/.habi/habi.log                               │
+│                                                             │
+│  Commands:                                                  │
+│    habi-agent status     — check service health             │
+│    habi-agent stop       — stop the service                 │
+│    habi-agent restart    — restart the service              │
+│    habi-agent uninstall  — remove the service               │
+└─────────────────────────────────────────────────────────────┘
+`);
+  process.exit(0);
 }
 
 async function cmdStatus() {
-  try {
-    const health = await pingAgent('/health');
-    console.log('\n✓ HABI agent is RUNNING');
-    console.log(`  Port:    ${AGENT_PORT}`);
-    console.log(`  Version: ${health.version || 'unknown'}`);
+  banner();
+  console.log('');
 
+  const installed = isServiceInstalled();
+  console.log(`  Service:   ${installed ? '✓ INSTALLED (runs at login)' : '✗ Not installed'}`);
+
+  // Check if agent is responding
+  try {
+    const health   = await pingAgent('/health');
     const identity = await pingAgent('/identity') as Record<string, unknown>;
     const hw = identity.hardware as Record<string, unknown>;
-    const sess = identity.session as Record<string, unknown>;
+    console.log(`  Agent:     ✓ RUNNING on port ${AGENT_PORT}`);
+    console.log(`  Version:   ${health.version || HABI_VERSION}`);
     if (hw) {
-      console.log(`  Hardware: ${hw.fingerprint} (${hw.type})`);
-      console.log(`  Platform: ${hw.platform} / ${hw.hostname}`);
-    }
-    if (sess) {
-      const expiresAt = new Date(sess.expiresAt as number);
-      const active = sess.active ? '✓ ACTIVE' : '✗ EXPIRED';
-      console.log(`  Session:  ${(sess.fingerprint as string)?.slice(0, 16)}...`);
-      console.log(`  Status:   ${active} (expires ${expiresAt.toLocaleTimeString()})`);
+      console.log(`  Hardware:  ${hw.fingerprint} (${hw.type})`);
+      console.log(`  Platform:  ${hw.platform} / ${hw.hostname}`);
     }
-    console.log();
   } catch {
-    console.log('\n✗ HABI agent is NOT RUNNING');
-    console.log('  Start it with: npx ts-node src/cli.ts start\n');
-    process.exit(1);
+    console.log(`  Agent:     ✗ NOT RESPONDING on port ${AGENT_PORT}`);
+    if (installed) console.log(`             (may still be starting — check ~/.habi/habi.log)`);
   }
+
+  // Claude Code MCP check
+  const claudeSettings = findClaudeSettings();
+  if (claudeSettings) {
+    try {
+      const s = JSON.parse(fs.readFileSync(claudeSettings, 'utf8'));
+      const habiMcp = s.mcpServers?.habi;
+      console.log(`  Claude MCP: ${habiMcp ? '✓ CONFIGURED' : '✗ Not configured'}`);
+    } catch {
+      console.log(`  Claude MCP: ✗ Could not read settings`);
+    }
+  } else {
+    console.log(`  Claude MCP: — Claude Code not detected`);
+  }
+
+  if (fs.existsSync(LOG_PATH)) {
+    console.log(`\n  Log file: ${LOG_PATH}`);
+    const lines = fs.readFileSync(LOG_PATH, 'utf8').split('\n').filter(Boolean).slice(-5);
+    console.log('  Last 5 log entries:');
+    lines.forEach(l => console.log(`    ${l}`));
+  }
+  console.log('');
 }
 
-async function cmdIdentity() {
-  try {
-    const hw = readHardwareIdentity();
-    const ctx = (() => {
-      try {
-        const p = findContextFile();
-        return p ? loadContext(p) : null;
-      } catch { return null; }
-    })();
+function cmdStop() {
+  console.log('[HABI] Stopping service...');
+  stopService();
+  console.log('[HABI] ✓ Service stopped. Run: habi-agent restart  to start again.');
+}
+
+async function cmdRestart() {
+  console.log('[HABI] Restarting service...');
+  restartService();
+  console.log('[HABI] ✓ Service restarted.');
+}
+
+async function cmdUninstall() {
+  console.log('[HABI] Uninstalling HABI background service...');
+  const answer = await ask('[HABI] This will stop HABI and remove it from startup. Continue? (y/N) ');
+  if (answer.toLowerCase() !== 'y') {
+    console.log('[HABI] Cancelled.');
+    return;
+  }
+  await uninstallService();
+
+  // Also remove MCP config from Claude Code if present
+  const claudeSettings = findClaudeSettings();
+  if (claudeSettings && fs.existsSync(claudeSettings)) {
+    try {
+      const s = JSON.parse(fs.readFileSync(claudeSettings, 'utf8'));
+      if (s.mcpServers?.habi) {
+        delete s.mcpServers.habi;
+        fs.writeFileSync(claudeSettings, JSON.stringify(s, null, 2));
+        console.log('[HABI] ✓ Removed HABI from Claude Code MCP config');
+      }
+    } catch { /* ignore */ }
+  }
+
+  console.log('[HABI] ✓ Service uninstalled');
+  console.log('[HABI]   Config and logs remain at ~/.habi/ — delete manually if desired');
+}
 
-    const contextHash = ctx?.contextHash ?? sha256('no-context');
-    const fingerprint = generateSessionFingerprint(hw.fingerprint, contextHash, Date.now());
+function cmdLogs() {
+  if (!fs.existsSync(LOG_PATH)) {
+    console.log('[HABI] No log file found at ' + LOG_PATH);
+    return;
+  }
+  const lines = fs.readFileSync(LOG_PATH, 'utf8').split('\n').filter(Boolean);
+  const last50 = lines.slice(-50);
+  console.log(`\n── HABI Log (last ${last50.length} lines) ──────────────────────\n`);
+  last50.forEach(l => console.log(l));
+  console.log('\n── End of log ──────────────────────────────────────────────\n');
+}
 
-    console.log('\n┌─ HABI Hardware Identity ──────────────────────────────────┐');
+function cmdIdentity() {
+  try {
+    const hw = readHardwareIdentity();
+    console.log('\n┌─ HABI Hardware Identity ───────────────────────────────────┐');
     console.log(`│  Hardware ID:  ${hw.fingerprint}`);
     console.log(`│  Type:         ${hw.type}`);
     console.log(`│  Interface:    ${hw.iface}`);
     console.log(`│  Platform:     ${hw.platform}`);
     console.log(`│  Hostname:     ${hw.hostname}`);
-    console.log(`│  Context:      ${ctx ? ctx.context.project : 'none loaded'}`);
-    console.log(`│  Context Hash: ${ctx ? ctx.contextHash.slice(0, 24) + '...' : 'n/a'}`);
-    console.log(`│  Session FP:   ${fingerprint.slice(0, 24)}...`);
     console.log('└────────────────────────────────────────────────────────────┘\n');
   } catch (err) {
-    console.error(`\n✗ Could not read hardware identity:\n  ${err}\n`);
+    console.error(`\n✗ Could not read hardware identity: ${err}\n`);
     process.exit(1);
   }
 }
 
-async function cmdInit() {
-  const hw = readHardwareIdentity();
-  const projectName = process.argv[3] || path.basename(process.cwd());
-
-  const contextContent = {
-    project: projectName,
-    version: '1.0.0',
-    owner: process.env.HABI_OWNER_EMAIL || process.env.USER || process.env.USERNAME || 'owner',
-    allowedMachines: [hw.fingerprint],
-    allowedOperations: {
-      paths: [`${process.cwd()}/**`],
-      cloudAccounts: [],
-      databases: [],
-      repos: [],
-    },
-    deniedOperations: [
-      'DROP TABLE',
-      'DROP DATABASE',
-      'rm -rf /',
-      'DELETE FROM audit_log',
-      'truncate audit_log',
-    ],
-  };
+function cmdHelp() {
+  console.log(`
+HABI Agent v${HABI_VERSION} — On Spot Solutions LLC
+Hardware-Anchored Binding Infrastructure
 
-  const contextHash = sha256(JSON.stringify(contextContent, null, 2));
-  const fullContext = { ...contextContent, contextHash };
-  const outPath = path.join(process.cwd(), 'habi-context.json');
+Usage:
+  npx habi-agent@latest start --key <key> --name <name>
+      Register device, install background service, configure Claude Code
 
-  if (fs.existsSync(outPath)) {
-    console.log(`\n⚠  Context file already exists: ${outPath}`);
-    console.log('   Delete it first if you want to regenerate.\n');
-    process.exit(1);
-  }
-
-  fs.writeFileSync(outPath, JSON.stringify(fullContext, null, 2), 'utf8');
+  habi-agent status         Show service status and agent health
+  habi-agent stop           Stop the background service
+  habi-agent restart        Restart the background service
+  habi-agent uninstall      Remove the background service and MCP config
+  habi-agent logs           Show the last 50 log entries
+  habi-agent identity       Print hardware fingerprint
 
-  console.log('\n✓ HABI Project Context File created');
-  console.log(`  File:     ${outPath}`);
-  console.log(`  Project:  ${projectName}`);
-  console.log(`  Machine:  ${hw.fingerprint} (${hw.type})`);
-  console.log(`  Hash:     ${contextHash.slice(0, 24)}...`);
-  console.log('\n⚠  IMPORTANT: Add habi-context.json to your .gitignore');
-  console.log('   This file contains your machine identity — do not commit it.\n');
-}
-
-async function cmdVerify() {
-  try {
-    const data = await pingAgent('/verify') as Record<string, unknown>;
-    if (data.verified) {
-      console.log('\n✓ Session VERIFIED');
-      console.log(`  Hardware ID: ${data.hardwareId}`);
-      console.log(`  Fingerprint: ${(data.fingerprint as string)?.slice(0, 24)}...`);
-      console.log(`  Expires in:  ${data.expiresIn}s\n`);
-    } else {
-      console.log('\n✗ Session NOT VERIFIED');
-      console.log(`  Reason: ${data.reason || data.error}\n`);
-      process.exit(1);
-    }
-  } catch {
-    console.log('\n✗ HABI agent not running. Start with: npx ts-node src/cli.ts start\n');
-    process.exit(1);
-  }
+Dashboard: https://imhabi.com
+`);
 }
 
-// ── Dispatch ────────────────────────────────────────────────────────────────
+// ── Dispatch ──────────────────────────────────────────────────────────────────
 (async () => {
   switch (command) {
-    case 'start':    await cmdStart();    break;
-    case 'status':   await cmdStatus();   break;
-    case 'identity': await cmdIdentity(); break;
-    case 'init':     await cmdInit();     break;
-    case 'verify':   await cmdVerify();   break;
-    default:
-      console.log(`\nHABI Agent v0.1.0 — On Spot Solutions LLC`);
-      console.log(`Usage: npx ts-node src/cli.ts <command>`);
-      console.log(`\nCommands:`);
-      console.log(`  start     Start the local identity daemon on port 7432`);
-      console.log(`  status    Show agent status and hardware identity`);
-      console.log(`  identity  Print hardware fingerprint`);
-      console.log(`  init      Generate a Project Context File for this directory`);
-      console.log(`  verify    Verify the current session\n`);
+    case 'start':     await cmdStart();     break;
+    case 'stop':            cmdStop();      break;
+    case 'restart':   await cmdRestart();   break;
+    case 'uninstall': await cmdUninstall(); break;
+    case 'status':    await cmdStatus();    break;
+    case 'logs':            cmdLogs();      break;
+    case 'identity':        cmdIdentity();  break;
+    default:                cmdHelp();      break;
   }
 })();