from app import app
from pyrogram import filters
from pyrogram.types import Message, ReplyKeyboardMarkup
from pyrogram.client import Client
from classes.errors.holo_user import UserNotFoundError, UserInvalidError
from classes.holo_user import HoloUser
from modules.logging import logWrite
from modules.utils import locale
from modules.database import col_spoilers
from modules import custom_filters

# Spoiler command ==============================================================================================================
@app.on_message(custom_filters.enabled_spoilers & ~filters.scheduled & filters.private & filters.command(["spoiler"], prefixes=["/"]))
async def cmd_spoiler(app: Client, msg: Message):

    try:
        holo_user = HoloUser(msg.from_user)
    except (UserInvalidError, UserNotFoundError):
        return

    if holo_user.application_state()[0] != "approved":

        await msg.reply_text(locale("not_member", "message", locale=msg.from_user))
        return

    if holo_user.application_state()[0] != "fill" and holo_user.sponsorship_state()[0] != "fill":

        if col_spoilers.find_one( {"user": msg.from_user.id, "completed": False} ) is None:

            col_spoilers.insert_one(
                {
                    "user": msg.from_user.id,
                    "completed": False,
                    "category": None,
                    "description": None,
                    "photo": None,
                    "video": None,
                    "audio": None,
                    "animation": None,
                    "document": None,
                    "caption": None,
                    "text": None
                }
            )

            await msg.reply_text(locale("spoiler_started", "message", locale=msg.from_user), reply_markup=ReplyKeyboardMarkup(locale("spoiler_categories", "keyboard"), resize_keyboard=True, one_time_keyboard=True))
            logWrite(f"User {msg.from_user.id} started creating new spoiler")

        else:
            await msg.reply_text(locale("spoiler_unfinished", "message", locale=msg.from_user))
# ==============================================================================================================================