import httpx
from config import BALE_TOKEN, BALE_API_BASE

BASE = f"{BALE_API_BASE}{BALE_TOKEN}"


async def _req(method: str, **kwargs):
    url = f"{BASE}/{method}"
    async with httpx.AsyncClient(timeout=60) as client:
        r = await client.post(url, json=kwargs)
        return r.json()


async def get_me():
    return await _req("getMe")


async def send_message(chat_id, text, reply_markup=None, parse_mode=None):
    kwargs = {"chat_id": chat_id, "text": text}
    if reply_markup:
        kwargs["reply_markup"] = reply_markup
    if parse_mode:
        kwargs["parse_mode"] = parse_mode
    return await _req("sendMessage", **kwargs)


async def send_photo(chat_id, photo, caption=None, reply_markup=None):
    kwargs = {"chat_id": chat_id, "photo": photo}
    if caption:
        kwargs["caption"] = caption
    if reply_markup:
        kwargs["reply_markup"] = reply_markup
    return await _req("sendPhoto", **kwargs)


async def send_document(chat_id, document, caption=None, reply_markup=None, filename="file"):
    url = f"{BASE}/sendDocument"
    files = {"document": (filename, document)}
    data = {"chat_id": str(chat_id)}
    if caption:
        data["caption"] = caption
    if reply_markup:
        data["reply_markup"] = reply_markup
    async with httpx.AsyncClient(timeout=300) as client:
        r = await client.post(url, data=data, files=files)
        return r.json()


async def send_document_by_url(chat_id, file_url, caption=None):
    kwargs = {"chat_id": chat_id, "document": file_url}
    if caption:
        kwargs["caption"] = caption
    return await _req("sendDocument", **kwargs)


async def send_chat_action(chat_id, action):
    return await _req("sendChatAction", chat_id=chat_id, action=action)


async def answer_callback_query(callback_query_id, text=None, show_alert=False):
    kwargs = {"callback_query_id": callback_query_id}
    if text:
        kwargs["text"] = text
    kwargs["show_alert"] = show_alert
    return await _req("answerCallbackQuery", **kwargs)


async def edit_message_text(chat_id, message_id, text, reply_markup=None):
    kwargs = {"chat_id": chat_id, "message_id": message_id, "text": text}
    if reply_markup:
        kwargs["reply_markup"] = reply_markup
    return await _req("editMessageText", **kwargs)


async def get_file(file_id):
    return await _req("getFile", file_id=file_id)


async def get_file_link(file_path: str):
    return f"https://tapi.bale.ai/file/bot{BALE_TOKEN}/{file_path}"


def make_keyboard(buttons):
    return {"inline_keyboard": [[{"text": b[0], "callback_data": b[1]}] for b in buttons]}


def make_reply_keyboard(buttons, resize=True):
    return {
        "keyboard": [[{"text": b}] for b in buttons],
        "resize_keyboard": resize,
    }


def remove_keyboard():
    return {"remove_keyboard": True}


MENU_BUTTONS = make_reply_keyboard([
    "🎬 YouTube Downloader",
    "📁 File Downloader",
    "🤖 ChatGPT",
    "🌐 Open Panel",
    "❓ Help",
])

YOUTUBE_QUALITY = make_keyboard([
    ("🎵 Audio (MP3)", "yt:audio"),
    ("🎬 Video (720p)", "yt:720p"),
    ("🎬 Video (1080p)", "yt:1080p"),
    ("🔙 Back to Menu", "menu:main"),
])

BACK_TO_MENU = make_keyboard([
    ("🔙 Back to Menu", "menu:main"),
])