toon-memory 1.7.0 → 2.0.0

package/dist/cli/setup.js

@@ -1,4 +1,4 @@
-import { existsSync, mkdirSync, readFileSync, writeFileSync, unlinkSync, readdirSync, statSync } from "fs";
+import { existsSync, mkdirSync, readFileSync, writeFileSync, unlinkSync, readdirSync, statSync, chmodSync, rmSync } from "fs";
 import { basename, dirname, join } from "path";
 import { fileURLToPath } from "url";
 import { createInterface } from "readline";
@@ -7,68 +7,170 @@ const __dirname = dirname(fileURLToPath(import.meta.url));
 const projectRoot = process.cwd();
 const sourceDir = join(__dirname, "..", "..", "src");
 const HOME = process.env.HOME || process.env.USERPROFILE || "~";
+const MEMORY_DIR = join(projectRoot, ".toon-memory", "memory");
+const HOOK_CONTENT = `#!/bin/bash
+echo "toon-memory: Use memory_recall BEFORE reading files for project context."
+exit 0
+`;
+const INSTRUCTION_CONTENT = `# toon-memory
+
+Persistent memory for this project. Use it to avoid re-investigating things.
+
+## At the START of every session
+1. Run memory_stats to see what's in memory.
+2. If the user asks something that might be in memory, run memory_recall BEFORE reading files.
+
+## When making decisions
+- Before implementing a non-trivial change: memory_remember(category='decision')
+- When you resolve a complex bug: memory_remember(category='bug')
+- When you observe a code pattern: memory_remember(category='pattern')
+
+## At the END of every session
+- Save important decisions, bugs resolved, and patterns observed.
+`;
 function detectAgents() {
   const agents2 = [];
-  const opencodeGlobal = join(HOME, ".config", "opencode", "opencode.json");
-  const opencodeLocal = join(projectRoot, ".opencode", "opencode.json");
   agents2.push({
     name: "opencode",
-    global: opencodeGlobal,
-    local: opencodeLocal,
-    mcpKey: "mcp"
+    global: join(HOME, ".config", "opencode", "opencode.json"),
+    local: join(projectRoot, ".opencode", "opencode.json"),
+    mcpKey: "mcp",
+    format: "json",
+    needsHooks: false,
+    needsInstructions: true,
+    instructionFile: join(projectRoot, "AGENTS.md")
   });
-  const vscodeLocal = join(projectRoot, ".vscode", "mcp.json");
   agents2.push({
     name: "vscode/copilot",
-    local: vscodeLocal,
-    mcpKey: "servers"
+    local: join(projectRoot, ".vscode", "mcp.json"),
+    mcpKey: "servers",
+    format: "json",
+    needsHooks: false,
+    needsInstructions: false
   });
-  const claudeGlobal = join(HOME, ".claude", "settings.json");
-  const claudeLocal = join(projectRoot, ".claude", "settings.json");
   agents2.push({
     name: "claude",
-    global: claudeGlobal,
-    local: claudeLocal,
-    mcpKey: "mcpServers"
+    global: join(HOME, ".claude", "settings.json"),
+    local: join(projectRoot, ".claude", "settings.json"),
+    mcpKey: "mcpServers",
+    format: "json",
+    needsHooks: true,
+    needsInstructions: true,
+    instructionFile: join(projectRoot, ".claude", "AGENTS.md")
   });
-  const cursorLocal = join(projectRoot, ".cursor", "mcp.json");
   agents2.push({
     name: "cursor",
-    local: cursorLocal,
-    mcpKey: "mcpServers"
+    local: join(projectRoot, ".cursor", "mcp.json"),
+    mcpKey: "mcpServers",
+    format: "json",
+    needsHooks: false,
+    needsInstructions: false
   });
-  const windsurfGlobal = join(HOME, ".codeium", "windsurf", "mcp_config.json");
   agents2.push({
     name: "windsurf",
-    global: windsurfGlobal,
-    mcpKey: "mcpServers"
+    global: join(HOME, ".codeium", "windsurf", "mcp_config.json"),
+    mcpKey: "mcpServers",
+    format: "json",
+    needsHooks: false,
+    needsInstructions: false
   });
-  const clineLocal = join(projectRoot, ".cline", "mcp.json");
   agents2.push({
     name: "cline",
-    local: clineLocal,
-    mcpKey: "mcpServers"
+    local: join(projectRoot, ".cline", "mcp.json"),
+    mcpKey: "mcpServers",
+    format: "json",
+    needsHooks: false,
+    needsInstructions: false
   });
-  const continueLocal = join(projectRoot, ".continue", "config.json");
   agents2.push({
     name: "continue",
-    local: continueLocal,
-    mcpKey: "mcpServers"
+    local: join(projectRoot, ".continue", "config.json"),
+    mcpKey: "mcpServers",
+    format: "json",
+    needsHooks: false,
+    needsInstructions: false
+  });
+  agents2.push({
+    name: "codex",
+    local: join(projectRoot, ".codex", "config.toml"),
+    mcpKey: "mcpServers",
+    format: "toml",
+    needsHooks: true,
+    needsInstructions: true,
+    instructionFile: join(projectRoot, ".codex", "AGENTS.md")
+  });
+  agents2.push({
+    name: "gemini",
+    local: join(projectRoot, ".gemini", "settings.json"),
+    mcpKey: "mcpServers",
+    format: "json",
+    needsHooks: true,
+    needsInstructions: true,
+    instructionFile: join(projectRoot, ".gemini", "GEMINI.md")
+  });
+  agents2.push({
+    name: "zed",
+    global: join(HOME, ".config", "zed", "settings.json"),
+    mcpKey: "mcp_servers",
+    format: "jsonc",
+    needsHooks: false,
+    needsInstructions: false
+  });
+  agents2.push({
+    name: "antigravity",
+    local: join(projectRoot, ".gemini", "config", "mcp_config.json"),
+    mcpKey: "mcpServers",
+    format: "json",
+    needsHooks: true,
+    needsInstructions: true,
+    instructionFile: join(projectRoot, "antigravity-cli", "AGENTS.md")
+  });
+  agents2.push({
+    name: "aider",
+    mcpKey: "",
+    format: "none",
+    needsHooks: false,
+    needsInstructions: true,
+    instructionFile: join(projectRoot, "CONVENTIONS.md")
+  });
+  agents2.push({
+    name: "kilocode",
+    global: join(HOME, ".kilocode", "mcp_settings.json"),
+    mcpKey: "mcpServers",
+    format: "json",
+    needsHooks: false,
+    needsInstructions: true,
+    instructionFile: join(HOME, ".kilocode", "rules", "toon-memory.md")
+  });
+  agents2.push({
+    name: "openclaw",
+    local: join(projectRoot, ".openclaw.json"),
+    mcpKey: "mcpServers",
+    format: "json",
+    needsHooks: false,
+    needsInstructions: false
+  });
+  agents2.push({
+    name: "kiro",
+    local: join(projectRoot, ".kiro", "settings", "mcp.json"),
+    mcpKey: "mcpServers",
+    format: "json",
+    needsHooks: false,
+    needsInstructions: false
   });
   return agents2;
 }
-function installOpenCodeTools() {
-  const memoryDir = join(projectRoot, ".opencode", "memory");
-  const memoryFile = join(memoryDir, "data.toon");
-  if (!existsSync(memoryDir)) mkdirSync(memoryDir, { recursive: true });
+function installMemoryDir() {
+  const memoryFile = join(MEMORY_DIR, "data.toon");
+  if (!existsSync(MEMORY_DIR)) mkdirSync(MEMORY_DIR, { recursive: true });
   if (!existsSync(memoryFile)) {
     writeFileSync(memoryFile, "version: 1\n[0|]\n");
-    console.log("  Created .opencode/memory/data.toon");
+    console.log("  Created .toon-memory/memory/data.toon");
   }
 }
 function ensureGitignore() {
   const gitignorePath = join(projectRoot, ".gitignore");
-  const entry = ".opencode/memory/";
+  const entry = ".toon-memory/memory/";
   if (!existsSync(gitignorePath)) {
     writeFileSync(gitignorePath, `${entry}
 `);
@@ -80,10 +182,10 @@ function ensureGitignore() {
     writeFileSync(gitignorePath, `${content.trim()}
 ${entry}
 `);
-    console.log("  Added .opencode/memory/ to .gitignore");
+    console.log("  Added .toon-memory/memory/ to .gitignore");
   }
 }
-function installMCPConfig(agent, scope) {
+function installJSONConfig(agent, scope) {
   const configPath = scope === "global" ? agent.global : agent.local;
   if (!configPath) {
     console.log(`  No ${scope} config path for ${agent.name}`);
@@ -116,46 +218,217 @@ function installMCPConfig(agent, scope) {
   writeFileSync(configPath, JSON.stringify(config, null, 2));
   console.log(`  MCP server added to ${configPath}`);
 }
+function installTOMLConfig(agent) {
+  const configPath = agent.local;
+  if (!configPath) return;
+  const configDir = dirname(configPath);
+  if (!existsSync(configDir)) mkdirSync(configDir, { recursive: true });
+  const toml = `[mcpServers.toon-memory]
+command = "npx"
+args = ["-y", "toon-memory", "mcp"]
+`;
+  writeFileSync(configPath, toml);
+  console.log(`  MCP server added to ${configPath}`);
+}
+function installZedConfig(agent) {
+  const configPath = agent.global;
+  if (!configPath) return;
+  const configDir = dirname(configPath);
+  if (!existsSync(configDir)) mkdirSync(configDir, { recursive: true });
+  let config = {};
+  if (existsSync(configPath)) {
+    try {
+      const raw = readFileSync(configPath, "utf-8");
+      const stripped = raw.replace(/\/\/.*$/gm, "").replace(/\/\*[\s\S]*?\*\//g, "");
+      config = JSON.parse(stripped);
+    } catch {
+      config = {};
+    }
+  }
+  if (!config.mcp_servers) config.mcp_servers = {};
+  config.mcp_servers["toon-memory"] = {
+    command: "npx",
+    args: ["-y", "toon-memory", "mcp"]
+  };
+  writeFileSync(configPath, JSON.stringify(config, null, 2));
+  console.log(`  MCP server added to ${configPath}`);
+}
+function installMCPConfig(agent, scope) {
+  if (agent.format === "none") {
+    console.log(`  ${agent.name}: instructions only (no MCP)`);
+    return;
+  }
+  if (agent.format === "toml") {
+    installTOMLConfig(agent);
+  } else if (agent.format === "jsonc") {
+    installZedConfig(agent);
+  } else {
+    installJSONConfig(agent, scope);
+  }
+}
+function installInstructions(agent) {
+  if (!agent.needsInstructions || !agent.instructionFile) return;
+  const filePath = agent.instructionFile;
+  const fileDir = dirname(filePath);
+  if (!existsSync(fileDir)) mkdirSync(fileDir, { recursive: true });
+  if (existsSync(filePath)) {
+    const existing = readFileSync(filePath, "utf-8");
+    if (existing.includes("toon-memory") || existing.includes("memory_recall")) {
+      console.log(`  Instructions already present in ${filePath}`);
+      return;
+    }
+    writeFileSync(filePath, `${existing.trim()}
+
+${INSTRUCTION_CONTENT}`);
+    console.log(`  Appended toon-memory instructions to ${filePath}`);
+  } else {
+    writeFileSync(filePath, INSTRUCTION_CONTENT);
+    console.log(`  Created ${filePath}`);
+  }
+}
+function installHooks(agent) {
+  if (!agent.needsHooks) return;
+  const hookDir = join(projectRoot, ".toon-memory", "hooks");
+  if (!existsSync(hookDir)) mkdirSync(hookDir, { recursive: true });
+  const hookPath = join(hookDir, `session-start-${agent.name}.sh`);
+  writeFileSync(hookPath, HOOK_CONTENT);
+  chmodSync(hookPath, 493);
+  console.log(`  Hook script created at ${hookPath}`);
+  if (agent.format === "toml" && agent.local) {
+    registerHookTOML(agent, hookPath);
+  } else if (agent.format === "json" || agent.format === "jsonc") {
+    registerHookJSON(agent, hookPath);
+  }
+}
+function registerHookTOML(agent, hookPath) {
+  const configPath = agent.local;
+  if (!configPath || !existsSync(configPath)) return;
+  let content = readFileSync(configPath, "utf-8");
+  if (content.includes("session_start")) {
+    console.log(`  Hook already registered in ${configPath}`);
+    return;
+  }
+  content += `
+[hooks.session_start]
+command = "${hookPath}"
+`;
+  writeFileSync(configPath, content);
+  console.log(`  Hook registered in ${configPath}`);
+}
+function registerHookJSON(agent, hookPath) {
+  const configPath = agent.format === "jsonc" ? agent.global : agent.local || agent.global;
+  if (!configPath) return;
+  const configDir = dirname(configPath);
+  if (!existsSync(configDir)) mkdirSync(configDir, { recursive: true });
+  let config = {};
+  if (existsSync(configPath)) {
+    try {
+      if (agent.format === "jsonc") {
+        const raw = readFileSync(configPath, "utf-8");
+        const stripped = raw.replace(/\/\/.*$/gm, "").replace(/\/\*[\s\S]*?\*\//g, "");
+        config = JSON.parse(stripped);
+      } else {
+        config = JSON.parse(readFileSync(configPath, "utf-8"));
+      }
+    } catch {
+      config = {};
+    }
+  }
+  if (agent.name === "claude") {
+    if (!config.hooks) config.hooks = {};
+    if (!config.hooks.SessionStart) config.hooks.SessionStart = [];
+    if (!config.hooks.SessionStart.some((h) => h.command === hookPath)) {
+      config.hooks.SessionStart.push({ command: hookPath });
+      writeFileSync(configPath, JSON.stringify(config, null, 2));
+      console.log(`  Hook registered in ${configPath}`);
+    }
+    return;
+  }
+  if (!config.session_start_hooks) config.session_start_hooks = [];
+  if (!config.session_start_hooks.includes(hookPath)) {
+    config.session_start_hooks.push(hookPath);
+    writeFileSync(configPath, JSON.stringify(config, null, 2));
+    console.log(`  Hook registered in ${configPath}`);
+  }
+}
+function installForAgent(agent, scope) {
+  console.log(`${agent.name}:`);
+  installMCPConfig(agent, scope);
+  installInstructions(agent);
+  installHooks(agent);
+}
+function init(scope = "local") {
+  console.log("\n\u{1F9E0} toon-memory init\n");
+  installMemoryDir();
+  const agents2 = detectAgents();
+  for (const agent of agents2) {
+    installForAgent(agent, scope);
+    console.log("");
+  }
+  ensureGitignore();
+  console.log("Done! Restart your agent to use memory tools.\n");
+}
 function uninstall() {
   console.log("\n\u{1F9E0} toon-memory uninstaller\n");
   const agents2 = detectAgents();
   for (const agent of agents2) {
-    const configs = [agent.global, agent.local].filter(Boolean);
-    for (const configPath of configs) {
-      if (!existsSync(configPath)) continue;
+    if (agent.format === "json" || agent.format === "jsonc") {
+      const configs = [agent.global, agent.local].filter(Boolean);
+      for (const configPath of configs) {
+        if (!existsSync(configPath)) continue;
+        try {
+          let config;
+          if (agent.format === "jsonc") {
+            const raw = readFileSync(configPath, "utf-8");
+            const stripped = raw.replace(/\/\/.*$/gm, "").replace(/\/\*[\s\S]*?\*\//g, "");
+            config = JSON.parse(stripped);
+          } else {
+            config = JSON.parse(readFileSync(configPath, "utf-8"));
+          }
+          const mcpKey = agent.mcpKey || "mcpServers";
+          if (config[mcpKey]?.["toon-memory"]) {
+            delete config[mcpKey]["toon-memory"];
+            writeFileSync(configPath, JSON.stringify(config, null, 2));
+            console.log(`  Removed MCP from ${agent.name} (${configPath})`);
+          }
+        } catch {
+        }
+      }
+    }
+    if (agent.format === "toml" && agent.local && existsSync(agent.local)) {
+      try {
+        let content = readFileSync(agent.local, "utf-8");
+        content = content.replace(/\[mcpServers\.toon-memory\][\s\S]*?(?=\n\[|$)/, "").trim() + "\n";
+        writeFileSync(agent.local, content);
+        console.log(`  Removed MCP from ${agent.name} (${agent.local})`);
+      } catch {
+      }
+    }
+  }
+  for (const agent of agents2) {
+    if (agent.instructionFile && existsSync(agent.instructionFile)) {
       try {
-        const config = JSON.parse(readFileSync(configPath, "utf-8"));
-        const mcpKey = agent.mcpKey || "mcpServers";
-        if (config[mcpKey]?.["toon-memory"]) {
-          delete config[mcpKey]["toon-memory"];
-          writeFileSync(configPath, JSON.stringify(config, null, 2));
-          console.log(`  \u2705 Removed from ${agent.name} (${configPath})`);
+        const content = readFileSync(agent.instructionFile, "utf-8");
+        if (content.includes(INSTRUCTION_CONTENT.trim()) && content.length < INSTRUCTION_CONTENT.length + 100) {
+          unlinkSync(agent.instructionFile);
+          console.log(`  Removed ${agent.instructionFile}`);
         }
       } catch {
       }
     }
   }
+  const hookDir = join(projectRoot, ".toon-memory", "hooks");
+  if (existsSync(hookDir)) {
+    rmSync(hookDir, { recursive: true, force: true });
+    console.log("  Removed .toon-memory/hooks/");
+  }
   const toolsFile = join(projectRoot, ".opencode", "tools", "memory.ts");
   if (existsSync(toolsFile)) {
     unlinkSync(toolsFile);
-    console.log("  \u2705 Removed .opencode/tools/memory.ts");
+    console.log("  Removed .opencode/tools/memory.ts");
   }
   console.log("\n\u2705 toon-memory uninstalled from all agents\n");
 }
-function init(scope = "local") {
-  console.log("\n\u{1F9E0} toon-memory init\n");
-  const agents2 = detectAgents();
-  for (const agent of agents2) {
-    console.log(`${agent.name}:`);
-    if (agent.name === "opencode") {
-      installOpenCodeTools();
-    }
-    installMCPConfig(agent, scope);
-    console.log("");
-  }
-  ensureGitignore();
-  console.log("Done! Restart your agent to use memory tools.\n");
-}
 function status() {
   console.log("\n\u{1F9E0} toon-memory status\n");
   try {
@@ -164,7 +437,7 @@ function status() {
   } catch {
     console.log("Version: unknown");
   }
-  const memoryFile = join(projectRoot, ".opencode", "memory", "data.toon");
+  const memoryFile = join(MEMORY_DIR, "data.toon");
   if (existsSync(memoryFile)) {
     const data = readFileSync(memoryFile, "utf-8");
     const lines = data.split("\n").filter((l) => l.startsWith("  ") && l.includes("|"));
@@ -175,22 +448,41 @@ function status() {
   const agents2 = detectAgents();
   console.log("\nAgent configs:");
   for (const agent of agents2) {
-    const configs = [agent.global, agent.local].filter(Boolean);
-    let found = false;
-    for (const configPath of configs) {
-      if (!existsSync(configPath)) continue;
-      try {
-        const config = JSON.parse(readFileSync(configPath, "utf-8"));
-        const mcpKey = agent.mcpKey || "mcpServers";
-        if (config[mcpKey]?.["toon-memory"]) {
-          console.log(`  \u2705 ${agent.name} (${configPath})`);
-          found = true;
+    let configured = false;
+    if (agent.format === "json" || agent.format === "jsonc") {
+      const configs = [agent.global, agent.local].filter(Boolean);
+      for (const configPath of configs) {
+        if (!existsSync(configPath)) continue;
+        try {
+          let config;
+          if (agent.format === "jsonc") {
+            const raw = readFileSync(configPath, "utf-8");
+            const stripped = raw.replace(/\/\/.*$/gm, "").replace(/\/\*[\s\S]*?\*\//g, "");
+            config = JSON.parse(stripped);
+          } else {
+            config = JSON.parse(readFileSync(configPath, "utf-8"));
+          }
+          if (config[agent.mcpKey]?.["toon-memory"]) {
+            configured = true;
+          }
+        } catch {
         }
-      } catch {
       }
     }
-    if (!found) {
-      console.log(`  \u274C ${agent.name} (not configured)`);
+    if (agent.format === "toml" && agent.local && existsSync(agent.local)) {
+      const content = readFileSync(agent.local, "utf-8");
+      if (content.includes("toon-memory")) configured = true;
+    }
+    const hasInstructions = agent.instructionFile ? existsSync(agent.instructionFile) : false;
+    const hookPath = join(projectRoot, ".toon-memory", "hooks", `session-start-${agent.name}.sh`);
+    const hasHooks = existsSync(hookPath);
+    if (agent.format === "none") {
+      console.log(`  ${hasInstructions ? "\u2705" : "\u274C"} ${agent.name} (instructions only)`);
+    } else {
+      const mcpStatus = configured ? "\u2705" : "\u274C";
+      const instrStatus = agent.needsInstructions ? hasInstructions ? " \u{1F4DD}" : "" : "";
+      const hookStatus = agent.needsHooks ? hasHooks ? " \u{1FA9D}" : "" : "";
+      console.log(`  ${mcpStatus} ${agent.name}${instrStatus}${hookStatus}`);
     }
   }
   console.log("");
@@ -212,7 +504,7 @@ function upgrade() {
 }
 function stats() {
   console.log("\n\u{1F9E0} toon-memory stats\n");
-  const memoryFile = join(projectRoot, ".opencode", "memory", "data.toon");
+  const memoryFile = join(MEMORY_DIR, "data.toon");
   if (!existsSync(memoryFile)) {
     console.log("Memory not initialized. Run 'npx toon-memory init' first.\n");
     return;
@@ -243,7 +535,7 @@ function stats() {
 }
 function exportMemory() {
   console.log("\n\u{1F9E0} toon-memory export\n");
-  const memoryFile = join(projectRoot, ".opencode", "memory", "data.toon");
+  const memoryFile = join(MEMORY_DIR, "data.toon");
   if (!existsSync(memoryFile)) {
     console.log("Memory not initialized. Run 'npx toon-memory init' first.\n");
     return;
@@ -298,9 +590,8 @@ function importMemory() {
     console.log("Invalid format: missing 'entries' array\n");
     return;
   }
-  const memoryDir = join(projectRoot, ".opencode", "memory");
-  const memoryFile = join(memoryDir, "data.toon");
-  if (!existsSync(memoryDir)) mkdirSync(memoryDir, { recursive: true });
+  const memoryFile = join(MEMORY_DIR, "data.toon");
+  if (!existsSync(MEMORY_DIR)) mkdirSync(MEMORY_DIR, { recursive: true });
   let existingKeys = [];
   if (existsSync(memoryFile)) {
     const existing = readFileSync(memoryFile, "utf-8");
@@ -317,9 +608,10 @@ function importMemory() {
   }).join("\n");
   if (existsSync(memoryFile)) {
     const existing = readFileSync(memoryFile, "utf-8");
+    const existingCount = existing.split("\n").filter((l) => l.startsWith("  ") && l.includes("|")).length;
     const updated = existing.replace(
       /entries\[\d+\|]/,
-      `entries[${newEntries.length}|]`
+      `entries[${existingCount + newEntries.length}|]`
     ) + "\n" + newLines;
     writeFileSync(memoryFile, updated);
   } else {
@@ -346,7 +638,7 @@ function parseWatchOptions(args2) {
       opts.compress = true;
     } else if (arg === "--log" || arg === "-l") {
       opts.logFile = true;
-      opts.logPath = args2[++i] || join(projectRoot, ".opencode", "memory", "watch.log");
+      opts.logPath = args2[++i] || join(MEMORY_DIR, "watch.log");
     } else if (arg === "--max-backups" || arg === "-m") {
       opts.maxBackups = parseInt(args2[++i]) || 10;
     } else if (!arg.startsWith("-")) {
@@ -384,8 +676,8 @@ function decompressData(data) {
 }
 function watch() {
   console.log("\n\u{1F9E0} toon-memory watch\n");
-  const memoryFile = join(projectRoot, ".opencode", "memory", "data.toon");
-  const backupDir = join(projectRoot, ".opencode", "memory", "backups");
+  const memoryFile = join(MEMORY_DIR, "data.toon");
+  const backupDir = join(MEMORY_DIR, "backups");
   if (!existsSync(memoryFile)) {
     console.log("Memory not initialized. Run 'npx toon-memory init' first.\n");
     return;
@@ -495,25 +787,46 @@ if (args[0] === "watch") {
 }
 const agents = detectAgents();
 console.log("\n\u{1F9E0} toon-memory installer\n");
-console.log("Supported agents:");
-agents.forEach((a, i) => console.log(`  ${i + 1}. ${a.name}`));
+console.log("Available agents:");
+agents.forEach((a, i) => {
+  const hasConfig = a.format !== "none" && (a.local && existsSync(a.local) || a.global && existsSync(a.global));
+  const indicator = hasConfig ? "\u2713" : "\xB7";
+  console.log(`  ${indicator} ${i + 1}. ${a.name}`);
+});
 console.log("");
 const rl = createInterface({ input: process.stdin, output: process.stdout });
-rl.question("Install (1) Local or (2) Global? [1/2]: ", (answer) => {
-  const scope = answer === "2" ? "global" : "local";
+rl.question("Select agents (numbers separated by commas, 'all', or Enter for all): ", (answer) => {
+  let selected;
+  const trimmed = answer.trim().toLowerCase();
+  if (trimmed === "all" || trimmed === "") {
+    selected = agents;
+  } else if (trimmed === "none") {
+    selected = [];
+  } else {
+    const indices = trimmed.split(",").map((s) => parseInt(s.trim(), 10) - 1).filter((i) => i >= 0 && i < agents.length);
+    selected = indices.map((i) => agents[i]);
+  }
+  if (selected.length === 0) {
+    console.log("\nNo agents selected. Nothing installed.\n");
+    rl.close();
+    return;
+  }
   console.log(`
+Selected: ${selected.map((a) => a.name).join(", ")}
+`);
+  rl.question("Installation scope \u2014 (1) Local or (2) Global? [1/2]: ", (scopeAnswer) => {
+    const scope = scopeAnswer === "2" ? "global" : "local";
+    console.log(`
 Installing ${scope}ly...
 `);
-  for (const agent of agents) {
-    console.log(`${agent.name}:`);
-    if (agent.name === "opencode") {
-      installOpenCodeTools();
+    installMemoryDir();
+    for (const agent of selected) {
+      installForAgent(agent, scope);
+      console.log("");
     }
-    installMCPConfig(agent, scope);
-    console.log("");
-  }
-  ensureGitignore();
-  console.log("Done! Restart your agent to use memory tools.");
-  console.log("Run 'npx toon-memory uninstall' to remove.\n");
-  rl.close();
+    ensureGitignore();
+    console.log("Done! Restart your agent to use memory tools.");
+    console.log("Run 'npx toon-memory uninstall' to remove.\n");
+    rl.close();
+  });
 });