openAgent

1export * as DatabaseMigration from "./migration"
2 
3import { sql } from "drizzle-orm"
4import { Effect, Semaphore } from "effect"
5import type { EffectDrizzleSqlite } from "@opencode-ai/effect-drizzle-sqlite"
6import { migrations } from "./migration.gen"
7import schema from "./schema.gen"
8 
9type Database = EffectDrizzleSqlite.EffectSQLiteDatabase
10type Transaction = Parameters<Parameters<Database["transaction"]>[0]>[0]
11const lock = Semaphore.makeUnsafe(1)
12 
13export type Migration = {
14  id: string
15  up: (tx: Transaction) => Effect.Effect<void, unknown>
16}
17 
18export function apply(db: Database) {
19  return lock.withPermit(
20    Effect.gen(function* () {
21      const tables = yield* db.all<{ name: string }>(
22        sql`SELECT name FROM sqlite_master WHERE type = 'table' AND name NOT LIKE 'sqlite_%'`,
23      )
24      if (tables.some((table) => table.name === "session")) return yield* applyOnly(db, migrations)
25      if (tables.length > 0) return yield* Effect.die("Database is not empty and has no session table")
26      yield* db.transaction((tx) =>
27        Effect.gen(function* () {
28          yield* schema.up(tx)
29          yield* tx.run(
30            sql`CREATE TABLE ${sql.identifier("migration")} (id TEXT PRIMARY KEY, time_completed INTEGER NOT NULL)`,
31          )
32          yield* Effect.forEach(migrations, (migration) =>
33            tx.run(
34              sql`INSERT INTO ${sql.identifier("migration")} (id, time_completed) VALUES (${migration.id}, ${Date.now()})`,
35            ),
36          )
37        }),
38      )
39    }),
40  )
41}
42 
43export function applyOnly(db: Database, input: Migration[]) {
44  return Effect.gen(function* () {
45    yield* db.run(
46      sql`CREATE TABLE IF NOT EXISTS ${sql.identifier("migration")} (id TEXT PRIMARY KEY, time_completed INTEGER NOT NULL)`,
47    )
48    let completed = new Set(
49      (yield* db.all<{ id: string }>(sql`SELECT id FROM ${sql.identifier("migration")}`)).map((row) => row.id),
50    )
51    if (completed.size === 0) {
52      // Existing installs used Drizzle's migration journal. Seed the new
53      // journal once so TypeScript migrations don't replay old SQL.
54      if (
55        yield* db.get(sql`SELECT name FROM sqlite_master WHERE type = 'table' AND name = ${"__drizzle_migrations"}`)
56      ) {
57        yield* db.run(sql`
58          INSERT OR IGNORE INTO ${sql.identifier("migration")} (id, time_completed)
59          SELECT name, ${Date.now()}
60          FROM ${sql.identifier("__drizzle_migrations")}
61          WHERE name IS NOT NULL
62        `)
63        completed = new Set(
64          (yield* db.all<{ id: string }>(sql`SELECT id FROM ${sql.identifier("migration")}`)).map((row) => row.id),
65        )
66      }
67    }
68 
69    for (const migration of input) {
70      if (completed.has(migration.id)) continue
71      yield* db.transaction((tx) =>
72        Effect.gen(function* () {
73          yield* migration.up(tx)
74          yield* tx.run(
75            sql`INSERT INTO ${sql.identifier("migration")} (id, time_completed) VALUES (${migration.id}, ${Date.now()})`,
76          )
77        }),
78      )
79    }
80  })
81}
82