agent-working-memory 0.4.1 → 0.4.3

package/src/cli.ts

@@ -1,355 +1,385 @@
-#!/usr/bin/env node
-
-/**
- * CLI entrypoint for AgentWorkingMemory.
- *
- * Commands:
- *   awm setup    — configure MCP for the current project
- *   awm mcp      — start the MCP server (called by Claude Code)
- *   awm serve    — start the HTTP API server
- *   awm health   — check if a running server is healthy
- */
-
-import { readFileSync, writeFileSync, mkdirSync, existsSync } from 'node:fs';
-import { resolve, basename, join, dirname } from 'node:path';
-import { execSync } from 'node:child_process';
-import { randomBytes } from 'node:crypto';
-import { fileURLToPath } from 'node:url';
-import { homedir as osHomedir } from 'node:os';
-
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = dirname(__filename);
-
-// Load .env if present
-try {
-  const envPath = resolve(process.cwd(), '.env');
-  const envContent = readFileSync(envPath, 'utf-8');
-  for (const line of envContent.split('\n')) {
-    const trimmed = line.trim();
-    if (!trimmed || trimmed.startsWith('#')) continue;
-    const eqIdx = trimmed.indexOf('=');
-    if (eqIdx === -1) continue;
-    const key = trimmed.slice(0, eqIdx).trim();
-    const val = trimmed.slice(eqIdx + 1).trim().replace(/^["']|["']$/g, '');
-    if (!process.env[key]) process.env[key] = val;
-  }
-} catch { /* No .env file */ }
-
-const args = process.argv.slice(2);
-const command = args[0];
-
-function printUsage() {
-  console.log(`
-AgentWorkingMemory — Cognitive memory for AI agents
-
-Usage:
-  awm setup [--global] [--agent-id <id>] [--db-path <path>] [--no-claude-md]
-            [--no-hooks] [--hook-port <port>]       Configure MCP for Claude Code
-  awm mcp                                           Start MCP server (used by Claude Code)
-  awm serve [--port <port>]                         Start HTTP API server
-  awm health [--port <port>]                        Check server health
-
-Setup:
-  awm setup --global     Recommended. Writes ~/.mcp.json so AWM is available
-                         in every project — one brain across all your work.
-
-  awm setup              Project-level. Writes .mcp.json in the current directory
-                         and appends workflow instructions to CLAUDE.md.
-
-  --no-claude-md    Skip CLAUDE.md modification
-  --no-hooks        Skip hook installation (no auto-checkpoint)
-  --hook-port PORT  Sidecar port for hooks (default: 8401)
-
-  Restart Claude Code after setup to pick up the new MCP server.
-`.trim());
-}
-
-// ─── SETUP ──────────────────────────────────────
-
-function setup() {
-  const cwd = process.cwd();
-  const projectName = basename(cwd).toLowerCase().replace(/[^a-z0-9-]/g, '-');
-
-  // Parse flags
-  let agentId = projectName;
-  let dbPath: string | null = null;
-  let skipClaudeMd = false;
-  let isGlobal = false;
-  let skipHooks = false;
-  let hookPort = '8401';
-
-  for (let i = 1; i < args.length; i++) {
-    if (args[i] === '--agent-id' && args[i + 1]) {
-      agentId = args[++i];
-    } else if (args[i] === '--db-path' && args[i + 1]) {
-      dbPath = args[++i];
-    } else if (args[i] === '--no-claude-md') {
-      skipClaudeMd = true;
-    } else if (args[i] === '--no-hooks') {
-      skipHooks = true;
-    } else if (args[i] === '--hook-port' && args[i + 1]) {
-      hookPort = args[++i];
-    } else if (args[i] === '--global') {
-      isGlobal = true;
-      agentId = 'claude'; // unified agent ID for global setup
-    }
-  }
-
-  // Find the package root (where src/mcp.ts lives)
-  const packageRoot = resolve(__dirname, '..');
-  const mcpScript = join(packageRoot, 'src', 'mcp.ts');
-  const mcpDist = join(packageRoot, 'dist', 'mcp.js');
-
-  // Determine DB path — default to <awm-root>/data/memory.db (shared across projects)
-  if (!dbPath) {
-    dbPath = join(packageRoot, 'data', 'memory.db');
-  }
-  const dbDir = dirname(dbPath);
-
-  // Ensure data directory exists
-  if (!existsSync(dbDir)) {
-    mkdirSync(dbDir, { recursive: true });
-    console.log(`Created data directory: ${dbDir}`);
-  }
-
-  // Generate hook secret (or reuse existing one)
-  let hookSecret = '';
-  const secretPath = join(dirname(dbPath!), '.awm-hook-secret');
-  if (existsSync(secretPath)) {
-    hookSecret = readFileSync(secretPath, 'utf-8').trim();
-  }
-  if (!hookSecret) {
-    hookSecret = randomBytes(32).toString('hex');
-    mkdirSync(dirname(secretPath), { recursive: true });
-    writeFileSync(secretPath, hookSecret + '\n');
-  }
-
-  // Determine command based on platform and whether dist exists
-  const isWindows = process.platform === 'win32';
-  const hasDist = existsSync(mcpDist);
-
-  const envVars: Record<string, string> = {
-    AWM_DB_PATH: (isWindows ? dbPath!.replace(/\\/g, '/') : dbPath!),
-    AWM_AGENT_ID: agentId,
-    AWM_HOOK_PORT: hookPort,
-    AWM_HOOK_SECRET: hookSecret,
-  };
-
-  let mcpConfig: { command: string; args: string[]; env: Record<string, string> };
-
-  if (hasDist) {
-    mcpConfig = {
-      command: 'node',
-      args: [mcpDist.replace(/\\/g, '/')],
-      env: envVars,
-    };
-  } else if (isWindows) {
-    mcpConfig = {
-      command: 'cmd',
-      args: ['/c', 'npx', 'tsx', mcpScript.replace(/\\/g, '/')],
-      env: envVars,
-    };
-  } else {
-    mcpConfig = {
-      command: 'npx',
-      args: ['tsx', mcpScript],
-      env: envVars,
-    };
-  }
-
-  // Read or create .mcp.json
-  const mcpJsonPath = isGlobal ? join(osHomedir(), '.mcp.json') : join(cwd, '.mcp.json');
-  let existing: any = { mcpServers: {} };
-  if (existsSync(mcpJsonPath)) {
-    try {
-      existing = JSON.parse(readFileSync(mcpJsonPath, 'utf-8'));
-      if (!existing.mcpServers) existing.mcpServers = {};
-    } catch {
-      existing = { mcpServers: {} };
-    }
-  }
-
-  existing.mcpServers['agent-working-memory'] = mcpConfig;
-  writeFileSync(mcpJsonPath, JSON.stringify(existing, null, 2) + '\n');
-
-  // Auto-append CLAUDE.md snippet unless --no-claude-md
-  let claudeMdAction = '';
-  const claudeMdSnippet = `
-
-## Memory (AWM)
-You have persistent memory via the agent-working-memory MCP server.
-- At conversation start: ALWAYS call memory_restore to recover previous context
-- When you learn something important: call memory_write PROACTIVELY — do not wait to be asked
-- When you need past context: call memory_recall (use the "query" parameter)
-- Before long operations: ALWAYS call memory_task_begin to checkpoint and recall context
-- When finishing a task: call memory_task_end with a summary of what was accomplished
-- After using a recalled memory: call memory_feedback (useful/not-useful)
-- To retract incorrect info: call memory_retract
-- To manage tasks: call memory_task_add, memory_task_update, memory_task_list, memory_task_next
-- Auto-checkpoint: hooks auto-save on context compaction and session end (no action needed)
-`;
-
-  // For global: write to ~/.claude/CLAUDE.md (loaded by Claude Code in every session)
-  // For project: write to ./CLAUDE.md in the current directory
-  const claudeMdPath = isGlobal
-    ? join(osHomedir(), '.claude', 'CLAUDE.md')
-    : join(cwd, 'CLAUDE.md');
-
-  // Ensure parent directory exists (for ~/.claude/CLAUDE.md)
-  const claudeMdDir = dirname(claudeMdPath);
-  if (!existsSync(claudeMdDir)) {
-    mkdirSync(claudeMdDir, { recursive: true });
-  }
-
-  if (skipClaudeMd) {
-    claudeMdAction = '  CLAUDE.md: skipped (--no-claude-md)';
-  } else if (existsSync(claudeMdPath)) {
-    const content = readFileSync(claudeMdPath, 'utf-8');
-    if (content.includes('## Memory (AWM)')) {
-      claudeMdAction = '  CLAUDE.md: already has AWM section (skipped)';
-    } else {
-      writeFileSync(claudeMdPath, content.trimEnd() + '\n' + claudeMdSnippet);
-      claudeMdAction = '  CLAUDE.md: appended AWM workflow section';
-    }
-  } else {
-    const title = isGlobal ? '# Global Instructions' : `# ${basename(cwd)}`;
-    writeFileSync(claudeMdPath, `${title}\n${claudeMdSnippet}`);
-    claudeMdAction = '  CLAUDE.md: created with AWM workflow section';
-  }
-
-  // --- Hook configuration ---
-  let hookAction = '';
-  if (skipHooks) {
-    hookAction = '  Hooks: skipped (--no-hooks)';
-  } else {
-    // Write hooks to Claude Code settings (~/.claude/settings.json)
-    const settingsPath = join(osHomedir(), '.claude', 'settings.json');
-    let settings: any = {};
-    if (existsSync(settingsPath)) {
-      try {
-        settings = JSON.parse(readFileSync(settingsPath, 'utf-8'));
-      } catch {
-        settings = {};
-      }
-    }
-
-    if (!settings.hooks) settings.hooks = {};
-
-    const hookUrl = `http://127.0.0.1:${hookPort}/hooks/checkpoint`;
-
-    // PreCompact — auto-checkpoint before context compaction
-    settings.hooks.PreCompact = [{
-      matcher: '',
-      hooks: [{
-        type: 'command',
-        command: `curl -sf -X POST ${hookUrl} -H "Content-Type: application/json" -H "Authorization: Bearer ${hookSecret}" -d "{\\"hook_event_name\\":\\"PreCompact\\"}" --max-time 5`,
-        timeout: 10,
-      }],
-    }];
-
-    // SessionEnd — auto-checkpoint on session close
-    settings.hooks.SessionEnd = [{
-      matcher: '',
-      hooks: [{
-        type: 'command',
-        command: `curl -sf -X POST ${hookUrl} -H "Content-Type: application/json" -H "Authorization: Bearer ${hookSecret}" -d "{\\"hook_event_name\\":\\"SessionEnd\\"}" --max-time 5`,
-        timeout: 10,
-      }],
-    }];
-
-    // Ensure settings directory exists
-    const settingsDir = dirname(settingsPath);
-    if (!existsSync(settingsDir)) {
-      mkdirSync(settingsDir, { recursive: true });
-    }
-
-    writeFileSync(settingsPath, JSON.stringify(settings, null, 2) + '\n');
-    hookAction = `  Hooks: PreCompact + SessionEnd → auto-checkpoint (port ${hookPort})`;
-  }
-
-  const scope = isGlobal ? 'globally (all projects)' : cwd;
-  console.log(`
-AWM configured ${isGlobal ? 'globally' : 'for: ' + cwd}
-
-  Agent ID:    ${agentId}
-  DB path:     ${dbPath}
-  MCP config:  ${mcpJsonPath}
-  Hook port:   ${hookPort}
-  Hook secret: ${hookSecret.slice(0, 8)}...
-${claudeMdAction}
-${hookAction}
-
-Next steps:
-  1. Restart Claude Code to pick up the MCP server
-  2. The memory tools will appear automatically
-  3. Hooks auto-checkpoint on context compaction and session end${isGlobal ? '\n  4. One brain across all your projects — no per-project setup needed' : ''}
-`.trim());
-}
-
-// ─── MCP ──────────────────────────────────────
-
-async function mcp() {
-  // Dynamic import to avoid loading heavy deps for setup/health commands
-  await import('./mcp.js');
-}
-
-// ─── SERVE ──────────────────────────────────────
-
-async function serve() {
-  // Parse --port flag
-  for (let i = 1; i < args.length; i++) {
-    if (args[i] === '--port' && args[i + 1]) {
-      process.env.AWM_PORT = args[++i];
-    }
-  }
-  await import('./index.js');
-}
-
-// ─── HEALTH ──────────────────────────────────────
-
-function health() {
-  let port = '8400';
-  for (let i = 1; i < args.length; i++) {
-    if (args[i] === '--port' && args[i + 1]) {
-      port = args[++i];
-    }
-  }
-
-  try {
-    const result = execSync(`curl -sf http://localhost:${port}/health`, {
-      encoding: 'utf8',
-      timeout: 5000,
-    });
-    const data = JSON.parse(result);
-    console.log(`OK — v${data.version} (${data.timestamp})`);
-  } catch {
-    console.error(`Cannot reach AWM server on port ${port}`);
-    process.exit(1);
-  }
-}
-
-// ─── Dispatch ──────────────────────────────────────
-
-switch (command) {
-  case 'setup':
-    setup();
-    break;
-  case 'mcp':
-    mcp();
-    break;
-  case 'serve':
-    serve();
-    break;
-  case 'health':
-    health();
-    break;
-  case '--help':
-  case '-h':
-  case undefined:
-    printUsage();
-    break;
-  default:
-    console.error(`Unknown command: ${command}`);
-    printUsage();
-    process.exit(1);
-}
+#!/usr/bin/env node
+// Copyright 2026 Robert Winter / Complete Ideas
+// SPDX-License-Identifier: Apache-2.0
+
+/**
+ * CLI entrypoint for AgentWorkingMemory.
+ *
+ * Commands:
+ *   awm setup    — configure MCP for the current project
+ *   awm mcp      — start the MCP server (called by Claude Code)
+ *   awm serve    — start the HTTP API server
+ *   awm health   — check if a running server is healthy
+ */
+
+import { readFileSync, writeFileSync, mkdirSync, existsSync } from 'node:fs';
+import { resolve, basename, join, dirname } from 'node:path';
+import { execSync } from 'node:child_process';
+import { randomBytes } from 'node:crypto';
+import { fileURLToPath } from 'node:url';
+import { homedir as osHomedir } from 'node:os';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+// Load .env if present
+try {
+  const envPath = resolve(process.cwd(), '.env');
+  const envContent = readFileSync(envPath, 'utf-8');
+  for (const line of envContent.split('\n')) {
+    const trimmed = line.trim();
+    if (!trimmed || trimmed.startsWith('#')) continue;
+    const eqIdx = trimmed.indexOf('=');
+    if (eqIdx === -1) continue;
+    const key = trimmed.slice(0, eqIdx).trim();
+    const val = trimmed.slice(eqIdx + 1).trim().replace(/^["']|["']$/g, '');
+    if (!process.env[key]) process.env[key] = val;
+  }
+} catch { /* No .env file */ }
+
+const args = process.argv.slice(2);
+const command = args[0];
+
+function printUsage() {
+  console.log(`
+AgentWorkingMemory — Cognitive memory for AI agents
+
+Usage:
+  awm setup [--global] [--agent-id <id>] [--db-path <path>] [--no-claude-md]
+            [--no-hooks] [--hook-port <port>]       Configure MCP for Claude Code
+  awm mcp                                           Start MCP server (used by Claude Code)
+  awm serve [--port <port>]                         Start HTTP API server
+  awm health [--port <port>]                        Check server health
+
+Setup:
+  awm setup --global     Recommended. Writes ~/.mcp.json so AWM is available
+                         in every project — one brain across all your work.
+
+  awm setup              Project-level. Writes .mcp.json in the current directory
+                         and appends workflow instructions to CLAUDE.md.
+
+  --no-claude-md    Skip CLAUDE.md modification
+  --no-hooks        Skip hook installation (no auto-checkpoint)
+  --hook-port PORT  Sidecar port for hooks (default: 8401)
+
+  Restart Claude Code after setup to pick up the new MCP server.
+`.trim());
+}
+
+// ─── SETUP ──────────────────────────────────────
+
+function setup() {
+  const cwd = process.cwd();
+  const projectName = basename(cwd).toLowerCase().replace(/[^a-z0-9-]/g, '-');
+
+  // Parse flags
+  let agentId = projectName;
+  let dbPath: string | null = null;
+  let skipClaudeMd = false;
+  let isGlobal = false;
+  let skipHooks = false;
+  let hookPort = '8401';
+
+  for (let i = 1; i < args.length; i++) {
+    if (args[i] === '--agent-id' && args[i + 1]) {
+      agentId = args[++i];
+    } else if (args[i] === '--db-path' && args[i + 1]) {
+      dbPath = args[++i];
+    } else if (args[i] === '--no-claude-md') {
+      skipClaudeMd = true;
+    } else if (args[i] === '--no-hooks') {
+      skipHooks = true;
+    } else if (args[i] === '--hook-port' && args[i + 1]) {
+      hookPort = args[++i];
+    } else if (args[i] === '--global') {
+      isGlobal = true;
+      agentId = 'claude'; // unified agent ID for global setup
+    }
+  }
+
+  // Find the package root (where src/mcp.ts lives)
+  const packageRoot = resolve(__dirname, '..');
+  const mcpScript = join(packageRoot, 'src', 'mcp.ts');
+  const mcpDist = join(packageRoot, 'dist', 'mcp.js');
+
+  // Determine DB path — default to <awm-root>/data/memory.db (shared across projects)
+  if (!dbPath) {
+    dbPath = join(packageRoot, 'data', 'memory.db');
+  }
+  const dbDir = dirname(dbPath);
+
+  // Ensure data directory exists
+  if (!existsSync(dbDir)) {
+    mkdirSync(dbDir, { recursive: true });
+    console.log(`Created data directory: ${dbDir}`);
+  }
+
+  // Generate hook secret (or reuse existing one)
+  let hookSecret = '';
+  const secretPath = join(dirname(dbPath!), '.awm-hook-secret');
+  if (existsSync(secretPath)) {
+    hookSecret = readFileSync(secretPath, 'utf-8').trim();
+  }
+  if (!hookSecret) {
+    hookSecret = randomBytes(32).toString('hex');
+    mkdirSync(dirname(secretPath), { recursive: true });
+    writeFileSync(secretPath, hookSecret + '\n');
+  }
+
+  // Determine command based on platform and whether dist exists
+  const isWindows = process.platform === 'win32';
+  const hasDist = existsSync(mcpDist);
+
+  const envVars: Record<string, string> = {
+    AWM_DB_PATH: (isWindows ? dbPath!.replace(/\\/g, '/') : dbPath!),
+    AWM_AGENT_ID: agentId,
+    AWM_HOOK_PORT: hookPort,
+    AWM_HOOK_SECRET: hookSecret,
+  };
+
+  let mcpConfig: { command: string; args: string[]; env: Record<string, string> };
+
+  if (hasDist) {
+    mcpConfig = {
+      command: 'node',
+      args: [mcpDist.replace(/\\/g, '/')],
+      env: envVars,
+    };
+  } else if (isWindows) {
+    mcpConfig = {
+      command: 'cmd',
+      args: ['/c', 'npx', 'tsx', mcpScript.replace(/\\/g, '/')],
+      env: envVars,
+    };
+  } else {
+    mcpConfig = {
+      command: 'npx',
+      args: ['tsx', mcpScript],
+      env: envVars,
+    };
+  }
+
+  // Read or create .mcp.json
+  const mcpJsonPath = isGlobal ? join(osHomedir(), '.mcp.json') : join(cwd, '.mcp.json');
+  let existing: any = { mcpServers: {} };
+  if (existsSync(mcpJsonPath)) {
+    try {
+      existing = JSON.parse(readFileSync(mcpJsonPath, 'utf-8'));
+      if (!existing.mcpServers) existing.mcpServers = {};
+    } catch {
+      existing = { mcpServers: {} };
+    }
+  }
+
+  existing.mcpServers['agent-working-memory'] = mcpConfig;
+  writeFileSync(mcpJsonPath, JSON.stringify(existing, null, 2) + '\n');
+
+  // Auto-append CLAUDE.md snippet unless --no-claude-md
+  let claudeMdAction = '';
+  const claudeMdSnippet = `
+
+## Memory (AWM)
+You have persistent memory via the agent-working-memory MCP server.
+
+### Lifecycle (always do these)
+- Session start: call memory_restore to recover previous context
+- Starting a task: call memory_task_begin (checkpoints + recalls relevant memories)
+- Finishing a task: call memory_task_end with a summary
+- Auto-checkpoint: hooks handle compaction, session end, and 15-min timer (no action needed)
+
+### Write memory when:
+- A project decision is made or changed
+- A root cause is discovered after debugging
+- A reusable implementation pattern is established
+- A user preference, constraint, or requirement is clarified
+- A prior assumption is found to be wrong
+- A significant piece of work is completed
+
+### Recall memory when:
+- Starting work on a new task or subsystem
+- Re-entering code you haven't touched recently
+- After a failed attempt — check if there's prior knowledge
+- Before refactoring or making architectural changes
+- When a topic comes up that you might have prior context on
+
+### Also:
+- After using a recalled memory: call memory_feedback (useful/not-useful)
+- To correct wrong info: call memory_retract
+- To track work items: memory_task_add, memory_task_update, memory_task_list, memory_task_next
+`;
+
+  // For global: write to ~/.claude/CLAUDE.md (loaded by Claude Code in every session)
+  // For project: write to ./CLAUDE.md in the current directory
+  const claudeMdPath = isGlobal
+    ? join(osHomedir(), '.claude', 'CLAUDE.md')
+    : join(cwd, 'CLAUDE.md');
+
+  // Ensure parent directory exists (for ~/.claude/CLAUDE.md)
+  const claudeMdDir = dirname(claudeMdPath);
+  if (!existsSync(claudeMdDir)) {
+    mkdirSync(claudeMdDir, { recursive: true });
+  }
+
+  if (skipClaudeMd) {
+    claudeMdAction = '  CLAUDE.md: skipped (--no-claude-md)';
+  } else if (existsSync(claudeMdPath)) {
+    const content = readFileSync(claudeMdPath, 'utf-8');
+    if (content.includes('## Memory (AWM)')) {
+      claudeMdAction = '  CLAUDE.md: already has AWM section (skipped)';
+    } else {
+      writeFileSync(claudeMdPath, content.trimEnd() + '\n' + claudeMdSnippet);
+      claudeMdAction = '  CLAUDE.md: appended AWM workflow section';
+    }
+  } else {
+    const title = isGlobal ? '# Global Instructions' : `# ${basename(cwd)}`;
+    writeFileSync(claudeMdPath, `${title}\n${claudeMdSnippet}`);
+    claudeMdAction = '  CLAUDE.md: created with AWM workflow section';
+  }
+
+  // --- Hook configuration ---
+  let hookAction = '';
+  if (skipHooks) {
+    hookAction = '  Hooks: skipped (--no-hooks)';
+  } else {
+    // Write hooks to Claude Code settings (~/.claude/settings.json)
+    const settingsPath = join(osHomedir(), '.claude', 'settings.json');
+    let settings: any = {};
+    if (existsSync(settingsPath)) {
+      try {
+        settings = JSON.parse(readFileSync(settingsPath, 'utf-8'));
+      } catch {
+        settings = {};
+      }
+    }
+
+    if (!settings.hooks) settings.hooks = {};
+
+    const hookUrl = `http://127.0.0.1:${hookPort}/hooks/checkpoint`;
+
+    // Stop — remind Claude to write/recall/switch tasks after each response
+    settings.hooks.Stop = [{
+      matcher: '',
+      hooks: [{
+        type: 'command',
+        command: 'echo "MEMORY: (1) Did you learn anything new? Call memory_write. (2) Are you about to work on a topic you might have prior knowledge about? Call memory_recall. (3) Switching tasks? Call memory_task_begin."',
+        timeout: 5,
+        async: true,
+      }],
+    }];
+
+    // PreCompact — auto-checkpoint before context compaction
+    settings.hooks.PreCompact = [{
+      matcher: '',
+      hooks: [{
+        type: 'command',
+        command: `curl -sf -X POST ${hookUrl} -H "Content-Type: application/json" -H "Authorization: Bearer ${hookSecret}" -d "{\\"hook_event_name\\":\\"PreCompact\\"}" --max-time 2`,
+        timeout: 3,
+      }],
+    }];
+
+    // SessionEnd — auto-checkpoint on session close (fast timeout to avoid cancellation)
+    settings.hooks.SessionEnd = [{
+      matcher: '',
+      hooks: [{
+        type: 'command',
+        command: `curl -sf -X POST ${hookUrl} -H "Content-Type: application/json" -H "Authorization: Bearer ${hookSecret}" -d "{\\"hook_event_name\\":\\"SessionEnd\\"}" --max-time 2`,
+        timeout: 3,
+      }],
+    }];
+
+    // Ensure settings directory exists
+    const settingsDir = dirname(settingsPath);
+    if (!existsSync(settingsDir)) {
+      mkdirSync(settingsDir, { recursive: true });
+    }
+
+    writeFileSync(settingsPath, JSON.stringify(settings, null, 2) + '\n');
+    hookAction = `  Hooks: Stop (memory reminder) + PreCompact + SessionEnd → auto-checkpoint (port ${hookPort})`;
+  }
+
+  const scope = isGlobal ? 'globally (all projects)' : cwd;
+  console.log(`
+AWM configured ${isGlobal ? 'globally' : 'for: ' + cwd}
+
+  Agent ID:    ${agentId}
+  DB path:     ${dbPath}
+  MCP config:  ${mcpJsonPath}
+  Hook port:   ${hookPort}
+  Hook secret: ${hookSecret.slice(0, 8)}...
+${claudeMdAction}
+${hookAction}
+
+Next steps:
+  1. Restart Claude Code to pick up the MCP server
+  2. The memory tools will appear automatically
+  3. Hooks auto-checkpoint on context compaction and session end${isGlobal ? '\n  4. One brain across all your projects — no per-project setup needed' : ''}
+`.trim());
+}
+
+// ─── MCP ──────────────────────────────────────
+
+async function mcp() {
+  // Dynamic import to avoid loading heavy deps for setup/health commands
+  await import('./mcp.js');
+}
+
+// ─── SERVE ──────────────────────────────────────
+
+async function serve() {
+  // Parse --port flag
+  for (let i = 1; i < args.length; i++) {
+    if (args[i] === '--port' && args[i + 1]) {
+      process.env.AWM_PORT = args[++i];
+    }
+  }
+  await import('./index.js');
+}
+
+// ─── HEALTH ──────────────────────────────────────
+
+function health() {
+  let port = '8400';
+  for (let i = 1; i < args.length; i++) {
+    if (args[i] === '--port' && args[i + 1]) {
+      port = args[++i];
+    }
+  }
+
+  try {
+    const result = execSync(`curl -sf http://localhost:${port}/health`, {
+      encoding: 'utf8',
+      timeout: 5000,
+    });
+    const data = JSON.parse(result);
+    console.log(`OK — v${data.version} (${data.timestamp})`);
+  } catch {
+    console.error(`Cannot reach AWM server on port ${port}`);
+    process.exit(1);
+  }
+}
+
+// ─── Dispatch ──────────────────────────────────────
+
+switch (command) {
+  case 'setup':
+    setup();
+    break;
+  case 'mcp':
+    mcp();
+    break;
+  case 'serve':
+    serve();
+    break;
+  case 'health':
+    health();
+    break;
+  case '--help':
+  case '-h':
+  case undefined:
+    printUsage();
+    break;
+  default:
+    console.error(`Unknown command: ${command}`);
+    printUsage();
+    process.exit(1);
+}