brainbank 0.9.0 → 0.9.1

package/dist/cli.js

@@ -1,6 +1,7 @@
 #!/usr/bin/env node
 import {
   SUPPORTED_EXTENSIONS,
+  VERSION,
   args,
   c,
   createBrain,
@@ -15,7 +16,7 @@ import {
   printResults,
   registerConfigCollections,
   stripFlags
-} from "./chunk-Z54MHEYW.js";
+} from "./chunk-6NM6WRDX.js";
 import {
   DEFAULT_PORT,
   isServerRunning,
@@ -28,15 +29,121 @@ import {
 } from "./chunk-7QVYU63E.js";
 
 // src/cli/commands/index.ts
-import * as fs2 from "fs";
-import * as path2 from "path";
+import * as fs3 from "fs";
+import * as path3 from "path";
 
-// src/cli/commands/scan.ts
+// src/cli/commands/mcp-export.ts
 import * as fs from "fs";
 import * as path from "path";
+import { fileURLToPath } from "url";
+var __filename = fileURLToPath(import.meta.url);
+var __dirname = path.dirname(__filename);
+var TARGETS = {
+  antigravity: {
+    configPath: path.join(process.env.HOME ?? "~", ".gemini", "antigravity", "mcp_config.json"),
+    label: "Gemini Antigravity"
+  }
+};
+function buildBrainbankMcpBlock(config) {
+  const nodeBin = process.execPath;
+  const globalCliJs = path.join(path.dirname(nodeBin), "..", "lib", "node_modules", "brainbank", "dist", "cli.js");
+  const localCliJs = path.resolve(__dirname, "..", "..", "dist", "cli.js");
+  const resolvedCliJs = fs.existsSync(globalCliJs) ? globalCliJs : localCliJs;
+  const env = {};
+  const keys = config?.keys;
+  const perplexityKey = keys?.perplexity ?? process.env.PERPLEXITY_API_KEY;
+  const anthropicKey = keys?.anthropic ?? process.env.ANTHROPIC_API_KEY;
+  const openaiKey = keys?.openai ?? process.env.OPENAI_API_KEY;
+  if (perplexityKey) env.PERPLEXITY_API_KEY = perplexityKey;
+  if (anthropicKey) env.ANTHROPIC_API_KEY = anthropicKey;
+  if (openaiKey) env.OPENAI_API_KEY = openaiKey;
+  const block = {
+    command: nodeBin,
+    args: [resolvedCliJs, "mcp"]
+  };
+  if (Object.keys(env).length > 0) {
+    block.env = env;
+  }
+  return block;
+}
+__name(buildBrainbankMcpBlock, "buildBrainbankMcpBlock");
+function mergeAndWrite(targetPath, block) {
+  let existing = { mcpServers: {} };
+  const created = !fs.existsSync(targetPath);
+  if (!created) {
+    try {
+      const raw = fs.readFileSync(targetPath, "utf-8");
+      existing = JSON.parse(raw);
+      if (!existing.mcpServers) existing.mcpServers = {};
+    } catch {
+      existing = { mcpServers: {} };
+    }
+  }
+  existing.mcpServers.brainbank = block;
+  fs.mkdirSync(path.dirname(targetPath), { recursive: true });
+  fs.writeFileSync(targetPath, JSON.stringify(existing, null, 2) + "\n");
+  return { created };
+}
+__name(mergeAndWrite, "mergeAndWrite");
+function hasBrainbankMcpEntry(targetPath) {
+  if (!fs.existsSync(targetPath)) return false;
+  try {
+    const raw = fs.readFileSync(targetPath, "utf-8");
+    const config = JSON.parse(raw);
+    return !!config.mcpServers?.brainbank;
+  } catch {
+    return false;
+  }
+}
+__name(hasBrainbankMcpEntry, "hasBrainbankMcpEntry");
+async function autoExportMcp(repoPath) {
+  const target = TARGETS.antigravity;
+  if (!target) return;
+  const antigravityDir = path.dirname(target.configPath);
+  if (!fs.existsSync(antigravityDir)) return;
+  if (hasBrainbankMcpEntry(target.configPath)) return;
+  const config = await getConfig(repoPath);
+  const block = buildBrainbankMcpBlock(config);
+  mergeAndWrite(target.configPath, block);
+  console.log(`  ${c.green("\u2713")} Exported MCP config to ${c.dim(path.relative(process.env.HOME ?? "", target.configPath))}`);
+}
+__name(autoExportMcp, "autoExportMcp");
+async function cmdMcpExport() {
+  const targetName = args[1] || getFlag("target") || "antigravity";
+  const repoPath = getFlag("repo") || ".";
+  const target = TARGETS[targetName];
+  if (!target) {
+    console.error(c.red(`Unknown export target: ${targetName}`));
+    console.error(c.dim(`  Available: ${Object.keys(TARGETS).join(", ")}`));
+    process.exit(1);
+  }
+  const config = await getConfig(repoPath);
+  const block = buildBrainbankMcpBlock(config);
+  const { created } = mergeAndWrite(target.configPath, block);
+  console.log(c.bold(`
+\u2501\u2501\u2501 MCP Export: ${target.label} \u2501\u2501\u2501
+`));
+  console.log(`  ${c.green("\u2713")} ${created ? "Created" : "Updated"} ${c.dim(target.configPath)}`);
+  console.log(`  ${c.dim("Node:")}    ${block.command}`);
+  console.log(`  ${c.dim("CLI:")}     ${block.args[0]}`);
+  const envKeys = block.env ? Object.keys(block.env) : [];
+  if (envKeys.length > 0) {
+    console.log(`  ${c.dim("Keys:")}    ${envKeys.join(", ")}`);
+  } else {
+    console.log(`  ${c.yellow("\u26A0")} No API keys found. Set env vars or add keys to .brainbank/config.json`);
+  }
+  console.log(`
+  ${c.dim("Restart your IDE to apply changes.")}
+`);
+}
+__name(cmdMcpExport, "cmdMcpExport");
+
+// src/cli/commands/scan.ts
+import * as fs2 from "fs";
+import * as path2 from "path";
 import { execSync } from "child_process";
 function scanRepo(repoPath) {
-  const resolved = path.resolve(repoPath);
+  const resolved = path2.resolve(repoPath);
   const gitSubdirs = scanGitSubdirs(resolved);
   return {
     repoPath: resolved,
@@ -61,15 +168,15 @@ function scanCodeModule(repoPath) {
   function walk(dir) {
     let entries;
     try {
-      entries = fs.readdirSync(dir, { withFileTypes: true });
+      entries = fs2.readdirSync(dir, { withFileTypes: true });
     } catch {
       return;
     }
     for (const entry of entries) {
-      const fullPath = path.join(dir, entry.name);
+      const fullPath = path2.join(dir, entry.name);
       const isDir = entry.isDirectory() || entry.isSymbolicLink() && (() => {
         try {
-          return fs.statSync(fullPath).isDirectory();
+          return fs2.statSync(fullPath).isDirectory();
         } catch {
           return false;
         }
@@ -79,7 +186,7 @@ function scanCodeModule(repoPath) {
         walk(fullPath);
       } else if (entry.isFile()) {
         if (isIgnoredFile(entry.name)) continue;
-        const ext = path.extname(entry.name).toLowerCase();
+        const ext = path2.extname(entry.name).toLowerCase();
         const lang = SUPPORTED_EXTENSIONS[ext];
         if (!lang) continue;
         byLanguage.set(lang, (byLanguage.get(lang) ?? 0) + 1);
@@ -158,7 +265,7 @@ function scanDocsModule(repoPath) {
 }
 __name(scanDocsModule, "scanDocsModule");
 function scanGitStats(repoPath, gitSubdirs) {
-  if (fs.existsSync(path.join(repoPath, ".git"))) {
+  if (fs2.existsSync(path2.join(repoPath, ".git"))) {
     return gitStats(repoPath);
   }
   if (gitSubdirs.length === 0) return null;
@@ -166,7 +273,7 @@ function scanGitStats(repoPath, gitSubdirs) {
   let latestMessage = "";
   let latestDate = "";
   for (const sub of gitSubdirs) {
-    const stats = gitStats(path.join(repoPath, sub.name));
+    const stats = gitStats(path2.join(repoPath, sub.name));
     if (stats) {
       totalCommits += stats.commitCount;
       if (!latestMessage) {
@@ -192,15 +299,15 @@ __name(gitStats, "gitStats");
 function scanDocsCollections(repoPath) {
   const results = [];
   const seen = /* @__PURE__ */ new Set();
-  const configPath = path.join(repoPath, ".brainbank", "config.json");
+  const configPath = path2.join(repoPath, ".brainbank", "config.json");
   try {
-    if (fs.existsSync(configPath)) {
-      const config = JSON.parse(fs.readFileSync(configPath, "utf-8"));
+    if (fs2.existsSync(configPath)) {
+      const config = JSON.parse(fs2.readFileSync(configPath, "utf-8"));
       const docsCfg = config?.docs;
       const collections = docsCfg?.collections;
       if (collections) {
         for (const coll of collections) {
-          const absPath = path.resolve(repoPath, coll.path);
+          const absPath = path2.resolve(repoPath, coll.path);
           results.push({ name: coll.name, path: coll.path, fileCount: countDocs(absPath) });
           seen.add(absPath);
         }
@@ -213,11 +320,11 @@ function scanDocsCollections(repoPath) {
     results.push({ name: "(root)", path: ".", fileCount: rootDocs });
   }
   try {
-    for (const entry of fs.readdirSync(repoPath, { withFileTypes: true })) {
+    for (const entry of fs2.readdirSync(repoPath, { withFileTypes: true })) {
       if (!entry.isDirectory()) continue;
       if (isIgnoredDir(entry.name)) continue;
       if (entry.name.startsWith(".")) continue;
-      const dirPath = path.join(repoPath, entry.name);
+      const dirPath = path2.join(repoPath, entry.name);
       if (seen.has(dirPath)) continue;
       const count = countDocs(dirPath);
       if (count > 0) {
@@ -232,11 +339,11 @@ __name(scanDocsCollections, "scanDocsCollections");
 function countDocs(dir) {
   let count = 0;
   try {
-    for (const e of fs.readdirSync(dir, { withFileTypes: true })) {
-      const ePath = path.join(dir, e.name);
+    for (const e of fs2.readdirSync(dir, { withFileTypes: true })) {
+      const ePath = path2.join(dir, e.name);
       const isDir = e.isDirectory() || e.isSymbolicLink() && (() => {
         try {
-          return fs.statSync(ePath).isDirectory();
+          return fs2.statSync(ePath).isDirectory();
         } catch {
           return false;
         }
@@ -256,7 +363,7 @@ __name(countDocs, "countDocs");
 function countDocsShallow(dir) {
   let count = 0;
   try {
-    for (const e of fs.readdirSync(dir, { withFileTypes: true })) {
+    for (const e of fs2.readdirSync(dir, { withFileTypes: true })) {
       if (e.isFile() && /\.mdx?$/i.test(e.name)) count++;
     }
   } catch {
@@ -265,10 +372,10 @@ function countDocsShallow(dir) {
 }
 __name(countDocsShallow, "countDocsShallow");
 function scanConfig(repoPath) {
-  const configPath = path.join(repoPath, ".brainbank", "config.json");
-  if (!fs.existsSync(configPath)) return { exists: false };
+  const configPath = path2.join(repoPath, ".brainbank", "config.json");
+  if (!fs2.existsSync(configPath)) return { exists: false };
   try {
-    const config = JSON.parse(fs.readFileSync(configPath, "utf-8"));
+    const config = JSON.parse(fs2.readFileSync(configPath, "utf-8"));
     const codeCfg = config?.code;
     return {
       exists: true,
@@ -281,10 +388,10 @@ function scanConfig(repoPath) {
 }
 __name(scanConfig, "scanConfig");
 function scanDb(repoPath) {
-  const dbPath = path.join(repoPath, ".brainbank", "data", "brainbank.db");
-  if (!fs.existsSync(dbPath)) return { exists: false, sizeMB: 0 };
+  const dbPath = path2.join(repoPath, ".brainbank", "data", "brainbank.db");
+  if (!fs2.existsSync(dbPath)) return { exists: false, sizeMB: 0 };
   try {
-    const stat = fs.statSync(dbPath);
+    const stat = fs2.statSync(dbPath);
     return {
       exists: true,
       sizeMB: Math.round(stat.size / 1024 / 1024 * 10) / 10,
@@ -296,19 +403,19 @@ function scanDb(repoPath) {
 }
 __name(scanDb, "scanDb");
 function scanGitSubdirs(repoPath) {
-  if (fs.existsSync(path.join(repoPath, ".git"))) return [];
+  if (fs2.existsSync(path2.join(repoPath, ".git"))) return [];
   try {
-    let subdirs = fs.readdirSync(repoPath, { withFileTypes: true }).filter((e) => {
+    let subdirs = fs2.readdirSync(repoPath, { withFileTypes: true }).filter((e) => {
       if (e.name.startsWith(".")) return false;
       const isDir = e.isDirectory() || e.isSymbolicLink() && (() => {
         try {
-          return fs.statSync(path.join(repoPath, e.name)).isDirectory();
+          return fs2.statSync(path2.join(repoPath, e.name)).isDirectory();
         } catch {
           return false;
         }
       })();
       return isDir;
-    }).filter((e) => fs.existsSync(path.join(repoPath, e.name, ".git"))).map((e) => ({ name: e.name }));
+    }).filter((e) => fs2.existsSync(path2.join(repoPath, e.name, ".git"))).map((e) => ({ name: e.name }));
     const configRepos = readReposFromConfig(repoPath);
     if (configRepos) {
       subdirs = subdirs.filter((s) => configRepos.includes(s.name));
@@ -320,10 +427,10 @@ function scanGitSubdirs(repoPath) {
 }
 __name(scanGitSubdirs, "scanGitSubdirs");
 function readReposFromConfig(repoPath) {
-  const configPath = path.join(repoPath, ".brainbank", "config.json");
+  const configPath = path2.join(repoPath, ".brainbank", "config.json");
   try {
-    if (!fs.existsSync(configPath)) return null;
-    const config = JSON.parse(fs.readFileSync(configPath, "utf-8"));
+    if (!fs2.existsSync(configPath)) return null;
+    const config = JSON.parse(fs2.readFileSync(configPath, "utf-8"));
     const repos = config.repos;
     if (Array.isArray(repos) && repos.every((r) => typeof r === "string")) {
       return repos;
@@ -383,8 +490,8 @@ async function cmdIndex() {
   const config = await getConfig(repoPath);
   await registerConfigCollections(brain, repoPath, config);
   if (docsPath) {
-    const absDocsPath = path2.resolve(docsPath);
-    const collName = path2.basename(absDocsPath);
+    const absDocsPath = path3.resolve(docsPath);
+    const collName = path3.basename(absDocsPath);
     try {
       const docsPlugin = findDocsPlugin(brain);
       await docsPlugin?.addCollection({
@@ -428,6 +535,7 @@ async function cmdIndex() {
     console.log(`    ${name}: ${entries}`);
   }
   brain.close();
+  await autoExportMcp(repoPath);
 }
 __name(cmdIndex, "cmdIndex");
 function printScanTree(scan, depth) {
@@ -506,7 +614,7 @@ function capitalizeFirst(s) {
 }
 __name(capitalizeFirst, "capitalizeFirst");
 async function saveConfig(repoPath, modules) {
-  const { select } = await import("@inquirer/prompts");
+  const { select, confirm } = await import("@inquirer/prompts");
   const envEmbedding = process.env.BRAINBANK_EMBEDDING;
   const embedding = await select({
     message: "Embedding provider:",
@@ -534,18 +642,18 @@ async function saveConfig(repoPath, modules) {
     message: "Noise pruner:",
     choices: [
       {
-        name: "none   \u2014 no pruning (default)",
-        value: "none"
+        name: "haiku  \u2014 AI-powered noise filter (recommended)",
+        value: "haiku"
       },
       {
-        name: "haiku  \u2014 AI-powered noise filter (Claude Haiku)",
-        value: "haiku"
+        name: "none   \u2014 no pruning",
+        value: "none"
       }
     ],
-    default: "none"
+    default: "haiku"
   });
-  const configDir = path2.join(repoPath, ".brainbank");
-  const configPath = path2.join(configDir, "config.json");
+  const configDir = path3.join(repoPath, ".brainbank");
+  const configPath = path3.join(configDir, "config.json");
   const config = {
     plugins: modules,
     embedding
@@ -553,9 +661,32 @@ async function saveConfig(repoPath, modules) {
   if (pruner !== "none") {
     config.pruner = pruner;
   }
-  fs2.mkdirSync(configDir, { recursive: true });
-  fs2.writeFileSync(configPath, JSON.stringify(config, null, 2) + "\n");
-  console.log(c.green(`  \u2713 Saved ${path2.relative(process.cwd(), configPath)}`));
+  const detectedKeys = {};
+  const needsPerplexity = embedding.startsWith("perplexity");
+  const needsAnthropic = pruner === "haiku";
+  const needsOpenai = embedding === "openai";
+  if (needsPerplexity && process.env.PERPLEXITY_API_KEY) {
+    detectedKeys.perplexity = process.env.PERPLEXITY_API_KEY;
+  }
+  if (needsAnthropic && process.env.ANTHROPIC_API_KEY) {
+    detectedKeys.anthropic = process.env.ANTHROPIC_API_KEY;
+  }
+  if (needsOpenai && process.env.OPENAI_API_KEY) {
+    detectedKeys.openai = process.env.OPENAI_API_KEY;
+  }
+  if (Object.keys(detectedKeys).length > 0) {
+    const keyNames = Object.keys(detectedKeys).join(", ");
+    const saveKeys = await confirm({
+      message: `Save API keys (${keyNames}) to config.json? (portable, no env vars needed)`,
+      default: true
+    });
+    if (saveKeys) {
+      config.keys = detectedKeys;
+    }
+  }
+  fs3.mkdirSync(configDir, { recursive: true });
+  fs3.writeFileSync(configPath, JSON.stringify(config, null, 2) + "\n");
+  console.log(c.green(`  \u2713 Saved ${path3.relative(process.cwd(), configPath)}`));
 }
 __name(saveConfig, "saveConfig");
 
@@ -564,12 +695,12 @@ async function cmdCollection() {
   const pos = stripFlags(args);
   const sub = pos[1];
   if (sub === "add") {
-    const path3 = pos[2];
+    const path4 = pos[2];
     const name = getFlag("name");
     const pattern = getFlag("pattern") ?? "**/*.md";
     const context = getFlag("context");
     const ignoreRaw = getFlag("ignore");
-    if (!path3 || !name) {
+    if (!path4 || !name) {
       console.log(c.red('Usage: brainbank collection add <path> --name <name> [--pattern "**/*.md"] [--ignore "glob"] [--context "description"]'));
       process.exit(1);
     }
@@ -581,12 +712,12 @@ async function cmdCollection() {
     }
     await docsPlugin.addCollection({
       name,
-      path: path3,
+      path: path4,
       pattern,
       ignore: ignoreRaw ? ignoreRaw.split(",") : [],
       context: context ?? void 0
     });
-    console.log(c.green(`\u2713 Collection '${name}' added: ${path3} (${pattern})`));
+    console.log(c.green(`\u2713 Collection '${name}' added: ${path4} (${pattern})`));
     if (context) console.log(c.dim(`  Context: ${context}`));
     brain.close();
     return;
@@ -964,12 +1095,12 @@ async function serverHealth() {
   }
 }
 __name(serverHealth, "serverHealth");
-function httpPost(port, path3, body) {
-  return new Promise((resolve3, reject) => {
+function httpPost(port, path4, body) {
+  return new Promise((resolve4, reject) => {
     const req = http.request({
       hostname: "127.0.0.1",
       port,
-      path: path3,
+      path: path4,
       method: "POST",
       headers: {
         "Content-Type": "application/json",
@@ -980,7 +1111,7 @@ function httpPost(port, path3, body) {
     }, (res) => {
       const chunks = [];
       res.on("data", (chunk) => chunks.push(chunk));
-      res.on("end", () => resolve3(Buffer.concat(chunks).toString("utf8")));
+      res.on("end", () => resolve4(Buffer.concat(chunks).toString("utf8")));
     });
     req.on("error", reject);
     req.on("timeout", () => {
@@ -992,18 +1123,18 @@ function httpPost(port, path3, body) {
   });
 }
 __name(httpPost, "httpPost");
-function httpGet(port, path3) {
-  return new Promise((resolve3, reject) => {
+function httpGet(port, path4) {
+  return new Promise((resolve4, reject) => {
     const req = http.request({
       hostname: "127.0.0.1",
       port,
-      path: path3,
+      path: path4,
       method: "GET",
       timeout: 5e3
     }, (res) => {
       const chunks = [];
       res.on("data", (chunk) => chunks.push(chunk));
-      res.on("end", () => resolve3(Buffer.concat(chunks).toString("utf8")));
+      res.on("end", () => resolve4(Buffer.concat(chunks).toString("utf8")));
     });
     req.on("error", reject);
     req.on("timeout", () => {
@@ -1062,9 +1193,9 @@ async function cmdContext() {
   const sub = pos[1];
   if (sub === "add") {
     const collection = pos[2];
-    const path3 = pos[3];
+    const path4 = pos[3];
     const desc = pos.slice(4).join(" ");
-    if (!collection || !path3 || !desc) {
+    if (!collection || !path4 || !desc) {
       console.log(c.red("Usage: brainbank context add <collection> <path> <description>"));
       process.exit(1);
     }
@@ -1075,8 +1206,8 @@ async function cmdContext() {
       console.log(c.red("Docs plugin not loaded."));
       process.exit(1);
     }
-    docsPlugin.addContext(collection, path3, desc);
-    console.log(c.green(`\u2713 Context added: ${collection}:${path3} \u2192 "${desc}"`));
+    docsPlugin.addContext(collection, path4, desc);
+    console.log(c.green(`\u2713 Context added: ${collection}:${path4} \u2192 "${desc}"`));
     brain2.close();
     return;
   }
@@ -1482,11 +1613,13 @@ function showHelp() {
   console.log(`  ${c.cyan("reembed")}                            Re-embed all vectors`);
   console.log(`  ${c.cyan("watch")}                              Watch files, auto-re-index`);
   console.log(`  ${c.cyan("mcp")}                                Start MCP server (stdio)`);
+  console.log(`  ${c.cyan("mcp:export")} [target]                Export MCP config (antigravity)`);
   console.log(`  ${c.cyan("daemon")}                             Start HTTP daemon (foreground)`);
   console.log(`  ${c.cyan("daemon start")}                       Start HTTP daemon (background)`);
   console.log(`  ${c.cyan("daemon stop")}                        Stop background daemon`);
   console.log(`  ${c.cyan("daemon restart")}                     Restart background daemon`);
   console.log(`  ${c.cyan("status")}                             Show daemon status`);
+  console.log(`  ${c.cyan("--version")} ${c.dim("(-v)")}                      Show version`);
   console.log("");
   console.log(c.bold("Options:"));
   console.log(`  ${c.dim("--repo <path>")}           Repository path (default: .)`);
@@ -1515,6 +1648,7 @@ function showHelp() {
   console.log(c.dim('  brainbank context "auth flow" | pbcopy               # \u2192 clipboard'));
   console.log(c.dim("  brainbank daemon start                               # background HTTP"));
   console.log(c.dim("  brainbank mcp                                        # MCP stdio"));
+  console.log(c.dim("  brainbank mcp:export antigravity                     # export MCP config"));
 }
 __name(showHelp, "showHelp");
 
@@ -1522,6 +1656,10 @@ __name(showHelp, "showHelp");
 var command = args[0];
 async function main() {
   switch (command) {
+    case "--version":
+    case "-v":
+      console.log(`brainbank v${VERSION}`);
+      break;
     case "i":
     case "index":
       return cmdIndex();
@@ -1551,6 +1689,8 @@ async function main() {
       return cmdWatch();
     case "mcp":
       return cmdMcp();
+    case "mcp:export":
+      return cmdMcpExport();
     case "serve":
       return cmdMcp();
     // backward compat