brackt/server/db.ts

import { drizzle } from "drizzle-orm/postgres-js";
import postgres from "postgres";
import * as schema from "~/database/schema";

// Cache the client on globalThis so HMR re-evaluations and multiple server
// modules share a single connection pool instead of each opening their own.
declare global {
  // eslint-disable-next-line no-var
  var __pgClient: ReturnType<typeof postgres> | undefined;
  // eslint-disable-next-line no-var
  var __pgDb: ReturnType<typeof drizzle<typeof schema>> | undefined;
}

function getDb() {
  if (!globalThis.__pgDb) {
    if (!process.env.DATABASE_URL) throw new Error("DATABASE_URL is required");
    globalThis.__pgClient ??= postgres(process.env.DATABASE_URL, {
      prepare: process.env.PGBOUNCER !== "true",
    });
    globalThis.__pgDb = drizzle(globalThis.__pgClient, { schema });
  }
  return globalThis.__pgDb;
}

// Export a Proxy so callers can use `db.query.foo` etc. without calling getDb() themselves.
// The proxy defers the DATABASE_URL check until the first actual property access.
export const db = new Proxy({} as ReturnType<typeof drizzle<typeof schema>>, {
  get(_target, prop) {
    return getDb()[prop as keyof ReturnType<typeof drizzle<typeof schema>>];
  },
});