filter.py

# -*- coding: utf-8 -*-

# Module author: @ftgmodulesbyfl1yd

from .. import loader, utils


@loader.tds
class FiltersMod(loader.Module):
    """Filters module"""

    strings = {"name": "Filters"}

    async def client_ready(self, client, db):
        self.db = db

    async def filtercmd(self, message):
        """Adds a filter into the list."""
        filters = self.db.get("Filters", "filters", {})
        key = utils.get_args_raw(message).lower()
        reply = await message.get_reply_message()
        chatid = str(message.chat_id)

        if not key and not reply:
            return await message.edit("<b>No args or reply.</b>")

        if chatid not in filters:
            filters.setdefault(chatid, {})

        if key in filters[chatid]:
            return await message.edit("<b>Such a filter already exists.</b>")

        if reply:
            if key:
                msgid = await self.db.store_asset(reply)
            else:
                return await message.edit(
                    "<b>You need arguments to save the filter!</b>"
                )
        else:
            try:
                msgid = (
                    await message.client.send_message(
                        f"friendly-{(await message.client.get_me()).id}-assets",
                        key.split("/")[1],
                    )
                ).id
                key = key.split("/")[0]
            except IndexError:
                return await message.edit(
                    "<b>Need a second argument (through / ) or a reply.</b>"
                )

        filters[chatid].setdefault(key, msgid)
        self.db.set("Filters", "filters", filters)
        await message.edit(f'<b>Filter "{key}" saved!</b>')

    async def stopcmd(self, message):
        """Removes a filter from the list."""
        filters = self.db.get("Filters", "filters", {})
        args = utils.get_args_raw(message)
        chatid = str(message.chat_id)

        if chatid not in filters:
            return await message.edit("<b>There are no filters in this chat.</b>")

        if not args:
            return await message.edit("<b>No args.</b>")

        if args:
            try:
                filters[chatid].pop(args)
                self.db.set("Filters", "filters", filters)
                await message.edit(f'<b>Filter "{args}" removed from chat list!</b>')
            except KeyError:
                return await message.edit(f'<b>No "{args}" filter.</b>')
        else:
            return await message.edit("<b>No args.</b>")

    async def stopallcmd(self, message):
        """Clears out the filter list."""
        filters = self.db.get("Filters", "filters", {})
        chatid = str(message.chat_id)

        if chatid not in filters:
            return await message.edit("<b>There are no filters in this chat.</b>")

        filters.pop(chatid)
        self.db.set("Filters", "filters", filters)
        await message.edit("<b>All filters have been removed from the chat list!</b>")

    async def filterscmd(self, message):
        """Shows saved filters."""
        filters = self.db.get("Filters", "filters", {})
        chatid = str(message.chat_id)

        if chatid not in filters:
            return await message.edit("<b>There are no filters in this chat.</b>")

        msg = ""
        for _ in filters[chatid]:
            msg += f"<b>• {_}</b>\n"
        await message.edit(
            f"<b>List of filters in this chat: {len(filters[chatid])}\n\n{msg}</b>"
        )

    async def watcher(self, message):
        try:
            filters = self.db.get("Filters", "filters", {})
            chatid = str(message.chat_id)
            m = message.text.lower()
            if chatid not in filters:
                return

            for _ in filters[chatid]:
                msg = await self.db.fetch_asset(filters[chatid][_])
                def_pref = self.db.get("friendly-telegram.main", "command_prefix")
                pref = "." if not def_pref else def_pref[0]

                if len(_.split()) == 1:
                    if _ in m.split():
                        await self.exec_comm(msg, message, pref)
                else:
                    if _ in m:
                        await self.exec_comm(msg, message, pref)
        except:
            pass

    async def exec_comm(self, msg, message, pref):
        try:
            if msg.text[0] == pref:
                smsg = msg.text.split()
                return await self.allmodules.commands[smsg[0][1:]](
                    await message.reply(
                        smsg[0] + " ".join(_ for _ in smsg if len(smsg) > 1)
                    )
                )
            else:
                pass
        except:
            pass
        await message.reply(msg)