myrlin-workbook 0.9.4 → 0.9.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "myrlin-workbook",
-  "version": "0.9.4",
+  "version": "0.9.5",
   "description": "Browser-based project manager for Claude Code sessions - session discovery, multi-terminal, cost tracking, docs, and kanban board",
   "main": "src/index.js",
   "bin": {

package/src/web/public/app.js

@@ -7740,13 +7740,8 @@ class CWMApp {
 
     list.innerHTML = html;
 
-    // Fire off async cost fetches for visible sessions (best-effort, non-blocking)
-    const visibleSessionIds = (this.state.allSessions || this.state.sessions)
-      .filter(s => s.status === 'running' || s.status === 'idle')
-      .map(s => s.id);
-    if (visibleSessionIds.length > 0) {
-      this._fetchSessionCostsAsync(visibleSessionIds);
-    }
+    // Fetch all session costs in a single batch request (non-blocking)
+    this._fetchSessionCostsAsync();
 
 
     this.els.workspaceCount.textContent = `${workspaces.length} project${workspaces.length !== 1 ? 's' : ''}`;
@@ -14745,30 +14740,34 @@ class CWMApp {
     return null;
   }
 
-  _fetchSessionCostsAsync(sessionIds) {
+  /**
+   * Fetch costs for all sessions in a single batch request instead of N+1
+   * individual requests. Results are cached for 5 minutes. Only re-renders
+   * the sidebar once after all costs are received.
+   */
+  _fetchSessionCostsAsync() {
     if (!this._costCache) this._costCache = {};
-    if (!this._costFetchInFlight) this._costFetchInFlight = new Set();
-
-    sessionIds.forEach(sid => {
-      // Don't re-fetch if already in flight or recently cached
-      if (this._costFetchInFlight.has(sid)) return;
-      const entry = this._costCache[sid];
-      if (entry && (Date.now() - entry.ts < 300000)) return;
-
-      this._costFetchInFlight.add(sid);
-      this.api('GET', `/api/sessions/${sid}/cost`).then(data => {
-        this._costFetchInFlight.delete(sid);
-        if (data && (data.totalCost !== undefined || data.cost !== undefined)) {
-          const cost = data.totalCost ?? (data.cost && typeof data.cost === 'object' ? data.cost.total : data.cost) ?? null;
-          this._costCache[sid] = { cost, ts: Date.now() };
-          // Trigger a soft re-render of workspaces to show updated cost badges
-          this.renderWorkspaces();
-        }
-      }).catch(() => {
-        this._costFetchInFlight.delete(sid);
-        // Cache a null so we don't keep retrying for 5 minutes
-        this._costCache[sid] = { cost: null, ts: Date.now() };
-      });
+    // Skip if a batch fetch is already in flight or cache is fresh
+    if (this._costBatchInFlight) return;
+    if (this._costBatchTs && (Date.now() - this._costBatchTs < 300000)) return;
+
+    this._costBatchInFlight = true;
+    this.api('GET', '/api/cost/batch').then(data => {
+      this._costBatchInFlight = false;
+      this._costBatchTs = Date.now();
+      if (data && data.costs) {
+        let changed = false;
+        for (const [sid, entry] of Object.entries(data.costs)) {
+          const prev = this._costCache[sid];
+          if (!prev || prev.cost !== entry.cost) changed = true;
+          this._costCache[sid] = { cost: entry.cost, ts: Date.now() };
+        }
+        // Only re-render if any cost value actually changed
+        if (changed) this.renderWorkspaces();
+      }
+    }).catch(() => {
+      this._costBatchInFlight = false;
+      this._costBatchTs = Date.now();
     });
   }
 

package/src/web/server.js

@@ -2444,6 +2444,57 @@ app.get('/api/sessions/:id/cost', requireAuth, (req, res) => {
   }
 });
 
+/**
+ * GET /api/cost/batch
+ * Returns cost totals for all sessions in a single response, avoiding N+1 requests.
+ * Each entry includes sessionId, totalCost, and lastActive for sidebar badge rendering.
+ * Uses the same cache as per-session cost endpoints.
+ */
+app.get('/api/cost/batch', requireAuth, (req, res) => {
+  try {
+    const store = getStore();
+    const allWorkspaces = store.getAllWorkspacesList();
+    const costs = {};
+
+    for (const workspace of allWorkspaces) {
+      const sessions = store.getWorkspaceSessions(workspace.id);
+      for (const session of sessions) {
+        const resumeSessionId = session.resumeSessionId;
+        if (!resumeSessionId) continue;
+        const jsonlPath = findJsonlFile(resumeSessionId);
+        if (!jsonlPath) continue;
+
+        try {
+          const stat = fs.statSync(jsonlPath);
+          if (stat.size >= 500 * 1024 * 1024) continue;
+          const mtimeMs = stat.mtimeMs;
+          const cached = _costCache.get(resumeSessionId);
+          const now = Date.now();
+          let costData;
+
+          if (cached && cached.mtimeMs === mtimeMs && (now - cached.timestamp) < COST_CACHE_TTL) {
+            costData = cached.result;
+          } else {
+            costData = calculateSessionCost(jsonlPath);
+            const result = { sessionId: session.id, resumeSessionId, ...costData };
+            _costCache.set(resumeSessionId, { mtimeMs, timestamp: now, result });
+            costData = result;
+          }
+
+          costs[session.id] = {
+            cost: costData.cost ? costData.cost.total : 0,
+            lastActive: costData.lastMessage || session.lastActive || null,
+          };
+        } catch (_) {}
+      }
+    }
+
+    return res.json({ costs });
+  } catch (err) {
+    return res.status(500).json({ error: 'Batch cost failed: ' + err.message });
+  }
+});
+
 /**
  * GET /api/quota-overview
  * Returns all sessions ranked by context window size (heaviness).