limoka/ZetGoHack/nullmod/HaremManager.py

__version__ = (1,1,7)
#░░░███░███░███░███░███
#░░░░░█░█░░░░█░░█░░░█░█
#░░░░█░░███░░█░░█░█░█░█
#░░░█░░░█░░░░█░░█░█░█░█
#░░░███░███░░█░░███░███
# H:Mods Team [💎]
# meta developer: @nullmod

# -      main      - #
from .. import loader, utils
# -      func      - #
import asyncio
import logging
import time
import re
# -    func(tl)    - #
from telethon.tl.functions.chatlists import CheckChatlistInviteRequest, JoinChatlistInviteRequest, LeaveChatlistRequest
from telethon.tl.functions.messages import ImportChatInviteRequest, CheckChatInviteRequest
from telethon.tl.functions.channels import JoinChannelRequest, LeaveChannelRequest
from telethon.tl.functions.contacts import BlockRequest, UnblockRequest
# -      types     - #
from telethon.tl.types import InputChatlistDialogFilter, UpdateDialogFilter
# -      errors    - #
from telethon.errors import YouBlockedUserError, InviteRequestSentError
# -      end       - #

logger = logging.getLogger(__name__)

@loader.tds
class HaremManager(loader.Module):
    """Module for harem bots: Gif Harem, Waifu Harem, Horny Harem"""

    strings = {
        "name": "HaremManager"
    }

    def __init__(self):
        self.config = loader.ModuleConfig(
            loader.ConfigValue(
                "ignore-chats",
                [],
                "Список чатов, где модуль НЕ будет ловить вайфу. Указывайте ID чатов в виде 123456789",
                validator=loader.validators.Series(
                    validator=loader.validators.Integer(),
                )
            ),
            loader.ConfigValue(
                "whitelist-chats",
                [],
                "Список чатов, где модуль будет ловить вайфу. Указывайте ID чатов в виде 123456789. Если что-то указано, то модуль будет ловить вайфу только в этих чатах",
                validator=loader.validators.Series(
                    validator=loader.validators.Integer(),
                )
            ),
            loader.ConfigValue(
                "interval-horny",
                4,
                "Интервал между автобонусом",
                validator=loader.validators.Float(2.0)
            ),
            loader.ConfigValue(
                "interval-waifu",
                4,
                "Интервал между автобонусом",
                validator=loader.validators.Float(2.0)
            ),
            loader.ConfigValue(
                "interval-gif",
                4,
                "Интервал между автобонусом",
                validator=loader.validators.Float(2.0)
            ),
        )

    async def client_ready(self):
        self.harems = {
            "horny": "@Horny_GaremBot",
            "waifu": "@garem_chatbot",
            "gif": "@GIFgarem_bot",
        }
        self.harems_ids = {
            "horny": 7896566560,
            "waifu": 6704842953,
            "gif": 7084965046,
        }

        temp_values = [
            "config",
            "ab-horny",
            "catch-horny",
            "out-horny",
            "ab-waifu",
            "catch-waifu",
            "out-waifu",
            "ab-gif",
            "catch-gif",
            "out-gif"
        ]
        if not self.get("config", None):
            for value in temp_values:
                self.set(value, False if value not in "config" else True)

    @loader.loop(interval=1, autostart=True)
    async def loop(self):
        for bot in self.harems:
            if self.get(f"ab-{bot}", None):
                if (not self.get(f"ab-t-{bot}") or (time.time() - self.get(f"ab-t-{bot}")) >= int(3600*self.config[f"interval-{bot}"])):
                    await self._autobonus(self.harems[bot], bot)

    @loader.watcher("only_messages")
    async def watcher(self, message):
        """Watcher"""
        chatid = int(str(message.chat_id).replace("-100", ""))
        for bot in self.harems:
            if bot == "waifu": continue
            if message.sender_id == self.harems_ids[bot] and self.get(f"catch-{bot}", None):
                if self.config["whitelist-chats"]:
                    if chatid not in self.config["whitelist-chats"]:
                        return
                elif chatid in self.config["ignore-chats"]:
                    return
                if (not self.get(f"catcher_time-{bot}") or int(time.time()) - int(self.get(f"catcher_time-{bot}")) > 14400):
                    if "заблудилась" in message.text.lower():
                        try:
                            await message.click()
                            await asyncio.sleep(5)
                            msgs = await message.client.get_messages(chatid, limit=10)
                            for msg in msgs:
                                if msg.mentioned and "забрали" in msg.text and msg.sender_id == self.harems_ids[bot]:
                                    if self.get(f"out-{bot}", None):
                                        match = re.search(r", Вы забрали (.+?)\. Вайфу", msg.text)
                                        waifu = match.group(1)
                                        caption = f"{waifu} в вашем гареме! <emoji document_id=5395592707580127159>😎</emoji>"
                                        await self.client.send_file(self.harems[bot], caption=caption, file=message.media)
                                    self.set(f"catcher_time-{bot}", int(time.time()))
                        except Exception as e:
                            logger.error(f"Ошибка при ловле вайфу для {bot}(не критично): {e}")


    def _main_markup(self):
        return [
                [
                    {
                        "text": "[✔️] Horny Harem" if self.get("ab-horny") else "[❌] Horny Harem",
                        "callback": self.callback_handler,
                        "args": ("horny",)
                    },
                    {
                        "text": "[✔️] Waifu Harem" if self.get("ab-waifu") else "[❌] Waifu Harem",
                        "callback": self.callback_handler,
                        "args": ("waifu",)
                    },
                   ],
                   [
                    {
                        "text": "[✔️] Gif Harem" if self.get("ab-gif") else "[❌] Gif Harem",
                        "callback": self.callback_handler,
                        "args": ("gif",)
                    }
                   ],
                   [
                    {
                        "text": "🔻 Закрыть",
                        "action": "close",
                    }
                   ]
               ]

    def _menu_markup(self, bot):
        markup = [[],[]]
        markup[0].append({
                            "text": "[✔️] Автобонус" if self.get(f"ab-{bot}", None) else "[❌] Автобонус", 
                            "callback": self.callback_handler,
                            "args": (f"ab-{bot}",)
                        })
        if "waifu" not in bot:
            markup[0].append({
                                "text": "[✔️] Автоловля" if self.get(f"catch-{bot}", None) else "[❌] Автоловля",
                                "callback": self.callback_handler,
                                "args": (f"catch-{bot}",)
                            })
            markup[1].append({
                                "text": "[✔️] Вывод от ловца" if self.get(f"out-{bot}", None) else "[❌] Вывод от ловца",
                                "callback": self.callback_handler,
                                "args": (f"out-{bot}",)
                            })
        markup.append([
                    {
                        "text":"🔁 Перезапустить автобонус",
                        "callback": self.callback_handler,
                        "args": (f"restart-{bot}",)
                    },
                ])
        markup.append([
                    {
                        "text":"↩️ Назад", 
                        "callback":self.callback_handler,
                        "args": ("back",)
                    }
                ])
        return markup

    async def _set_menu(self, message):
        await utils.answer(
            message,
            f"❤️ Выберите бота для управления\n\n✅ <i>- означает, что автобонус включён</i>\
\n\nБольше настроек в конфиге модуля(<code>{self.get_prefix()}config HaremManager</code>)",
            reply_markup=self._main_markup()
        )

    async def callback_handler(self, call, data):
        if data == "back":
            await self._set_menu(call)
            return
        elif data.startswith("restart-"):
            bot = data.split("-")[-1]
            await call.answer(f"Перезапуск бонуса для {self.harems[bot]}...")
            await self._autobonus(self.harems[bot], bot)
            return
        elif data.startswith("ab-"):
            bot = data.split("-")[-1]
            self.set(data, not self.get(data, None))
            await utils.answer(call, f"Меню <code>{self.harems[bot]}</code>", reply_markup=self._menu_markup(bot))
        elif data.startswith("catch-"):
            bot = data.split("-")[-1]
            self.set(data, not self.get(data, None))
            await utils.answer(call, f"Меню <code>{self.harems[bot]}</code>", reply_markup=self._menu_markup(bot))
        elif data.startswith("out-"):
            bot = data.split("-")[-1]
            self.set(data, not self.get(data, None))
            await utils.answer(call, f"Меню <code>{self.harems[bot]}</code>", reply_markup=self._menu_markup(bot))
        else:
            bot = data
            await utils.answer(call, f"Меню <code>{self.harems[bot]}</code>", reply_markup=self._menu_markup(bot))

    async def _autobonus(self, id, bot):
        wait_boost = False
        async with self._client.conversation(id) as conv:
            try:
                await conv.send_message("/bonus")
            except YouBlockedUserError:
                await self.client(UnblockRequest(id))
                await conv.send_message("/bonus")
            r = None
            try:
                r = await conv.get_response(timeout=5*60)
            except:
                tryings = 5
                while tryings > 0:
                    tryings -= 1
                    try:
                        await conv.send_message("/bonus")
                        r = await conv.get_response(5*60)
                        break
                    except:
                        pass
                if r is None:
                    logger.warning("Ответ от бота не получен. Вероятно, он снова лёг\n\nПерезапустите автобонус, когда бот очнётся")
                    self.set(f"ab-{bot}", False)
                    return
            self.set(f"ab-t-{bot}", int(time.time()))
            if "Доступен бонус за подписки" in r.text:
                await conv.send_message("/start ad_bonus")
                r = await conv.get_response()
                if "проверка пройдена" not in r.text:
                    to_leave, to_block, folders, chats_in_folders = [], [], [], []
                    wait_boost = False
                    if r.reply_markup:
                        a = r.buttons
                        for i in a:
                            for button in i: # каждая кнопка...
                                if button.url:
                                    alr = False # "уже зашёл"
                                    if "addlist/" in button.url: # добавление папок
                                        slug = button.url.split("addlist/")[-1]
                                        peers = await self.client(CheckChatlistInviteRequest(slug=slug))
                                        if peers:
                                            peers = peers.peers
                                            try:
                                                a = await self.client(JoinChatlistInviteRequest(slug=slug, peers=peers))
                                                chats_in_folders.append(peers) # для выхода
                                                for update in a.updates:
                                                    if isinstance(update, UpdateDialogFilter):
                                                        folders.append(InputChatlistDialogFilter(filter_id=update.id)) # для удаления папки
                                            except: pass
                                        continue
                                    if "t.me/boost" in button.url: # бустить не обязательно
                                        wait_boost = True
                                        continue
                                    if not bool(re.match(r"^https?:\/\/t\.me\/[^\/]+\/?$", button.url)): # дополнительные вложения отметаем
                                        continue
                                    if "t.me/+" in button.url: # приватные чаты
                                        try:
                                            a = await self.client(CheckChatInviteRequest(button.url.split("+")[-1]))
                                            if not hasattr(a, "request_needed") or not a.request_needed: # получить айди приватного чата/канала с приглашениями без входа невозможно
                                                pass
                                            else:
                                                url = button.url.split("?")[0] if "?" in button.url else button.url
                                                try:
                                                    await self.client(ImportChatInviteRequest(button.url.split("+")[-1]))
                                                except InviteRequestSentError: pass
                                                await asyncio.sleep(3)
                                                try:
                                                    entity = await self.client.get_entity(url)
                                                except ValueError:
                                                    try:
                                                        await asyncio.sleep(15)
                                                        entity = await self.client.get_entity(url)
                                                    except: 
                                                        continue
                                                except:
                                                    pass
                                                alr = True
                                        except: continue
                                    url = button.url.split("?")[0] if "?" in button.url else button.url
                                    if not alr:
                                        try:
                                            entity = await self.client.get_entity(url)
                                        except:
                                            entity = (await self.client(ImportChatInviteRequest(button.url.split("+")[-1]))).chats[0] #gotten class Updates
                                            alr = True
                                    if hasattr(entity, "broadcast"):
                                        if not alr:
                                            await self.client(JoinChannelRequest(button.url))
                                            to_leave.append(entity.id)
                                        else:
                                            to_leave.append(entity.chat.id) if hasattr(entity,"chat") else to_leave.append(entity.id) if hasattr(entity,"id") else None
                                    elif hasattr(entity, "bot"):
                                        username = entity.username if entity.username is not None else entity.usernames[0].username
                                        try:
                                            await self.client(UnblockRequest(username))
                                        except: print("блин")
                                        await self.client.send_message(entity, "/start")
                                        to_block.append(username)
                        flyer_messages = await self.client.get_messages(id, limit=1)
                        if wait_boost:
                            await asyncio.sleep(150)
                        for m in flyer_messages:
                            await asyncio.sleep(5)
                            await m.click(-1)
                            await asyncio.sleep(5)
                        for folder, chats in zip(folders, chats_in_folders):
                            await self.client(LeaveChatlistRequest(peers=chats, chatlist=folder))
                        for bot in to_block:
                            await self.client(BlockRequest(bot))
                            await self.client.delete_dialog(bot)
                        for channel in to_leave:
                            try:
                                await self.client(LeaveChannelRequest(channel))
                            except Exception as e:
                                pass
                count = 0
                if not self.get(f"last_lout-{bot}") or int(time.time()) - self.get(f"last_lout-{bot}") > 43200:
                    while count <= 3: # на всякий случай 4 попытки. Бот может забагаться и не выдать завершающий ответ
                        await conv.send_message("/lout")
                        r = await conv.get_response()
                        if r.reply_markup:
                            pattern = self._parse(r)
                            clicks = self._solution(pattern)
                            for i in range(len(clicks)):
                                if clicks[i] == 1:
                                    await r.click(i)
                            self.set(f"last_lout-{bot}", int(time.time()))
                            count += 1
                        else:
                            break

    def _parse(self, r):
        a = r.buttons
        pattern = []
        for i in a:
            for m in i:
                t = m.text
                if t == "🌚":
                    pattern.append(0)
                elif t == "🌞":
                    pattern.append(1)
                else:
                    pass
        return pattern

    def _solution(self, pole):
        n = len(pole)
        for num in range(2**n):
            binary_string = bin(num)[2:].zfill(n)
            presses = [int(char) for char in binary_string]
            temp = pole[:]

            for i in range(n):
                if presses[i]:
                    temp[i] ^= 1
                    if i % 3 > 0: temp[i - 1] ^= 1
                    if i % 3 < 2: temp[i + 1] ^= 1
                    if i >= 3: temp[i - 3] ^= 1
                    if i < 6: temp[i + 3] ^= 1

            if sum(temp) == 0:
                return presses

        return None

    @loader.command()
    async def Harems(self, message):
        """Открыть меню управления"""
        await self._set_menu(message)

    @loader.command()
    async def lightsout(self, message):
        """[ответ на соо с полем] Автоматически решает Lights Out"""
        if message.is_reply:
            r = await message.get_reply_message()
            if r.reply_markup:
                pattern = self._parse(r)
            else:
                await utils.answer(message, "<emoji document_id=5299030091735525430>❗️</emoji> Не вижу поля игры. Это точно то сообщение?")
                return

        else:
            
            await utils.answer(message, "<emoji document_id=5299030091735525430>❗️</emoji> Пропиши команду в ответ на игру.")
            return
        if pattern:
            await utils.answer(message, "<emoji document_id=5472146462362048818>💡</emoji>")
            clicks = self._solution(pattern)
            if not clicks:
                await utils.answer(message, "Иди код трейси гений.")
                return #*смачный пинок кодеру под зад.*
            for i in range(len(clicks)):
                if clicks[i] == 1:
                    await r.click(i)
            await utils.answer(message, "<emoji document_id=5395592707580127159>😎</emoji> Готово.")
        else:
            await utils.answer(message, "<emoji document_id=5299030091735525430>❗️</emoji> Ты ответил не на поле игры.")
            return