vigthoria-cli 1.0.0

package/src/utils/session.ts

@@ -0,0 +1,167 @@
+/**
+ * Session Manager - Persist and resume conversations
+ * Similar to Claude Code's session persistence
+ */
+
+import * as fs from 'fs';
+import * as path from 'path';
+import * as os from 'os';
+import { ChatMessage } from './api.js';
+
+export interface Session {
+  id: string;
+  name: string;
+  project: string;
+  model: string;
+  messages: ChatMessage[];
+  createdAt: string;
+  updatedAt: string;
+  agentMode: boolean;
+}
+
+export class SessionManager {
+  private sessionsDir: string;
+
+  constructor() {
+    this.sessionsDir = path.join(os.homedir(), '.vigthoria', 'sessions');
+    this.ensureDir();
+  }
+
+  private ensureDir(): void {
+    if (!fs.existsSync(this.sessionsDir)) {
+      fs.mkdirSync(this.sessionsDir, { recursive: true });
+    }
+  }
+
+  /**
+   * Generate unique session ID
+   */
+  private generateId(): string {
+    const timestamp = Date.now().toString(36);
+    const random = Math.random().toString(36).substring(2, 8);
+    return `${timestamp}-${random}`;
+  }
+
+  /**
+   * Create a new session
+   */
+  create(project: string, model: string, agentMode: boolean = false): Session {
+    const session: Session = {
+      id: this.generateId(),
+      name: `Session ${new Date().toLocaleString()}`,
+      project,
+      model,
+      messages: [],
+      createdAt: new Date().toISOString(),
+      updatedAt: new Date().toISOString(),
+      agentMode,
+    };
+
+    this.save(session);
+    return session;
+  }
+
+  /**
+   * Save session to disk
+   */
+  save(session: Session): void {
+    session.updatedAt = new Date().toISOString();
+    const filePath = path.join(this.sessionsDir, `${session.id}.json`);
+    fs.writeFileSync(filePath, JSON.stringify(session, null, 2), 'utf-8');
+  }
+
+  /**
+   * Load session by ID
+   */
+  load(id: string): Session | null {
+    const filePath = path.join(this.sessionsDir, `${id}.json`);
+    
+    if (!fs.existsSync(filePath)) {
+      return null;
+    }
+
+    try {
+      const content = fs.readFileSync(filePath, 'utf-8');
+      return JSON.parse(content) as Session;
+    } catch {
+      return null;
+    }
+  }
+
+  /**
+   * Get the most recent session for a project
+   */
+  getLatest(project: string): Session | null {
+    const sessions = this.list();
+    const projectSessions = sessions
+      .filter(s => s.project === project)
+      .sort((a, b) => new Date(b.updatedAt).getTime() - new Date(a.updatedAt).getTime());
+
+    return projectSessions.length > 0 ? this.load(projectSessions[0].id) : null;
+  }
+
+  /**
+   * List all sessions (metadata only)
+   */
+  list(): Omit<Session, 'messages'>[] {
+    const files = fs.readdirSync(this.sessionsDir)
+      .filter(f => f.endsWith('.json'));
+
+    return files.map(f => {
+      try {
+        const content = fs.readFileSync(path.join(this.sessionsDir, f), 'utf-8');
+        const session = JSON.parse(content) as Session;
+        // Return without messages for efficiency
+        const { messages, ...metadata } = session;
+        return { ...metadata, messages: [] };
+      } catch {
+        return null;
+      }
+    }).filter(Boolean) as Omit<Session, 'messages'>[];
+  }
+
+  /**
+   * Delete a session
+   */
+  delete(id: string): boolean {
+    const filePath = path.join(this.sessionsDir, `${id}.json`);
+    
+    if (fs.existsSync(filePath)) {
+      fs.unlinkSync(filePath);
+      return true;
+    }
+    
+    return false;
+  }
+
+  /**
+   * Clear all sessions
+   */
+  clearAll(): number {
+    const files = fs.readdirSync(this.sessionsDir)
+      .filter(f => f.endsWith('.json'));
+
+    files.forEach(f => fs.unlinkSync(path.join(this.sessionsDir, f)));
+    return files.length;
+  }
+
+  /**
+   * Add message to session
+   */
+  addMessage(session: Session, message: ChatMessage): void {
+    session.messages.push(message);
+    this.save(session);
+  }
+
+  /**
+   * Get session summary for display
+   */
+  getSummary(session: Session): string {
+    const userMessages = session.messages.filter(m => m.role === 'user');
+    const preview = userMessages.length > 0 
+      ? userMessages[0].content.substring(0, 50) + '...'
+      : 'Empty session';
+
+    return `[${session.id}] ${preview} (${userMessages.length} messages)`;
+  }
+}