managarten/apps/picture/apps/backend/src/db/connection.ts

import { drizzle } from 'drizzle-orm/postgres-js';
import * as schema from './schema';

// Use require for postgres to avoid ESM/CommonJS interop issues
// eslint-disable-next-line @typescript-eslint/no-var-requires
const postgres = require('postgres');

let connection: ReturnType<typeof postgres> | null = null;
let db: ReturnType<typeof drizzle> | null = null;

export function getConnection(databaseUrl: string) {
  if (!connection) {
    connection = postgres(databaseUrl, {
      max: 10,
      idle_timeout: 20,
      connect_timeout: 10,
    });
  }
  return connection;
}

export function getDb(databaseUrl: string) {
  if (!db) {
    const conn = getConnection(databaseUrl);
    db = drizzle(conn, { schema });
  }
  return db;
}

export async function closeConnection() {
  if (connection) {
    await connection.end();
    connection = null;
    db = null;
  }
}

export type Database = ReturnType<typeof getDb>;