dave-code 1.0.4 → 1.2.0

package/bin/sessionManager.js

@@ -0,0 +1,182 @@
+/**
+ * ── DESIGN PHILOSOPHY / 视觉设计与交互规范 ──
+ * 为了向用户提供极致、顺滑的交互体验，Dave Code 规定：
+ * 1. 所有耗时的 API 请求、大文件处理和本地工具调用，都必须具有动画化（Animated）或动态（Dynamic）的终端 UI。
+ * 2. 严禁使用静态、刷屏式的连续 Log 输出。对于多步骤操作（如分片消化），应使用单行原地更新（In-place update）的进度条和加载动画。
+ * 3. 使用富文本终端色彩（如精心调配的 HSL/ANSI 颜色）和微型字符动画（如 Spinner）来提供实时的视觉反馈。
+ * 后续维护与功能续写时，必须严格遵守此设计理念。
+ */
+
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+import crypto from 'crypto';
+
+export let SESSIONS_DIR = path.join(os.homedir(), '.dave-code-sessions');
+
+export function setSessionsDirForTesting(directory) {
+  SESSIONS_DIR = directory;
+}
+
+function normalizeMessage(message) {
+  if (!message || typeof message !== 'object') return null;
+  if (!['user', 'assistant', 'tool'].includes(message.role)) return null;
+  if (typeof message.content !== 'string') return null;
+  return { ...message, role: message.role, content: message.content };
+}
+
+function normalizeSession(session) {
+  if (!session || typeof session !== 'object' || !session.id || !Array.isArray(session.messages)) return null;
+  if (!/^[A-Za-z0-9_-]+$/.test(String(session.id))) return null;
+  const messages = session.messages.map(normalizeMessage).filter(Boolean);
+  const rawPlan = session.contextPlan && typeof session.contextPlan === 'object' ? session.contextPlan : null;
+  const contextPlan = rawPlan && Number.isFinite(rawPlan.budgetTokens) ? {
+    version: 1,
+    requestFingerprint: String(rawPlan.requestFingerprint || '').slice(0, 100),
+    snapshotId: String(rawPlan.snapshotId || '').slice(0, 100),
+    contextWindowTokens: Math.max(8192, Number(rawPlan.contextWindowTokens) || 32768),
+    usableTokens: Math.max(4096, Number(rawPlan.usableTokens) || Number(rawPlan.budgetTokens)),
+    budgetTokens: Math.max(4096, Number(rawPlan.budgetTokens)),
+    reserveTokens: Math.max(0, Number(rawPlan.reserveTokens) || 0),
+    rationale: String(rawPlan.rationale || '').slice(0, 2000),
+    recommendedFiles: Array.isArray(rawPlan.recommendedFiles) ? rawPlan.recommendedFiles.slice(0, 100) : [],
+    degraded: rawPlan.degraded === true,
+    createdAt: Number(rawPlan.createdAt) || Date.now()
+  } : null;
+  return {
+    ...session,
+    id: String(session.id),
+    title: String(session.title || 'New Chat'),
+    timestamp: Number(session.timestamp) || Date.now(),
+    messages,
+    workspaceRoot: typeof session.workspaceRoot === 'string' ? session.workspaceRoot : null,
+    activeOpenFile: typeof session.activeOpenFile === 'string' ? session.activeOpenFile : null,
+    workMode: session.workMode === 'thunder' ? 'thunder' : 'highway',
+    contextPlan
+  };
+}
+
+function atomicWriteSession(filePath, sessionData) {
+  const tempPath = `${filePath}.${process.pid}.${crypto.randomBytes(5).toString('hex')}.tmp`;
+  fs.writeFileSync(tempPath, JSON.stringify(sessionData, null, 2), { encoding: 'utf8', mode: 0o600 });
+  try {
+    fs.chmodSync(tempPath, 0o600);
+  } catch {
+    // Best effort on Windows.
+  }
+  try {
+    fs.renameSync(tempPath, filePath);
+  } catch (error) {
+    if (process.platform !== 'win32' || !fs.existsSync(filePath)) throw error;
+    fs.unlinkSync(filePath);
+    fs.renameSync(tempPath, filePath);
+  } finally {
+    if (fs.existsSync(tempPath)) fs.unlinkSync(tempPath);
+  }
+}
+
+export function ensureSessionsDir() {
+  try {
+    if (!fs.existsSync(SESSIONS_DIR)) {
+      fs.mkdirSync(SESSIONS_DIR, { recursive: true, mode: 0o700 });
+    }
+    try {
+      fs.chmodSync(SESSIONS_DIR, 0o700);
+    } catch {
+      // Best effort on Windows.
+    }
+  } catch (e) {
+    // Ignore folder creation errors (handled gracefully during saves)
+  }
+}
+
+/**
+ * List all saved chat sessions sorted by timestamp descending (newest first).
+ * @returns {Array} Array of session objects
+ */
+export function listSessions() {
+  ensureSessionsDir();
+  try {
+    if (!fs.existsSync(SESSIONS_DIR)) return [];
+    
+    const files = fs.readdirSync(SESSIONS_DIR);
+    const sessions = [];
+    for (const file of files) {
+      if (file.endsWith('.json')) {
+        try {
+          const filePath = path.join(SESSIONS_DIR, file);
+          const content = fs.readFileSync(filePath, 'utf8');
+          const session = normalizeSession(JSON.parse(content));
+          if (session && session.id) {
+            sessions.push(session);
+          }
+        } catch (e) {
+          // Skip invalid JSON files
+        }
+      }
+    }
+    return sessions.sort((a, b) => (b.timestamp || 0) - (a.timestamp || 0));
+  } catch (e) {
+    return [];
+  }
+}
+
+/**
+ * Save or update a session file.
+ * @param {string} sessionId Unique session identifier
+ * @param {object} sessionData Session object containing messages, title, timestamp, and workspace metadata
+ */
+export function saveSession(sessionId, sessionData) {
+  ensureSessionsDir();
+  try {
+    const safeId = String(sessionId || '');
+    if (!/^[A-Za-z0-9_-]+$/.test(safeId)) return false;
+    const normalized = normalizeSession({ ...sessionData, id: safeId });
+    if (!normalized) return false;
+    const filePath = path.join(SESSIONS_DIR, `${safeId}.json`);
+    atomicWriteSession(filePath, normalized);
+    return true;
+  } catch (e) {
+    return false;
+  }
+}
+
+/**
+ * Load a session from its identifier.
+ * @param {string} sessionId Unique session identifier
+ * @returns {object|null} Loaded session or null if not found
+ */
+export function loadSession(sessionId) {
+  ensureSessionsDir();
+  try {
+    if (!/^[A-Za-z0-9_-]+$/.test(String(sessionId || ''))) return null;
+    const filePath = path.join(SESSIONS_DIR, `${sessionId}.json`);
+    if (fs.existsSync(filePath)) {
+      const content = fs.readFileSync(filePath, 'utf8');
+      return normalizeSession(JSON.parse(content));
+    }
+  } catch (e) {
+    // Return null on parsing or read failures
+  }
+  return null;
+}
+
+/**
+ * Delete a session from storage.
+ * @param {string} sessionId Unique session identifier
+ * @returns {boolean} True if successfully deleted
+ */
+export function deleteSession(sessionId) {
+  ensureSessionsDir();
+  try {
+    if (!/^[A-Za-z0-9_-]+$/.test(String(sessionId || ''))) return false;
+    const filePath = path.join(SESSIONS_DIR, `${sessionId}.json`);
+    if (fs.existsSync(filePath)) {
+      fs.unlinkSync(filePath);
+      return true;
+    }
+  } catch (e) {
+    // Ignore unlink failures
+  }
+  return false;
+}