Something went wrong. Try again.
Monorepo for Aesthetic.Computer aesthetic.computer
Something went wrong. Try again.
JavaScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410#!/usr/bin/env node// memory-mcp.mjs — an MCP server over the local-first agent memory, so any// session can SEARCH OUR PROMPT HISTORY across the fleet by name, instead of// remembering that `node memory/cli.mjs` exists on each Mac separately.//// The stores are local-first and encrypted per machine (AES-256-GCM under a key// that never leaves the Mac that wrote it), so there is no central index to// query. The fan-out inverts that: we ship the QUERY to each machine and let it// decrypt and search its own store, then merge the (redacted) hits here. No key// ever crosses the network, and a machine that's asleep simply drops out of the// results instead of failing the search.//// Machines come from the fleet registry first, with the older frame/puppet// registry layered on top for SSH/CDP overrides. A machine's name doubles as// its ssh host unless it sets `sshHost`/`ssh.alias`/`ssh`; set `repo` or// `repoPath` to override where the checkout lives.//// Hand-rolled JSON-RPC over stdio, matching the house style of// slab/bin/frame-mcp.mjs and artery/emacs-mcp.mjs — no SDK, node builtins only.import { execFile } from "node:child_process";import { existsSync, readFileSync } from "node:fs";import { homedir, hostname } from "node:os";import { dirname, join } from "node:path";import { fileURLToPath } from "node:url";
import { httpPort, serveHttp, serveStdio } from "../toolchain/mcp/http-front.mjs";import { shortWhen, toon } from "../shared/toon.mjs";
const HERE = dirname(fileURLToPath(import.meta.url));const HOME = homedir();const REPO = join(HERE, "..");const FLEET_REGISTRIES = [ process.env.FLEET_MACHINES, join(HOME, "aesthetic-computer-vault", "machines.normalized.json"), join(HOME, "aesthetic-computer-vault", "machines.json"), join(REPO, "aesthetic-computer-vault", "machines.normalized.json"), join(REPO, "aesthetic-computer-vault", "machines.json"),].filter(Boolean);const PUPPET_REGISTRY = process.env.SLAB_PUPPET_CONFIG || join(HOME, ".config", "slab", "puppet.json");const LEDGER_LOCAL = join(HOME, ".config", "slab", "ledger", "local.json");const DEFAULT_REPO = "~/aesthetic-computer";const SELF = localHostName(); // e.g. neo/blueberry — what `local` actually is
function readJson(path) { if (!path || !existsSync(path)) return null; try { return JSON.parse(readFileSync(path, "utf8")); } catch { return null; }}
function localHostName() { const ledger = readJson(LEDGER_LOCAL); if (ledger?.host) return ledger.host; return hostname().split(".")[0];}
function loadFleet() { for (const path of FLEET_REGISTRIES) { const data = readJson(path); const machines = data?.machines || data; if (machines && typeof machines === "object") return { path, machines }; } return { path: null, machines: {} };}
function loadPuppet() { const data = readJson(PUPPET_REGISTRY); return data?.machines || data || {};}
function sshHost(name, spec) { const ssh = spec.ssh; return ( spec.sshHost || (typeof ssh === "string" ? ssh : ssh?.alias || ssh?.host) || spec.host || spec.tailscale?.name || spec.hostname || name );}
function repoPath(spec) { return spec.repo || spec.repoPath || DEFAULT_REPO;}
function memoryCandidate(name, spec, puppet) { if (name === SELF || spec.local || puppet[name]) return true; const caps = spec.capabilities || []; return ( caps.includes("macos-automation") || caps.includes("tailnet-api") || caps.includes("mail") || spec.designation === "agent-endpoint" );}
function normalizeMachine(name, spec = {}, puppet = {}) { return { ...spec, ...puppet, sourceName: name, local: spec.local || name === SELF || puppet.local === true, sshHost: sshHost(name, { ...spec, ...puppet }), repo: repoPath({ ...spec, ...puppet }), };}
// The fleet, as the canonical registry knows it. `local` is always present —// this Mac searches its own store in-process, no ssh.function machines() { const { machines: fleet } = loadFleet(); const puppet = loadPuppet(); const out = {};
for (const [name, spec] of Object.entries(fleet)) { if (name.startsWith("_")) continue; if (!memoryCandidate(name, spec, puppet)) continue; out[name] = normalizeMachine(name, spec, puppet[name]); }
// Puppet-only machines are still valid; they may be work boxes not promoted // into the fleet registry yet. for (const [name, spec] of Object.entries(puppet)) { if (name.startsWith("_") || out[name]) continue; out[name] = normalizeMachine(name, {}, spec); }
// This Mac already answers as `local`. If the fleet list also names it, drop // the named copy — otherwise its store gets searched twice and every hit // shows up doubled. delete out[SELF]; out.local = normalizeMachine("local", { local: true }, {}); return out;}
function registryLabel() { const { path } = loadFleet(); return `${path || "(no fleet registry)"}; puppet overrides: ${PUPPET_REGISTRY}`;}
function isLocal(name, spec) { return name === "local" || spec?.local === true;}
// Run the memory CLI somewhere and get its --json envelope back.//// Remote commands go over stdin to `bash -s` rather than as an ssh argument:// the remote runs a login shell, and blueberry's is fish, which can't parse the// bash we'd otherwise send. The query is base64'd across the wire so no quoting// or shell metacharacter in a prompt ("$(", backticks, quotes — all common in// our prompts!) can reach the remote shell.function runCli(name, spec, args, { timeoutMs = 20000 } = {}) { const argv = ["memory/cli.mjs", ...args, "--json"];
if (isLocal(name, spec)) { return execFileAsync("node", [join(HERE, "cli.mjs"), ...args, "--json"], { cwd: REPO, timeoutMs, }); }
const repo = spec.repo || DEFAULT_REPO; const host = spec.sshHost || name; const encoded = Buffer.from(JSON.stringify(argv), "utf8").toString("base64"); const script = [ `cd ${repo} || exit 3`, `printf %s '${encoded}' | base64 --decode > /tmp/.memory-mcp-argv.json`, `node -e 'const a=require("/tmp/.memory-mcp-argv.json");require("child_process").spawnSync(process.execPath,a,{stdio:"inherit"})'`, `rm -f /tmp/.memory-mcp-argv.json`, ].join("\n");
return execFileAsync( "ssh", [ "-o", "ControlMaster=auto", "-o", `ControlPath=${join(HOME, ".ssh", `cm-${host}`)}`, "-o", "ControlPersist=300", "-o", "BatchMode=yes", "-o", "ConnectTimeout=6", host, "bash -s", ], { input: script, timeoutMs }, );}
function execFileAsync(file, args, { cwd, input, timeoutMs } = {}) { return new Promise((resolve, reject) => { const child = execFile( file, args, { cwd, timeout: timeoutMs, encoding: "utf8", maxBuffer: 32 * 1024 * 1024 }, (err, stdout, stderr) => { if (err && !stdout) return reject(new Error((stderr || err.message).trim())); resolve(stdout); }, ); if (input != null) { child.stdin.write(input); child.stdin.end(); } });}
// Ask every machine at once; an unreachable one contributes an error line, not// a failure. A search that silently dropped a sleeping Mac would be worse than// useless — you'd conclude the prompt was never written.// A login shell on a fleet Mac is not a clean pipe: neo's node version manager// greets every non-interactive command with the active fnm Node alias on// STDOUT. So take the JSON envelope, not the whole stream — parse from the first// brace/bracket. If there's none, the remote said something that isn't a result// (usage text, a stack trace); surface its first line as the error.function parseEnvelope(stdout) { const at = stdout.search(/[[{]/); if (at < 0) throw new Error(stdout.trim().split("\n")[0] || "no output"); return JSON.parse(stdout.slice(at));}
async function fanOut(target, args) { const fleet = machines(); const names = target ? [target] : Object.keys(fleet); const unknown = names.filter((name) => !fleet[name]); if (unknown.length) throw new Error(`unknown machine: ${unknown.join(", ")} (have: ${Object.keys(fleet).join(", ")})`);
const settled = await Promise.all( names.map(async (name) => { try { return { name, result: parseEnvelope(await runCli(name, fleet[name], args)) }; } catch (error) { return { name, error: String(error.message || error).split("\n")[0] }; } }), ); return settled;}
function flag(name, value) { return value == null || value === false || value === "" ? [] : value === true ? [`--${name}`] : [`--${name}`, String(value)];}
async function toolSearch(a) { if (!a.query) throw new Error("`query` is required"); const args = [ "search", "--query", String(a.query), ...flag("limit", a.limit ?? 10), ...flag("since", a.since), ...flag("project", a.project), ...flag("role", a.role), ...flag("regex", a.regex), ];
const results = await fanOut(a.machine, args); const lines = []; const failures = []; const hits = [];
for (const entry of results) { if (entry.error) { failures.push(` ${entry.name}: ${entry.error}`); continue; } // "[local]" is meaningless once a result is read back later — say which Mac. const label = entry.name === "local" ? SELF : entry.name; for (const hit of entry.result.hits || []) hits.push({ ...hit, machine: label }); }
hits.sort((a, b) => new Date(b.when || 0) - new Date(a.when || 0));
// One TOON row per hit; `ref` is session#seq, what memory_search takes back. const rows = hits.map((hit) => ({ when: shortWhen(hit.when), machine: hit.machine, role: hit.role || "?", ref: `${hit.session_id}#${hit.seq}`, snippet: hit.snippet, })); const notes = [`query "${a.query}" over ${results.length - failures.length} machine(s)`]; if (failures.length) notes.push(`UNREACHABLE (not searched): ${failures.map((f) => f.trim()).join("; ")}`); if (!hits.length) notes.push("nothing — try fewer words, regex:true, or a wider since"); lines.push(toon("hits", rows, ["when", "machine", "role", "ref", "snippet"], { note: notes.join(" · ") }));
return [{ type: "text", text: lines.join("\n") }];}
async function toolSessions(a) { const args = ["list", ...flag("limit", a.limit ?? 15), ...flag("project", a.project)]; const results = await fanOut(a.machine, args); const rows = []; const failures = [];
for (const entry of results) { if (entry.error) { failures.push(`${entry.name}: ${entry.error}`); continue; } const machine = entry.name === "local" ? SELF : entry.name; for (const s of entry.result || []) { rows.push({ machine, updated: shortWhen(s.updated_at), sortTime: Date.parse(s.updated_at) || 0, id: s.session_id, seq: s.last_seq, project: s.project, title: s.title }); } } rows.sort((x, y) => y.sortTime - x.sortTime); const notes = []; if (failures.length) notes.push(`UNREACHABLE (not listed): ${failures.join("; ")}`); notes.push("search inside one: memory_search with query (+ project/since)"); return [{ type: "text", text: toon("sessions", rows, ["machine", "updated", "id", "seq", "project", "title"], { note: notes.join(" · ") }) }];}
async function toolMachines() { const fleet = machines(); const lines = Object.entries(fleet).map(([name, spec]) => isLocal(name, spec) ? `local (= ${SELF}, this Mac — searched in-process, no ssh)` : `${name} ssh=${spec.sshHost || name} repo=${spec.repo || DEFAULT_REPO}`, ); lines.push(`\nregistry: ${registryLabel()} (fleet first; puppet supplies overrides and legacy machines)`); return [{ type: "text", text: lines.join("\n") }];}
const TOOLS = [ { name: "memory_search", description: "Search OUR PROMPT HISTORY — every prompt @jeffrey has typed to Claude/Codex — across the fleet (this Mac + neo + any other registered machine). Each machine decrypts and searches its own local store, so nothing is missed just because it was said on a different Mac. Use this to answer 'when did we decide X', 'what did I say about Y', 'have we hit this bug before'. Results are snippets with machine, session id, and timestamp; secrets are redacted. Unreachable machines are named explicitly so you know what ISN'T covered.", inputSchema: { type: "object", properties: { query: { type: "string", description: "Text to find (case-insensitive substring, or a regex with regex:true)." }, machine: { type: "string", description: "Limit to one machine (see memory_machines). Omit to search the whole fleet." }, limit: { type: "number", description: "Max hits per machine (default 10)." }, since: { type: "string", description: "Only prompts at/after this time — ISO or YYYY-MM-DD." }, project: { type: "string", description: "Filter to one project." }, role: { type: "string", description: "Filter by role, e.g. 'user' for things @jeffrey typed." }, regex: { type: "boolean", description: "Treat query as a regular expression." }, }, required: ["query"], }, }, { name: "memory_sessions", description: "List recent agent sessions (title, project, last activity) per machine — the index behind memory_search. Use it to find a session id to search within, or to see what's been worked on lately and where.", inputSchema: { type: "object", properties: { machine: { type: "string", description: "One machine, or omit for the whole fleet." }, limit: { type: "number", description: "Sessions per machine (default 15)." }, project: { type: "string", description: "Filter to one project." }, }, }, }, { name: "memory_machines", description: "List the machines whose prompt history is searchable, and where each one's store lives. Machines come from the shared frame/puppet registry at ~/.config/slab/puppet.json.", inputSchema: { type: "object", properties: {} }, },];
async function callTool(name, args) { switch (name) { case "memory_search": return toolSearch(args || {}); case "memory_sessions": return toolSessions(args || {}); case "memory_machines": return toolMachines(); default: throw new Error(`Unknown tool: ${name}`); }}
async function handleMessage(message) { const { id, method, params } = message; try { switch (method) { case "initialize": return { jsonrpc: "2.0", id, result: { protocolVersion: "2024-11-05", capabilities: { tools: {} }, serverInfo: { name: "memory-mcp", version: "1.0.0" }, }, }; case "initialized": case "notifications/initialized": return null; // notification — no response case "ping": return { jsonrpc: "2.0", id, result: {} }; case "tools/list": return { jsonrpc: "2.0", id, result: { tools: TOOLS } }; case "tools/call": { const content = await callTool(params?.name, params?.arguments); return { jsonrpc: "2.0", id, result: { content } }; } default: return { jsonrpc: "2.0", id, error: { code: -32601, message: `Method not found: ${method}` } }; } } catch (error) { // Tool failures come back as an error-flagged result (visible to the model) // rather than a protocol error, so the agent can read and react to them. if (method === "tools/call") { return { jsonrpc: "2.0", id, result: { isError: true, content: [{ type: "text", text: String(error.message || error) }] } }; } return { jsonrpc: "2.0", id, error: { code: -32000, message: String(error.message || error) } }; }}
const port = httpPort(process.argv, 7772); // 7769 is puppet'sif (port) serveHttp({ handleMessage, port, banner: "🧠 memory-mcp shared daemon" });else serveStdio({ handleMessage, banner: "🧠 memory-mcp server started (memory_search, memory_sessions, memory_machines)" });