import json
import logging
import uuid

import httpx

from config import RPC_SECRET, ARIA2_RPC_URL

logger = logging.getLogger(__name__)


async def send_request(method, params=None):
    request_id = uuid.uuid4().hex
    payload = {
        "jsonrpc": "2.0",
        "id": request_id,
        "method": method,
        "params": [f"token:{RPC_SECRET}"] + (params or []),
    }

    # specify the internal API call don't use proxy
    async with httpx.AsyncClient(
        mounts={"all://": httpx.AsyncHTTPTransport()}
    ) as client:
        response = await client.post(ARIA2_RPC_URL, json=payload)
    try:
        return response.json()
    except json.JSONDecodeError as e:
        logger.warning(
            f"aria2 request failed, response: {response.status_code} {response.text}"
        )
        raise e


async def add_download(url, save_dir="/app/cache"):
    method = "aria2.addUri"
    params = [[url], {"dir": save_dir, "header": []}]
    response = await send_request(method, params)
    return response["result"]


async def pause_download(gid):
    method = "aria2.pause"
    params = [gid]
    response = await send_request(method, params)
    return response["result"]


async def resume_download(gid):
    method = "aria2.unpause"
    params = [gid]
    response = await send_request(method, params)
    return response["result"]


async def get_status(gid):
    method = "aria2.tellStatus"
    params = [gid]
    response = await send_request(method, params)
    return response["result"]


async def list_downloads():
    method = "aria2.tellActive"
    response = await send_request(method)
    return response["result"]