fix(store): add explicit v8->v9 report schema migration

Author: Him188

hub/src/store/index.ts

@@ -43,7 +43,7 @@ export { UserStore } from './userStore'
 export { GroupStore } from './groupStore'
 export { ReportStore } from './reportStore'
 
-const SCHEMA_VERSION: number = 8
+const SCHEMA_VERSION: number = 9
 const REQUIRED_TABLES = [
     'sessions',
     'machines',
@@ -192,6 +192,13 @@ export class Store {
             return
         }
 
+        if (currentVersion === 8 && SCHEMA_VERSION >= 9) {
+            this.migrateFromV8ToV9()
+            this.setUserVersion(9)
+            this.initSchema()
+            return
+        }
+
         if (currentVersion !== SCHEMA_VERSION) {
             throw this.buildSchemaMismatchError(currentVersion)
         }
@@ -794,6 +801,69 @@ export class Store {
         }
     }
 
+    private migrateFromV8ToV9(): void {
+        try {
+            this.db.exec('BEGIN')
+            this.db.exec(`
+                CREATE TABLE IF NOT EXISTS reports (
+                    id TEXT PRIMARY KEY,
+                    namespace TEXT NOT NULL DEFAULT 'default',
+                    session_id TEXT,
+                    task_id TEXT,
+                    title TEXT NOT NULL,
+                    status TEXT NOT NULL DEFAULT 'unknown',
+                    markdown TEXT NOT NULL DEFAULT '',
+                    metadata TEXT,
+                    created_at INTEGER NOT NULL,
+                    updated_at INTEGER NOT NULL,
+                    FOREIGN KEY (session_id) REFERENCES sessions(id) ON DELETE SET NULL
+                );
+                CREATE INDEX IF NOT EXISTS idx_reports_namespace_updated
+                    ON reports(namespace, updated_at DESC);
+                CREATE INDEX IF NOT EXISTS idx_reports_session_namespace
+                    ON reports(session_id, namespace);
+
+                CREATE TABLE IF NOT EXISTS report_assets (
+                    id TEXT PRIMARY KEY,
+                    report_id TEXT NOT NULL,
+                    namespace TEXT NOT NULL DEFAULT 'default',
+                    file_name TEXT NOT NULL,
+                    storage_key TEXT NOT NULL,
+                    mime_type TEXT NOT NULL,
+                    size INTEGER NOT NULL,
+                    caption TEXT,
+                    created_at INTEGER NOT NULL,
+                    FOREIGN KEY (report_id) REFERENCES reports(id) ON DELETE CASCADE
+                );
+                CREATE INDEX IF NOT EXISTS idx_report_assets_report
+                    ON report_assets(report_id, created_at ASC);
+                CREATE UNIQUE INDEX IF NOT EXISTS idx_report_assets_storage_key
+                    ON report_assets(report_id, storage_key);
+
+                CREATE TABLE IF NOT EXISTS report_shares (
+                    id TEXT PRIMARY KEY,
+                    report_id TEXT NOT NULL,
+                    namespace TEXT NOT NULL DEFAULT 'default',
+                    token TEXT NOT NULL UNIQUE,
+                    created_by TEXT,
+                    created_at INTEGER NOT NULL,
+                    expires_at INTEGER,
+                    revoked_at INTEGER,
+                    FOREIGN KEY (report_id) REFERENCES reports(id) ON DELETE CASCADE
+                );
+                CREATE INDEX IF NOT EXISTS idx_report_shares_report
+                    ON report_shares(report_id, created_at DESC);
+                CREATE INDEX IF NOT EXISTS idx_report_shares_token
+                    ON report_shares(token);
+            `)
+            this.db.exec('COMMIT')
+        } catch (error) {
+            this.db.exec('ROLLBACK')
+            const message = error instanceof Error ? error.message : String(error)
+            throw new Error(`SQLite schema migration v8->v9 failed: ${message}`)
+        }
+    }
+
     private getMachineColumnNames(): Set<string> {
         const rows = this.db.prepare('PRAGMA table_info(machines)').all() as Array<{ name: string }>
         return new Set(rows.map((row) => row.name))

hub/src/store/schemaRepair.test.ts

@@ -16,6 +16,12 @@ const GROUP_TABLES = [
     'group_notes'
 ] as const
 
+const REPORT_TABLES = [
+    'reports',
+    'report_assets',
+    'report_shares'
+] as const
+
 function closeStore(store: Store): void {
     const db = (store as unknown as { db: Database }).db
     db.close()
@@ -63,4 +69,36 @@ describe('Store schema repair', () => {
         closeStore(repaired)
         rmSync(dir, { recursive: true, force: true })
     })
+
+    it('runs v8 to v9 migration and creates report tables', () => {
+        const dir = mkdtempSync(join(tmpdir(), 'haqi-store-repair-'))
+        const dbPath = join(dir, 'hapi.db')
+
+        const seeded = new Store(dbPath)
+        const seededDb = (seeded as unknown as { db: Database }).db
+        seededDb.exec(`
+            DROP TABLE IF EXISTS report_shares;
+            DROP TABLE IF EXISTS report_assets;
+            DROP TABLE IF EXISTS reports;
+            PRAGMA user_version = 8;
+        `)
+        closeStore(seeded)
+
+        const migrated = new Store(dbPath)
+        const migratedDb = (migrated as unknown as { db: Database }).db
+        const versionRow = migratedDb.prepare('PRAGMA user_version').get() as { user_version: number } | undefined
+        expect(versionRow?.user_version).toBe(9)
+
+        const placeholders = REPORT_TABLES.map(() => '?').join(', ')
+        const rows = migratedDb.prepare(
+            `SELECT name FROM sqlite_master WHERE type = 'table' AND name IN (${placeholders})`
+        ).all(...REPORT_TABLES) as Array<{ name: string }>
+        const tableSet = new Set(rows.map((row) => row.name))
+        for (const table of REPORT_TABLES) {
+            expect(tableSet.has(table)).toBe(true)
+        }
+
+        closeStore(migrated)
+        rmSync(dir, { recursive: true, force: true })
+    })
 })