@myrialabs/clopen 0.1.4 → 0.1.6

package/backend/lib/chat/stream-manager.ts

@@ -268,6 +268,14 @@ class StreamManager extends EventEmitter {
 		// Track user message ID for stream-end snapshot capture
 		let userMessageId: string | undefined;
 
+		// Initialize session baseline for snapshot system (non-blocking)
+		if (requestData.projectPath && requestData.chatSessionId) {
+			snapshotService.initializeSessionBaseline(
+				requestData.projectPath,
+				requestData.chatSessionId
+			).catch(err => debug.error('snapshot', 'Failed to initialize session baseline:', err));
+		}
+
 		try {
 			const { projectPath, prompt, chatSessionId, engine: engineType = 'claude-code', model, temperature, claudeAccountId } = requestData;
 

package/backend/lib/database/migrations/022_add_snapshot_changes_column.ts

@@ -0,0 +1,35 @@
+/**
+ * Migration: Add session_changes column to message_snapshots
+ * Purpose: Store session-scoped file changes (old/new hash per file)
+ * instead of full project state. This enables:
+ * - Session-scoped restore (only undo changes within the session)
+ * - Cross-session conflict detection
+ * - Dramatically reduced storage (only changed files stored)
+ *
+ * session_changes JSON format:
+ * {
+ *   "filepath": { "oldHash": "sha256...", "newHash": "sha256..." },
+ *   ...
+ * }
+ */
+
+import type { DatabaseConnection } from '$shared/types/database/connection';
+import { debug } from '$shared/utils/logger';
+
+export const description = 'Add session_changes for session-scoped snapshot deltas';
+
+export const up = (db: DatabaseConnection): void => {
+	debug.log('migration', 'Adding session_changes column to message_snapshots...');
+
+	db.exec(`
+		ALTER TABLE message_snapshots
+		ADD COLUMN session_changes TEXT
+	`);
+
+	debug.log('migration', 'session_changes column added');
+};
+
+export const down = (db: DatabaseConnection): void => {
+	debug.log('migration', 'Removing session_changes column...');
+	debug.warn('migration', 'Rollback not implemented for session_changes (SQLite limitation)');
+};

package/backend/lib/database/migrations/index.ts

@@ -20,6 +20,7 @@ import * as migration018 from './018_create_claude_accounts_table';
 import * as migration019 from './019_add_claude_account_to_sessions';
 import * as migration020 from './020_add_snapshot_tree_hash';
 import * as migration021 from './021_drop_prompt_templates_table';
+import * as migration022 from './022_add_snapshot_changes_column';
 
 // Export all migrations in order
 export const migrations = [
@@ -148,6 +149,12 @@ export const migrations = [
 		description: migration021.description,
 		up: migration021.up,
 		down: migration021.down
+	},
+	{
+		id: '022',
+		description: migration022.description,
+		up: migration022.up,
+		down: migration022.down
 	}
 ];
 

package/backend/lib/database/queries/snapshot-queries.ts

@@ -27,6 +27,7 @@ export const snapshotQueries = {
 		deletions?: number;
 		branch_id?: string;
 		tree_hash?: string; // Blob store tree hash (new format)
+		session_changes?: any; // SessionScopedChanges object
 	}): MessageSnapshot {
 		const db = getDatabase();
 		const id = data.id || `snapshot_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
@@ -48,7 +49,8 @@ export const snapshotQueries = {
 			deletions: data.deletions || 0,
 			is_deleted: 0,
 			branch_id: data.branch_id || null,
-			tree_hash: data.tree_hash || null
+			tree_hash: data.tree_hash || null,
+			session_changes: data.session_changes ? JSON.stringify(data.session_changes) : null
 		};
 
 		db.prepare(`
@@ -57,8 +59,8 @@ export const snapshotQueries = {
 				files_snapshot, project_metadata, created_at,
 				snapshot_type, parent_snapshot_id, delta_changes,
 				files_changed, insertions, deletions,
-				is_deleted, branch_id, tree_hash
-			) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 0, ?, ?)
+				is_deleted, branch_id, tree_hash, session_changes
+			) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 0, ?, ?, ?)
 		`).run(
 			snapshot.id,
 			snapshot.message_id,
@@ -74,7 +76,8 @@ export const snapshotQueries = {
 			snapshot.insertions,
 			snapshot.deletions,
 			snapshot.branch_id || null,
-			snapshot.tree_hash || null
+			snapshot.tree_hash || null,
+			snapshot.session_changes || null
 		);
 
 		return snapshot;

package/backend/lib/files/file-watcher.ts

@@ -82,6 +82,36 @@ interface ProjectWatcher {
 class FileWatcherManager {
 	private watchers = new Map<string, ProjectWatcher>();
 
+	/**
+	 * Per-project dirty file tracking for snapshot system.
+	 * Accumulates changed file relative paths between snapshot captures.
+	 */
+	private dirtyFiles = new Map<string, Set<string>>();
+
+	/**
+	 * Get dirty files accumulated since last clear for a project.
+	 */
+	getDirtyFiles(projectId: string): Set<string> {
+		return this.dirtyFiles.get(projectId) || new Set();
+	}
+
+	/**
+	 * Clear dirty files after snapshot capture.
+	 */
+	clearDirtyFiles(projectId: string): void {
+		this.dirtyFiles.delete(projectId);
+	}
+
+	/**
+	 * Track a file as dirty for snapshot purposes.
+	 */
+	private trackDirtyFile(projectId: string, relativePath: string): void {
+		if (!this.dirtyFiles.has(projectId)) {
+			this.dirtyFiles.set(projectId, new Set());
+		}
+		this.dirtyFiles.get(projectId)!.add(relativePath);
+	}
+
 	/**
 	 * Start watching a project directory
 	 */
@@ -238,6 +268,10 @@ class FileWatcherManager {
 			}
 		}
 
+		// Track dirty file for snapshot system
+		const relativePath = relative(projectPath, fullPath).replace(/\\/g, '/');
+		this.trackDirtyFile(projectId, relativePath);
+
 		// Create file change object
 		const fileChange: FileChange = {
 			path: fullPath,

package/backend/lib/project/status-manager.ts

@@ -18,11 +18,13 @@ const INTERACTIVE_TOOLS = new Set(['AskUserQuestion']);
 function detectStreamWaitingInput(stream: StreamState): boolean {
   if (stream.status !== 'active') return false;
 
+  // SSEEventData.message is SDKMessage: { type, message: { content: [...] } }
+  // Content blocks live at msg.message.content, NOT msg.content
   const answeredToolIds = new Set<string>();
   for (const event of stream.messages) {
     const msg = event.message;
-    if (!msg || (msg as any).type !== 'user' || !(msg as any).content) continue;
-    const content = Array.isArray((msg as any).content) ? (msg as any).content : [];
+    if (!msg || (msg as any).type !== 'user') continue;
+    const content = Array.isArray((msg as any).message?.content) ? (msg as any).message.content : [];
     for (const item of content) {
       if (item.type === 'tool_result' && item.tool_use_id) {
         answeredToolIds.add(item.tool_use_id);
@@ -32,8 +34,8 @@ function detectStreamWaitingInput(stream: StreamState): boolean {
 
   for (const event of stream.messages) {
     const msg = event.message;
-    if (!msg || (msg as any).type !== 'assistant' || !(msg as any).content) continue;
-    const content = Array.isArray((msg as any).content) ? (msg as any).content : [];
+    if (!msg || (msg as any).type !== 'assistant') continue;
+    const content = Array.isArray((msg as any).message?.content) ? (msg as any).message.content : [];
     if (content.some((item: any) =>
       item.type === 'tool_use' && INTERACTIVE_TOOLS.has(item.name) && item.id && !answeredToolIds.has(item.id)
     )) {