@oh-my-pi/omp-stats 18.1.22 → 18.2.1

package/src/trace.ts

@@ -854,22 +854,54 @@ export async function getTraceEntry(fileParam: string, entryId: string): Promise
 // ---------------------------------------------------------------------------
 // Session list
 
-const titleCache = new Map<string, { mtimeMs: number; title: string | null }>();
+interface SessionMetadata {
+	title: string | null;
+	cwd: string | null;
+}
+
+const metadataCache = new Map<string, SessionMetadata & { mtimeMs: number }>();
 
-/** Read the session title from the fixed title slot or the header line. */
-async function readSessionTitle(file: string): Promise<string | null> {
+/** Read enough decompressed transcript bytes to cover the title slot and session header. */
+async function readSessionMetadataHead(file: string): Promise<string> {
+	if (!file.endsWith(".gz")) return Bun.file(file).slice(0, TITLE_SCAN_BYTES).text();
+
+	const reader = Bun.file(file).stream().pipeThrough(new DecompressionStream("gzip")).getReader();
+	const decoder = new TextDecoder();
+	let remaining = TITLE_SCAN_BYTES;
+	let head = "";
+	try {
+		while (remaining > 0) {
+			const { done, value } = await reader.read();
+			if (done) {
+				head += decoder.decode();
+				break;
+			}
+			const chunk = value.subarray(0, remaining);
+			remaining -= chunk.byteLength;
+			head += decoder.decode(chunk, { stream: remaining > 0 && chunk.byteLength === value.byteLength });
+			if (chunk.byteLength < value.byteLength) break;
+		}
+	} finally {
+		await reader.cancel();
+	}
+	return head;
+}
+
+/** Read list metadata from the fixed title slot and session header. */
+async function readSessionMetadata(file: string): Promise<SessionMetadata> {
 	let mtimeMs: number;
 	try {
 		mtimeMs = (await fs.stat(file)).mtimeMs;
 	} catch {
-		return null;
+		return { title: null, cwd: null };
 	}
-	const cached = titleCache.get(file);
-	if (cached && cached.mtimeMs === mtimeMs) return cached.title;
+	const cached = metadataCache.get(file);
+	if (cached && cached.mtimeMs === mtimeMs) return cached;
 
 	let title: string | null = null;
+	let cwd: string | null = null;
 	try {
-		const head = await Bun.file(file).slice(0, TITLE_SCAN_BYTES).text();
+		const head = await readSessionMetadataHead(file);
 		for (const line of head.split("\n")) {
 			if (!line.trim()) continue;
 			let parsed: EntryView;
@@ -881,18 +913,20 @@ async function readSessionTitle(file: string): Promise<string | null> {
 			}
 			if (parsed.type === "title" && typeof parsed.title === "string") {
 				title = parsed.title;
-				break;
+				continue;
 			}
 			if (parsed.type === "session") {
-				if (typeof parsed.title === "string") title = parsed.title;
+				if (title === null && typeof parsed.title === "string") title = parsed.title;
+				if (typeof parsed.cwd === "string") cwd = parsed.cwd;
 				break;
 			}
 		}
 	} catch {
-		// Unreadable head (gc'd, gz, permission) → keep the row, title null.
+		// Unreadable head (gc'd, gz, permission) → keep the row with path-derived metadata.
 	}
-	titleCache.set(file, { mtimeMs, title });
-	return title;
+	const metadata = { mtimeMs, title, cwd };
+	metadataCache.set(file, metadata);
+	return metadata;
 }
 
 interface SummaryFold extends SessionSummary {
@@ -1027,7 +1061,9 @@ export async function listSessionSummaries(limit = 100, q?: string): Promise<Ses
 	const folds = [...byRoot.values()].sort((a, b) => b.endedAt - a.endedAt).slice(0, LIST_FOLD_LIMIT);
 	await Promise.all(
 		folds.map(async fold => {
-			fold.title = await readSessionTitle(fold.file);
+			const metadata = await readSessionMetadata(fold.file);
+			fold.title = metadata.title;
+			fold.folder = metadata.cwd ?? extractFolderFromPath(fold.file);
 			fold.models = [...fold.modelSet].sort();
 		}),
 	);