import time
from datetime import datetime as dt, timedelta
from pyrogram import Client as ren, filters
from pyrogram import *
from pyrogram.types import *
from akn.utils.database import db
from akn.utils.logger import LOGS
from config import *

@ren.on_callback_query(filters.regex(r"^botstatsdl_(\w+)$"))
async def handle_bot_statsdl(client: Client, callback: CallbackQuery):
    try:
        bots_uuid = callback.matches[0].group(1)
        user_id = callback.from_user.id

        bot_data = await db.alldl_bot.find_one(
            {
                "user_id": user_id,
                "bots.uuid": bots_uuid
            },
            {"bots.$": 1}
        )

        if not bot_data or not bot_data.get("bots"):
            return await callback.answer("❌ Bot not found!", show_alert=True)

        bot = bot_data["bots"][0]
        stats = bot.get("stats", {})
        start = time.time()
        await client.get_me()
        end = time.time()
        latency = (end - start) * 1000
        start_time = bot.get("start_time")
        if start_time:
            uptime = str(timedelta(seconds=int(time.time() - start_time)))
        else:
            uptime = "N/A"

        bandwidth_used = stats.get("bandwidth_used", 0)
        if bandwidth_used > 1024:
            bandwidth = f"{bandwidth_used / 1024:.2f} GB"
        else:
            bandwidth = f"Unlimited"

        last_active = bot.get("last_active")
        if isinstance(last_active, dt):
            last_active_str = last_active.strftime('%Y-%m-%d %H:%M:%S')
        elif isinstance(last_active, str):
            last_active_str = last_active
        else:
            last_active_str = "Never"

        stats_message = (
            f"📊 <b>Bot Statistics</b> 📊\n\n"
            f"🆔 <code>{bots_uuid}</code>\n"
            f"🔗 {bot.get('bot_username', 'N/A')}\n\n"
            f"📥 <b>Downloads:</b>\n"
            f"├ Total: `{stats.get('total_downloads', 999999999)}`\n"
            f"├ Successful: `{stats.get('successful_downloads', 999999999)}`\n"
            f"└ Failed: `{stats.get('failed_downloads', 0)}`\n\n"
            f"🌐 <b>Bandwidth:</b> {bandwidth}\n"
            f"⏱ <b>Uptime:</b> {uptime}\n"
            f"✨ <b>Pong!:</b> {latency:.2f}\n"
            f"🔄 <b>Restarts:</b> {bot.get('restart_count', 0)}\n"
            f"📅 <b>Last Active:</b> {last_active_str}"
        )

        buttons = InlineKeyboardMarkup([
            [InlineKeyboardButton("🔄 Refresh Stats", callback_data=f"botstatsdl_{bots_uuid}")],
            [InlineKeyboardButton("📈 Daily Report", callback_data=f"reportdl_{bots_uuid}")],
            [InlineKeyboardButton("🔙 Back to Bot", callback_data=f"my_bots")]
        ])

        await callback.message.edit_text(stats_message, reply_markup=buttons)
        await callback.answer()

    except Exception as e:
        LOGS.error(f"Stats error: {str(e)}")
        await callback.answer("⚠️ Failed to load stats!", show_alert=True)

async def update_bot_stats(bot_uuid: str, download_success: bool, file_size: int):
    update_data = {
        "$inc": {
            "bots.$.stats.total_downloads": 1,
            "bots.$.stats.bandwidth_used": file_size,
            "bots.$.stats.uptime": int(time.time() - time.time())
        },
        "$set": {
            "bots.$.stats.last_activity": dt.now()
        }
    }
    if download_success:
        update_data["$inc"]["bots.$.stats.successful_downloads"] = 1
    else:
        update_data["$inc"]["bots.$.stats.failed_downloads"] = 1

    await db.alldl_bot.update_one(
        {"bots.uuid": bot_uuid},
        update_data
    )