labgate 0.5.13 → 0.5.14

package/dist/lib/ui.html

@@ -627,6 +627,59 @@
       line-height: 1.4;
     }
 
+    .dataset-card-stats {
+      display: flex;
+      align-items: center;
+      gap: 6px;
+      padding-left: 42px;
+      margin-bottom: 8px;
+      font-size: 0.75rem;
+      color: var(--text-muted);
+    }
+
+    .dataset-card-stats .stat-dot {
+      color: var(--border-color);
+    }
+
+    .dataset-card-stats .stat-warning {
+      color: #d97706;
+      font-weight: 500;
+    }
+
+    .dataset-card-stats .stat-loading {
+      color: var(--text-muted);
+      font-style: italic;
+    }
+
+    .dataset-card-stats .stat-scan-btn {
+      background: none;
+      border: 1px solid var(--border-color);
+      color: var(--text-secondary);
+      font-size: 0.6875rem;
+      font-family: inherit;
+      padding: 1px 8px;
+      border-radius: 4px;
+      cursor: pointer;
+      transition: background 0.15s, color 0.15s, border-color 0.15s;
+    }
+
+    .dataset-card-stats .stat-scan-btn:hover {
+      background: var(--bg-secondary);
+      color: var(--text-primary);
+      border-color: #B8CBB8;
+    }
+
+    .dataset-card-stats .stat-scan-btn:disabled {
+      opacity: 0.5;
+      cursor: default;
+    }
+
+    .dataset-card-stats .stat-scanned-at {
+      color: var(--text-muted);
+      font-size: 0.6875rem;
+      margin-left: 2px;
+    }
+
     .dataset-card-paths {
       display: flex;
       align-items: center;
@@ -3173,7 +3226,7 @@ function renderDatasets() {
     var descHtml = ds.description
       ? '<div class="dataset-card-desc">' + escapeHtml(ds.description) + '</div>'
       : '';
-    return '<div class="dataset-card">'
+    return '<div class="dataset-card" data-ds-name="' + escapeHtml(ds.name) + '">'
       + '<div class="dataset-card-header">'
       + '<div class="dataset-card-icon">' + dbIcon + '</div>'
       + '<div class="dataset-card-title">'
@@ -3183,6 +3236,7 @@ function renderDatasets() {
       + '<button class="remove-btn" data-index="' + i + '" title="Remove dataset">&times;</button>'
       + '</div>'
       + descHtml
+      + '<div class="dataset-card-stats"><span class="stat-loading">Loading stats\u2026</span></div>'
       + '<div class="dataset-card-paths">'
       + '<span class="dataset-path">' + escapeHtml(ds.path) + '</span>'
       + '<span class="dataset-path-arrow">' + arrowIcon + '</span>'
@@ -3198,6 +3252,98 @@ function renderDatasets() {
       removeDataset(idx);
     });
   });
+  fetchDatasetStats();
+}
+
+function renderStatsContent(statsEl, name, s) {
+  if (!s.exists) {
+    statsEl.innerHTML = '<span class="stat-warning">Path not found</span>';
+    return;
+  }
+  if (!s.scanned_at) {
+    statsEl.innerHTML = '<span style="color:var(--text-muted)">Not initialized</span>'
+      + ' <button class="stat-scan-btn" data-scan-name="' + escapeHtml(name) + '" title="Initialize dataset">Scan</button>'
+      + '<span class="stat-scanned-at" style="margin-left:4px">or run: <code style="font-size:0.6875rem">labgate dataset init ' + escapeHtml(name) + '</code></span>';
+    bindScanButtons(statsEl);
+    return;
+  }
+  var ago = timeAgo(s.scanned_at);
+  statsEl.innerHTML = '<span>' + s.file_count + ' file' + (s.file_count !== 1 ? 's' : '') + '</span>'
+    + '<span class="stat-dot">&middot;</span>'
+    + '<span>' + escapeHtml(s.total_size_formatted) + '</span>'
+    + '<span class="stat-scanned-at">(' + ago + ')</span>'
+    + ' <button class="stat-scan-btn" data-scan-name="' + escapeHtml(name) + '" title="Rescan">&#8635;</button>';
+  bindScanButtons(statsEl);
+}
+
+function bindScanButtons(root) {
+  root.querySelectorAll('.stat-scan-btn[data-scan-name]').forEach(function(btn) {
+    btn.addEventListener('click', function() {
+      scanDataset(btn.getAttribute('data-scan-name'));
+    });
+  });
+}
+
+function timeAgo(isoStr) {
+  var diff = Date.now() - new Date(isoStr).getTime();
+  var sec = Math.floor(diff / 1000);
+  if (sec < 60) return 'just now';
+  var min = Math.floor(sec / 60);
+  if (min < 60) return min + 'm ago';
+  var hr = Math.floor(min / 60);
+  if (hr < 24) return hr + 'h ago';
+  var days = Math.floor(hr / 24);
+  return days + 'd ago';
+}
+
+function scanDataset(name) {
+  var card = document.querySelector('.dataset-card[data-ds-name="' + name + '"]');
+  if (!card) return;
+  var statsEl = card.querySelector('.dataset-card-stats');
+  if (statsEl) statsEl.innerHTML = '<span class="stat-loading">Scanning\u2026</span>';
+
+  fetch('/api/dataset-scan', {
+    method: 'POST',
+    headers: apiWriteHeaders(),
+    body: JSON.stringify({ name: name })
+  })
+    .then(function(r) { return r.json(); })
+    .then(function(data) {
+      if (!statsEl) return;
+      if (!data.ok) {
+        statsEl.innerHTML = '<span class="stat-warning">' + escapeHtml(data.error || 'Scan failed') + '</span>';
+        return;
+      }
+      renderStatsContent(statsEl, name, {
+        exists: true,
+        file_count: data.stats.file_count,
+        total_size_formatted: data.stats.total_size_formatted,
+        scanned_at: data.stats.scanned_at
+      });
+      showToast('Dataset "' + name + '" scanned: ' + data.stats.file_count + ' files, ' + data.stats.total_size_formatted, 'success');
+    })
+    .catch(function() {
+      if (statsEl) statsEl.innerHTML = '<span class="stat-warning">Scan failed</span>';
+    });
+}
+
+function fetchDatasetStats() {
+  fetch('/api/dataset-stats', { headers: apiReadHeaders() })
+    .then(function(r) { return r.json(); })
+    .then(function(data) {
+      if (!data || !data.stats) return;
+      var cards = document.querySelectorAll('.dataset-card[data-ds-name]');
+      cards.forEach(function(card) {
+        var name = card.getAttribute('data-ds-name');
+        var statsEl = card.querySelector('.dataset-card-stats');
+        if (!statsEl || !name || !data.stats[name]) return;
+        renderStatsContent(statsEl, name, data.stats[name]);
+      });
+    })
+    .catch(function() {
+      var els = document.querySelectorAll('.dataset-card-stats .stat-loading');
+      els.forEach(function(el) { el.textContent = ''; });
+    });
 }
 
 function addDataset() {
@@ -3565,12 +3711,14 @@ function checkDatasetRestartNeeded() {
     var notice = document.getElementById('datasetNotice');
     var text = notice ? notice.querySelector('.dataset-notice-text') : null;
     if (needsRestart.length > 0) {
+      if (notice) notice.style.display = '';
       if (text) text.textContent = needsRestart.length + ' active session' + (needsRestart.length > 1 ? 's' : '') + ' running with outdated config. Restart them now to apply dataset changes.';
       if (btn) { btn.style.display = ''; btn.disabled = false; btn.textContent = 'Restart Sessions Now'; }
     } else if (sessions.length > 0) {
-      if (text) text.textContent = 'Dataset changes are saved. Active sessions keep old mounts until you restart them.';
+      if (notice) notice.style.display = 'none';
       if (btn) btn.style.display = 'none';
     } else {
+      if (notice) notice.style.display = '';
       if (text) text.textContent = 'No active sessions. Dataset changes will be used the next time you start a session.';
       if (btn) btn.style.display = 'none';
     }

package/dist/lib/ui.js

@@ -1502,6 +1502,129 @@ async function handleValidatePath(req, res) {
         json(res, { valid: false, error: err.message ?? String(err) }, 400);
     }
 }
+// ── Dataset stats helpers ───────────────────────────────
+function formatBytesUI(bytes) {
+    if (bytes === 0)
+        return '0 B';
+    const units = ['B', 'KB', 'MB', 'GB', 'TB'];
+    const i = Math.floor(Math.log(bytes) / Math.log(1024));
+    const val = bytes / Math.pow(1024, i);
+    return `${val.toFixed(i === 0 ? 0 : 1)} ${units[i]}`;
+}
+function walkDirShallow(dir, maxDepth, depth = 0) {
+    let files = 0;
+    let totalSize = 0;
+    if (depth > maxDepth)
+        return { files, totalSize };
+    let entries;
+    try {
+        entries = (0, fs_1.readdirSync)(dir);
+    }
+    catch {
+        return { files, totalSize };
+    }
+    for (const entry of entries) {
+        if (entry.startsWith('.') && depth === 0)
+            continue;
+        const fullPath = (0, path_1.join)(dir, entry);
+        try {
+            const st = (0, fs_1.statSync)(fullPath);
+            if (st.isDirectory()) {
+                if (depth < maxDepth) {
+                    const sub = walkDirShallow(fullPath, maxDepth, depth + 1);
+                    files += sub.files;
+                    totalSize += sub.totalSize;
+                }
+            }
+            else {
+                files++;
+                totalSize += st.size;
+            }
+        }
+        catch { /* skip inaccessible entries */ }
+    }
+    return { files, totalSize };
+}
+function handleGetDatasetStats(_req, res) {
+    const config = (0, config_js_1.loadConfig)();
+    const datasets = config.datasets || [];
+    const stats = {};
+    for (const ds of datasets) {
+        const hostPath = ds.path.replace(/^~/, (0, os_1.homedir)());
+        const pathExists = (0, fs_1.existsSync)(hostPath);
+        if (ds.stats) {
+            stats[ds.name] = {
+                exists: pathExists,
+                file_count: ds.stats.file_count,
+                total_size: ds.stats.total_size,
+                total_size_formatted: ds.stats.total_size_formatted,
+                scanned_at: ds.stats.scanned_at,
+            };
+        }
+        else {
+            stats[ds.name] = {
+                exists: pathExists,
+                file_count: 0,
+                total_size: 0,
+                total_size_formatted: '0 B',
+                scanned_at: null,
+            };
+        }
+    }
+    json(res, { stats });
+}
+async function handlePostDatasetScan(req, res) {
+    try {
+        const body = await readBody(req);
+        const { name } = JSON.parse(body);
+        if (!name || typeof name !== 'string') {
+            json(res, { ok: false, error: 'Missing dataset name' }, 400);
+            return;
+        }
+        const configPath = (0, config_js_1.getConfigPath)();
+        let obj = {};
+        if ((0, fs_1.existsSync)(configPath)) {
+            const rawText = (0, fs_1.readFileSync)(configPath, 'utf-8');
+            const stripped = rawText
+                .split('\n')
+                .filter(line => !line.trimStart().startsWith('//'))
+                .join('\n');
+            try {
+                obj = JSON.parse(stripped);
+            }
+            catch {
+                obj = {};
+            }
+        }
+        const datasets = (obj.datasets || []);
+        const idx = datasets.findIndex((d) => d?.name && String(d.name).toLowerCase() === name.toLowerCase());
+        if (idx < 0) {
+            json(res, { ok: false, error: `Dataset "${name}" not found` }, 404);
+            return;
+        }
+        const ds = datasets[idx];
+        const hostPath = String(ds.path || '').replace(/^~/, (0, os_1.homedir)());
+        if (!(0, fs_1.existsSync)(hostPath)) {
+            json(res, { ok: false, error: `Path does not exist: ${hostPath}` }, 400);
+            return;
+        }
+        const walked = walkDirShallow(hostPath, 1);
+        const statsObj = {
+            file_count: walked.files,
+            total_size: walked.totalSize,
+            total_size_formatted: formatBytesUI(walked.totalSize),
+            scanned_at: new Date().toISOString(),
+        };
+        ds.stats = statsObj;
+        obj.datasets = datasets;
+        (0, fs_1.writeFileSync)(configPath, JSON.stringify(obj, null, 2) + '\n', { encoding: 'utf-8', mode: config_js_1.PRIVATE_FILE_MODE });
+        (0, config_js_1.ensurePrivateFile)(configPath);
+        json(res, { ok: true, stats: statsObj });
+    }
+    catch (err) {
+        json(res, { ok: false, error: err.message ?? String(err) }, 500);
+    }
+}
 function handleGetLogs(_req, res) {
     const config = (0, config_js_1.loadConfig)();
     if (!config.audit.enabled) {
@@ -2700,6 +2823,12 @@ function startUI(optsOrPort = {}, standaloneArg = true) {
             else if (pathname === '/api/validate-path' && method === 'POST') {
                 await handleValidatePath(req, res);
             }
+            else if (pathname === '/api/dataset-stats' && method === 'GET') {
+                handleGetDatasetStats(req, res);
+            }
+            else if (pathname === '/api/dataset-scan' && method === 'POST') {
+                await handlePostDatasetScan(req, res);
+            }
             else if (/^\/api\/sessions\/[^/]+\/instructions$/.test(pathname) && method === 'GET') {
                 handleGetSessionInstructions(pathname, reqUrl, res);
             }