@sna-sdk/core 0.6.1 → 0.7.2

package/dist/server/session-manager.js

@@ -12,6 +12,7 @@ class SessionManager {
     this.lifecycleListeners = /* @__PURE__ */ new Set();
     this.configChangedListeners = /* @__PURE__ */ new Set();
     this.stateChangedListeners = /* @__PURE__ */ new Set();
+    this.metadataChangedListeners = /* @__PURE__ */ new Set();
     this.maxSessions = options.maxSessions ?? DEFAULT_MAX_SESSIONS;
     this.restoreFromDb();
   }
@@ -64,26 +65,11 @@ class SessionManager {
     } catch {
     }
   }
-  /** Create a new session. Throws if max sessions reached. */
+  /** Create a new session. Throws if session already exists or max sessions reached. */
   createSession(opts = {}) {
     const id = opts.id ?? crypto.randomUUID().slice(0, 8);
     if (this.sessions.has(id)) {
-      const existing = this.sessions.get(id);
-      let changed = false;
-      if (opts.cwd && opts.cwd !== existing.cwd) {
-        existing.cwd = opts.cwd;
-        changed = true;
-      }
-      if (opts.label && opts.label !== existing.label) {
-        existing.label = opts.label;
-        changed = true;
-      }
-      if (opts.meta !== void 0 && opts.meta !== existing.meta) {
-        existing.meta = opts.meta ?? null;
-        changed = true;
-      }
-      if (changed) this.persistSession(existing);
-      return existing;
+      throw new Error(`Session "${id}" already exists`);
     }
     const aliveCount = Array.from(this.sessions.values()).filter((s) => s.process?.alive).length;
     if (aliveCount >= this.maxSessions) {
@@ -107,6 +93,17 @@ class SessionManager {
     this.persistSession(session);
     return session;
   }
+  /** Update an existing session's metadata. Throws if session not found. */
+  updateSession(id, opts) {
+    const session = this.sessions.get(id);
+    if (!session) throw new Error(`Session "${id}" not found`);
+    if (opts.label !== void 0) session.label = opts.label;
+    if (opts.meta !== void 0) session.meta = opts.meta;
+    if (opts.cwd !== void 0) session.cwd = opts.cwd;
+    this.persistSession(session);
+    this.emitMetadataChanged(id);
+    return session;
+  }
   /** Get a session by ID. */
   getSession(id) {
     return this.sessions.get(id);
@@ -128,27 +125,45 @@ class SessionManager {
     const session = this.sessions.get(sessionId);
     if (!session) throw new Error(`Session "${sessionId}" not found`);
     session.process = proc;
-    this.setSessionState(sessionId, session, "processing");
     session.lastActivityAt = Date.now();
+    session.eventBuffer.length = 0;
+    try {
+      const db = getDb();
+      const row = db.prepare(
+        `SELECT COUNT(*) as c FROM chat_messages WHERE session_id = ?`
+      ).get(sessionId);
+      session.eventCounter = row.c;
+    } catch {
+    }
     proc.on("event", (e) => {
-      if (e.type === "init" && e.data?.sessionId && !session.ccSessionId) {
-        session.ccSessionId = e.data.sessionId;
-        this.persistSession(session);
+      if (e.type === "init") {
+        if (e.data?.sessionId && !session.ccSessionId) {
+          session.ccSessionId = e.data.sessionId;
+          this.persistSession(session);
+        }
+        this.setSessionState(sessionId, session, "waiting");
+      }
+      if (e.type === "thinking" || e.type === "tool_use" || e.type === "assistant_delta") {
+        this.setSessionState(sessionId, session, "processing");
+      } else if (e.type === "complete" || e.type === "error" || e.type === "interrupted") {
+        this.setSessionState(sessionId, session, "waiting");
       }
-      if (e.type !== "assistant_delta") {
+      const persisted = this.persistEvent(sessionId, e);
+      if (persisted) {
+        session.eventCounter++;
         session.eventBuffer.push(e);
         if (session.eventBuffer.length > MAX_EVENT_BUFFER) {
           session.eventBuffer.splice(0, session.eventBuffer.length - MAX_EVENT_BUFFER);
         }
-      }
-      session.eventCounter++;
-      if (e.type === "complete" || e.type === "error" || e.type === "interrupted") {
-        this.setSessionState(sessionId, session, "waiting");
-      }
-      this.persistEvent(sessionId, e);
-      const listeners = this.eventListeners.get(sessionId);
-      if (listeners) {
-        for (const cb of listeners) cb(session.eventCounter, e);
+        const listeners = this.eventListeners.get(sessionId);
+        if (listeners) {
+          for (const cb of listeners) cb(session.eventCounter, e);
+        }
+      } else if (e.type === "assistant_delta") {
+        const listeners = this.eventListeners.get(sessionId);
+        if (listeners) {
+          for (const cb of listeners) cb(-1, e);
+        }
       }
     });
     proc.on("exit", (code) => {
@@ -186,11 +201,17 @@ class SessionManager {
     for (const cb of this.skillEventListeners) cb(event);
   }
   /** Push a synthetic event into a session's event stream (for user message broadcast). */
+  /**
+   * Push an externally-persisted event into the session.
+   * The caller is responsible for DB persistence — this method only updates
+   * the in-memory counter/buffer and notifies listeners.
+   * eventCounter increments to stay in sync with the DB row count.
+   */
   pushEvent(sessionId, event) {
     const session = this.sessions.get(sessionId);
     if (!session) return;
-    session.eventBuffer.push(event);
     session.eventCounter++;
+    session.eventBuffer.push(event);
     if (session.eventBuffer.length > MAX_EVENT_BUFFER) {
       session.eventBuffer.splice(0, session.eventBuffer.length - MAX_EVENT_BUFFER);
     }
@@ -223,6 +244,14 @@ class SessionManager {
   emitConfigChanged(sessionId, config) {
     for (const cb of this.configChangedListeners) cb({ session: sessionId, config });
   }
+  // ── Session metadata change pub/sub ─────────────────────────────
+  onMetadataChanged(cb) {
+    this.metadataChangedListeners.add(cb);
+    return () => this.metadataChangedListeners.delete(cb);
+  }
+  emitMetadataChanged(sessionId) {
+    for (const cb of this.metadataChangedListeners) cb(sessionId);
+  }
   // ── Agent status change pub/sub ────────────────────────────────
   onStateChanged(cb) {
     this.stateChangedListeners.add(cb);
@@ -303,7 +332,6 @@ class SessionManager {
       extraArgs: overrides.extraArgs ?? base.extraArgs
     };
     if (session.process?.alive) session.process.kill();
-    session.eventBuffer.length = 0;
     const proc = spawnFn(config);
     this.setProcess(id, proc);
     session.lastStartConfig = config;
@@ -408,6 +436,7 @@ class SessionManager {
       return { messageCount: 0, lastMessage: null };
     }
   }
+  /** Persist an agent event to chat_messages. Returns true if a row was inserted. */
   persistEvent(sessionId, e) {
     try {
       const db = getDb();
@@ -415,29 +444,34 @@ class SessionManager {
         case "assistant":
           if (e.message) {
             db.prepare(`INSERT INTO chat_messages (session_id, role, content) VALUES (?, 'assistant', ?)`).run(sessionId, e.message);
+            return true;
           }
-          break;
+          return false;
         case "thinking":
           if (e.message) {
             db.prepare(`INSERT INTO chat_messages (session_id, role, content) VALUES (?, 'thinking', ?)`).run(sessionId, e.message);
+            return true;
           }
-          break;
+          return false;
         case "tool_use": {
           const toolName = e.data?.toolName ?? e.message ?? "tool";
           db.prepare(`INSERT INTO chat_messages (session_id, role, content, meta) VALUES (?, 'tool', ?, ?)`).run(sessionId, toolName, JSON.stringify(e.data ?? {}));
-          break;
+          return true;
         }
         case "tool_result":
           db.prepare(`INSERT INTO chat_messages (session_id, role, content, meta) VALUES (?, 'tool_result', ?, ?)`).run(sessionId, e.message ?? "", JSON.stringify(e.data ?? {}));
-          break;
+          return true;
         case "complete":
           db.prepare(`INSERT INTO chat_messages (session_id, role, content, meta) VALUES (?, 'status', '', ?)`).run(sessionId, JSON.stringify({ status: "complete", ...e.data }));
-          break;
+          return true;
         case "error":
           db.prepare(`INSERT INTO chat_messages (session_id, role, content, meta) VALUES (?, 'error', ?, ?)`).run(sessionId, e.message ?? "Error", JSON.stringify({ status: "error" }));
-          break;
+          return true;
+        default:
+          return false;
       }
     } catch {
+      return false;
     }
   }
   /** Kill all sessions. Used during shutdown. */