PromoClaimer.py

# ---------------------------------------------------лл------------------------------
# ░█▀▄░▄▀▀▄░█▀▄░█▀▀▄░█▀▀▄░█▀▀▀░▄▀▀▄░░░█▀▄▀█
# ░█░░░█░░█░█░█░█▄▄▀░█▄▄█░█░▀▄░█░░█░░░█░▀░█
# ░▀▀▀░░▀▀░░▀▀░░▀░▀▀░▀░░▀░▀▀▀▀░░▀▀░░░░▀░░▒▀
# Name: PromoClaimer
# Description: Automatically claim https://t.me/StableWaifuBot promo from any chat
# Author: @codrago_m
# Ported with Wine Hikka for FoxUserbot
# ---------------------------------------------------------------------------------
# 🔒    Licensed under the GNU AGPLv3
# 🌐 https://www.gnu.org/licenses/agpl-3.0.html
# ---------------------------------------------------------------------------------
# Author: @codrago
# Commands: checktokens
# scope: hikka_only
# meta developer: @codrago_m
# ---------------------------------------------------------------------------------

import asyncio
import os
import logging
from pyrogram import Client, filters
from pyrogram.enums import ParseMode
import re
from command import fox_command, fox_sudo, who_message, get_text

logger = logging.getLogger(__name__)

filename = os.path.basename(__file__)
Module_Name = 'PromoClaimer'

LANGUAGES = {
    "en": {
        "module_name": "PromoClaimer",
        "claimed_promo": "[PromoClaimer] 👌 I successfully claimed promo {promo} for {amount} tokens!",
        "error_watcher": "[PromoClaimer] ⛔️ An error occurred while watching for messages:\n{e}",
        "invalid_promo": "[PromoClaimer] 😢 Promo code {promo} is invalid or has expired!",
        "already_claimed": "[PromoClaimer] 😢 Promo code {promo} has already been claimed!",
        "checking_tokens": "[PromoClaimer] Checking tokens balance...",
        "watcher_started": "[PromoClaimer] Watcher started for StableWaifuBot promos",
        "command_description": "| Check tokens balance"
    },
    "ru": {
        "module_name": "PromoClaimer",
        "claimed_promo": "[PromoClaimer] 👌 Я успешно активировал промокод {promo} на {amount} токен(-ов)!",
        "error_watcher": "[PromoClaimer] ⛔️ Во время отслеживания сообщений произошла ошибка:\n{e}",
        "invalid_promo": "[PromoClaimer] 😢 Промокод {promo} недействителен, либо уже истек!",
        "already_claimed": "[PromoClaimer] 😢 Промокод {promo} уже активирован!",
        "checking_tokens": "[PromoClaimer] Проверка баланса токенов...",
        "watcher_started": "[PromoClaimer] Старт отслеживания промокодов StableWaifuBot",
        "command_description": "| Посмотреть баланс токенов"
    }
}



@Client.on_message(fox_command("checktokens", Module_Name, filename) & fox_sudo())
async def checktokens(client, message):
    message = await who_message(client, message)
    await message.edit(
        get_text("PromoClaimer", "checking_tokens", LANGUAGES=LANGUAGES),
        parse_mode=ParseMode.MARKDOWN
    )
    
    try:
        bot_username = "StableWaifuBot"
        sent_message = await client.send_message(bot_username, "/tokens")
        response = None
        
        for _ in range(15):
            await asyncio.sleep(1)
            async for msg in client.get_chat_history(bot_username, limit=1):
                if msg.from_user and not msg.from_user.is_self and msg.id != sent_message.id:
                    response = msg
                    break
            if response:
                break
        
        if response and response.text:
            # Проверяем, есть ли информация о токенах в ответе
            if "не нашел" in response.text.lower() or "not found" in response.text.lower():
                tokens = f"❌ {response.text}"
            else:
                # Показываем весь ответ от бота
                tokens = response.text
        else:
            tokens = "❌ No response from bot"
        
        await message.edit(tokens, parse_mode=ParseMode.MARKDOWN)
        
    except Exception as e:
        await message.edit(f"❌ Error: {e}", parse_mode=ParseMode.MARKDOWN)

@Client.on_message(filters.text & ~filters.me)
async def watcher(client, message):
    try:
        print(message.text)
        if not message.text:
            return
            
        pattern = r'https://t\.me/StableWaifuBot\?start=promo_(\w+)'
        matches = re.findall(pattern, message.text)
        if not matches:
            return
            
        bot_username = "StableWaifuBot"
        
        for match in matches:
            promo = 'promo_' + match
            
            sent_message = await client.send_message(bot_username, f'/start {promo}')
            response = None
            
            for _ in range(15):
                await asyncio.sleep(1)
                async for msg in client.get_chat_history(bot_username, limit=1):
                    if msg.from_user and not msg.from_user.is_self and msg.id != sent_message.id:
                        response = msg
                        break
                if response:
                    break
            
            if not response:
                logger.error(f"No response for promo {promo}")
                continue
            
            if 'недействителен' in response.text or 'истёк' in response.text or 'неверный' in response.text:
                logger.info(get_text("PromoClaimer", "invalid_promo", LANGUAGES=LANGUAGES, promo=promo))
            elif 'уже активирован' in response.text:
                logger.info(get_text("PromoClaimer", "already_claimed", LANGUAGES=LANGUAGES, promo=promo))
            else:
                try:
                    amount = response.text.split('(+')[1]
                    logger.info(get_text("PromoClaimer", "claimed_promo", LANGUAGES=LANGUAGES, promo=promo, amount=amount))
                except (IndexError, ValueError):
                    logger.info(get_text("PromoClaimer", "claimed_promo", LANGUAGES=LANGUAGES, promo=promo, amount="?"))
                    
    except Exception as e:
        logger.error(get_text("PromoClaimer", "error_watcher", LANGUAGES=LANGUAGES, e=str(e)))