feat: pg support

Author: luoling8192

packages/drizzle-orm-browser-migrator/package.json

@@ -24,6 +24,11 @@
       "types": "./dist/pglite/index.d.mts",
       "import": "./dist/pglite/index.mjs",
       "node": "./dist/pglite/index.mjs"
+    },
+    "./pg": {
+      "types": "./dist/pg/index.d.mts",
+      "import": "./dist/pg/index.mjs",
+      "node": "./dist/pg/index.mjs"
     }
   },
   "main": "./dist/index.mjs",
@@ -44,7 +49,7 @@
     "attw": "attw --pack . --profile esm-only --ignore-rules cjs-resolves-to-esm"
   },
   "peerDependencies": {
-    "drizzle-orm": ">=0.40.0"
+    "drizzle-orm": ">=0.44.0"
   },
   "dependencies": {
     "@guiiai/logg": "^1.0.10"

packages/drizzle-orm-browser-migrator/src/pg/index.ts

@@ -0,0 +1,86 @@
+import type { PostgresJsDatabase } from 'drizzle-orm/postgres-js'
+
+import { Format, useLogg } from '@guiiai/logg'
+import { sql } from 'drizzle-orm'
+
+import packageJSON from '../../package.json' with { type: 'json' }
+
+async function listTables<TSchema extends Record<string, unknown>>(db: PostgresJsDatabase<TSchema>) {
+  const res = await db.execute(sql`
+    SELECT
+      table_name
+    FROM information_schema.tables
+    WHERE table_schema = 'public';
+  `)
+
+  return res.rows
+}
+
+export async function migrate<TSchema extends Record<string, unknown>>(db: PostgresJsDatabase<TSchema>, bundledMigrations: {
+  idx: number
+  when: number
+  tag: string
+  hash: string
+  sql: string[]
+}[]) {
+  const log = useLogg(packageJSON.name).withFormat(Format.Pretty)
+  const TABLE_NAME = sql.identifier('__drizzle_migrations')
+
+  await db.execute(
+    sql`CREATE TABLE IF NOT EXISTS ${TABLE_NAME} (
+      id bigserial PRIMARY KEY NOT NULL,
+      hash text NOT NULL,
+      tag text NOT NULL,
+      created_at bigint NOT NULL
+    );`,
+  )
+
+  const deployments = await db.execute<{
+    id: number
+    hash: string
+    created_at: number
+  }>(
+    sql`SELECT
+      id,
+      hash,
+      created_at
+    FROM ${TABLE_NAME}
+    ORDER BY created_at DESC
+    LIMIT 1;`,
+  )
+
+  const deployment = deployments.rows.at(0)
+  const migrations = bundledMigrations.filter((migration) => {
+    const timestamp = deployment?.created_at ?? 0
+    return !deployment || Number(timestamp) < migration.when
+  })
+  if (migrations.length === 0) {
+    log.withField('tables', await listTables(db)).debug('no pending migrations')
+    log.log('no pending migrations to apply')
+    return
+  }
+
+  await db.transaction(async (tx) => {
+    for (let i = 0; i < migrations.length; i++) {
+      const migration = migrations[i]
+
+      log.log(`${i + 1}. Deploying migration:`)
+      log.log(`     tag  => ${migration.tag}`)
+      log.log(`     hash => ${migration.hash}`)
+      for (const stmt of migration.sql) {
+        await tx.execute(stmt)
+      }
+
+      await tx.execute(sql`
+        INSERT INTO ${TABLE_NAME} ("hash", "created_at", "tag") VALUES (
+          ${sql.raw(`'${migration.hash}'`)},
+          ${sql.raw(`${migration.when}`)},
+          ${sql.raw(`'${migration.tag}'`)}
+        );
+      `)
+    }
+  })
+
+  log.withField('tables', await listTables(db)).debug('migration successful')
+  log.log(`all ${migrations.length} pending migrations applied!`)
+}

packages/drizzle-orm-browser-migrator/tsdown.config.ts

@@ -4,6 +4,7 @@ export default defineConfig({
   entry: {
     'index': './src/index.ts',
     'pglite/index': './src/pglite/index.ts',
+    'pg/index': './src/pg/index.ts',
   },
   unused: true,
   fixedExtension: true,