core: show visual progress bar during database migration so users can see real-time status of projects, sessions, and messages being converted

Author: thdxr

packages/opencode/src/index.ts

@@ -82,7 +82,31 @@ const cli = yargs(hideBin(process.argv))
     const marker = path.join(Global.Path.data, "opencode.db")
     if (!(await Bun.file(marker).exists())) {
       console.log("Performing one time database migration, may take a few minutes...")
-      await JsonMigration.run(Database.Client().$client)
+      const tty = process.stdout.isTTY
+      const width = 36
+      const orange = "\x1b[38;5;214m"
+      const muted = "\x1b[0;2m"
+      const reset = "\x1b[0m"
+      let last = -1
+      if (tty) process.stdout.write("\x1b[?25l")
+      try {
+        await JsonMigration.run(Database.Client().$client, {
+          progress: (event) => {
+            if (!tty) return
+            const percent = Math.floor((event.current / event.total) * 100)
+            if (percent === last && event.current !== event.total) return
+            last = percent
+            const fill = Math.round((percent / 100) * width)
+            const bar = `${"■".repeat(fill)}${"･".repeat(width - fill)}`
+            process.stdout.write(
+              `\r${orange}${bar} ${percent.toString().padStart(3)}%${reset} ${muted}${event.label.padEnd(12)} ${event.current}/${event.total}${reset}`,
+            )
+            if (event.current === event.total) process.stdout.write("\n")
+          },
+        })
+      } finally {
+        if (tty) process.stdout.write("\x1b[?25h")
+      }
       console.log("Database migration complete.")
     }
   })

packages/opencode/src/storage/json-migration.ts

@@ -11,7 +11,17 @@ import { existsSync } from "fs"
 export namespace JsonMigration {
   const log = Log.create({ service: "json-migration" })
 
-  export async function run(sqlite: Database) {
+  export type Progress = {
+    current: number
+    total: number
+    label: string
+  }
+
+  type Options = {
+    progress?: (event: Progress) => void
+  }
+
+  export async function run(sqlite: Database, options?: Options) {
     const storageDir = path.join(Global.Path.data, "storage")
 
     if (!existsSync(storageDir)) {
@@ -120,6 +130,25 @@ export namespace JsonMigration {
       shares: shareFiles.length,
     })
 
+    const total = Math.max(
+      1,
+      projectFiles.length +
+        sessionFiles.length +
+        messageFiles.length +
+        partFiles.length +
+        todoFiles.length +
+        permFiles.length +
+        shareFiles.length,
+    )
+    const progress = options?.progress
+    let current = 0
+    const step = (label: string, count: number) => {
+      current = Math.min(total, current + count)
+      progress?.({ current, total, label })
+    }
+
+    progress?.({ current, total, label: "starting" })
+
     sqlite.exec("BEGIN TRANSACTION")
 
     // Migrate projects first (no FK deps)
@@ -151,6 +180,7 @@ export namespace JsonMigration {
         })
       }
       stats.projects += insert(projectValues, ProjectTable, "project")
+      step("projects", end - i)
     }
     log.info("migrated projects", { count: stats.projects, duration: Math.round(performance.now() - start) })
 
@@ -195,6 +225,7 @@ export namespace JsonMigration {
         })
       }
       stats.sessions += insert(sessionValues, SessionTable, "session")
+      step("sessions", end - i)
     }
     log.info("migrated sessions", { count: stats.sessions })
     if (orphans.sessions > 0) {
@@ -241,6 +272,7 @@ export namespace JsonMigration {
       }
       values.length = count
       stats.messages += insert(values, MessageTable, "message")
+      step("messages", end - i)
     }
     log.info("migrated messages", { count: stats.messages })
 
@@ -281,6 +313,7 @@ export namespace JsonMigration {
       }
       values.length = count
       stats.parts += insert(values, PartTable, "part")
+      step("parts", end - i)
     }
     log.info("migrated parts", { count: stats.parts })
 
@@ -317,6 +350,7 @@ export namespace JsonMigration {
         }
       }
       stats.todos += insert(values, TodoTable, "todo")
+      step("todos", end - i)
     }
     log.info("migrated todos", { count: stats.todos })
     if (orphans.todos > 0) {
@@ -341,6 +375,7 @@ export namespace JsonMigration {
         permValues.push({ project_id: projectID, data })
       }
       stats.permissions += insert(permValues, PermissionTable, "permission")
+      step("permissions", end - i)
     }
     log.info("migrated permissions", { count: stats.permissions })
     if (orphans.permissions > 0) {
@@ -369,6 +404,7 @@ export namespace JsonMigration {
         shareValues.push({ session_id: sessionID, id: data.id, secret: data.secret, url: data.url })
       }
       stats.shares += insert(shareValues, SessionShareTable, "session_share")
+      step("shares", end - i)
     }
     log.info("migrated session shares", { count: stats.shares })
     if (orphans.shares > 0) {
@@ -393,6 +429,8 @@ export namespace JsonMigration {
       log.warn("migration errors", { errors: stats.errors.slice(0, 20) })
     }
 
+    progress?.({ current: total, total, label: "complete" })
+
     return stats
   }
 }