@nklisch/pi-enhanced 0.4.3 → 0.5.0

package/node_modules/@nklisch/pi-astral-pocket/src/sessions.ts

@@ -1,5 +1,5 @@
 import { createReadStream, existsSync, readdirSync, statSync } from "node:fs";
-import { join } from "node:path";
+import { join, resolve } from "node:path";
 import { createInterface } from "node:readline";
 
 import { ASTRA_MODEL_ID, ASTRA_PROVIDER } from "./activation.js";
@@ -18,6 +18,12 @@ export interface SessionFileInfo extends SessionRevision {
   astra: boolean;
 }
 
+export interface SessionExclusion { id?: string; path?: string }
+
+export function isExcludedSession(session: { id: string; path: string }, excluded?: SessionExclusion): boolean {
+  return Boolean((excluded?.id && session.id === excluded.id) || (excluded?.path && resolve(session.path) === resolve(excluded.path)));
+}
+
 export interface SessionSearchHit {
   sessionId: string;
   project: string;
@@ -112,7 +118,7 @@ function truncate(text: string, cap: number): string {
 export async function searchAstraSessions(
   sessionsDir: string,
   query: string,
-  options: { full?: boolean; limit?: number; maxAgeDays?: number; projectId?: string; recallScope?: "current" | "all" } = {},
+  options: { full?: boolean; limit?: number; maxAgeDays?: number; projectId?: string; recallScope?: "current" | "all"; excludeSession?: SessionExclusion } = {},
 ): Promise<SessionSearchHit[]> {
   const terms = query.toLowerCase().split(/\s+/).filter(Boolean);
   if (terms.length === 0) return [];
@@ -121,8 +127,9 @@ export async function searchAstraSessions(
   const hits: SessionSearchHit[] = [];
   const sessions: SessionFileInfo[] = [];
   for (const file of listSessionFiles(sessionsDir, options.maxAgeDays)) {
+    if (isExcludedSession({ id: "", path: file.path }, options.excludeSession)) continue;
     const info = await identifySession(file.path, file);
-    if (!info.astra) continue;
+    if (!info.astra || isExcludedSession(info, options.excludeSession)) continue;
     if ((options.recallScope ?? "current") !== "all" &&
       (!info.cwd || options.projectId === undefined || resolveProjectIdentity(info.cwd) !== options.projectId)) continue;
     sessions.push(info);