troxy-cli 1.29.50 → 1.29.51

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "troxy-cli",
-  "version": "1.29.50",
+  "version": "1.29.51",
   "description": "A secure control layer for AI agents: policies across payments, messages, logins, destructive actions, model usage, and secrets, all enforceable from the CLI",
   "homepage": "https://troxy.io",
   "bugs": {

package/src/mods/troxy-guard/hooks/register.js

@@ -280,9 +280,14 @@ function cancelCompact() {
 // stayAwake: hold off system sleep (the screen still sleeps) until the timer fires, so a
 // big session compacts on time instead of when the user is back. macOS caffeinate; where
 // it is missing the spawn fails quietly and compact-on-return in scorePrompt covers it.
-function armCompact($, ms, { stayAwake = false } = {}) {
+function armCompact($, ms, { stayAwake = false, persist = false } = {}) {
   cancelCompact();
   if (stayAwake) keepAwake($, ms + 2 * 60 * 1000);
+  // Persisted so a restart (which loses lastSent and the timer) still finds the idle start.
+  if (persist) try {
+    Promise.all([$.session.id(), $.clock.now()])
+      .then(([sid, at]) => $.store.set(`idle_started_${sid}`, at)).catch(() => {});
+  } catch {}
   compactTimer = $.clock.after(ms, () => {
     compactTimer = null;
     idleCompact($).catch(() => {}).finally(stopAwake);
@@ -346,10 +351,16 @@ async function scorePrompt($, e, next) {
   cancelCompact();
   if ((await $.state.get(COMPACTED)).value) await $.state.set(COMPACTED, null);
   const sid = await $.session.id();
-  // Idle past the timer by the wall clock: a sleeping Mac freezes the timer, so the
-  // compaction it missed runs here, before this prompt goes out on the big session.
+  // Idle past the timer by the wall clock: a sleeping Mac freezes the timer and a restart
+  // loses it, so the compaction it missed runs here, before this prompt goes out.
+  const now = await $.clock.now();
   const ls = lastSent.get(sid);
-  if (ls && (await $.clock.now()) - ls.at >= COMPACT_IDLE_MS) {
+  let idleSince = ls ? ls.at : null;
+  try {
+    const stored = await $.store.get(`idle_started_${sid}`);
+    if (typeof stored === 'number' && (idleSince == null || stored > idleSince)) idleSince = stored;
+  } catch {}
+  if (idleSince != null && now - idleSince >= COMPACT_IDLE_MS) {
     try { await idleCompact($, { late: true }); } catch {} // fail open: send the prompt as-is
   }
   turn = { start: await snapshot($), ratio: 1 };
@@ -502,7 +513,7 @@ export function register(on) {
     } catch {} // a missed fetch only leaves the band as it was
     let stayAwake = false;
     try { stayAwake = await bigSession($); } catch {}
-    armCompact($, COMPACT_IDLE_MS, { stayAwake });
+    armCompact($, COMPACT_IDLE_MS, { stayAwake, persist: true });
     return r;
   });
 }

package/src/tests/mods-troxy-guard.test.js

@@ -233,6 +233,28 @@ test('scorePrompt: back after 55+ min by the wall clock (Mac slept through the t
   assert.ok(await runTurn($, opus));       // a failed compaction never blocks the prompt
 });
 
+test('scorePrompt: idle 55+ min across a Claude Code restart still compacts, from the saved idle start', async () => {
+  const { armCompact, cancelCompact } = _internal;
+  const $ = routingApi();
+  const db = new Map();
+  let now = 1000, compacts = 0;
+  $.store = { get: async k => db.get(k), set: async (k, v) => { db.set(k, v); } };
+  $.clock.now = async () => now;
+  $.clock.after = () => ({ cancel() {} });
+  $.session.usage = async () => ({ context: { tokens: 174000 }, rateLimits: [] });
+  $.session.compact = async () => { compacts++; return { tokensBefore: 174000, tokensAfter: 20000 }; };
+  armCompact($, 55 * 60 * 1000, { persist: true }); // end of the last turn before the restart
+  await new Promise(r => setImmediate(r));
+  assert.equal(typeof [...db.values()][0], 'number');
+  now += 40 * 60 * 1000;
+  armCompact($, 55 * 60 * 1000);                     // session.start after the restart: must not reset it
+  await new Promise(r => setImmediate(r));
+  now += 16 * 60 * 1000;
+  await runTurn($, _internal.MODELS.opus);           // fresh process: no lastSent, only the store
+  cancelCompact();
+  assert.equal(compacts, 1);
+});
+
 test('armCompact: a big session keeps the Mac awake until the timer fires, then lets it sleep', async () => {
   const { armCompact, cancelCompact } = _internal;
   const spawned = [], stopped = [];