1 files changed, 10 insertions, 0 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();