import logging
from io import BytesIO

from convopyro import listen_message
from pykeyboard import ReplyButton, ReplyKeyboard
from pyrogram import filters
from pyrogram.types import Message, ReplyKeyboardRemove
from ujson import dumps

from classes.pyroclient import PyroClient

logger = logging.getLogger(__name__)


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

    user_data = BytesIO(
        dumps(await user.checkout(), escape_forward_slashes=False).encode()
    )

    await message.reply_document(
        user_data,
        file_name="user_data.json",
    )

    # Data deletion request
    keyboard_delete = ReplyKeyboard(
        row_width=1, resize_keyboard=True, one_time_keyboard=True
    )
    keyboard_delete.add(
        ReplyButton("Yes, I want to delete it"),
        ReplyButton("No, I don't want to delete it"),
    )

    await message.reply_text(
        "Here's pretty much all the data bot has. Please, use these buttons to choose whether you want to delete your data from the bot.",
        reply_markup=keyboard_delete,
    )

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

        if answer_delete is None or answer_delete.text == "/cancel":
            await message.reply_text(
                "Cancelled.",
                reply_markup=ReplyKeyboardRemove(),
            )
            return

        if answer_delete.text not in [
            "Yes, I want to delete it",
            "No, I don't want to delete it",
        ]:
            await answer_delete.reply_text(
                "Invalid answer provided. Use /cancel if you want to cancel this operation."
            )
            continue

        if answer_delete.text in [
            "No, I don't want to delete it",
        ]:
            await answer_delete.reply_text(
                "Alright, cancelled.", reply_markup=ReplyKeyboardRemove()
            )
            return

        break

    # Confirmation
    keyboard_confirm = ReplyKeyboard(
        row_width=1, resize_keyboard=True, one_time_keyboard=True
    )
    keyboard_confirm.add(ReplyButton("I agree and want to proceed"))

    await message.reply_text(
        "Alright. Please, confirm that you want to delete your data from the bot.\n\nFollowing data will be deleted:\nSelected location, preferred language of the messages, notifications time and your notifications offset.",
        reply_markup=keyboard_confirm,
    )

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

        if answer_confirm is None or answer_confirm.text == "/cancel":
            await message.reply_text(
                "Cancelled.",
                reply_markup=ReplyKeyboardRemove(),
            )
            return

        if answer_confirm.text not in ["I agree and want to proceed"]:
            await answer_confirm.reply_text(
                "Invalid answer provided. Use /cancel if you want to cancel this operation."
            )
            continue

        break

    await user.delete()
    await answer_confirm.reply_text(
        "Your data has been deleted. If you want to start using this bot again, please use /setup command. Otherwise delete/block the bot and do not interact with it anymore.",
        reply_markup=ReplyKeyboardRemove(),
    )