import postgres from "postgres";

const sql = postgres(process.env.DATABASE_URL ?? "postgres://pingql:pingql@localhost:5432/pingql", {
  max: 20,
  idle_timeout: 30,
  connect_timeout: 10,
});

export default sql;

export async function migrate() {
  await sql`CREATE EXTENSION IF NOT EXISTS pgcrypto`;
  await sql`
    CREATE TABLE IF NOT EXISTS accounts (
      id          UUID PRIMARY KEY DEFAULT gen_random_uuid(),
      key         TEXT NOT NULL UNIQUE,
      email_hash  TEXT,
      created_at  TIMESTAMPTZ DEFAULT now()
    )
  `;

  await sql`
    CREATE TABLE IF NOT EXISTS monitors (
      id               TEXT PRIMARY KEY DEFAULT encode(gen_random_bytes(8), 'hex'),
      account_id       UUID NOT NULL REFERENCES accounts(id) ON DELETE CASCADE,
      name             TEXT NOT NULL,
      url              TEXT NOT NULL,
      method           TEXT NOT NULL DEFAULT 'GET',
      request_headers  JSONB,
      request_body     TEXT,
      timeout_ms       INTEGER NOT NULL DEFAULT 30000,
      interval_s       INTEGER NOT NULL DEFAULT 60,
      query            JSONB,
      enabled          BOOLEAN NOT NULL DEFAULT true,
      created_at       TIMESTAMPTZ DEFAULT now()
    )
  `;

  await sql`
    CREATE TABLE IF NOT EXISTS pings (
      id           BIGSERIAL PRIMARY KEY,
      monitor_id   TEXT NOT NULL REFERENCES monitors(id) ON DELETE CASCADE,
      checked_at   TIMESTAMPTZ NOT NULL DEFAULT now(),
      scheduled_at TIMESTAMPTZ,
      jitter_ms    INTEGER,
      status_code  INTEGER,
      latency_ms   INTEGER,
      up           BOOLEAN NOT NULL,
      error        TEXT,
      meta         JSONB
    )
  `;

  // Migrations for existing deployments
  await sql`ALTER TABLE pings ADD COLUMN IF NOT EXISTS scheduled_at TIMESTAMPTZ`;
  await sql`ALTER TABLE pings ADD COLUMN IF NOT EXISTS jitter_ms INTEGER`;
  await sql`ALTER TABLE monitors ADD COLUMN IF NOT EXISTS regions TEXT[] NOT NULL DEFAULT '{}'`;
  await sql`ALTER TABLE pings ADD COLUMN IF NOT EXISTS region TEXT`;
  await sql`ALTER TABLE pings ADD COLUMN IF NOT EXISTS run_id TEXT`;

  await sql`ALTER TABLE accounts ADD COLUMN IF NOT EXISTS plan TEXT NOT NULL DEFAULT 'free'`;
  await sql`ALTER TABLE accounts ADD COLUMN IF NOT EXISTS plan_expires_at TIMESTAMPTZ`;

  await sql`CREATE INDEX IF NOT EXISTS idx_pings_monitor ON pings(monitor_id, checked_at DESC)`;
  await sql`CREATE INDEX IF NOT EXISTS idx_pings_checked_at ON pings(checked_at)`;

  await sql`
    CREATE TABLE IF NOT EXISTS api_keys (
      id           UUID PRIMARY KEY DEFAULT gen_random_uuid(),
      key          TEXT NOT NULL UNIQUE,
      account_id   UUID NOT NULL REFERENCES accounts(id) ON DELETE CASCADE,
      label        TEXT NOT NULL,
      created_at   TIMESTAMPTZ DEFAULT now(),
      last_used_at TIMESTAMPTZ
    )
  `;

  console.log("DB ready");
}