import logging
from datetime import datetime

from convopyro import listen_message
from pyrogram import filters
from pyrogram.types import ForceReply, Message, ReplyKeyboardRemove

from classes.pyroclient import PyroClient

logger = logging.getLogger(__name__)


@PyroClient.on_message(
    ~filters.scheduled & filters.private & filters.command(["set_time"], prefixes=["/"])  # type: ignore
)
async def command_set_time(app: PyroClient, message: Message):
    user = await app.find_user(message.from_user)

    await message.reply_text(
        "Alright. Please, send your desired time in HH:MM format.",
        reply_markup=ForceReply(placeholder="Time as HH:MM"),
    )

    while True:
        answer = await listen_message(app, message.chat.id, 300)

        if answer is None or answer.text == "/cancel":
            await message.reply_text(
                app._("cancelled", "messages", locale=user.locale),
                reply_markup=ReplyKeyboardRemove(),
            )
            return

        try:
            datetime.strptime(answer.text, "%H:%M")
        except ValueError:
            await answer.reply_text(
                "Please, provide a valid time in HH:MM format. Use /cancel if you want to cancel this operation."
            )
            continue

        break

    user_time = datetime.strptime(answer.text, "%H:%M")

    await user.update_time(hour=user_time.hour, minute=user_time.minute)

    logger.info(
        "User %s has selected notification time of %s",
        user.id,
        user_time.strftime("%H:%M"),
    )

    notice = "" if user.enabled else "Execute /toggle to enable notifications."

    garbage_time = user_time.strftime(app._("time", "formats"))

    await answer.reply_text(
        f"Notifications time has been updated! You will now receive notification about collection {user.offset} d. before the collection at **{garbage_time}**. {notice}",
        reply_markup=ReplyKeyboardRemove(),
    )