wavebrain 0.2.0 → 0.3.0

package/src/cli.mjs

@@ -1,353 +0,0 @@
-import { spawn } from "node:child_process";
-import os from "node:os";
-import readline from "node:readline";
-import { Conflict, Session, hasBlock, readText } from "./files.mjs";
-import { loadState, ownedBy, saveState, statePath } from "./state.mjs";
-import { stageCapture, unstageCapture } from "./capture.mjs";
-import { KEY_ENV, allTargets, targetById, tilde } from "./targets.mjs";
-
-export const VERSION = "0.2.0";
-const DEFAULT_URL = "https://getwavebrain.com/api/mcp";
-const SETUP_URL = "https://getwavebrain.com/dashboard/setup";
-
-const HELP = `wavebrain ${VERSION}: connect your AI agents to WaveBrain (https://getwavebrain.com)
-
-Usage:
-  npx wavebrain init      Pick agents and connect them to WaveBrain
-  npx wavebrain remove    Undo exactly what init added
-  npx wavebrain status    Show which agents are configured
-
-Options:
-  --agent <name>          Agent to set up (repeatable). Skips the picker. Names:
-                          claude-code, codex, cursor, windsurf, claude-desktop,
-                          vscode, hermes, hermes:<profile>
-  --key <key>             API key (or set ${KEY_ENV}). Create one at
-                          ${SETUP_URL}
-  --dry-run               Show what would change without writing anything
-  --no-hooks              MCP server only: skip the hooks, skill and Hermes plugin
-                          that record actions the model forgets to log
-  --no-claude-permission  Don't pre-approve WaveBrain's tools in Claude Code
-  -y, --yes               Don't ask for confirmation
-  --url <url>             MCP server URL (default ${DEFAULT_URL})
-  -h, --help, -v, --version
-
-For Claude Code, Codex, Cursor and Hermes, init also installs auto-capture: hooks that
-record deploys, pushes, releases, payments, emails and database writes even when the model
-doesn't call log_action, plus a short skill telling it when to log. Secrets are redacted
-before anything is sent. Turn capture off any time with WAVEBRAIN_HOOKS=off.
-
-Every file is backed up to <file>.bak-wavebrain before it's changed.`;
-
-export function parseArgs(argv) {
-  const o = { command: null, agents: [], key: null, dryRun: false, yes: false, claudePermission: null, url: DEFAULT_URL, hooks: true };
-  for (let i = 0; i < argv.length; i++) {
-    const a = argv[i];
-    const val = () => {
-      const v = argv[++i];
-      if (v === undefined || v.startsWith("--")) throw new UsageError(`${a} needs a value`);
-      return v;
-    };
-    if (a === "--agent" || a === "-a") o.agents.push(val());
-    else if (a.startsWith("--agent=")) o.agents.push(a.slice(8));
-    else if (a === "--key") o.key = val();
-    else if (a.startsWith("--key=")) o.key = a.slice(6);
-    else if (a === "--url") o.url = val();
-    else if (a === "--dry-run") o.dryRun = true;
-    else if (a === "--yes" || a === "-y") o.yes = true;
-    else if (a === "--no-claude-permission") o.claudePermission = false;
-    else if (a === "--no-hooks") o.hooks = false;
-    else if (a === "--claude-permission") o.claudePermission = true;
-    else if (a === "-h" || a === "--help") o.command = "help";
-    else if (a === "-v" || a === "--version") o.command = "version";
-    else if (!a.startsWith("-") && !o.command) o.command = a;
-    else throw new UsageError(`Unknown option: ${a}`);
-  }
-  o.command ??= "help";
-  return o;
-}
-
-class UsageError extends Error {}
-
-export const mask = (key) => (key.length > 12 ? `${key.slice(0, 10)}…` : "al_…");
-const validKey = (k) => /^al_[A-Za-z0-9_-]{20,}$/.test(k);
-
-/** io: { out(line), interactive, ask(q), askSecret(q), open(url) } — swapped out in tests. */
-export async function run(argv, ctx = defaultCtx(), io = defaultIo()) {
-  let o;
-  try {
-    o = parseArgs(argv);
-  } catch (e) {
-    if (!(e instanceof UsageError)) throw e;
-    io.out(`${e.message}\n\n${HELP}`);
-    return 2;
-  }
-  try {
-    if (o.command === "help") return io.out(HELP), 0;
-    if (o.command === "version") return io.out(VERSION), 0;
-    if (o.command === "status") return status(ctx, io);
-    if (o.command === "init") return await init(o, ctx, io);
-    if (o.command === "remove" || o.command === "uninstall") return await remove(o, ctx, io);
-    throw new UsageError(`Unknown command: ${o.command}`);
-  } catch (e) {
-    if (e instanceof UsageError) {
-      io.out(e.message);
-      return 2;
-    }
-    throw e;
-  }
-}
-
-function resolveTargets(ctx, names) {
-  return names.map((n) => {
-    const t = targetById(ctx, n.trim().toLowerCase());
-    if (!t) throw new UsageError(`Unknown agent "${n}". Run with --help for the list.`);
-    return t;
-  });
-}
-
-async function init(o, ctx, io) {
-  let targets;
-  if (o.agents.length) {
-    targets = resolveTargets(ctx, o.agents);
-  } else {
-    if (!io.interactive) throw new UsageError("Not a terminal: pass --agent <name> (repeatable) to choose agents.");
-    targets = await pick(ctx, io);
-    if (!targets.length) return io.out("Nothing selected; nothing changed."), 0;
-  }
-
-  let key = o.key ?? null;
-  let keyFromEnv = false;
-  if (!key && ctx.env[KEY_ENV]) {
-    key = ctx.env[KEY_ENV];
-    keyFromEnv = true;
-  }
-  if (!key) {
-    if (!io.interactive) throw new UsageError(`No API key: pass --key or set ${KEY_ENV}. Create one at ${SETUP_URL}`);
-    key = await askKey(io);
-  }
-  key = key.trim();
-  if (!validKey(key)) throw new UsageError(`That doesn't look like a WaveBrain API key (they start with "al_"). Create one at ${SETUP_URL}`);
-
-  let claudePermission = o.claudePermission;
-  if (claudePermission === null && targets.some((t) => t.id === "claude-code")) {
-    if (io.interactive && !o.yes) {
-      io.out(
-        "\nClaude Code: runs started with a narrow --allowedTools (claude -p, browser automations, Hermes delegations)\n" +
-          "can't call log_action unless WaveBrain's tools are pre-approved (mcp__wavebrain__* in ~/.claude/settings.json).",
-      );
-      claudePermission = !/^n/i.test((await io.ask("Allow logging in non-interactive runs? (Y/n) ")).trim());
-    } else claudePermission = true;
-  }
-
-  io.out(`\n${o.dryRun ? "Dry run: nothing will be written.\n" : ""}Using key ${mask(key)} and ${o.url}\n`);
-  const state = loadState(ctx);
-  const session = new Session({ dryRun: o.dryRun });
-  const opts = { url: o.url, key, keyFromEnv, claudePermission, hooks: o.hooks };
-  let failed = 0;
-  const nextSteps = [];
-
-  for (const t of targets) {
-    const owned = ownedBy(state, t.id);
-    try {
-      const results = t.install(ctx, session, opts, owned);
-      for (const r of results) {
-        if (r.kind !== "unchanged" || owned.has(r.ref)) {
-          state.records = state.records.filter((x) => x.ref !== r.ref);
-          state.records.push({
-            ref: r.ref,
-            target: t.id,
-            file: r.file,
-            createdFile: Boolean(r.createdFile),
-            ...(r.pruneFrom !== undefined && { pruneFrom: r.pruneFrom }),
-          });
-        }
-      }
-      const changed = results.filter((r) => r.kind !== "unchanged");
-      if (!changed.length) io.out(`• ${t.label}: already set up`);
-      else {
-        io.out(`✓ ${t.label}`);
-        for (const r of changed) io.out(`    ${o.dryRun ? "would " + verb(r.kind) : verb(r.kind, true)} ${describe(ctx, r)}`);
-        nextSteps.push(`${t.label}: ${t.next}`);
-      }
-      if (t.id === "codex" && keyFromEnv) io.out(`    Codex reads the key from ${KEY_ENV}; keep it exported in your shell.`);
-    } catch (e) {
-      if (!(e instanceof Conflict)) throw e;
-      failed++;
-      io.out(`! ${t.label}: ${e.message}`);
-    }
-  }
-
-  // The hook runner and the key it uses, shared by every agent with auto-capture.
-  if (o.hooks && targets.some((t) => t.capture)) {
-    const results = stageCapture(ctx, session, opts);
-    const changed = results.filter((r) => r.kind !== "unchanged");
-    for (const r of results) {
-      const prev = state.records.find((x) => x.ref === r.ref);
-      state.records = state.records.filter((x) => x.ref !== r.ref);
-      state.records.push({ ref: r.ref, target: "capture", file: r.file, createdFile: prev?.createdFile ?? Boolean(r.createdFile) });
-    }
-    if (changed.length) {
-      io.out(`✓ Auto-capture`);
-      for (const r of changed) io.out(`    ${o.dryRun ? "would " + verb(r.kind) : verb(r.kind, true)} ${describe(ctx, r)}`);
-    }
-  }
-
-  if (!o.dryRun) saveState(ctx, state);
-  if (nextSteps.length) io.out(`\nNext:\n${nextSteps.map((s) => `  ${s}`).join("\n")}`);
-  if (!o.dryRun && session.changes.some((c) => c.backup)) io.out(`\nBackups end in .bak-wavebrain. Undo with: npx wavebrain remove`);
-  return failed ? 1 : 0;
-}
-
-function verb(kind, past) {
-  return { created: past ? "created" : "create", updated: past ? "updated" : "update", deleted: past ? "deleted" : "delete" }[kind] ?? kind;
-}
-
-function describe(ctx, r) {
-  const ref = r.ref;
-  const what = ref.endsWith(":permission")
-    ? " (allow mcp__wavebrain__*)"
-    : ref.endsWith(":env")
-      ? ` (${KEY_ENV})`
-      : ref.endsWith(":hooks")
-        ? " (auto-capture hooks)"
-        : ref.endsWith(":features")
-          ? " ([features] hooks = true)"
-          : ref.endsWith(":skill")
-            ? " (when-to-log skill)"
-            : ref.includes(":plugin:")
-              ? " (WaveBrain plugin)"
-              : ref.endsWith(":plugin-enabled")
-                ? " (plugins.enabled: wavebrain)"
-                : ref === "capture:config"
-                  ? " (API key for the hooks)"
-                  : ref.startsWith("capture:")
-                    ? " (hook runner)"
-                    : " (wavebrain server)";
-  return tilde(ctx, r.file) + what;
-}
-
-async function pick(ctx, io) {
-  const targets = allTargets(ctx);
-  const found = targets.filter((t) => t.detect(ctx));
-  const list = found.length ? found : targets;
-  io.out(found.length ? "Agents found on this machine:" : "No agents detected. Available:");
-  list.forEach((t, i) => io.out(`  ${i + 1}) ${t.label}${t.configured(ctx) ? "  (already has wavebrain)" : ""}`));
-  const answer = (await io.ask("\nWhich should log to WaveBrain? Numbers separated by spaces, 'all', or Enter for none: ")).trim();
-  if (!answer) return [];
-  if (answer.toLowerCase() === "all") return list;
-  const picked = [];
-  for (const part of answer.split(/[\s,]+/)) {
-    const t = list[Number(part) - 1];
-    if (!t) throw new UsageError(`"${part}" isn't one of the numbers above.`);
-    if (!picked.includes(t)) picked.push(t);
-  }
-  io.out(`Selected: ${picked.map((t) => t.label).join(", ")}`);
-  return picked;
-}
-
-async function askKey(io) {
-  io.out(`\nPaste a WaveBrain API key, or press Enter to open ${SETUP_URL} and create one.`);
-  let key = (await io.askSecret("API key: ")).trim();
-  if (!key) {
-    io.open(SETUP_URL);
-    io.out(`Opened ${SETUP_URL}. Click "Create a key", copy it, and paste it here.`);
-    key = (await io.askSecret("API key: ")).trim();
-  }
-  return key;
-}
-
-async function remove(o, ctx, io) {
-  const state = loadState(ctx);
-  const ids = o.agents.length
-    ? resolveTargets(ctx, o.agents).map((t) => t.id)
-    : [...new Set([...state.records.map((r) => r.target), ...allTargets(ctx).filter((t) => markerOnly(ctx, t)).map((t) => t.id)])];
-  if (!ids.length) return io.out("Nothing to remove: wavebrain init hasn't added anything here."), 0;
-  const targets = ids.map((id) => targetById(ctx, id)).filter(Boolean);
-
-  if (io.interactive && !o.yes && !o.dryRun) {
-    const ok = (await io.ask(`Remove WaveBrain from ${targets.map((t) => t.label).join(", ")}? (y/N) `)).trim();
-    if (!/^y/i.test(ok)) return io.out("Nothing changed."), 0;
-  }
-
-  const session = new Session({ dryRun: o.dryRun });
-  for (const t of targets) {
-    try {
-      const results = t.uninstall(ctx, session, ownedBy(state, t.id)).filter((r) => r.kind !== "unchanged");
-      if (!results.length) io.out(`• ${t.label}: nothing added by wavebrain init`);
-      else {
-        io.out(`✓ ${t.label}`);
-        for (const r of results) io.out(`    ${o.dryRun ? "would " + verb(r.kind) : verb(r.kind, true)} ${describe(ctx, r)}`);
-      }
-      state.records = state.records.filter((r) => r.target !== t.id);
-    } catch (e) {
-      if (!(e instanceof Conflict)) throw e;
-      io.out(`! ${t.label}: ${e.message}`);
-    }
-  }
-  // Last agent with auto-capture gone: remove the hook runner and its key too.
-  const captureLeft = state.records.some((r) => r.target !== "capture" && targetById(ctx, r.target)?.capture);
-  if (!captureLeft && state.records.some((r) => r.target === "capture")) {
-    const results = unstageCapture(ctx, session).filter((r) => r.kind !== "unchanged");
-    if (results.length) {
-      io.out(`✓ Auto-capture`);
-      for (const r of results) io.out(`    ${o.dryRun ? "would " + verb(r.kind) : verb(r.kind, true)} ${tilde(ctx, r.file)}`);
-    }
-    state.records = state.records.filter((r) => r.target !== "capture");
-  }
-  if (!o.dryRun) saveState(ctx, state);
-  return 0;
-}
-
-// Codex and Hermes edits are marked in the file itself, so they're found even without the state file.
-function markerOnly(ctx, t) {
-  return (t.id === "codex" || t.id.startsWith("hermes")) && t.configured(ctx) && t.files(ctx).some((f) => hasMarker(f));
-}
-const hasMarker = (f) => hasBlock(readText(f));
-
-function status(ctx, io) {
-  const state = loadState(ctx);
-  io.out("Agent                         Detected  WaveBrain");
-  for (const t of allTargets(ctx)) {
-    const ours = state.records.some((r) => r.target === t.id) || markerOnly(ctx, t);
-    const conf = t.configured(ctx) ? (ours ? "configured (by wavebrain init)" : "configured") : "not configured";
-    io.out(`${t.label.padEnd(30)}${(t.detect(ctx) ? "yes" : "no").padEnd(10)}${conf}`);
-  }
-  io.out(`\nState: ${tilde(ctx, statePath(ctx))}`);
-  return 0;
-}
-
-export function defaultCtx() {
-  return { home: os.homedir(), env: process.env, platform: process.platform };
-}
-
-export function defaultIo() {
-  const interactive = Boolean(process.stdin.isTTY && process.stdout.isTTY);
-  const ask = (q, secret = false) =>
-    new Promise((resolve) => {
-      const rl = readline.createInterface({ input: process.stdin, output: process.stdout, terminal: true });
-      if (secret) {
-        process.stdout.write(q);
-        rl._writeToOutput = (s) => {
-          if (s.includes("\n") || s.includes("\r")) process.stdout.write("\n");
-          else process.stdout.write("*".repeat(s.length));
-        };
-      }
-      rl.question(secret ? "" : q, (a) => {
-        rl.close();
-        resolve(a);
-      });
-    });
-  return {
-    interactive,
-    out: (line) => console.log(line),
-    ask: (q) => ask(q),
-    askSecret: (q) => ask(q, true),
-    open(url) {
-      const [cmd, args] =
-        process.platform === "darwin" ? ["open", [url]] : process.platform === "win32" ? ["cmd", ["/c", "start", "", url]] : ["xdg-open", [url]];
-      try {
-        spawn(cmd, args, { stdio: "ignore", detached: true }).on("error", () => {}).unref();
-      } catch {}
-    },
-  };
-}

package/src/files.mjs

@@ -1,292 +0,0 @@
-// File editing with backups, atomic writes and dry runs, plus small pure text transforms for the
-// formats agents use (JSON, TOML, YAML, .env). Transforms never see or return anything but text.
-
-import fs from "node:fs";
-import path from "node:path";
-
-export const BACKUP_SUFFIX = ".bak-wavebrain";
-export const MARK_START = ">>> wavebrain (added by npx wavebrain init)";
-export const MARK_END = "<<< wavebrain";
-
-export class Conflict extends Error {}
-
-export function readText(file) {
-  try {
-    return fs.readFileSync(file, "utf8");
-  } catch (e) {
-    if (e.code === "ENOENT") return null;
-    throw e;
-  }
-}
-
-/** Applies edits to files; records what changed. In a dry run nothing is written. */
-export class Session {
-  constructor({ dryRun = false } = {}) {
-    this.dryRun = dryRun;
-    this.changes = [];
-  }
-
-  /**
-   * fn(text|null) returns the new text, null to delete the file, or the same text for no change.
-   * Throws Conflict to refuse. Returns "created" | "updated" | "deleted" | "unchanged".
-   */
-  edit(file, fn) {
-    const before = readText(file);
-    const after = fn(before);
-    if (after === before) return "unchanged";
-    const kind = before === null ? "created" : after === null ? "deleted" : "updated";
-    // Files that are entirely ours (in a wavebrain folder) aren't backed up: a backup of
-    // ~/.wavebrain/config.json would leave the API key behind after `remove`.
-    const ownedFile = /[\\/](\.wavebrain|wavebrain)[\\/]/.test(file);
-    const backup = before === null || ownedFile ? null : file + BACKUP_SUFFIX;
-    if (!this.dryRun) {
-      let mode;
-      if (before !== null) {
-        mode = fs.statSync(file).mode & 0o777;
-        if (backup) {
-          fs.copyFileSync(file, backup);
-          fs.chmodSync(backup, mode);
-        }
-      }
-      if (after === null) {
-        fs.unlinkSync(file);
-        // Folders that only ever held our files (skills/wavebrain, plugins/wavebrain, ~/.wavebrain/hooks).
-        const dir = path.dirname(file);
-        if (/(^|[\\/])(wavebrain|hooks)$/.test(dir) && /wavebrain/.test(dir)) {
-          try {
-            fs.rmdirSync(dir);
-          } catch {}
-        }
-      } else writeAtomic(file, after, mode);
-    }
-    this.changes.push({ file, kind, backup });
-    return kind;
-  }
-}
-
-function writeAtomic(file, text, mode) {
-  fs.mkdirSync(path.dirname(file), { recursive: true });
-  const tmp = `${file}.wavebrain-tmp-${process.pid}`;
-  fs.writeFileSync(tmp, text, { mode: mode ?? 0o600 });
-  if (mode !== undefined) fs.chmodSync(tmp, mode);
-  fs.renameSync(tmp, file);
-}
-
-// ---------- JSON ----------
-
-export function parseJson(text, file) {
-  if (text === null || text.trim() === "") return {};
-  try {
-    const v = JSON.parse(text);
-    if (!v || typeof v !== "object" || Array.isArray(v)) throw new Error("not an object");
-    return v;
-  } catch {
-    throw new Conflict(`couldn't parse ${file} as plain JSON (comments or trailing commas?); add it by hand instead`);
-  }
-}
-
-const dump = (obj) => JSON.stringify(obj, null, 2) + "\n";
-
-export function getIn(obj, keys) {
-  let cur = obj;
-  for (const k of keys) {
-    if (!cur || typeof cur !== "object") return undefined;
-    cur = cur[k];
-  }
-  return cur;
-}
-
-/** Sets obj[...keys] = value. */
-export function jsonSet(text, keys, value, file) {
-  const obj = parseJson(text, file);
-  let cur = obj;
-  for (const k of keys.slice(0, -1)) {
-    if (cur[k] === undefined) cur[k] = {};
-    if (!cur[k] || typeof cur[k] !== "object" || Array.isArray(cur[k])) {
-      throw new Conflict(`${file}: "${k}" isn't an object; add it by hand instead`);
-    }
-    cur = cur[k];
-  }
-  const last = keys[keys.length - 1];
-  if (JSON.stringify(cur[last]) === JSON.stringify(value)) return text;
-  cur[last] = value;
-  return dump(obj);
-}
-
-/** Index of the first key in the path that doesn't exist yet (keys.length if all do). */
-export function firstMissing(text, keys, file) {
-  let cur = parseJson(text, file);
-  for (let i = 0; i < keys.length; i++) {
-    if (!cur || typeof cur !== "object" || !(keys[i] in cur)) return i;
-    cur = cur[keys[i]];
-  }
-  return keys.length;
-}
-
-/**
- * Deletes obj[...keys], then any now-empty parents from index `pruneFrom` down (parents we created).
- * Returns null (delete the file) if that leaves only empty objects and deleteIfEmpty.
- */
-export function jsonDelete(text, keys, file, { deleteIfEmpty = false, pruneFrom = keys.length } = {}) {
-  if (text === null) return null;
-  const obj = parseJson(text, file);
-  const parent = getIn(obj, keys.slice(0, -1));
-  const last = keys[keys.length - 1];
-  if (!parent || typeof parent !== "object" || !(last in parent)) return text;
-  delete parent[last];
-  for (let i = keys.length - 2; i >= pruneFrom; i--) {
-    const v = getIn(obj, keys.slice(0, i + 1));
-    const empty = Array.isArray(v) ? v.length === 0 : v && typeof v === "object" && Object.keys(v).length === 0;
-    if (!empty) break;
-    delete getIn(obj, keys.slice(0, i))[keys[i]];
-  }
-  if (deleteIfEmpty && isEmptyTree(obj)) return null;
-  return dump(obj);
-}
-
-const isEmptyTree = (v) => v && typeof v === "object" && !Array.isArray(v) && Object.values(v).every(isEmptyTree);
-
-/** Adds a string to the array at keys if missing. */
-export function jsonArrayAdd(text, keys, item, file) {
-  const obj = parseJson(text, file);
-  const arr = getIn(obj, keys);
-  if (arr !== undefined && !Array.isArray(arr)) throw new Conflict(`${file}: ${keys.join(".")} isn't a list`);
-  if (arr?.includes(item)) return text;
-  return jsonSet(text, keys, [...(arr ?? []), item], file);
-}
-
-export function jsonArrayRemove(text, keys, item, file, opts) {
-  if (text === null) return null;
-  const obj = parseJson(text, file);
-  const arr = getIn(obj, keys);
-  if (!Array.isArray(arr) || !arr.includes(item)) return text;
-  const rest = arr.filter((x) => x !== item);
-  if (rest.length || (opts?.pruneFrom ?? keys.length) >= keys.length) return jsonSet(text, keys, rest, file);
-  return jsonDelete(text, keys, file, opts);
-}
-
-// ---------- marked text blocks (TOML, YAML, .env) ----------
-
-const markerLine = (indent, mark) => `${indent}# ${mark}`;
-
-/** Index range [start, end] of our marked block, or null. */
-export function findBlock(lines) {
-  const start = lines.findIndex((l) => l.trim().startsWith(`# ${MARK_START}`));
-  if (start < 0) return null;
-  const end = lines.findIndex((l, i) => i > start && l.trim() === `# ${MARK_END}`);
-  return end < 0 ? null : [start, end];
-}
-
-export function hasBlock(text) {
-  return text !== null && findBlock(text.split("\n")) !== null;
-}
-
-/** The text with our block removed (and a trailing blank line we added). */
-export function removeBlock(text, { restore } = {}) {
-  if (text === null) return null;
-  const lines = text.split("\n");
-  const range = findBlock(lines);
-  if (!range) return text;
-  const [s, e] = range;
-  const was = lines[s].match(/; was "(.*)"\)?\s*$/);
-  const replacement = was ? [lines[s].match(/^\s*/)[0] + was[1].replace(/\\"/g, '"')] : [];
-  lines.splice(s, e - s + 1, ...replacement);
-  // Drop one blank line left behind where we appended with a separator.
-  if (!was && s > 0 && lines[s - 1] === "" && (s === lines.length || lines[s] === "")) lines.splice(s - 1, 1);
-  const out = lines.join("\n");
-  return restore === "delete-if-empty" && out.trim() === "" ? null : out;
-}
-
-function block(indent, body) {
-  return [markerLine(indent, MARK_START), ...body, markerLine(indent, MARK_END)];
-}
-
-/** Appends (or replaces) our block at the end of the file. */
-export function upsertAppendedBlock(text, body) {
-  const base = removeBlock(text) ?? "";
-  const trimmed = base.replace(/\n+$/, "");
-  const lines = block("", body);
-  return (trimmed ? trimmed + "\n\n" : "") + lines.join("\n") + "\n";
-}
-
-// ---------- TOML (Codex) ----------
-
-export function tomlString(s) {
-  return JSON.stringify(s);
-}
-
-export function tomlUpsertServer(text, name, entries) {
-  const without = removeBlock(text) ?? "";
-  if (new RegExp(`^\\s*\\[mcp_servers\\.${name}(\\.|\\])`, "m").test(without)) {
-    throw new Conflict(`a [mcp_servers.${name}] server is already configured (not by wavebrain init); left as is`);
-  }
-  const body = [`[mcp_servers.${name}]`, ...entries.map(([k, v]) => `${k} = ${v}`)];
-  const next = upsertAppendedBlock(text, body);
-  return next === text ? text : next;
-}
-
-// ---------- YAML (Hermes) ----------
-
-/** Adds `name` under top-level mcp_servers, keeping the file's own indentation. */
-export function yamlUpsertServer(text, name, bodyLines) {
-  const base = removeBlock(text) ?? "";
-  const lines = base.split("\n");
-  const keyIdx = lines.findIndex((l) => /^mcp_servers:/.test(l));
-  const childRe = new RegExp(`^\\s+${name}:`);
-
-  if (keyIdx < 0) {
-    const trimmed = base.replace(/\n+$/, "");
-    const out = block("", ["mcp_servers:", ...indentAll(bodyLines(2, name), 2)]);
-    const next = (trimmed ? trimmed + "\n\n" : "") + out.join("\n") + "\n";
-    return next === text ? text : next;
-  }
-
-  const keyLine = lines[keyIdx];
-  if (/^mcp_servers:\s*(#.*)?$/.test(keyLine)) {
-    // Block mapping: find its children and their indent.
-    let end = keyIdx + 1;
-    let indent = 0;
-    while (end < lines.length && (lines[end].trim() === "" || /^\s/.test(lines[end]))) {
-      if (!indent && lines[end].trim() && !lines[end].trim().startsWith("#")) indent = lines[end].match(/^\s*/)[0].length;
-      if (childRe.test(lines[end]) && lines[end].match(/^\s*/)[0].length === (indent || 2)) {
-        throw new Conflict(`mcp_servers already has a "${name}" server (not added by wavebrain init); left as is`);
-      }
-      end++;
-    }
-    indent ||= 2;
-    lines.splice(keyIdx + 1, 0, ...block(" ".repeat(indent), indentAll(bodyLines(indent, name), indent)));
-    const next = lines.join("\n");
-    return next === text ? text : next;
-  }
-
-  if (/^mcp_servers:\s*(\{\s*\}|null|~)?\s*$/.test(keyLine)) {
-    const was = keyLine.trim().replace(/"/g, '\\"');
-    const out = [
-      `# ${MARK_START}; was "${was}"`,
-      "mcp_servers:",
-      ...indentAll(bodyLines(2, name), 2),
-      `# ${MARK_END}`,
-    ];
-    lines.splice(keyIdx, 1, ...out);
-    return lines.join("\n");
-  }
-  throw new Conflict("mcp_servers is written in a form this tool can't edit safely; add it by hand instead");
-}
-
-function indentAll(lines, n) {
-  return lines.map((l) => " ".repeat(n) + l);
-}
-
-// ---------- .env ----------
-
-export function envUpsert(text, key, value) {
-  const without = removeBlock(text) ?? "";
-  const existing = without.match(new RegExp(`^\\s*(?:export\\s+)?${key}=(.*)$`, "m"));
-  if (existing) {
-    const v = existing[1].trim().replace(/^["']|["']$/g, "");
-    if (v === value) return text; // Already set to this key by the user: leave it.
-    throw new Conflict(`${key} is already set to a different key (not by wavebrain init); left as is`);
-  }
-  const next = upsertAppendedBlock(text, [`${key}=${value}`]);
-  return next === text ? text : next;
-}

package/src/state.mjs

@@ -1,36 +0,0 @@
-// What `wavebrain init` added, so `wavebrain remove` undoes exactly that and nothing the user
-// configured themselves. Stores file paths and entry names only, never keys.
-
-import fs from "node:fs";
-import path from "node:path";
-
-export function statePath(ctx) {
-  return path.join(ctx.home, ".wavebrain", "installed.json");
-}
-
-export function loadState(ctx) {
-  try {
-    const s = JSON.parse(fs.readFileSync(statePath(ctx), "utf8"));
-    return Array.isArray(s.records) ? s : { version: 1, records: [] };
-  } catch {
-    return { version: 1, records: [] };
-  }
-}
-
-export function saveState(ctx, state) {
-  const p = statePath(ctx);
-  if (!state.records.length) {
-    fs.rmSync(p, { force: true });
-    try {
-      fs.rmdirSync(path.dirname(p));
-    } catch {}
-    return;
-  }
-  fs.mkdirSync(path.dirname(p), { recursive: true });
-  fs.writeFileSync(p, JSON.stringify(state, null, 2) + "\n", { mode: 0o600 });
-}
-
-/** Records owned by one target, keyed by ref. */
-export function ownedBy(state, targetId) {
-  return new Map(state.records.filter((r) => r.target === targetId).map((r) => [r.ref, r]));
-}