airgate 0.2.2 → 0.2.3

package/README.md

@@ -25,7 +25,7 @@ macOS and Linux, Node.js 20.10 or newer.
 airgate login     # approve this terminal in your browser
 airgate           # the full-screen terminal app
 airgate launch    # a new pod
-airgate up        # move the session you're in to a pod; airgate down brings the work back
+airgate up        # move the session you're in to a pod (files, conversation, memory); airgate down brings it back
 airgate help      # every command
 ```
 
@@ -68,7 +68,7 @@ Guide: https://airgate.sh/docs/teams
 ## For your agent
 
 Claude Code or Codex on your laptop can send work to pods on its own: one task in a fresh pod
-with the result back here, or the whole project and conversation up and back.
+with the result back here, or the whole project, conversation and memory up and back.
 
 ```sh
 airgate run "get the failing tests passing"   # a fresh pod; the result comes back as a patch

package/dist/airgate.mjs

@@ -813,6 +813,9 @@ var Api = class {
   resumeAgentSession(id, body) {
     return this.request("POST", `/v1/pods/${enc(id)}/agent-session/resume`, { body });
   }
+  exportAgentMemory(id, dir) {
+    return this.request("GET", `/v1/pods/${enc(id)}/agent-memory?${new URLSearchParams({ dir })}`);
+  }
   exportAgentSession(id, q) {
     const qs = new URLSearchParams({ agent: q.agent, dir: q.dir, ...q.id ? { id: q.id } : {} });
     return this.request("GET", `/v1/pods/${enc(id)}/agent-session?${qs}`);
@@ -7559,7 +7562,7 @@ import { rmSync as rmSync2, writeFileSync as writeFileSync2 } from "node:fs";
 import { resolve as resolve4 } from "node:path";
 
 // src/version.ts
-var VERSION = "0.2.2";
+var VERSION = "0.2.3";
 
 // src/commands/team.ts
 async function checkSimilar(ctx, task2) {
@@ -8754,6 +8757,79 @@ function bringBack(sent, podContent, now = /* @__PURE__ */ new Date()) {
   atomic(file, body);
   return { agent: sent.agent, id, file, outcome: "new", sha256: sha256(body) };
 }
+var TEXTISH = /\.(md|markdown|txt|json|jsonl|ya?ml|toml)$/i;
+function findClaudeMemory(home, projectDir, env = {}) {
+  for (const p of spellings(projectDir)) {
+    const dir = join3(claudeHome(home, env), "projects", claudeKey(p), "memory");
+    try {
+      if (statSync(dir).isDirectory() && listFiles(dir).length) return dir;
+    } catch {
+    }
+  }
+  return null;
+}
+function listFiles(dir, rel = "") {
+  const out = [];
+  for (const x of readdirSync2(join3(dir, rel), { withFileTypes: true })) {
+    const r = rel ? `${rel}/${x.name}` : x.name;
+    if (x.isDirectory()) out.push(...listFiles(dir, r));
+    else if (x.isFile()) out.push(r);
+  }
+  return out;
+}
+async function packMemory(dir, projectDir, podPath, limit = UPLOAD_LIMIT) {
+  const froms = spellings(projectDir);
+  const stage = mkdtempSync(join3(tmpdir(), "airgate-memory-"));
+  try {
+    const files2 = {};
+    for (const rel of listFiles(dir)) {
+      const body = readFileSync4(join3(dir, rel));
+      files2[rel] = sha256(body);
+      const out = join3(stage, "memory", rel);
+      mkdirSync2(dirname(out), { recursive: true });
+      writeFileSync3(out, TEXTISH.test(rel) ? rewriteAll(body.toString("utf8"), froms, podPath) : body);
+    }
+    const archive = await runTar(stage, ["-czf", "-", "memory"], null, limit);
+    return { podDir: `.claude/projects/${claudeKey(podPath)}`, archive, sent: { dir, localPath: projectDir, podPath, files: files2 }, count: Object.keys(files2).length };
+  } finally {
+    rmSync3(stage, { recursive: true, force: true });
+  }
+}
+function bringBackMemory(sent, pod) {
+  const out = { added: [], updated: [], conflicts: [], sent: { ...sent, files: { ...sent.files } } };
+  const write2 = (file, body) => {
+    mkdirSync2(dirname(file), { recursive: true });
+    const tmp = `${file}.airgate-${process.pid}`;
+    writeFileSync3(tmp, body, { mode: 384 });
+    renameSync2(tmp, file);
+  };
+  for (const f of pod) {
+    const rel = f.path.replace(/\\/g, "/");
+    if (!rel || rel.startsWith("/") || rel.split("/").some((p) => p === ".." || p === "")) continue;
+    const body = TEXTISH.test(rel) ? Buffer.from(rewritePath(f.body.toString("utf8"), sent.podPath, sent.localPath)) : f.body;
+    const file = join3(sent.dir, rel);
+    const local = existsSync2(file) ? readFileSync4(file) : null;
+    if (local && local.equals(body)) {
+      out.sent.files[rel] = sha256(local);
+      continue;
+    }
+    if (!local) {
+      write2(file, body);
+      out.added.push(rel);
+      out.sent.files[rel] = sha256(body);
+    } else if (sent.files[rel] && sha256(local) === sent.files[rel]) {
+      write2(file, body);
+      out.updated.push(rel);
+      out.sent.files[rel] = sha256(body);
+    } else {
+      const dot = rel.lastIndexOf(".");
+      const savedAs = dot > rel.lastIndexOf("/") ? `${rel.slice(0, dot)}.from-pod${rel.slice(dot)}` : `${rel}.from-pod`;
+      write2(join3(sent.dir, savedAs), body);
+      out.conflicts.push({ file: rel, savedAs });
+    }
+  }
+  return out;
+}
 
 // src/patch.ts
 var RULES2 = [
@@ -8963,7 +9039,7 @@ function podDirName(projectDir) {
 var up = {
   name: "up",
   group: "up \xB7 down",
-  usage: "[dir] [--pod p] [--agent a] [--with-config] [--env [.env]] [--session id | --no-session]",
+  usage: "[dir] [--pod p] [--agent a] [--with-config] [--env [.env]] [--session id | --no-session] [--no-memory]",
   summary: "move this project (with uncommitted work) into a pod",
   example: "airgate up --with-config",
   notes: [
@@ -8972,6 +9048,7 @@ var up = {
     "--with-config also sends ~/.claude settings.json, CLAUDE.md, skills/, agents/, commands/ (never credentials).",
     "remembers the pod in .airgate/state.json (kept out of git via .git/info/exclude) so `airgate down` knows where to look.",
     "also sends your latest Claude Code / Codex conversation in this directory and resumes it in the pod (paths rewritten); --no-session to skip, --session <id> to pick one. dropped with a warning if it is over the size limit.",
+    "and Claude Code's memory for this project (~/.claude/projects/<project>/memory); --no-memory to skip. down brings new and changed memories back, never overwriting one you changed here.",
     "--env [file] sends the keys of a dotenv file (default .env) as secrets of that pod; the names sent are listed."
   ],
   flags: {
@@ -8983,6 +9060,7 @@ var up = {
     env: { type: "string", value: "file", implicit: ".env", help: "send a dotenv file's keys as this pod's secrets (default .env)" },
     session: { type: "string", value: "id", help: "send this Claude Code / Codex conversation (default: the latest here)" },
     "no-session": { type: "boolean", help: "don't send the agent conversation" },
+    "no-memory": { type: "boolean", help: "don't send Claude Code's memory for this project" },
     yes: { type: "boolean", short: "y", help: "don't ask" }
   },
   async run(ctx) {
@@ -9026,6 +9104,16 @@ var up = {
       for (const f of [findClaudeSession(home, projectDir, sessionId, ctx.io.env), findCodexSession(home, projectDir, sessionId, ctx.io.env)]) if (f) found.push(f);
       if (sessionId && !found.length) throw new CliError(`no Claude Code or Codex session ${sessionId} for ${projectDir}`, "leave out --session to take the latest one");
     }
+    let memory2 = null;
+    const memDir = bool(ctx.args, "no-memory") ? null : findClaudeMemory(home, projectDir, ctx.io.env);
+    if (memDir) {
+      try {
+        memory2 = await packMemory(memDir, projectDir, podPath);
+      } catch (e) {
+        if (!(e instanceof TooLargeError)) throw e;
+        ui2.err(ui2.ce("warn", `Claude Code's memory for this project is over ${bytes(UPLOAD_LIMIT)} packed: not sent`));
+      }
+    }
     const sessions = [];
     for (const f of found) {
       try {
@@ -9072,6 +9160,15 @@ var up = {
     let envSent = null;
     const envFile = str(ctx.args, "env");
     if (envFile) envSent = await ui2.spin(`sending ${envFile} as secrets`, sendDotenv(ctx, envFile, pod.id));
+    let memorySent;
+    if (memory2) {
+      try {
+        await ui2.spin(`sending Claude Code's memory (${memory2.count} file${memory2.count === 1 ? "" : "s"})`, ctx.api.upload(pod.id, memory2.podDir, memory2.archive));
+        memorySent = memory2.sent;
+      } catch (e) {
+        ui2.err(ui2.ce("warn", `couldn't send the memory: ${e.message}`));
+      }
+    }
     const sent = [];
     const resumed = [];
     for (const p of sessions) {
@@ -9101,7 +9198,7 @@ var up = {
     } catch {
       based = false;
     }
-    const state = { gateway: ctx.cfg.gateway, podId: pod.id, podName: pod.spec.name, dir: podDir, at: Date.now(), ...sent.length ? { sessions: sent } : {} };
+    const state = { gateway: ctx.cfg.gateway, podId: pod.id, podName: pod.spec.name, dir: podDir, at: Date.now(), ...sent.length ? { sessions: sent } : {}, ...memorySent ? { memory: memorySent } : {} };
     writeState(projectDir, state);
     if (ctx.json) {
       return ui2.json({ pod, dir: podDir, bytes: packed.archive.length, files: packed.files, skipped: packed.skipped, config: config?.parts ?? [], baseSet: based, sessions: resumed, secrets: envSent });
@@ -9114,6 +9211,7 @@ var up = {
       ui2.out(`secrets for ${pod.spec.name}: ${envSent.sent.length ? envSent.sent.join(", ") : "none"}${envSent.skipped.length ? ui2.c("dim", ` (skipped: ${envSent.skipped.join(", ")})`) : ""}`);
       for (const f of envSent.failed) ui2.err(ui2.ce("warn", `not sent: ${f}`));
     }
+    if (memorySent) ui2.out(`memory: ${Object.keys(memorySent.files).length} file(s) of Claude Code's memory for this project`);
     for (const r of resumed) {
       ui2.out(`conversation: ${r.command}${r.running ? ui2.c("dim", ` \xB7 resuming in the pod (cd ~/${podDir})`) : ui2.c("dim", ` \xB7 sent; in the pod: cd ~/${podDir} && ${r.command}`)}`);
     }
@@ -9138,8 +9236,35 @@ async function downSessions(ctx, root, state, pod, quiet2 = false) {
     }
   }
   if (state.sessions?.length) writeState(root, { ...state, sessions: next });
-  if (!quiet2 && !ctx.json) printSessions(ctx, out);
-  return out;
+  let memory2 = null;
+  if (state.memory) {
+    try {
+      const x = await ctx.api.exportAgentMemory(pod.id, state.dir);
+      memory2 = bringBackMemory(state.memory, x.files.map((f) => ({ path: f.path, body: Buffer.from(f.b64, "base64") })));
+      writeState(root, { ...readState(root, state), memory: memory2.sent });
+    } catch (e) {
+      if (!(e instanceof ApiError && e.status === 404)) ctx.ui.err(ctx.ui.ce("warn", `couldn't bring Claude Code's memory back: ${e.message}`));
+    }
+  }
+  if (!quiet2 && !ctx.json) {
+    printSessions(ctx, out);
+    printMemory(ctx, memory2);
+  }
+  return { convo: out, memory: memory2 };
+}
+function readState(root, fallback) {
+  try {
+    return JSON.parse(readFileSync5(statePath(root), "utf8"));
+  } catch {
+    return fallback;
+  }
+}
+function printMemory(ctx, m) {
+  if (!m) return;
+  const n2 = m.added.length + m.updated.length;
+  if (!n2 && !m.conflicts.length) return ctx.ui.out(ctx.ui.c("dim", "memory: nothing new in the pod"));
+  if (n2) ctx.ui.out(`memory: ${[m.added.length ? `${m.added.length} new` : "", m.updated.length ? `${m.updated.length} updated` : ""].filter(Boolean).join(", ")} from the pod`);
+  for (const c of m.conflicts) ctx.ui.out(`memory: ${c.file} changed here and in the pod; yours kept, the pod's saved as ${c.savedAs}`);
 }
 function printSessions(ctx, list2) {
   for (const r of list2) {
@@ -9261,11 +9386,13 @@ ${check.stderr.trim()}`, "commit or stash local edits to those files, or look wi
     } catch {
       acked = false;
     }
-    const convo = takeSessions && found ? await downSessions(ctx, found.root, found.state, pod, true) : [];
-    if (ctx.json) return ui2.json({ pod: pod.id, base, files: files2, outside, refused, applied: true, acked, sessions: convo });
+    const back = takeSessions && found ? await downSessions(ctx, found.root, found.state, pod, true) : { convo: [], memory: null };
+    const convo = back.convo;
+    if (ctx.json) return ui2.json({ pod: pod.id, base, files: files2, outside, refused, applied: true, acked, sessions: convo, memory: back.memory && { added: back.memory.added, updated: back.memory.updated, conflicts: back.memory.conflicts } });
     ui2.out(`${ui2.c("ok", "down")}: applied ${files2.length} file(s). review: git status \xB7 git diff`);
     if (!acked) ui2.err(ui2.ce("warn", "couldn't tell the pod what was taken; the next down will show these again"));
     printSessions(ctx, convo);
+    printMemory(ctx, back.memory);
   }
 };
 
@@ -10404,7 +10531,7 @@ var TOOLS = [
   },
   {
     name: "pod_down",
-    description: "Apply a pod\u2019s work to the local repo as a git patch, with the CLI\u2019s safety checks (refuses symlinks and paths outside the repo; all or nothing). Look with pod_diff first and tell the user about risky files.",
+    description: "Apply a pod\u2019s work to the local repo as a git patch (and bring back the conversation and Claude Code\u2019s memory that up sent), with the CLI\u2019s safety checks (refuses symlinks and paths outside the repo; all or nothing). Look with pod_diff first and tell the user about risky files.",
     inputSchema: obj({ pod: s("pod name or id (default: the pod from airgate up in dir)"), dir: s("the local project directory (default: where the server started)") }),
     async call(ctx, a) {
       const argv = ["--yes"];
@@ -10447,7 +10574,7 @@ var TOOLS = [
   },
   {
     name: "pod_up",
-    description: "Move the local project into a pod: its files (git-tracked and untracked-but-not-ignored), and by default the latest Claude Code / Codex conversation here, which resumes in the pod. Launches a new pod named after the folder unless pod is given. Bring the work back later with pod_diff / pod_down.",
+    description: "Move the local project into a pod: its files (git-tracked and untracked-but-not-ignored), and by default the latest Claude Code / Codex conversation here, which resumes in the pod, plus Claude Code\u2019s memory for the project. Launches a new pod named after the folder unless pod is given. Bring the work back later with pod_diff / pod_down.",
     inputSchema: obj({
       dir: s("the local project directory (default: where the server started)"),
       pod: s("send into this existing pod instead of launching one"),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "airgate",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "description": "Give your AI its own computer: the airgate command line.",
   "type": "module",
   "bin": {