mirror of https://github.com/coddrago/Heroku
660 lines
32 KiB
Python
660 lines
32 KiB
Python
# █ █ ▀ █▄▀ ▄▀█ █▀█ ▀
|
||
# █▀█ █ █ █ █▀█ █▀▄ █
|
||
# © Copyright 2022
|
||
# https://t.me/hikariatama
|
||
#
|
||
# 🔒 Licensed under the GNU AGPLv3
|
||
# 🌐 https://www.gnu.org/licenses/agpl-3.0.html
|
||
|
||
import asyncio
|
||
import io
|
||
import json
|
||
import logging
|
||
import time
|
||
|
||
from telethon.tl.types import Message
|
||
from telethon.tl import functions
|
||
from telethon.tl.tlobject import TLRequest
|
||
|
||
from .. import loader, utils
|
||
from ..inline.types import InlineCall
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
GROUPS = [
|
||
"auth",
|
||
"account",
|
||
"users",
|
||
"contacts",
|
||
"messages",
|
||
"updates",
|
||
"photos",
|
||
"upload",
|
||
"help",
|
||
"channels",
|
||
"bots",
|
||
"payments",
|
||
"stickers",
|
||
"phone",
|
||
"langpack",
|
||
"folders",
|
||
"stats",
|
||
]
|
||
|
||
|
||
def decapitalize(string: str) -> str:
|
||
return string[0].lower() + string[1:]
|
||
|
||
|
||
CONSTRUCTORS = {
|
||
decapitalize(
|
||
method.__class__.__name__.rsplit("Request", 1)[0]
|
||
): method.CONSTRUCTOR_ID
|
||
for method in utils.array_sum(
|
||
[
|
||
[
|
||
method
|
||
for method in dir(getattr(functions, group))
|
||
if isinstance(method, TLRequest)
|
||
]
|
||
for group in GROUPS
|
||
]
|
||
)
|
||
}
|
||
|
||
|
||
@loader.tds
|
||
class APIRatelimiterMod(loader.Module):
|
||
"""Helps userbot avoid spamming Telegram API"""
|
||
|
||
strings = {
|
||
"name": "APILimiter",
|
||
"warning": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji>"
|
||
" <b>WARNING!</b>\n\nYour account exceeded the limit of requests, specified"
|
||
" in config. In order to prevent Telegram API Flood, userbot has been"
|
||
" <b>fully frozen</b> for {} seconds. Further info is provided in attached"
|
||
" file. \n\nIt is recommended to get help in <code>{prefix}support</code>"
|
||
" group!\n\nIf you think, that it is an intended behavior, then wait until"
|
||
" userbot gets unlocked and next time, when you will be going to perform"
|
||
" such an operation, use <code>{prefix}suspend_api_protect</code> <time"
|
||
" in seconds>"
|
||
),
|
||
"args_invalid": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Invalid arguments</b>"
|
||
),
|
||
"suspended_for": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>API Flood Protection"
|
||
" is disabled for {} seconds</b>"
|
||
),
|
||
"test": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>This action will"
|
||
" expose your account to flooding Telegram API.</b> <i>In order to confirm,"
|
||
" that you really know, what you are doing, complete this simple test -"
|
||
" find the emoji, differing from others</i>"
|
||
),
|
||
"on": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>Protection enabled</b>"
|
||
),
|
||
"off": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>Protection"
|
||
" disabled</b>"
|
||
),
|
||
"u_sure": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Are you sure?</b>"
|
||
),
|
||
"_cfg_time_sample": "Time sample through which the bot will count requests",
|
||
"_cfg_threshold": "Threshold of requests to trigger protection",
|
||
"_cfg_local_floodwait": (
|
||
"Freeze userbot for this amount of time, if request limit exceeds"
|
||
),
|
||
"_cfg_forbidden_methods": (
|
||
"Forbid specified methods from being executed throughout external modules"
|
||
),
|
||
"btn_no": "🚫 No",
|
||
"btn_yes": "✅ Yes",
|
||
}
|
||
|
||
strings_ru = {
|
||
"warning": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji>"
|
||
" <b>ВНИМАНИЕ!</b>\n\nАккаунт вышел за лимиты запросов, указанные в"
|
||
" конфиге. С целью предотвращения флуда Telegram API, юзербот был"
|
||
" <b>полностью заморожен</b> на {} секунд. Дополнительная информация"
|
||
" прикреплена в файле ниже. \n\nРекомендуется обратиться за помощью в"
|
||
" <code>{prefix}support</code> группу!\n\nЕсли ты считаешь, что это"
|
||
" запланированное поведение юзербота, просто подожди, пока закончится"
|
||
" таймер и в следующий раз, когда запланируешь выполнять такую"
|
||
" ресурсозатратную операцию, используй"
|
||
" <code>{prefix}suspend_api_protect</code> <время в секундах>"
|
||
),
|
||
"args_invalid": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Неверные"
|
||
" аргументы</b>"
|
||
),
|
||
"suspended_for": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>Защита API отключена"
|
||
" на {} секунд</b>"
|
||
),
|
||
"test": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Это действие"
|
||
" открывает юзерботу возможность флудить Telegram API.</b> <i>Для того,"
|
||
" чтобы убедиться, что ты действительно уверен в том, что делаешь - реши"
|
||
" простенький тест - найди отличающийся эмодзи.</i>"
|
||
),
|
||
"on": "<emoji document_id=5458450833857322148>👌</emoji> <b>Защита включена</b>",
|
||
"off": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>Защита отключена</b>"
|
||
),
|
||
"u_sure": "<emoji document_id=6319093650693293883>☣️</emoji> <b>Ты уверен?</b>",
|
||
"_cfg_time_sample": (
|
||
"Временной промежуток, по которому будет считаться количество запросов"
|
||
),
|
||
"_cfg_threshold": "Порог запросов, при котором будет срабатывать защита",
|
||
"_cfg_local_floodwait": (
|
||
"Заморозить юзербота на это количество секунд, если лимит запросов превышен"
|
||
),
|
||
"_cfg_forbidden_methods": (
|
||
"Запретить выполнение указанных методов во всех внешних модулях"
|
||
),
|
||
"btn_no": "🚫 Нет",
|
||
"btn_yes": "✅ Да",
|
||
}
|
||
|
||
strings_de = {
|
||
"warning": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji>"
|
||
" <b>Achtung!</b>\n\nDas Konto hat die in der Konfiguration angegebenen"
|
||
" Grenzwerte für Anfragen überschritten. Um Telegram API-Flooding zu"
|
||
" verhindern, wurde der <b>ganze Userbot</b> für {} Sekunden"
|
||
" eingefroren. Weitere Informationen finden Sie im unten angefügten"
|
||
" Datei.\n\nWir empfehlen Ihnen, sich mit Hilfe der <code>{prefix}"
|
||
"support</code> Gruppe zu helfen!\n\nWenn du denkst, dass dies"
|
||
" geplantes Verhalten des Userbots ist, dann warte einfach, bis der"
|
||
" Timer abläuft und versuche beim nächsten Mal, eine so ressourcen"
|
||
" intensive Operation wie <code>{prefix}suspend_api_protect</code>"
|
||
" <Zeit in Sekunden> zu planen."
|
||
),
|
||
"args_invalid": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Ungültige"
|
||
" Argumente</b>"
|
||
),
|
||
"suspended_for": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>API Flood"
|
||
" Protection ist für {} Sekunden deaktiviert</b>"
|
||
),
|
||
"test": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Dieser"
|
||
" Vorgang wird deinen Account ermöglichen, die Telegram API zu"
|
||
" überfluten.</b> <i>Um sicherzustellen, dass du wirklich weißt, was"
|
||
" du tust, beende diesen einfachen Test - findest du das Emoji, das von"
|
||
" den anderen abweicht?</i>"
|
||
),
|
||
"on": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>Schutz aktiviert</b>"
|
||
),
|
||
"off": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>Schutz deaktiviert</b>"
|
||
),
|
||
"u_sure": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Bist du sicher?</b>"
|
||
),
|
||
"_cfg_time_sample": "Zeitintervall, in dem die Anfragen gezählt werden",
|
||
"_cfg_threshold": (
|
||
"Schwellenwert für Anfragen, ab dem der Schutz aktiviert wird"
|
||
),
|
||
"_cfg_local_floodwait": (
|
||
"Einfrieren des Userbots für diese Anzahl von Sekunden, wenn der Grenzwert"
|
||
" überschritten wird"
|
||
),
|
||
"_cfg_forbidden_methods": "Verbotene Methoden in allen externen Modulen",
|
||
"btn_no": "🚫 Nein",
|
||
"btn_yes": "✅ Ja",
|
||
}
|
||
|
||
strings_tr = {
|
||
"warning": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji>"
|
||
" <b>Dikkat!</b>\n\nHesabın ayarlarda belirtilmiş istek sınırını aştı."
|
||
" Telegram API Flood’unu önlemek için tüm <b>kullanıcı botu<b> {} saniye"
|
||
" boyunca durduruldu. Daha fazla bilgi almak için ekteki dosyayı"
|
||
" inceleyebilirsiniz. /n/ Ayrıca <code>{prefix}Destek</code> grubundan"
|
||
" yardım almanız önerilmektedir. Eğer bu işlemin kasıtlı bir işlem olduğunu"
|
||
" düşünüyorsanız, kullanıcı botunuzun açılmasının bekleyin ve bu tarz bir"
|
||
" işlem gerçekleştireceğiniz sıradaki sefer"
|
||
" <code>{prefix}suspend_api_protect</code> <saniye> kodunu kullanın."
|
||
),
|
||
"args_invalid": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Geçersiz"
|
||
" argümanlar</b>"
|
||
),
|
||
"suspended_for": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>API Flood koruması {}"
|
||
" saniyeliğine durduruldu.</b>"
|
||
),
|
||
"test": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Bu eylem"
|
||
" Thesabınızın Telegram API Flood’u yapabilmesine izin verecektir.</b>"
|
||
" <i>Ne yaptığını bildiğinizi onaylamak için bu basit testi çözün."
|
||
" - Diğerlerinden farklı olan emojiyi seç</i>"
|
||
),
|
||
"on": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>Koruma"
|
||
" aktifleştirildi.</b>"
|
||
),
|
||
"off": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>Koruma"
|
||
" de-aktifleştirildi</b>"
|
||
),
|
||
"u_sure": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Emin misin?</b>"
|
||
),
|
||
"_cfg_time_sample": "Saniyede sayılan isteklerin zaman aralığı",
|
||
"_cfg_threshold": "Korumanın etkinleşeceği sınır değeri",
|
||
"_cfg_local_floodwait": (
|
||
"Telegram API sınır değeri aşıldığında kullanıcı botu bir süre durdurulur"
|
||
),
|
||
"_cfg_forbidden_methods": (
|
||
"Belirtili metodların harici modüller tarafından çalıştırılmasını yasakla"
|
||
),
|
||
"btn_no": "🚫 Hayır",
|
||
"btn_yes": "✅ Evet",
|
||
}
|
||
|
||
strings_hi = {
|
||
"warning": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji>"
|
||
" <b>चेतावनी!</b>\n\nइस खाते के लिए विन्यास में निर्दिष्ट सीमा सीमा"
|
||
" पार कर गए हैं। टेलीग्राम एपीआई फ्लडिंग को रोकने के लिए, यह"
|
||
" <b>सभी userbot</b> को {} सेकंड तक जमा कर दिया गया है। अधिक"
|
||
" जानकारी के लिए नीचे दिए गए फ़ाइल पढ़ें।\n\nअपनी सहायता के लिए"
|
||
" <code>{prefix}support</code> समूह का उपयोग करें!\n\nयदि आपको लगता है"
|
||
" यह उपयोगकर्ता बॉट की योजित व्यवहार है, तो बस टाइमर समाप्त होने"
|
||
" तक इंतजार करें और अगली बार एक ऐसी संसाधन ज्यादा खर्च करने वाली"
|
||
" ऑपरेशन को योजित करने के लिए <code>{prefix}suspend_api_protect</code>"
|
||
" <सेकंड> का उपयोग करें।"
|
||
),
|
||
"args_invalid": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>अमान्य तर्क</b>"
|
||
),
|
||
"suspended_for": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>API Flood"
|
||
" सुरक्षा को {} सेकंड के लिए अक्षम कर दिया गया है</b>"
|
||
),
|
||
"test": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>यह ऑपरेशन"
|
||
" टेलीग्राम एपीआई को फ्लड करने की अनुमति देगा।</b> <i>आप क्या कर रहे हैं"
|
||
" यह सुनिश्चित करने के लिए एक आसान परीक्षण को हल करें, जिसमें अलग"
|
||
" एमोजी का पता लगाएं?</i>"
|
||
),
|
||
"on": "<emoji document_id=5458450833857322148>👌</emoji> <b>सुरक्षा सक्षम</b>",
|
||
"off": "<emoji document_id=5458450833857322148>👌</emoji> <b>सुरक्षा अक्षम</b>",
|
||
"u_sure": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>क्या आप"
|
||
" सुनिश्चित हैं?</b>"
|
||
),
|
||
"_cfg_time_sample": "प्रति सेकंड गिने जाने वाले अनुरोधों की समय सीमा",
|
||
"_cfg_threshold": "सुरक्षा सक्षम करने के लिए मान सीमित करें",
|
||
"_cfg_local_floodwait": (
|
||
"यूजरबॉट को इस संख्या के सेकंड के लिए फ्रीज करें जब सीमा मान पार हो जाए"
|
||
),
|
||
"_cfg_forbidden_methods": "सभी बाहरी मॉड्यूल में निषिद्ध तरीके",
|
||
"btn_no": "🚫 नहीं",
|
||
"btn_yes": "✅ हाँ",
|
||
}
|
||
|
||
strings_uz = {
|
||
"warning": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji>"
|
||
" <b>Ogohlantirish!</b>\n\nBu hisob uchun konfiguratsiyada ko'rsatilgan"
|
||
" chegaralar chegarani o'zgartirgan.\n\nTelegram API Flood"
|
||
" to'xtatish uchun, bu <b>hammasi userbot</b> uchun {} sekundni"
|
||
" blokirovka qilindi. Batafsil ma'lumot uchun pastdagi faylni o'qing.\n\n"
|
||
"Yordam uchun <code>{prefix}support</code> guruhidan foydalaning!\n\nAgar"
|
||
" siz hisobni botning yordamchisi bo'lishi kerak bo'lgan amalni bajarishga"
|
||
" imkoniyat berishga o'xshaysiz, unda faqat blokirovkani to'xtatish uchun"
|
||
" <code>{prefix}suspend_api_protect</code> <sekund> dan foydalaning."
|
||
),
|
||
"args_invalid": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Noto'g'ri"
|
||
" argument</b>"
|
||
),
|
||
"suspended_for": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>API Flood"
|
||
" himoya {} sekund uchun to'xtatildi</b>"
|
||
),
|
||
"test": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Ushbu amal Telegram"
|
||
" API-ni flood qilishga ruxsat beradi.</b> <i>Siz qanday ish"
|
||
" bajarayotganingizni tekshirish uchun oson testni bajarishga harakat"
|
||
" qiling, emojilarni aniqlash uchun?</i>"
|
||
),
|
||
"on": "<emoji document_id=5458450833857322148>👌</emoji> <b>Himoya yoqildi</b>",
|
||
"off": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>Himoya o'chirildi</b>"
|
||
),
|
||
"u_sure": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Siz"
|
||
" ishonchingiz komilmi?</b>"
|
||
),
|
||
"_cfg_time_sample": "Sekundda qabul qilinadigan so'rovlar soni chegarasi",
|
||
"_cfg_threshold": "Himoya yoqish uchun qiymatni chegaralash",
|
||
"_cfg_local_floodwait": (
|
||
"Foydalanuvchi botni ushbu soniya davomida blokirovka qiladi, agar"
|
||
" chegaralar qiymati oshsa"
|
||
),
|
||
"_cfg_forbidden_methods": "Barcha tashqi modullarda taqiqlangan usullar",
|
||
"btn_no": "🚫 Yo'q",
|
||
"btn_yes": "✅ Ha",
|
||
}
|
||
|
||
strings_ja = {
|
||
"warning": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji>"
|
||
" <b>警告！</b>\n\nこのアカウントの設定では、以下の制限が設定されています。\n\n"
|
||
"Telegram APIのフラッドを防ぐために、この<b>すべてのユーザーボット</b>は"
|
||
" {}秒間ブロックされます。詳細については、下記のファイルをご覧ください。\n\n"
|
||
"サポートについては、<code>{prefix}support</code>グループをご利用ください！\n\n"
|
||
"アカウントが実行する必要のあるアクションを許可する場合は、"
|
||
"<code>{prefix}suspend_api_protect</code>を使用してブロックを解除するだけです。"
|
||
),
|
||
"args_invalid": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>無効な引数</b>"
|
||
),
|
||
"suspended_for": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji>"
|
||
" <b>APIフラッド保護が{}秒間無効になりました</b>"
|
||
),
|
||
"test": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>このアクションは、"
|
||
"Telegram APIをフラッドさせることができます。</b> <i>あなたが何をしているかを"
|
||
"確認するために、簡単なテストを実行するには、次のように入力してください。</i>"
|
||
),
|
||
"on": "<emoji document_id=5458450833857322148>👌</emoji> <b>保護が有効になりました</b>",
|
||
"off": "<emoji document_id=5458450833857322148>👌</emoji> <b>保護が無効になりました</b>",
|
||
"u_sure": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>本当によろしいですか？</b>"
|
||
),
|
||
"_cfg_time_sample": "秒あたりの許可されたリクエスト数の制限",
|
||
"_cfg_threshold": "制限を超えた場合の値",
|
||
"_cfg_local_floodwait": "ユーザーがこの秒数以内にボットをブロックする場合",
|
||
"_cfg_forbidden_methods": "すべての外部モジュールで禁止されているメソッド",
|
||
"btn_no": "🚫 いいえ",
|
||
"btn_yes": "✅ はい",
|
||
}
|
||
|
||
strings_kr = {
|
||
"warning": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji>"
|
||
" <b>경고！</b>\n\n이 계정의 설정에 따르면, 다음 제한이 설정됩니다.\n\n"
|
||
"이 <b>모든 사용자 봇</b>은 Telegram API의 플러드를 방지하기 위해"
|
||
" {}초 동안 차단됩니다. 자세한 내용은 아래 파일을 참조하십시오.\n\n"
|
||
"지원에 대해서는 <code>{prefix}support</code> 그룹을 사용하십시오!\n\n"
|
||
"계정이 실행해야하는 작업을 허용하려면, <code>{prefix}suspend_api_protect</code>를"
|
||
"사용하여 차단을 해제하십시오."
|
||
),
|
||
"args_invalid": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>잘못된인수</b>"
|
||
),
|
||
"suspended_for": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji>"
|
||
" <b>API 플러드 보호가 {}초간 비활성화되었습니다</b>"
|
||
),
|
||
"test": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>이 작업은"
|
||
"Telegram API를 플러드시킬 수 있습니다.</b> <i>당신이 무엇을 하는지 확인하기 위해,"
|
||
"간단한 테스트를 실행하려면 다음과 같이 입력하십시오.</i>"
|
||
),
|
||
"on": "<emoji document_id=5458450833857322148>👌</emoji> <b>보호가 활성화되었습니다</b>",
|
||
"off": "<emoji document_id=5458450833857322148>👌</emoji> <b>보호가 비활성화되었습니다</b>",
|
||
"u_sure": "<emoji document_id=6319093650693293883>☣️</emoji> <b>확실합니까?</b>",
|
||
"_cfg_time_sample": "허용되는 요청 수의 제한 초",
|
||
"_cfg_threshold": "제한을 초과한 경우의 값",
|
||
"_cfg_local_floodwait": "사용자가 이 초 이내에 봇을 차단하는 경우",
|
||
"_cfg_forbidden_methods": "모든 외부 모듈에서 금지된 메서드",
|
||
"btn_no": "🚫 아니요",
|
||
"btn_yes": "✅ 예",
|
||
}
|
||
|
||
strings_ar = {
|
||
"warning": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>تحذير!</b>\n\nحسب"
|
||
" إعدادات هذا الحساب، فإن الحدود التالية ستتم تطبيقها.\n\nسيتم حظر <b>جميع"
|
||
" بوتات المستخدمين</b> لمدة {} ثانية لمنع تجاوز الحد الأقصى لمتطلبات"
|
||
" Telegram API. لمزيد من المعلومات، راجع الملف التالي.\n\nللمساعدة، استخدم"
|
||
" مجموعة <code>{prefix}support</code>!\n\nللسماح للحساب بالعمل، استخدم"
|
||
" <code>{prefix}suspend_api_protect</code> لإلغاء الحظر."
|
||
),
|
||
"args_invalid": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>معلمات غير صالحة</b>"
|
||
),
|
||
"suspended_for": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji>"
|
||
" <b>تم تعطيل حماية API لمدة {} ثانية</b>"
|
||
),
|
||
"test": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>هذا الأمر قد يؤدي"
|
||
" إلىتجاوز حدود Telegram API.</b> <i>للتحقق من ما تفعله، يمكنك تشغيل اختبار"
|
||
" بسيطبالإضافة إلى الأمر التالي.</i>"
|
||
),
|
||
"on": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>تم تفعيل الحماية</b>"
|
||
),
|
||
"off": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>تم تعطيل الحماية</b>"
|
||
),
|
||
"u_sure": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>هل أنت متأكد؟</b>"
|
||
),
|
||
"_cfg_time_sample": "المدة بالثواني التي يتم فيها تجاوزها حد المتطلبات",
|
||
"_cfg_threshold": "قيمة تجاوزها الحد",
|
||
"_cfg_local_floodwait": "المدة بالثواني التي يتم فيها حظر المستخدم للبوت",
|
||
"_cfg_forbidden_methods": "الأوامر الممنوعة من قبل كل الإضافات الخارجية",
|
||
"btn_no": "🚫 لا",
|
||
"btn_yes": "✅ نعم",
|
||
}
|
||
|
||
strings_es = {
|
||
"warning": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji>"
|
||
" <b>¡Advertencia!</b>\n\nDe acuerdo con la configuración de esta cuenta,"
|
||
" las siguientes limitaciones serán aplicadas.\n\nSe bloqueará <b>a todos"
|
||
" los bots de los usuarios</b> por {} segundos para evitar el exceso de las"
|
||
" limitaciones de Telegram API. Para más información, consulta el archivo"
|
||
" siguiente.\n\nPara obtener ayuda, use el grupo"
|
||
" <code>{prefix}support</code>!\n\nPara permitir que la cuenta funcione,"
|
||
" use <code>{prefix}suspend_api_protect</code> para desbloquear."
|
||
),
|
||
"args_invalid": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Argumentos"
|
||
" inválidos</b>"
|
||
),
|
||
"suspended_for": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji>"
|
||
" <b>Se ha desactivado la protección de API por {} segundos</b>"
|
||
),
|
||
"test": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>Este comando puede"
|
||
" llevar a exceder las limitaciones de Telegram API.</b> <i>Para comprobar"
|
||
" que estás haciendo, puedes ejecutar una prueba simple agregando el"
|
||
" siguiente comando.</i>"
|
||
),
|
||
"on": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>Protección"
|
||
" activada</b>"
|
||
),
|
||
"off": (
|
||
"<emoji document_id=5458450833857322148>👌</emoji> <b>Protección"
|
||
" desactivada</b>"
|
||
),
|
||
"u_sure": (
|
||
"<emoji document_id=6319093650693293883>☣️</emoji> <b>¿Estás seguro?</b>"
|
||
),
|
||
"_cfg_time_sample": (
|
||
"El tiempo en segundos durante el cual se exceden las limitaciones"
|
||
),
|
||
"_cfg_threshold": "El valor por encima del cual se exceden las limitaciones",
|
||
"_cfg_local_floodwait": (
|
||
"El tiempo en segundos durante el cual se bloquea al usuario para el bot"
|
||
),
|
||
"_cfg_forbidden_methods": (
|
||
"Los comandos prohibidos por todas las extensiones externas"
|
||
),
|
||
"btn_no": "🚫 No",
|
||
"btn_yes": "✅ Sí",
|
||
}
|
||
|
||
_ratelimiter = []
|
||
_suspend_until = 0
|
||
_lock = False
|
||
|
||
def __init__(self):
|
||
self.config = loader.ModuleConfig(
|
||
loader.ConfigValue(
|
||
"time_sample",
|
||
15,
|
||
lambda: self.strings("_cfg_time_sample"),
|
||
validator=loader.validators.Integer(minimum=1),
|
||
),
|
||
loader.ConfigValue(
|
||
"threshold",
|
||
100,
|
||
lambda: self.strings("_cfg_threshold"),
|
||
validator=loader.validators.Integer(minimum=10),
|
||
),
|
||
loader.ConfigValue(
|
||
"local_floodwait",
|
||
30,
|
||
lambda: self.strings("_cfg_local_floodwait"),
|
||
validator=loader.validators.Integer(minimum=10, maximum=3600),
|
||
),
|
||
loader.ConfigValue(
|
||
"forbidden_methods",
|
||
["joinChannel", "importChatInvite"],
|
||
lambda: self.strings("_cfg_forbidden_methods"),
|
||
validator=loader.validators.MultiChoice(
|
||
[
|
||
"sendReaction",
|
||
"joinChannel",
|
||
"importChatInvite",
|
||
]
|
||
),
|
||
on_change=lambda: self._client.forbid_constructors(
|
||
map(
|
||
lambda x: CONSTRUCTORS[x], self.config["forbidden_constructors"]
|
||
)
|
||
),
|
||
),
|
||
)
|
||
|
||
async def client_ready(self):
|
||
asyncio.ensure_future(self._install_protection())
|
||
|
||
async def _install_protection(self):
|
||
await asyncio.sleep(30) # Restart lock
|
||
if hasattr(self._client._call, "_old_call_rewritten"):
|
||
raise loader.SelfUnload("Already installed")
|
||
|
||
old_call = self._client._call
|
||
|
||
async def new_call(
|
||
sender: "MTProtoSender", # type: ignore
|
||
request: "TLRequest", # type: ignore
|
||
ordered: bool = False,
|
||
flood_sleep_threshold: int = None,
|
||
):
|
||
if time.perf_counter() > self._suspend_until and not self.get(
|
||
"disable_protection",
|
||
True,
|
||
):
|
||
request_name = type(request).__name__
|
||
self._ratelimiter += [[request_name, time.perf_counter()]]
|
||
|
||
self._ratelimiter = list(
|
||
filter(
|
||
lambda x: time.perf_counter() - x[1]
|
||
< int(self.config["time_sample"]),
|
||
self._ratelimiter,
|
||
)
|
||
)
|
||
|
||
if (
|
||
len(self._ratelimiter) > int(self.config["threshold"])
|
||
and not self._lock
|
||
):
|
||
self._lock = True
|
||
report = io.BytesIO(
|
||
json.dumps(
|
||
self._ratelimiter,
|
||
indent=4,
|
||
).encode("utf-8")
|
||
)
|
||
report.name = "local_fw_report.json"
|
||
|
||
await self.inline.bot.send_document(
|
||
self.tg_id,
|
||
report,
|
||
caption=self.strings("warning").format(
|
||
self.config["local_floodwait"],
|
||
prefix=self.get_prefix(),
|
||
),
|
||
)
|
||
|
||
# It is intented to use time.sleep instead of asyncio.sleep
|
||
time.sleep(int(self.config["local_floodwait"]))
|
||
self._lock = False
|
||
|
||
return await old_call(sender, request, ordered, flood_sleep_threshold)
|
||
|
||
self._client._call = new_call
|
||
self._client._old_call_rewritten = old_call
|
||
self._client._call._hikka_overwritten = True
|
||
logger.debug("Successfully installed ratelimiter")
|
||
|
||
async def on_unload(self):
|
||
if hasattr(self._client, "_old_call_rewritten"):
|
||
self._client._call = self._client._old_call_rewritten
|
||
delattr(self._client, "_old_call_rewritten")
|
||
logger.debug("Successfully uninstalled ratelimiter")
|
||
|
||
@loader.command(
|
||
ru_doc="<время в секундах> - Заморозить защиту API на N секунд",
|
||
de_doc="<Sekunden> - API-Schutz für N Sekunden einfrieren",
|
||
tr_doc="<saniye> - API korumasını N saniye dondur",
|
||
hi_doc="<सेकंड> - API सुरक्षा को N सेकंड जमा करें",
|
||
uz_doc="<soniya> - API himoyasini N soniya o'zgartirish",
|
||
)
|
||
async def suspend_api_protect(self, message: Message):
|
||
"""<time in seconds> - Suspend API Ratelimiter for n seconds"""
|
||
args = utils.get_args_raw(message)
|
||
|
||
if not args or not args.isdigit():
|
||
await utils.answer(message, self.strings("args_invalid"))
|
||
return
|
||
|
||
self._suspend_until = time.perf_counter() + int(args)
|
||
await utils.answer(message, self.strings("suspended_for").format(args))
|
||
|
||
@loader.command(
|
||
ru_doc="Включить/выключить защиту API",
|
||
de_doc="API-Schutz einschalten / ausschalten",
|
||
tr_doc="API korumasını aç / kapat",
|
||
hi_doc="API सुरक्षा चालू / बंद करें",
|
||
uz_doc="API himoyasini yoqish / o'chirish",
|
||
)
|
||
async def api_fw_protection(self, message: Message):
|
||
"""Toggle API Ratelimiter"""
|
||
await self.inline.form(
|
||
message=message,
|
||
text=self.strings("u_sure"),
|
||
reply_markup=[
|
||
{"text": self.strings("btn_no"), "action": "close"},
|
||
{"text": self.strings("btn_yes"), "callback": self._finish},
|
||
],
|
||
)
|
||
|
||
async def _finish(self, call: InlineCall):
|
||
state = self.get("disable_protection", True)
|
||
self.set("disable_protection", not state)
|
||
await call.edit(self.strings("on" if state else "off"))
|