@krtclcdy/tersio 2.25.4 → 2.25.6

package/dist/cli/dashboard.js

@@ -1,4 +1,4 @@
-// Serves the built Dashboard with local usage APIs. Binds 127.0.0.1 only; --export writes a file instead. No Promise.withResolvers: Node 20.12 lacks it.
+// Dashboard server + usage APIs. Binds 127.0.0.1; --export writes a file. No Promise.withResolvers (Node 20.12).
 import { spawn } from 'node:child_process';
 import { execFileSync } from 'node:child_process';
 import { existsSync, mkdirSync, readFileSync, statSync, writeFileSync } from 'node:fs';
@@ -6,17 +6,17 @@ import { promises as fs } from 'node:fs';
 import http from 'node:http';
 import path from 'node:path';
 import { fileURLToPath } from 'node:url';
-import { clearUsageLedger, markReset, readUsage } from '../extensions/shared/usage-ledger.js';
 import { pricesCachePath } from '../extensions/shared/pricing.js';
-import { summarizeUsage } from './usage.js';
-import { backupUsageDb, deleteUsageBackup, listUsageBackups, restoreUsageBackup, usageDbPath } from '../extensions/shared/usage-store.js';
 import { isCurrencyCode } from './currency.js';
 import { OMP_AGENT_DIR, OMP_PLUGINS_DIR, PACKAGE_VERSION, } from './common.js';
 import { BACKUP_SCHEDULES, formatCliStatus, storedProfile, writePluginSettings } from './profile.js';
 import { normalizeComboLevel } from '../extensions/shared/session-state.js';
-import { CAVEMAN_DEFAULTS, PONYTAIL_DEFAULTS } from './common.js';
+import { CAVEMAN_DEFAULTS, PONYTAIL_DEFAULTS, PSTACK_DEFAULTS } from './common.js';
 import { PACKAGE_NAME } from './common.js';
-import { findPackageRoot, resolveRtkBinary } from '../extensions/lib/utils.js';
+import { findPackageRoot, resolveRtkBinary, ruleBodyProblem } from '../extensions/shared/utils.js';
+import { BACKUP_KEEP, backupDir, backupDiskBytes, createDataBackup, dataSources, deleteDataBackup, listDataBackups, restoreDataBackup } from './backups.js';
+import { readBackupSchedule, syncUsageDb } from '../extensions/shared/usage-store.js';
+import { agentDbPath, emptyRequestEntry, isAgentView, isRangeKey, readAgentRequestEntry, readAgentStats, readSessionTrace, statsDbPath } from '../extensions/shared/agent-stats.js';
 const DASHBOARD_DIST = path.join(findPackageRoot(path.dirname(fileURLToPath(import.meta.url))), 'dashboard', 'dist');
 const DASHBOARD_INDEX = path.join(DASHBOARD_DIST, 'index.html');
 const DASHBOARD_BRAND = path.join(DASHBOARD_DIST, 'brand.webp');
@@ -49,8 +49,64 @@ async function brandDataUri() {
     const brand = await fs.readFile(DASHBOARD_BRAND);
     return `data:image/webp;base64,${brand.toString('base64')}`;
 }
-function dataJson() {
-    return JSON.stringify(summarizeUsage(readUsage()));
+/** The range a page opens on, matching the reference dashboard. */
+const DEFAULT_AGENT_RANGE = '24h';
+/** Poll memo TTL: every poll re-reads the aggregate, so a repeat poll reuses the last answer. */
+const AGENT_REFRESH_MS = 5000;
+/** A sync walks every transcript, so it runs at most this often even while a poll outruns it. */
+const AGENT_SYNC_MS = 15_000;
+let agentMemo = null;
+let lastSyncAt = 0;
+function fileStamp(file) {
+    try {
+        const stat = statSync(file);
+        return `${stat.mtimeMs}:${stat.size}`;
+    }
+    catch {
+        return '-';
+    }
+}
+/** Both sources: the usage store carries every host's rows, agent.db only omp's quota snapshots. */
+function storeStamp() {
+    return `${fileStamp(statsDbPath())}|${fileStamp(agentDbPath())}`;
+}
+/** Syncs before reading past the throttle; without it a page misses pi and OpenCode rows. */
+function agentStats(range, view = 'all', offset = 0) {
+    const held = agentMemo;
+    if (held && held.range === range && held.view === view && held.offset === offset && held.stamp === storeStamp() && Date.now() - held.at < AGENT_REFRESH_MS) {
+        return held.value;
+    }
+    if (Date.now() - lastSyncAt >= AGENT_SYNC_MS) {
+        lastSyncAt = Date.now();
+        try {
+            syncUsageDb();
+        }
+        catch { /* a store that will not write still reads the rows it already holds */ }
+    }
+    const value = readAgentStats(range, view, Date.now(), offset);
+    agentMemo = { range, view, offset, stamp: storeStamp(), at: Date.now(), value };
+    return value;
+}
+/** `?range=` on any of the omp routes; an unknown value falls back to the default. */
+function queryParam(url, name) {
+    return (url.split('?')[1] ?? '').split('&').find((part) => part.startsWith(`${name}=`))?.slice(name.length + 1) ?? null;
+}
+function rangeFrom(url) {
+    const raw = queryParam(url, 'range');
+    return isRangeKey(raw) ? raw : DEFAULT_AGENT_RANGE;
+}
+function viewFrom(url) {
+    const raw = queryParam(url, 'view');
+    return isAgentView(raw) ? raw : 'all';
+}
+/** `?offset=` in whole rows, floored; a later page starts further back instead of clamping at the cap. */
+function offsetFrom(url) {
+    const raw = queryParam(url, 'offset');
+    const value = raw === null ? 0 : Number(raw);
+    return Number.isFinite(value) && value > 0 ? Math.floor(value) : 0;
+}
+function agentJson(range, view, offset) {
+    return JSON.stringify({ range, view, offset, agent: agentStats(range, view, offset) });
 }
 // Runs outside any agent session, so it reports the persisted defaults.
 async function statusJson() {
@@ -61,6 +117,7 @@ async function statusJson() {
         caveman: profile.cavemanDefault,
         rtk: profile.rtkDefault ? 'on' : 'off',
         ponytail: profile.ponytailDefault,
+        pstack: profile.pstackDefault,
     });
 }
 // Local-only health: find each agent on PATH, run `--version`, report both.
@@ -245,18 +302,29 @@ function computeDoctorRows() {
     }
     catch { /* missing config is reported by extension rows */ }
     const duplicateExtensions = [...new Set(explicitEntries.filter((entry, index) => explicitEntries.indexOf(entry) !== index))];
-    addon('Unique config registrations', true, 'ok', 'Extensions');
-    // A listed extension whose file is gone makes the host warn on every load, so surface it rather than leaving the user with only a startup message.
+    addon('Unique config registrations', duplicateExtensions.length === 0, duplicateExtensions.length ? duplicateExtensions.join(', ') : 'ok', 'Extensions');
+    // A listed extension whose file is gone makes the host warn on every load, so surface it here.
     const dangling = explicitEntries
         .filter((entry) => entry.includes(`extensions${path.sep}`) || entry.includes('extensions/'))
         .filter((entry) => !ok(path.isAbsolute(entry) ? entry : path.resolve(OMP_AGENT_DIR, entry)));
     addon('Extension files present', dangling.length === 0, dangling.length ? dangling.join(', ') : 'ok', 'Extensions');
-    ext('Caveman extension', path.join(tersioPluginDir, 'extensions', 'caveman-session', 'index.ts'));
-    ext('RTK extension', path.join(tersioPluginDir, 'extensions', 'rtk-session', 'index.ts'));
-    const rule = path.join(tersioPluginDir, 'extensions', 'caveman-session', 'rule.md');
+    ext('Caveman extension', path.join(tersioPluginDir, 'extensions', 'caveman', 'index.ts'));
+    ext('pstack extension', path.join(tersioPluginDir, 'extensions', 'pstack', 'index.ts'));
+    ext('RTK extension', path.join(tersioPluginDir, 'extensions', 'rtk', 'session.ts'));
+    const rule = path.join(tersioPluginDir, 'extensions', 'caveman', 'rule.md');
     const ruleAge = ageStr(rule);
     const ruleAt = absTime(rule);
     addon('Caveman rule', ruleAge !== null, ruleAge && ruleAt ? `(updated ${ruleAge} · ${ruleAt})` : rule);
+    // The pstack body is a Tersio fork with no remote, so this row proves it is usable, not current.
+    const pstackRule = path.join(tersioPluginDir, 'extensions', 'pstack', 'rule.md');
+    let pstackProblem = 'missing — run: tersio install';
+    try {
+        pstackProblem = ruleBodyProblem(readFileSync(pstackRule, 'utf8'));
+    }
+    catch { /* missing stays */ }
+    const pstackAge = ageStr(pstackRule);
+    const pstackAt = absTime(pstackRule);
+    addon('pstack rule', pstackProblem === null, pstackProblem ?? `(updated ${pstackAge} · ${pstackAt})`);
     const rtkVer = rtkBin ? rtkVersion(rtkBin) : null;
     const rtkAge = rtkBin ? ageStr(rtkBin) : null;
     const rtkAt = rtkBin ? absTime(rtkBin) : null;
@@ -277,7 +345,7 @@ function computeDoctorRows() {
     addon('Prices feed', pricesAge !== null, pricesAge && pricesAt ? `(pulled ${pricesAge} · ${pricesAt})` : prices, 'Usage');
     return rows;
 }
-// Recompute when forced, missing, or past the schedule; a report holding a retired label is stale by definition.
+// A report holding any of these labels is stale by definition.
 const RETIRED_DIAG_LABELS = new Set([
     'OMP CLI',
     'Tersio CLI',
@@ -298,11 +366,10 @@ const RETIRED_DIAG_LABELS = new Set([
 function isRetiredReport(report) {
     return report.rows.some((r) => RETIRED_DIAG_LABELS.has(r.label));
 }
-// Recompute when forced, missing, retired, or past the schedule.
+// Recompute when forced, missing, retired, or past schedule; manual schedules recompute on open, dated ones reuse.
 function getDoctorReport(force) {
     const saved = readDiagReport();
     const ttl = saved ? DIAG_TTL_MS[saved.schedule] : 0;
-    // Manual recomputes on open; a dated schedule reuses the saved report.
     if (!force && saved && !isRetiredReport(saved) && ttl > 0 && Date.now() - saved.checkedAt < ttl)
         return saved;
     const report = { rows: computeDoctorRows(), checkedAt: Date.now(), schedule: saved?.schedule ?? 'manual' };
@@ -336,6 +403,12 @@ async function saveDefaults(body) {
             return { ok: false, error: 'unknown ponytail level' };
         profile.ponytailDefault = v;
     }
+    if (body.pstackDefault !== undefined) {
+        const v = String(body.pstackDefault);
+        if (!PSTACK_DEFAULTS.has(v))
+            return { ok: false, error: 'unknown pstack level' };
+        profile.pstackDefault = v;
+    }
     if (body.backupSchedule !== undefined) {
         const v = String(body.backupSchedule);
         if (!BACKUP_SCHEDULES.has(v))
@@ -355,6 +428,7 @@ async function saveDefaults(body) {
             cavemanDefault: profile.cavemanDefault,
             rtkDefault: profile.rtkDefault,
             ponytailDefault: profile.ponytailDefault,
+            pstackDefault: profile.pstackDefault,
             backupSchedule: profile.backupSchedule,
         },
     };
@@ -374,50 +448,6 @@ async function saveDashboardCurrency(raw) {
     await writePluginSettings(profile, {});
     return code;
 }
-export const EXPORT_FORMATS = ['json', 'jsonl', 'csv'];
-export function exportRows(report) {
-    return report.recent.map((r) => ({
-        timestamp: new Date(r.t).toISOString(),
-        local: new Date(r.t).toLocaleString(),
-        agent: r.h ?? 'pi',
-        model: r.m,
-        input: r.i,
-        output: r.o,
-        cacheRead: r.cr ?? 0,
-        cacheWrite: r.cw ?? 0,
-        costUsd: r.usd,
-        elapsedMs: r.d,
-        status: r.st,
-    }));
-}
-export function csvCell(v) {
-    const s = v === undefined || v === null ? '' : String(v);
-    return /[",\n]/.test(s) ? `"${s.replace(/"/g, '""')}"` : s;
-}
-export function exportBody(format, report) {
-    const stamp = new Date().toISOString().slice(0, 10);
-    if (format === 'json') {
-        return { body: JSON.stringify({ exportedAt: new Date().toISOString(), version: report.version, source: report.source, report }, null, 2), type: 'application/json' };
-    }
-    const rows = exportRows(report);
-    if (format === 'jsonl') {
-        return { body: rows.map((r) => JSON.stringify(r)).join('\n') + '\n', type: 'application/x-ndjson' };
-    }
-    const cols = Object.keys(rows[0] ?? { timestamp: '', agent: '' });
-    const head = cols.join(',');
-    const body = [head, ...rows.map((r) => cols.map((c) => csvCell(r[c])).join(','))].join('\n') + '\n';
-    return { body, type: 'text/csv; charset=utf-8' };
-}
-function serveExport(req, res) {
-    const format = ((req.url || '').split('?')[1] || '').match(/format=([a-z]+)/)?.[1];
-    const fmt = format && EXPORT_FORMATS.includes(format) ? format : 'json';
-    const { body, type } = exportBody(fmt, summarizeUsage(readUsage()));
-    res.writeHead(200, {
-        'Content-Type': type,
-        'Content-Disposition': `attachment; filename="tersio-usage-${new Date().toISOString().slice(0, 10)}.${fmt === 'jsonl' ? 'ndjson' : fmt}"`,
-    });
-    res.end(body);
-}
 function readBody(req) {
     return new Promise((resolve) => {
         let body = '';
@@ -430,7 +460,7 @@ function openBrowser(url) {
     const cmd = process.platform === 'darwin' ? 'open' : process.platform === 'win32' ? 'start' : 'xdg-open';
     spawn(cmd, [url], { detached: true, stdio: 'ignore' }).unref();
 }
-// Replacers throughout: session data holds `$'` (shell quoting) and literal `</script>`, which String.replace would mangle.
+// Replacers throughout: session data holds `$'` (shell quoting) and `</script>`, which String.replace would mangle.
 function escapeInline(json) {
     return json.replace(/<\/(script)/gi, '<\\/$1');
 }
@@ -448,73 +478,78 @@ function parseOwnJson(json, field) {
     }
     return parsed;
 }
-async function exportDashboard(exportFile) {
+async function exportDashboard(exportFile, currency) {
     requireDashboardBundle();
     const [bundle, icon] = await Promise.all([readSegment(DASHBOARD_INDEX), brandDataUri()]);
+    // Minified: fetch reads `fetch(\`brand.webp\`)` or `fetch("brand.webp")`; a mismatch loses the card's mark.
+    const b64 = icon.split(',')[1] ?? '';
+    // atob yields one byte per char, so a Uint8Array is needed or bytes >= 0x80 double and corrupt the webp.
+    const brandBytes = `(() => { const c = atob(${JSON.stringify(b64)}); const v = new Uint8Array(c.length); for (let i = 0; i < c.length; i++) v[i] = c.charCodeAt(i); return v; })()`;
     const inline = bundle
-        .replace('window.__TERSIO_SNAP = null;', () => `window.__TERSIO_SNAP = ${escapeInline(JSON.stringify({ data: parseOwnJson(dataJson(), 'usage data'), health: parseOwnJson(healthJson(), 'health'), doctor: getDoctorReport(false) }))};`)
-        .replace(/href="brand\.webp"/g, () => `href="${icon}"`)
-        .replace(/src="brand\.webp"/g, () => `src="${icon}"`)
-        .replace('fetch("brand.webp")', () => `Promise.resolve({ ok: true, blob: async () => new Blob([atob("${icon.split(',')[1]}")], { type: "image/webp" }) })`);
+        .replace('window.__TERSIO_SNAP = null;', () => `window.__TERSIO_SNAP = ${escapeInline(JSON.stringify({ currency, agent: agentStats(DEFAULT_AGENT_RANGE), health: parseOwnJson(healthJson(), 'health'), doctor: getDoctorReport(false) }))};`)
+        // A mark is an HTML attribute or JSX prop; match every quote or the file keeps an unresolvable reference.
+        .replace(/((?:src|href)[=:][ \t]*)([`"'])brand\.webp\2/g, (_all, head, quote) => `${head}${quote}${icon}${quote}`)
+        .replace(/fetch\(([`"])brand\.webp\1\)/g, () => `Promise.resolve({ ok: true, blob: async () => new Blob([${brandBytes}], { type: "image/webp" }) })`);
     await fs.writeFile(exportFile, inline, 'utf8');
     console.log(`[ok] Dashboard exported → ${exportFile}`);
 }
 async function runDashboard(options) {
     if (options.exportFile) {
-        await exportDashboard(options.exportFile);
+        await exportDashboard(options.exportFile, options.currency);
         return;
     }
     requireDashboardBundle();
     const server = http.createServer(async (req, res) => {
-        if (req.url === '/reset' && req.method === 'POST') {
-            const rows = clearUsageLedger();
-            markReset();
-            res.writeHead(200, { 'Content-Type': 'application/json' });
-            res.end(JSON.stringify({ ok: true, rows }));
-            return;
-        }
-        if (req.url === '/backups' && req.method === 'GET') {
-            const rows = listUsageBackups().map((b) => ({ file: path.basename(b.file), mtime: b.mtime, size: b.size }));
+        if (req.url === '/api/data' && req.method === 'GET') {
             res.writeHead(200, { 'Content-Type': 'application/json' });
-            res.end(JSON.stringify({ backups: rows }));
+            res.end(JSON.stringify({
+                dir: backupDir(),
+                keep: BACKUP_KEEP,
+                sources: dataSources(),
+                backups: listDataBackups(),
+                schedule: readBackupSchedule(),
+                bytes: backupDiskBytes(),
+            }));
             return;
         }
-        if (req.url === '/backups/create' && req.method === 'POST') {
-            try {
-                backupUsageDb(usageDbPath());
-                res.writeHead(200, { 'Content-Type': 'application/json' });
-                res.end(JSON.stringify({ ok: true }));
-            }
-            catch {
-                res.writeHead(500, { 'Content-Type': 'application/json' });
-                res.end(JSON.stringify({ ok: false, error: 'backup failed' }));
-            }
-            return;
-        }
-        if (req.url === '/backups/delete' && req.method === 'POST') {
-            let file = '';
+        if (req.url === '/api/data' && req.method === 'POST') {
+            let body = {};
             try {
-                file = String(JSON.parse(await readBody(req)).file ?? '');
+                body = JSON.parse(await readBody(req)) ?? {};
             }
             catch {
-                file = '';
+                body = {};
             }
-            const ok = deleteUsageBackup(file);
-            res.writeHead(ok ? 200 : 400, { 'Content-Type': 'application/json' });
-            res.end(JSON.stringify(ok ? { ok: true, file } : { ok: false, error: 'unknown backup' }));
-            return;
-        }
-        if (req.url === '/backups/restore' && req.method === 'POST') {
-            let file = '';
+            const action = String(body.action ?? '');
+            const name = String(body.name ?? '');
+            // A missing SQLite engine, or a locked file, is a 500 with a message rather than a silent no-op.
             try {
-                file = String(JSON.parse(await readBody(req)).file ?? '');
+                if (action === 'create') {
+                    const made = await createDataBackup();
+                    res.writeHead(200, { 'Content-Type': 'application/json' });
+                    res.end(JSON.stringify({ ok: true, name: made.name, files: made.files.length }));
+                    return;
+                }
+                if (action === 'restore') {
+                    const ok = await restoreDataBackup(name);
+                    res.writeHead(ok ? 200 : 400, { 'Content-Type': 'application/json' });
+                    res.end(JSON.stringify(ok ? { ok: true, name } : { ok: false, error: 'unknown snapshot' }));
+                    return;
+                }
+                if (action === 'delete') {
+                    const ok = await deleteDataBackup(name);
+                    res.writeHead(ok ? 200 : 400, { 'Content-Type': 'application/json' });
+                    res.end(JSON.stringify(ok ? { ok: true, name } : { ok: false, error: 'unknown snapshot' }));
+                    return;
+                }
             }
-            catch {
-                file = '';
+            catch (error) {
+                res.writeHead(500, { 'Content-Type': 'application/json' });
+                res.end(JSON.stringify({ ok: false, error: error.message || 'backup failed' }));
+                return;
             }
-            const ok = restoreUsageBackup(file);
-            res.writeHead(ok ? 200 : 400, { 'Content-Type': 'application/json' });
-            res.end(JSON.stringify(ok ? { ok: true, file } : { ok: false, error: 'unknown backup' }));
+            res.writeHead(400, { 'Content-Type': 'application/json' });
+            res.end(JSON.stringify({ ok: false, error: 'unknown action' }));
             return;
         }
         if (req.url === '/settings' && req.method === 'GET') {
@@ -525,7 +560,9 @@ async function runDashboard(options) {
                     cavemanDefault: profile.cavemanDefault,
                     rtkDefault: profile.rtkDefault,
                     ponytailDefault: profile.ponytailDefault,
+                    pstackDefault: profile.pstackDefault,
                     backupSchedule: profile.backupSchedule,
+                    currency: options.currency,
                 }));
             });
             return;
@@ -568,13 +605,34 @@ async function runDashboard(options) {
             }
             return;
         }
-        if (req.url === '/data.json') {
+        if (req.url === '/api/agent' || req.url?.startsWith('/api/agent?')) {
+            res.writeHead(200, { 'Content-Type': 'application/json' });
+            res.end(agentJson(rangeFrom(req.url), viewFrom(req.url), offsetFrom(req.url)));
+            return;
+        }
+        if (req.url?.startsWith('/api/agent/session')) {
+            const file = decodeURIComponent(queryParam(req.url, 'file') ?? '');
+            const body = file ? readSessionTrace(file) : {
+                sessionFile: '', project: '', title: '', cwd: '', startedAt: 0, endedAt: 0,
+                tracks: [], markers: [], summary: {
+                    wallMs: 0, modelMs: 0, toolMs: 0, idleMs: 0, turns: 0, requests: 0, toolCalls: 0, agents: 0,
+                    totalTokens: 0, costUsd: 0,
+                },
+                toolStats: [], truncated: false,
+            };
+            res.writeHead(200, { 'Content-Type': 'application/json' });
+            res.end(JSON.stringify(body));
+            return;
+        }
+        if (req.url?.startsWith('/api/agent/entry')) {
+            const file = decodeURIComponent(queryParam(req.url, 'file') ?? '');
+            const entry = decodeURIComponent(queryParam(req.url, 'entry') ?? '');
+            const body = file && entry ? readAgentRequestEntry(file, entry) : emptyRequestEntry();
             res.writeHead(200, { 'Content-Type': 'application/json' });
-            res.end(dataJson());
+            res.end(JSON.stringify(body));
             return;
         }
         if (req.url?.startsWith('/export')) {
-            serveExport(req, res);
             return;
         }
         if (req.url === '/health') {
@@ -664,4 +722,4 @@ async function runDashboard(options) {
         });
     });
 }
-export { runDashboard, saveDashboardCurrency, readDiagSchedule, setDiagSchedule };
+export { runDashboard, saveDashboardCurrency, readDiagSchedule, setDiagSchedule, offsetFrom };