import asyncio
from box import Box
from datetime import datetime as dt
from datetime import timedelta
from pytz import timezone
from pyrogram.types import *
from akn.utils.database import db
from akn.utils.logger import LOGS

WHY_TRY_DO = """
Hey, {user}

Sorry, your bot {bot} has expired. 

You can upgrade to premium for an extended period:
- 1 month or 2 months of premium
- But don’t worry, you still get a free 7-day trial every week!

If you're interested, feel free to purchase premium at any time!
"""

reply_markup = InlineKeyboardMarkup(
    [
        [
            InlineKeyboardButton(
                text="🆕 New expiry",
                callback_data="expiredtt"
            )
        ]
    ]
)

async def add_time_watch(
    user_id,
    first_name,
    username,
    bot_token,
    client_name,
    days,
    disconnected: bool = False
):
    try:
        now = dt.now(timezone("Asia/Jakarta"))
        expire_date = now + timedelta(days=days)
        LOGS.info(f"Setting expiration time for user {user_id}: {expire_date}")
        await db.set_expired_date(
            user_id,
            first_name,
            username,
            expire_date,
            bot_token,
            client_name,
            disconnected
        )
    except Exception as e:
        LOGS.info(f"Error: {e}")

async def watch_do_time(user_id, client, assistant):
    while True:
        now = dt.now(timezone("Asia/Jakarta"))
        time = now.strftime("%d-%m-%Y")
        
        exp = await db.get_expired_date(user_id)
        exp_obj = Box(exp or {})
        if not exp_obj:
            await asyncio.sleep(30)
            continue
        expired_on = exp_obj.expire_date
        if not expired_on:
            await asyncio.sleep(30)
            continue
        if isinstance(expired_on, str):
            expired_on = dt.strptime(expired_on, "%d-%m-%Y")

        if expired_on.tzinfo is None:
            expired_on = timezone("Asia/Jakarta").localize(expired_on)

        LOGS.info(f"Expiration date for user {user_id}: {expired_on.strftime('%d-%m-%Y')}")
        
        if now >= expired_on:
            send_mention = f"<a href='tg://user?id={exp_obj.user_id}'>{exp_obj.first_name}</a>"
            await client.stop()
            await assistant.send_message(
                exp_obj.user_id,
                WHY_TRY_DO.format(
                    user=send_mention,
                    bot=exp_obj.username
                )
            )
            await db.set_expired_date(
                user_id=exp_obj.user_id,
                first_name=exp_obj.first_name,
                username=exp_obj.username,
                expire_date=None,
                bot_token=exp_obj.user_client.bot_token,
                client_name=exp_obj.user_client.client_name,
                disconnected=True
            )
            LOGS.info(f"bot {exp_obj.username} has expired")
            break
        else:
            await asyncio.sleep(30)