lobehub/scripts/nextauth-to-betterauth/_internal/db.ts

import { Pool as NeonPool, neonConfig } from '@neondatabase/serverless';
import { drizzle as neonDrizzle } from 'drizzle-orm/neon-serverless';
import { drizzle as nodeDrizzle } from 'drizzle-orm/node-postgres';
import { Pool as NodePool } from 'pg';
import ws from 'ws';

// schema is the only dependency on project code, required for type-safe migrations
import * as schemaModule from '../../../packages/database/src/schemas';
import { getDatabaseDriver, getDatabaseUrl } from './config';

function createDatabase() {
  const databaseUrl = getDatabaseUrl();
  const driver = getDatabaseDriver();

  if (driver === 'node') {
    const pool = new NodePool({ connectionString: databaseUrl });
    const db = nodeDrizzle(pool, { schema: schemaModule });
    return { db, pool };
  }

  // neon driver (default)
  // https://github.com/neondatabase/serverless/blob/main/CONFIG.md#websocketconstructor-typeof-websocket--undefined
  neonConfig.webSocketConstructor = ws;
  const pool = new NeonPool({ connectionString: databaseUrl });
  const db = neonDrizzle(pool, { schema: schemaModule });
  return { db, pool };
}

const { db, pool } = createDatabase();

export { db, pool };
export * as schema from '../../../packages/database/src/schemas';