import json
from datetime import timedelta
from typing import Any

import asyncpg

JOBS_CHANGED_CHANNEL = "jobs_changed"


async def enqueue(
    pool: asyncpg.Pool,
    account_id: int,
    kind: str,
    params: dict[str, Any],
    *,
    cooldown: timedelta | None = None,
) -> int:
    params_json = json.dumps(params)
    existing = await pool.fetchval(
        "SELECT id FROM jobs WHERE account_id = $1 AND kind = $2 "
        "AND params = $3::jsonb AND (status IN ('pending', 'running') "
        "OR created_at > now() - $4::interval) LIMIT 1",
        account_id,
        kind,
        params_json,
        cooldown,
    )
    if existing is not None:
        return existing
    job_id = await pool.fetchval(
        "INSERT INTO jobs (account_id, kind, params) "
        "VALUES ($1, $2, $3::jsonb) RETURNING id",
        account_id,
        kind,
        params_json,
    )
    await pool.execute(f"NOTIFY {JOBS_CHANGED_CHANNEL}")
    return job_id