dsh-session-mng 0.1.4 → 0.2.1

package/lib/state.js

@@ -13,8 +13,13 @@
 import { zstdAvailable } from './store-jsonl.js'
 import { canReconcileMove } from './workspace-sync.js'
 
-/** Read a service without declaring it as a hard dependency. */
-function serviceOf(ctx, name) {
+/**
+ * Read a service without declaring it as a hard dependency.
+ *
+ * `ctx.get(name)` may throw for an unknown or not-yet-started service; an
+ * absent service must degrade one feature, never fail the request.
+ */
+export function serviceOf(ctx, name) {
   try {
     return ctx.get?.(name)
   } catch {
@@ -32,7 +37,7 @@ function liveSession(ctx, id) {
 }
 
 /** Best-effort running state for one id. */
-function isRunning(ctx, id) {
+export function isRunningSession(ctx, id) {
   try {
     const agent = ctx.get?.('agents')?.get?.(id)
     return agent?.status === 'running' || agent?.running === true
@@ -41,6 +46,39 @@ function isRunning(ctx, id) {
   }
 }
 
+/**
+ * Capabilities this host currently offers, shared by `/state` and the 0.2.0
+ * catalog so both surfaces answer the same question the same way.
+ */
+export function capabilitiesOf(ctx) {
+  const jsonl =
+    typeof ctx.sessionPersistence?.locate === 'function' && typeof ctx.sessionPersistence?.stat === 'function'
+  return {
+    title: Boolean(ctx.llm),
+    move: jsonl && zstdAvailable() && canReconcileMove(serviceOf(ctx, 'workspaceRegistry')),
+    delete: jsonl,
+    backend: jsonl ? 'jsonl' : 'other',
+  }
+}
+
+/**
+ * Per-row action flags derived from one row's state and the host capabilities.
+ *
+ * Keeping this shared is what makes `/state` rows and catalog rows offer the
+ * same buttons: a live or running session is never moved or deleted, a branch
+ * is never moved on its own, and deletion stays archived-only.
+ */
+export function canOf(row, capabilities) {
+  return {
+    title: capabilities.title,
+    archive: !row.archived,
+    restore: row.archived,
+    move: capabilities.move && !row.live && !row.running && row.kind === 'root',
+    delete: capabilities.delete && row.archived && !row.live,
+    open: true,
+  }
+}
+
 /** Best-effort snapshot list; an unavailable backend yields no snapshots. */
 async function persistenceSnapshots(ctx, signal) {
   try {
@@ -69,7 +107,7 @@ async function sessionSummaries(ctx, signal) {
       sessionId: record.header.id,
       updatedAt: record.header.createdAt,
       agentAvailable: false,
-      running: isRunning(ctx, record.header.id),
+      running: isRunningSession(ctx, record.header.id),
       blank: false,
       cwd: record.header.cwd,
       origin: record.header.origin,
@@ -241,7 +279,7 @@ export async function buildState(ctx, config, options = {}) {
       workspaceBySession.get(id) ??
       (typeof cwd === 'string' && cwd !== '' ? workspaceByPath.get(cwd) : undefined)
     const live = liveSession(ctx, id) !== undefined
-    const running = summary.running === true || isRunning(ctx, id)
+    const running = summary.running === true || isRunningSession(ctx, id)
     const title =
       typeof summary.projections?.values?.title === 'string' ? summary.projections.values.title : ''
     if (title === '') missingTitles.push(id)
@@ -284,7 +322,7 @@ export async function buildState(ctx, config, options = {}) {
       cwd: header.cwd,
       origin: header.origin,
       parentSessionId: header.parentSession,
-      running: isRunning(ctx, header.id),
+      running: isRunningSession(ctx, header.id),
       blank: false,
       formatStatus: snapshot.formatStatus,
     })
@@ -311,25 +349,8 @@ export async function buildState(ctx, config, options = {}) {
     row.forkCount = forkCountByParent.get(row.id) ?? 0
   }
 
-  const jsonl =
-    typeof ctx.sessionPersistence?.locate === 'function' && typeof ctx.sessionPersistence?.stat === 'function'
-  const capabilities = {
-    title: Boolean(ctx.llm),
-    move: jsonl && zstdAvailable() && canReconcileMove(serviceOf(ctx, 'workspaceRegistry')),
-    delete: jsonl,
-    backend: jsonl ? 'jsonl' : 'other',
-  }
-
-  for (const row of rows) {
-    row.can = {
-      title: capabilities.title,
-      archive: !row.archived,
-      restore: row.archived,
-      move: capabilities.move && !row.live && !row.running && row.kind === 'root',
-      delete: capabilities.delete && row.archived && !row.live,
-      open: true,
-    }
-  }
+  const capabilities = capabilitiesOf(ctx)
+  for (const row of rows) row.can = canOf(row, capabilities)
 
   // Archived ids whose log is gone can never be rendered; surface them so the
   // UI can offer to drop the stale archive marker. Every id the corpus knows

package/lib/store-jsonl.js

@@ -21,7 +21,7 @@
 
 import zlib from 'node:zlib'
 import { homedir } from 'node:os'
-import { dirname, join } from 'node:path'
+import { basename, dirname, join } from 'node:path'
 import { access, cp, mkdir, open, readdir, readFile, rename, rm, stat } from 'node:fs/promises'
 
 import { SessionMngError, messageOf } from './errors.js'
@@ -160,7 +160,81 @@ export async function currentLogFile(dir) {
     }
   }
   if (best === undefined) return undefined
-  return { path: join(dir, best.name), name: best.name, compressed: best.compressed }
+  return { path: join(dir, best.name), name: best.name, compressed: best.compressed, generation: bestGeneration }
+}
+
+/**
+ * Size and format facts of one stored session's current log, at the cost of
+ * exactly one readdir plus one stat.
+ *
+ * The directory comes from the backend's own zero-I/O locator, then
+ * {@link currentLogFile} picks the highest released generation inside it. That
+ * is deliberately NOT `findSessionDirs`/`stat(id)`: both fall back to scanning
+ * every project directory, which must never happen on a page request. The
+ * format status is inferred by comparing the selected generation with the one
+ * `locate()` names as current, so an older-generation log is reported honestly
+ * instead of being labelled current.
+ *
+ * @param {object} persistence - the host's JSONL persistence service.
+ * @param {string} id - stored session id.
+ * @param {string} [cwd] - the session's stored working directory.
+ * @returns {Promise<{sizeBytes: number, formatStatus: string, logName: string}|undefined>}
+ */
+export async function readLogFacts(persistence, id, cwd) {
+  let located
+  try {
+    located = persistence?.locate?.({ id, cwd })
+  } catch {
+    return undefined
+  }
+  const path = typeof located?.path === 'string' ? located.path : ''
+  if (path === '') return undefined
+  const currentGeneration = generationOf(basename(path))
+  const log = await currentLogFile(dirname(path))
+  if (log === undefined) return undefined
+  let size
+  try {
+    size = (await stat(log.path)).size
+  } catch {
+    return undefined
+  }
+  const migrationRequired = currentGeneration !== undefined && log.generation < currentGeneration
+  return {
+    sizeBytes: size,
+    formatStatus: migrationRequired ? 'migration-required' : 'current',
+    logName: log.name,
+  }
+}
+
+/** Maximum concurrent per-log stat batches; small enough to stay polite. */
+const FACT_CONCURRENCY = 8
+
+/**
+ * Read {@link readLogFacts} for a bounded batch (one page at most in practice).
+ *
+ * @param {object} persistence
+ * @param {{id: string, cwd?: string}[]} items
+ * @param {number} [concurrency]
+ * @returns {Promise<Map<string, {sizeBytes: number, formatStatus: string, logName: string}>>}
+ */
+export async function readLogFactsBatch(persistence, items, concurrency = FACT_CONCURRENCY) {
+  const found = new Map()
+  if (!Array.isArray(items) || items.length === 0) return found
+  let next = 0
+  const worker = async () => {
+    for (;;) {
+      const index = next
+      next += 1
+      if (index >= items.length) return
+      const item = items[index]
+      if (typeof item?.id !== 'string') continue
+      const facts = await readLogFacts(persistence, item.id, item.cwd).catch(() => undefined)
+      if (facts !== undefined) found.set(item.id, facts)
+    }
+  }
+  const width = Math.max(1, Math.min(Math.trunc(concurrency) || 1, items.length))
+  await Promise.all(Array.from({ length: width }, worker))
+  return found
 }
 
 /** Every offset where a Zstandard frame magic begins. */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-session-mng",
-  "version": "0.1.4",
+  "version": "0.2.1",
   "type": "module",
   "description": "Session management for DeepSeek Harness: one Settings page for every session — whole-session title generation, archive/restore, cross-workspace move, and permanent deletion of archived sessions.",
   "license": "MIT",