from hikkatl.types import Message
from .. import loader, utils

"""
    ███    ███ ██    ██ ██████  ██    ██ ██       ██████  ███████ ███████
    ████  ████ ██    ██ ██   ██ ██    ██ ██      ██    ██ ██      ██  
    ██ ████ ██ ██    ██ ██████  ██    ██ ██      ██    ██ ███████ █████
    ██  ██  ██ ██    ██ ██   ██ ██    ██ ██      ██    ██      ██ ██  
    ██      ██  ██████  ██   ██  ██████  ███████  ██████  ███████ ███████ 


    Filters
    📜 Licensed under the GNU AGPLv3	
"""

# meta desc: desc
# meta developer: @BruhHikkaModules


@loader.tds
class TextFilters(loader.Module):
    """- Module for filter text"""

    async def client_ready(self, db, client):
        self.db = db
        self.set("filter", "Dont care")

    strings = {
        "name": "TextFilters",
        "wrong": "Wrong Argument. \n\nupper - Capitalize\ncapitalize - Starts with a capital letter and the rest are small.\nlower - Reduces all letters\noff - Disable filters",
        "correct": "Filter changed to {}",
    }
    strings_ru = {
        "wrong": "Неверный аргумент. \n\nupper - Большие буквы\ncapitalize - Начинает с большой буквы, а остальные маленькие\nlower - Уменьшает все буквы\noff - Выключить фильтры",
        "correct": "Фильтр сменён на {}",
    }

    @loader.watcher()
    async def watcher(self, message):
        me = await self.client.get_me(id)
        try:
            if message.from_id == me.user_id and message.text:
                filter_txt = self.get("filter", "Dont care")

                if filter_txt == "lower":
                    try:
                        await message.edit(message.text.lower())
                    except Exception:
                        pass

                elif filter_txt == "upper":
                    try:
                        await message.edit(message.text.upper())
                    except Exception:
                        pass

                elif filter_txt == "capitalize":
                    try:
                        await message.edit(message.text.capitalize())
                    except Exception:
                        pass
                elif filter_txt == "off":
                    pass  # Ну а чо делать

        except AttributeError:
            pass  # Ну это херня с ивентами, поэтому да

    @loader.command(ru_doc=" - [lower / capitalize / upper / off] - Выбрать фильтр")
    async def filter_ch(self, message: Message):
        """- [lower / capitalize / upper / off] - Chose filter"""
        args = utils.get_args_raw(message)
        if args.lower() not in ["lower", "capitalize", "upper", "off"]:
            await utils.answer(message, self.strings("wrong"))
        else:
            await utils.answer(message, self.strings("correct").format(args))
            self.set("filter", args.lower())