@code-yeongyu/senpi 2026.6.6-3 → 2026.6.10-2

package/dist/core/session-manager.js

@@ -6,6 +6,7 @@ import { createInterface } from "readline";
 import { StringDecoder } from "string_decoder";
 import { getAgentDir as getDefaultAgentDir, getSessionsDir } from "../config.js";
 import { normalizePath, resolvePath } from "../utils/paths.js";
+import { ResidentStringStore } from "./session-resident-store.js";
 // Fork change: inlined UUIDv7 (upstream uses the `uuid` npm package). Keeps this
 // package self-contained so consumers don't need a transitive `uuid` install.
 function uuidv7() {
@@ -529,6 +530,7 @@ export class SessionManager {
         this.labelsById = new Map();
         this.labelTimestampsById = new Map();
         this.leafId = null;
+        this.residentStore = new ResidentStringStore();
         this.cwd = resolvePath(cwd);
         this.sessionDir = normalizePath(sessionDir);
         this.persist = persist;
@@ -545,6 +547,7 @@ export class SessionManager {
     /** Switch to a different session file (used for resume and branching) */
     setSessionFile(sessionFile) {
         this.sessionFile = resolvePath(sessionFile);
+        this.residentStore.clear();
         if (existsSync(this.sessionFile)) {
             this.fileEntries = loadEntriesFromFile(this.sessionFile);
             // If file was empty or corrupted (no valid header), truncate and start fresh
@@ -562,6 +565,7 @@ export class SessionManager {
             if (migrateToCurrentVersion(this.fileEntries)) {
                 this._rewriteFile();
             }
+            this.fileEntries = this.fileEntries.map((entry) => this.residentStore.externalize(entry));
             this._buildIndex();
             this.flushed = true;
         }
@@ -586,8 +590,10 @@ export class SessionManager {
             parentSession: options?.parentSession,
         };
         this.fileEntries = [header];
+        this.residentStore.clear();
         this.byId.clear();
         this.labelsById.clear();
+        this.labelTimestampsById.clear();
         this.leafId = null;
         this.flushed = false;
         if (this.persist) {
@@ -624,7 +630,7 @@ export class SessionManager {
         const fd = openSync(this.sessionFile, "w");
         try {
             for (const entry of this.fileEntries) {
-                writeFileSync(fd, `${JSON.stringify(entry)}\n`);
+                writeFileSync(fd, `${JSON.stringify(this.residentStore.materialize(entry))}\n`);
             }
         }
         finally {
@@ -649,13 +655,17 @@ export class SessionManager {
     getSessionFile() {
         return this.sessionFile;
     }
+    getResidentStoreStats() {
+        return this.residentStore.stats();
+    }
     _persist(entry) {
         if (!this.persist || !this.sessionFile)
             return;
+        const persistedEntry = this.residentStore.materialize(entry);
         const hasAssistant = this.fileEntries.some((e) => e.type === "message" && e.message.role === "assistant");
         if (!hasAssistant) {
             if (this.flushed) {
-                appendFileSync(this.sessionFile, `${JSON.stringify(entry)}\n`);
+                appendFileSync(this.sessionFile, `${JSON.stringify(persistedEntry)}\n`);
             }
             else {
                 // Mark as not flushed so when assistant arrives, all entries get written
@@ -667,7 +677,7 @@ export class SessionManager {
             const fd = openSync(this.sessionFile, "wx");
             try {
                 for (const e of this.fileEntries) {
-                    writeFileSync(fd, `${JSON.stringify(e)}\n`);
+                    writeFileSync(fd, `${JSON.stringify(this.residentStore.materialize(e))}\n`);
                 }
             }
             finally {
@@ -676,14 +686,15 @@ export class SessionManager {
             this.flushed = true;
         }
         else {
-            appendFileSync(this.sessionFile, `${JSON.stringify(entry)}\n`);
+            appendFileSync(this.sessionFile, `${JSON.stringify(persistedEntry)}\n`);
         }
     }
     _appendEntry(entry) {
-        this.fileEntries.push(entry);
-        this.byId.set(entry.id, entry);
-        this.leafId = entry.id;
-        this._persist(entry);
+        const residentEntry = this.residentStore.externalize(entry);
+        this.fileEntries.push(residentEntry);
+        this.byId.set(residentEntry.id, residentEntry);
+        this.leafId = residentEntry.id;
+        this._persist(residentEntry);
     }
     /** Append a message as child of current leaf, then advance leaf. Returns entry id.
      * Does not allow writing CompactionSummaryMessage and BranchSummaryMessage directly.
@@ -810,10 +821,12 @@ export class SessionManager {
         return this.leafId;
     }
     getLeafEntry() {
-        return this.leafId ? this.byId.get(this.leafId) : undefined;
+        const entry = this.leafId ? this.byId.get(this.leafId) : undefined;
+        return entry ? this.residentStore.materialize(entry) : undefined;
     }
     getEntry(id) {
-        return this.byId.get(id);
+        const entry = this.byId.get(id);
+        return entry ? this.residentStore.materialize(entry) : undefined;
     }
     /**
      * Get all direct children of an entry.
@@ -822,7 +835,7 @@ export class SessionManager {
         const children = [];
         for (const entry of this.byId.values()) {
             if (entry.parentId === parentId) {
-                children.push(entry);
+                children.push(this.residentStore.materialize(entry));
             }
         }
         return children;
@@ -871,7 +884,7 @@ export class SessionManager {
         const startId = fromId ?? this.leafId;
         let current = startId ? this.byId.get(startId) : undefined;
         while (current) {
-            path.unshift(current);
+            path.unshift(this.residentStore.materialize(current));
             current = current.parentId ? this.byId.get(current.parentId) : undefined;
         }
         return path;
@@ -881,14 +894,14 @@ export class SessionManager {
      * Uses tree traversal from current leaf.
      */
     buildSessionContext() {
-        return buildSessionContext(this.getEntries(), this.leafId, this.byId);
+        return buildSessionContext(this.getEntries(), this.leafId);
     }
     /**
      * Get session header.
      */
     getHeader() {
         const h = this.fileEntries.find((e) => e.type === "session");
-        return h ? h : null;
+        return h ? this.residentStore.materialize(h) : null;
     }
     /**
      * Get all session entries (excludes header). Returns a shallow copy.
@@ -896,7 +909,9 @@ export class SessionManager {
      * change the leaf pointer. Entries cannot be modified or deleted.
      */
     getEntries() {
-        return this.fileEntries.filter((e) => e.type !== "session");
+        return this.fileEntries
+            .filter((e) => e.type !== "session")
+            .map((entry) => this.residentStore.materialize(entry));
     }
     /**
      * Get the session as a tree structure. Returns a shallow defensive copy of all entries.
@@ -1037,7 +1052,8 @@ export class SessionManager {
                 labelEntries.push(labelEntry);
                 parentId = labelEntry.id;
             }
-            this.fileEntries = [header, ...pathWithoutLabels, ...labelEntries];
+            this.residentStore.clear();
+            this.fileEntries = [header, ...pathWithoutLabels, ...labelEntries].map((entry) => this.residentStore.externalize(entry));
             this.sessionId = newSessionId;
             this.sessionFile = newSessionFile;
             this._buildIndex();
@@ -1071,7 +1087,8 @@ export class SessionManager {
             labelEntries.push(labelEntry);
             parentId = labelEntry.id;
         }
-        this.fileEntries = [header, ...pathWithoutLabels, ...labelEntries];
+        this.residentStore.clear();
+        this.fileEntries = [header, ...pathWithoutLabels, ...labelEntries].map((entry) => this.residentStore.externalize(entry));
         this.sessionId = newSessionId;
         this._buildIndex();
         return undefined;