token-furnace 0.1.1 → 0.1.2

package/dist/cli.d.ts

@@ -11,8 +11,11 @@ interface SessionRepository {
      * gone from disk are dropped. The repository keeps the rest as they are.
      */
     changed(files: SourceFile[]): SourceFile[];
-    /** Every session held, in scan order: the order the analysis ranks ties by. */
-    all(): SessionInput[];
+    /**
+     * Every session held, in scan order: the order the analysis ranks ties by. With a cutoff, each
+     * session keeps only the turns at or after it, so the report reads no turn the analysis skips.
+     */
+    all(cutoff?: Date): SessionInput[];
     /** Store one parsed file, with its size: the two fields that say when to reparse it. */
     save(session: SessionInput, size: number): void;
 }

package/dist/cli.js

@@ -33,8 +33,12 @@ var MemorySessionRepository = class {
     for (const path of [...this.sessions.keys()]) if (!seen.has(path)) this.sessions.delete(path);
     return out;
   }
-  all() {
-    return [...this.sessions.values()].map((held) => held.session);
+  all(cutoff) {
+    const ms = cutoff?.getTime();
+    const filtered = ms !== void 0 && !Number.isNaN(ms);
+    return [...this.sessions.values()].map(
+      (held) => filtered ? { ...held.session, turns: held.session.turns.filter((turn) => turn.when.getTime() >= ms) } : held.session
+    );
   }
   save(session, size) {
     this.sessions.set(session.path, { session, size });
@@ -85,6 +89,7 @@ var SqliteSessionRepository = class {
   stmtInsertTurn = null;
   stmtAllFiles = null;
   stmtAllTurns = null;
+  stmtAllTurnsSince = null;
   constructor(dbPath) {
     this.dbPath = dbPath;
     this.ensureDb();
@@ -102,6 +107,9 @@ var SqliteSessionRepository = class {
     this.stmtAllTurns = this.db.prepare(
       "SELECT path, seq, project, session, model, at, base_in, cache_read, cache_write, output, tools, reads, wakes FROM turns ORDER BY path, seq;"
     );
+    this.stmtAllTurnsSince = this.db.prepare(
+      "SELECT path, seq, project, session, model, at, base_in, cache_read, cache_write, output, tools, reads, wakes FROM turns WHERE at >= ? ORDER BY path, seq;"
+    );
   }
   openDb(path) {
     const db = new DatabaseSync(path);
@@ -185,11 +193,13 @@ var SqliteSessionRepository = class {
     }
     return out;
   }
-  all() {
+  all(cutoff) {
     this.ensureDb();
     if (!this.db || !this.stmtAllFiles || !this.stmtAllTurns) return [];
+    const ms = cutoff?.getTime();
+    const filtered = ms !== void 0 && !Number.isNaN(ms);
     const fileRows = this.stmtAllFiles.all();
-    const turnRows = this.stmtAllTurns.iterate();
+    const turnRows = filtered && this.stmtAllTurnsSince ? this.stmtAllTurnsSince.iterate(ms) : this.stmtAllTurns.iterate();
     const turnsByPath = /* @__PURE__ */ new Map();
     for (const r of turnRows) {
       let list = turnsByPath.get(r.path);
@@ -1608,12 +1618,13 @@ var ReportService = class {
     const now = opts.now ?? /* @__PURE__ */ new Date();
     const live = this.live(source);
     if (!live.length) return null;
+    const cutoff = new Date(now.getTime() - days * 864e5);
     return buildReport({
       now,
       days,
       weeklyBudget: opts.weeklyBudget,
       live,
-      sessions: this.repo.all().sort((a, b) => (this.order.get(a.path) ?? Infinity) - (this.order.get(b.path) ?? Infinity)),
+      sessions: this.repo.all(cutoff).sort((a, b) => (this.order.get(a.path) ?? Infinity) - (this.order.get(b.path) ?? Infinity)),
       config: this.readConfig(live),
       planWeek: opts.planWeek ?? null,
       codexWeek: live.includes("codex") ? this.sources.find((s) => s.name === "codex")?.week?.(now) ?? null : null
@@ -1983,6 +1994,30 @@ LoopbackHostOnly = __decorateClass([
   Middleware({ type: "before", priority: 100 })
 ], LoopbackHostOnly);
 function reportController(app, options, page) {
+  let quota = null;
+  let fetching = null;
+  const plan = (now) => quota && quota.resets > now ? Promise.resolve(quota) : planWeek(app.env, now, false);
+  const refresh = () => {
+    fetching ??= planWeek(app.env, app.now(), true).then((week) => {
+      if (week && week.resets > app.now()) quota = week;
+    }).catch(() => {
+    }).finally(() => {
+      fetching = null;
+    });
+  };
+  async function build(query) {
+    const now = app.now();
+    const days = positiveInt(query.days, options.days);
+    const source = query.source === "claude" || query.source === "codex" ? query.source : options.source;
+    rescan(app);
+    const report = app.reports.getRaw(days, source, {
+      weeklyBudget: options.weeklyBudget,
+      planWeek: await plan(now),
+      now
+    });
+    if (options.planUsage) setImmediate(refresh);
+    return report;
+  }
   let ReportController = class {
     status(res) {
       if (!app.reports.hasScanned) warm(app);
@@ -1993,7 +2028,7 @@ function reportController(app, options, page) {
         warm(app);
         return res.status(202).json({ status: "scanning", progress: app.reports.progress });
       }
-      const report = await build(app, options, req.query);
+      const report = await build(req.query);
       return res.type("application/json").send(JSON.stringify(report ? reportView(report, app.now()) : null));
     }
     dashboard(res) {
@@ -2004,7 +2039,7 @@ function reportController(app, options, page) {
         warm(app);
         return res.type("text/html").send(loadingPage());
       }
-      const report = await build(app, options, req.query);
+      const report = await build(req.query);
       return res.type("text/html").send(report ? renderHtml(report, app.now(), page) : "No assistant turns.");
     }
   };
@@ -2037,17 +2072,6 @@ function warm(app) {
 function rescan(app) {
   if (Date.now() - app.reports.progress.updatedAt > 5e3) warm(app);
 }
-async function build(app, options, query) {
-  const now = app.now();
-  const days = positiveInt(query.days, options.days);
-  const source = query.source === "claude" || query.source === "codex" ? query.source : options.source;
-  rescan(app);
-  return app.reports.getRaw(days, source, {
-    weeklyBudget: options.weeklyBudget,
-    planWeek: await planWeek(app.env, now, options.planUsage),
-    now
-  });
-}
 function positiveInt(value, fallback) {
   const parsed = Number.parseInt(String(value ?? ""), 10);
   return parsed > 0 ? parsed : fallback;