3 files changed, 14 insertions, 1 deletions

diff --git a/src/main/db/index.ts b/src/main/db/index.ts
index a77cdd4..1613abc 100644
--- a/src/main/db/index.ts
+++ b/src/main/db/index.ts
@@ -20,9 +20,19 @@ export function getDb(): Database.Database {
   db.pragma("foreign_keys = ON");
 
   initSchema(db);
+  runMigrations(db);
+
   return db;
 }
 
+function runMigrations(db: Database.Database): void {
+  // v2: add git_branch column to sessions
+  const cols = db.pragma("table_info(sessions)") as { name: string }[];
+  if (!cols.find((c) => c.name === "git_branch")) {
+    db.exec("ALTER TABLE sessions ADD COLUMN git_branch TEXT");
+  }
+}
+
 export function closeDb() {
   if (db) {
     db.close();
diff --git a/src/main/db/schema.ts b/src/main/db/schema.ts
index c2093f9..39ee567 100644
--- a/src/main/db/schema.ts
+++ b/src/main/db/schema.ts
@@ -17,6 +17,7 @@ export function initSchema(db: Database.Database) {
       phase TEXT NOT NULL DEFAULT 'research',
       claude_session_id TEXT,
       permission_mode TEXT NOT NULL DEFAULT 'acceptEdits',
+      git_branch TEXT,
       created_at INTEGER NOT NULL DEFAULT (unixepoch()),
       updated_at INTEGER NOT NULL DEFAULT (unixepoch())
     );
diff --git a/src/main/db/sessions.ts b/src/main/db/sessions.ts
index 684bb9e..3e6352c 100644
--- a/src/main/db/sessions.ts
+++ b/src/main/db/sessions.ts
@@ -11,6 +11,7 @@ export interface Session {
   phase: Phase;
   claude_session_id: string | null;
   permission_mode: PermissionMode;
+  git_branch: string | null;
   created_at: number;
   updated_at: number;
 }
@@ -52,6 +53,7 @@ export function createSession(projectId: string, name: string): Session {
     phase: "research",
     claude_session_id: null,
     permission_mode: "acceptEdits",
+    git_branch: null,
     created_at: now,
     updated_at: now,
   };
@@ -59,7 +61,7 @@ export function createSession(projectId: string, name: string): Session {
 
 export function updateSession(
   id: string,
-  updates: Partial<Pick<Session, "name" | "phase" | "claude_session_id" | "permission_mode">>
+  updates: Partial<Pick<Session, "name" | "phase" | "claude_session_id" | "permission_mode" | "git_branch">>
 ): void {
   const db = getDb();
   const sets: string[] = [];