import "server-only";

import { drizzle } from "drizzle-orm/node-postgres";
import { Pool } from "pg";
import * as schema from "./schema";

/**
 * 数据库客户端（node-postgres 连接池）。
 *
 * ## 为什么用 pg 而不是 neon-http
 *
 * 脚手架原本使用 `drizzle-orm/neon-http` + `@neondatabase/serverless`。
 * 该驱动走的是 **Neon 的 HTTP 代理端点**，不是 Postgres 的 TCP 协议，
 * 因此无法连接本地或自建的普通 Postgres。
 *
 * 本项目改用 `pg`（node-postgres），它既能连本地 Postgres，
 * 也能连 Neon / Supabase / Railway 等任何标准 Postgres——
 * 部署到 Neon 时只需保留其 TCP 连接串（`*.neon.tech` 的 5432 端口）。
 *
 * ## 连接池与 Next.js 开发模式
 *
 * Next.js 开发模式下模块会被热重载，若不缓存连接池，每次改动都会新建一批
 * 连接，很快耗尽数据库的 max_connections。因此把 pool 挂到 globalThis 上做单例。
 */

const connectionString = process.env.DATABASE_URL;

if (!connectionString) {
	throw new Error(
		"缺少环境变量 DATABASE_URL。请在 .env.local 中配置 Postgres 连接串，" +
			"例如：postgresql://user:password@localhost:5432/blog",
	);
}

// 收窄为 string，供下方闭包使用（模块级已判空，但闭包内 TS 无法自动收窄）
const resolvedConnectionString: string = connectionString;

/** 在 globalThis 上缓存连接池，避免开发模式热重载导致的连接泄漏 */
const globalForDb = globalThis as unknown as {
	__postgresPool?: Pool;
};

function createPool(): Pool {
	const pool = new Pool({
		connectionString: resolvedConnectionString,
		// 本地开发（localhost）通常没配证书，强制 SSL 会直接失败；
		// 托管数据库（Neon / Supabase 等）则要求 SSL。
		ssl: shouldUseSsl(resolvedConnectionString)
			? { rejectUnauthorized: false }
			: undefined,
		// 单实例默认 10 连接足够；Serverless 环境建议调小或接 Neon 的 pooler
		max: Number(process.env.DATABASE_POOL_MAX ?? 10),
		idleTimeoutMillis: 30_000,
		connectionTimeoutMillis: 10_000,
		// 语句级超时，避免慢查询长期占用连接
		statement_timeout: Number(process.env.DATABASE_STATEMENT_TIMEOUT_MS ?? 15_000),
	});

	// 连接池错误不应导致进程崩溃（例如数据库重启期间的瞬时失败）
	pool.on("error", (error) => {
		console.error("[db] 空闲连接异常:", error.message);
	});

	return pool;
}

/** 判断是否需要启用 SSL：本地地址不启用，远端默认启用 */
function shouldUseSsl(url: string): boolean {
	try {
		const { hostname, searchParams } = new URL(url);
		const sslmode = searchParams.get("sslmode");

		// 显式声明优先
		if (sslmode === "disable") {
			return false;
		}
		if (sslmode === "require" || sslmode === "verify-full") {
			return true;
		}

		// 未显式声明时，按主机推断
		const isLocal =
			hostname === "localhost" ||
			hostname === "127.0.0.1" ||
			hostname === "::1" ||
			hostname.endsWith(".local");

		return !isLocal;
	} catch {
		// URL 解析失败时交给 pg 自己报错，这里默认不启用 SSL
		return false;
	}
}

const pool = globalForDb.__postgresPool ?? createPool();

// 生产环境不需要挂载（模块不会热重载），但挂载也无副作用
globalForDb.__postgresPool = pool;

export const db = drizzle(pool, { schema });

/** 底层连接池，供健康检查与脚本使用 */
export { pool };

export type DB = typeof db;

/** 检查数据库连通性，用于健康检查接口 */
export async function checkDatabaseHealth(): Promise<{
	ok: boolean;
	latencyMs: number;
	error?: string;
}> {
	const startedAt = Date.now();

	try {
		await pool.query("select 1");
		return { ok: true, latencyMs: Date.now() - startedAt };
	} catch (error) {
		return {
			ok: false,
			latencyMs: Date.now() - startedAt,
			error: error instanceof Error ? error.message : String(error),
		};
	}
}