From 2baff4118e28308b188c9bbde3d9cca001d35bd5 Mon Sep 17 00:00:00 2001 From: tema Date: Fri, 18 Mar 2022 21:25:50 +0200 Subject: [PATCH] Init --- .gitignore | 14 +++++ README.md | 3 + bot.py | 43 +++++++++++++ coc/__init__.py | 6 ++ coc/base.py | 26 ++++++++ coc/clan.py | 17 ++++++ coc/user.py | 12 ++++ config.py | 16 +++++ database.sqlite3 | Bin 0 -> 8192 bytes database/model.py | 18 ++++++ database/worker.py | 57 ++++++++++++++++++ example_config.ini | 26 ++++++++ filters/__init__.py | 5 ++ filters/main.py | 15 +++++ fsm/base.py | 5 ++ handlers/__init__.py | 2 + handlers/group/__init__.py | 3 + handlers/group/captcha.py | 52 ++++++++++++++++ handlers/group/main.py | 12 ++++ handlers/group/users.py | 18 ++++++ handlers/private/__init__.py | 1 + handlers/private/register.py | 114 +++++++++++++++++++++++++++++++++++ keyboards/__init__.py | 1 + keyboards/inline/__init__.py | 0 keyboards/inline/keyboard.py | 18 ++++++ load.py | 26 ++++++++ requirements.txt | 3 + utils/timer.py | 60 ++++++++++++++++++ 28 files changed, 573 insertions(+) create mode 100644 .gitignore create mode 100644 README.md create mode 100644 bot.py create mode 100644 coc/__init__.py create mode 100644 coc/base.py create mode 100644 coc/clan.py create mode 100644 coc/user.py create mode 100644 config.py create mode 100644 database.sqlite3 create mode 100644 database/model.py create mode 100644 database/worker.py create mode 100644 example_config.ini create mode 100644 filters/__init__.py create mode 100644 filters/main.py create mode 100644 fsm/base.py create mode 100644 handlers/__init__.py create mode 100644 handlers/group/__init__.py create mode 100644 handlers/group/captcha.py create mode 100644 handlers/group/main.py create mode 100644 handlers/group/users.py create mode 100644 handlers/private/__init__.py create mode 100644 handlers/private/register.py create mode 100644 keyboards/__init__.py create mode 100644 keyboards/inline/__init__.py create mode 100644 keyboards/inline/keyboard.py create mode 100644 load.py create mode 100644 requirements.txt create mode 100644 utils/timer.py diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..4bd2a4d --- /dev/null +++ b/.gitignore @@ -0,0 +1,14 @@ +# Byte-compiled / optimized / DLL files +__pycache__/ +*.py[cod] +*$py.class + +# C extensions +*.so + +venv/ +*.txt +!requirements.txt +*.ini +!example_config.ini +test.py \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..9a56e3f --- /dev/null +++ b/README.md @@ -0,0 +1,3 @@ +# clash-telegram-bot + +Бот для верификации юзеров. Проверка состояния в клане clash of clans по средствам clash of clans API \ No newline at end of file diff --git a/bot.py b/bot.py new file mode 100644 index 0000000..1c8b886 --- /dev/null +++ b/bot.py @@ -0,0 +1,43 @@ +import logging + +from aiogram import executor + +from load import dp, config +import filters +import handlers + +skip_updates=True + + +logging.basicConfig( + format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', + level=logging.INFO +) + +async def on_startup(dp): + webhook_url = "{host}{path}{token}".format( + host=config["WebHook"]["webhook_host"], + path=config["WebHook"]["webhook_path"], + token=config["Bot"]["token"] + ) + await dp.bot.set_webhook(url=webhook_url) + + +async def on_shutdown(dp): + await dp.bot.delete_webhook() + + +if __name__ == "__main__": + + if config["WebHook"]["use"].lower() in ["yes", "true"]: + executor.start_webhook( + dispatcher=dp, + skip_updates=skip_updates, + webhook_path="{}{}".format(config["WebHook"]["webhook_path"], config["Bot"]["token"]), + on_startup=on_startup, + on_shutdown=on_shutdown, + host=config["WebHook"]["app_host"], + port=config["WebHook"]["app_port"] + ) + else: + executor.start_polling(dp, skip_updates=skip_updates) diff --git a/coc/__init__.py b/coc/__init__.py new file mode 100644 index 0000000..251f393 --- /dev/null +++ b/coc/__init__.py @@ -0,0 +1,6 @@ +import imp +from .user import Player +from .clan import Clan + +class ClashOfClans(Player, Clan): + pass diff --git a/coc/base.py b/coc/base.py new file mode 100644 index 0000000..8d2bbc5 --- /dev/null +++ b/coc/base.py @@ -0,0 +1,26 @@ +import json +import requests +from abc import ABC + +class Base(ABC): + def __init__(self, api_token:str, clan_tag:str = None) -> None: + self.headers = { + "Content-Type": "application/json", + "Authorization": "Bearer {api_token}".format(api_token=api_token) + } + self.clan_tag = clan_tag + + def get(self, url: str, tag: str): + request = requests.get( + url.format(tag.replace("#", "")), + headers=self.headers + ) + return json.loads(request.text) + + def post(self, url: str, tag: str, payload: dict): + request = requests.post( + url=url.format(tag.replace("#", "")), + headers=self.headers, + data=json.dumps(payload) + ) + return json.loads(request.text) \ No newline at end of file diff --git a/coc/clan.py b/coc/clan.py new file mode 100644 index 0000000..b7ad36e --- /dev/null +++ b/coc/clan.py @@ -0,0 +1,17 @@ +from .base import Base + +class Clan(Base): + + def clan_members(self, clan_tag:str = None): + if self.clan_tag is not None: + clan_tag = self.clan_tag + + url = "https://api.clashofclans.com/v1/clans/%23{}/members" + return self.get(url, tag=clan_tag) + + def get_clan(self, clan_tag:str = None): + if self.clan_tag is not None: + clan_tag = self.clan_tag + + url = "https://api.clashofclans.com/v1/clans/%23{}" + return self.get(url, tag=clan_tag) \ No newline at end of file diff --git a/coc/user.py b/coc/user.py new file mode 100644 index 0000000..789e93b --- /dev/null +++ b/coc/user.py @@ -0,0 +1,12 @@ +from .base import Base + +class Player(Base): + + def get_player(self, tag: str): + url = "https://api.clashofclans.com/v1/players/%23{}" + return self.get(url, tag) + + def verify_token(self, tag: str, token: str): + payload = {"token":token} + url = "https://api.clashofclans.com/v1/players/%23{}/verifytoken" + return self.post(url, tag, payload) diff --git a/config.py b/config.py new file mode 100644 index 0000000..c360a6e --- /dev/null +++ b/config.py @@ -0,0 +1,16 @@ +from configparser import ConfigParser + +CONFIG_FILE ='./config.ini' + + + +data = ConfigParser() +data.read(CONFIG_FILE) + +config = dict() + +for section in data.sections(): + config[section] = dict() + + for key, value in data.items(section): + config[section][key] = value diff --git a/database.sqlite3 b/database.sqlite3 new file mode 100644 index 0000000000000000000000000000000000000000..8b84140fb3636039b84e4fe306a8bcae21a088a4 GIT binary patch literal 8192 zcmeI$O-lkn7y#fI5rU$&n@9-bmY@)_sFikVv&E#%YWx7{AY+ZT>=*2=NZ07nu|t2L zYnOgN&>^~Yuz%EOUzkAQ& z68*hl@e+O?i05$pQ;VVWp)xC?@Z19u7c>w60T2KI5C8!X009sH0T2KI5csXY**czz zL?ZaS8nCv-dp`I4(JfS@np)FHP2JQrB9G=|l9Y{OnUpIvt)x*>*{hLCUDwGzE$^zd zNp`g6@@s7CyEw7dF+4wLIn3rVX{fZgt9iP&6LJcx~zCesKMKIk+3#c6z3N>Ms_G zijr?AI^A)7+x4gC3RuK1phP@@F|YS7w#2!|u134e>9&qNE>0@E#b%>fD&#YYQng)^ MlTB_}yg$=^0gC8{F#rGn literal 0 HcmV?d00001 diff --git a/database/model.py b/database/model.py new file mode 100644 index 0000000..d2b18a0 --- /dev/null +++ b/database/model.py @@ -0,0 +1,18 @@ +from peewee import Model, BigIntegerField, CharField + +from load import db + + +class Users(Model): + user_id = BigIntegerField(null=False) + first_name = CharField(64) + last_name = CharField(64, null=True) + username = CharField(32, null=True) + + tag = CharField() + nickname = CharField() + townhall = BigIntegerField() + attackwins = BigIntegerField() + + class Meta: + database = db \ No newline at end of file diff --git a/database/worker.py b/database/worker.py new file mode 100644 index 0000000..c8efeda --- /dev/null +++ b/database/worker.py @@ -0,0 +1,57 @@ +from aiogram import types + +from .model import db, Users + + + +db.create_tables([Users]) + + +def register(user: types.User, data: dict): + if not Users.select().where(Users.user_id==user.id, Users.tag == data['tag']).exists(): + Users.create( + user_id = user.id, + first_name = user.first_name, + last_name = user.last_name, + username = user.username, + + tag = data['tag'], + nickname = data['name'], + townhall = data['townHallLevel'], + attackwins = data['attackWins'] + ) + else: + Users.update( + user_id = user.id, + first_name = user.first_name, + last_name = user.last_name, + username = user.username, + + tag = data['tag'], + nickname = data['name'], + townhall = data['townHallLevel'], + attackwins = data['attackWins'] + ).where(Users.user_id == user.id, Users.tag == data['tag']).execute() + +def delete(user_id: int): + Users.delete().where(Users.user_id == user_id).execute() + +def check_register(user_id: int): + return Users.select().where(Users.user_id==user_id).exists() + +def get_users(): + users = [] + for p in Users.select(): + users.append( + { + 'user_id': p.user_id, + 'first_name': p.first_name, + 'last_name': p.last_name, + 'username': p.username, + 'tag': p.tag, + 'nickname': p.nickname, + 'townhall': p.townhall, + 'attackwins': p.attackwins + } + ) + return users diff --git a/example_config.ini b/example_config.ini new file mode 100644 index 0000000..082964d --- /dev/null +++ b/example_config.ini @@ -0,0 +1,26 @@ +[Bot] +token = 123456789:ABCDEABCDEABCD +;; Token from @botfather +base_server = https://api.telegram.org +chat_id = -1001740735953 +verify_timeout = 120 + + +[WebHook] +use = false +webhook_host = http://127.0.0.1:3001 +webhook_path = /bot + +app_host = 127.0.0.1 +app_port = 3001 + + +[DataBase] +link = sqlite:///database.sqlite3 +;; http://docs.peewee-orm.com/en/latest/peewee/playhouse.html#database-url + +[API] +api_file = key.txt +;; file with clash of clans API Token +;; https://developer.clashofclans.com +clan_tag = #QWERTY1 \ No newline at end of file diff --git a/filters/__init__.py b/filters/__init__.py new file mode 100644 index 0000000..257a690 --- /dev/null +++ b/filters/__init__.py @@ -0,0 +1,5 @@ +from load import dp + +from .main import ChatFilter + +dp.filters_factory.bind(ChatFilter) \ No newline at end of file diff --git a/filters/main.py b/filters/main.py new file mode 100644 index 0000000..7c5978e --- /dev/null +++ b/filters/main.py @@ -0,0 +1,15 @@ +from aiogram.dispatcher.filters import BoundFilter +from aiogram import types + +from load import config + + +class ChatFilter(BoundFilter): + key = 'is_chat' + + def __init__(self, is_chat): + self.is_chat = is_chat + + async def check(self, message: types.Message): + if message.chat.id == int(config["Bot"]["chat_id"]): return True + return False \ No newline at end of file diff --git a/fsm/base.py b/fsm/base.py new file mode 100644 index 0000000..6ceefa9 --- /dev/null +++ b/fsm/base.py @@ -0,0 +1,5 @@ +from aiogram.dispatcher.filters.state import State, StatesGroup + +class Verifycation(StatesGroup): + tag = State() + token = State() diff --git a/handlers/__init__.py b/handlers/__init__.py new file mode 100644 index 0000000..8be81f1 --- /dev/null +++ b/handlers/__init__.py @@ -0,0 +1,2 @@ +from . import group +from . import private \ No newline at end of file diff --git a/handlers/group/__init__.py b/handlers/group/__init__.py new file mode 100644 index 0000000..2d3973d --- /dev/null +++ b/handlers/group/__init__.py @@ -0,0 +1,3 @@ +from . import main +from . import users +from . import captcha \ No newline at end of file diff --git a/handlers/group/captcha.py b/handlers/group/captcha.py new file mode 100644 index 0000000..ef10c6d --- /dev/null +++ b/handlers/group/captcha.py @@ -0,0 +1,52 @@ +from aiogram import types +from aiogram.types.chat_permissions import ChatPermissions + +from load import dp, bot, config +from keyboards.inline.keyboard import verify_button +from utils.timer import timer_manager, KickTimer + + + +async def captcha_kick_user(message: types.Message, chat_id: int, user_id:int): + print("asdasdasd") + try: + await bot.ban_chat_member(chat_id=chat_id, user_id=user_id) + except Exception: pass + await bot.unban_chat_member(chat_id, user_id, only_if_banned=True) + await message.delete() + try: + await bot.send_message(user_id, "Время вышло!") + except Exception: pass + state = dp.current_state(chat=user_id, user=user_id) + await state.finish() + + +@dp.message_handler(is_chat=True, content_types=["new_chat_members"]) +async def on_join(message: types.Message): + user_id = message.from_user.id + chat_id = message.chat.id + + try: + await bot.restrict_chat_member( + chat_id=chat_id, + user_id=user_id, + permissions=ChatPermissions(can_send_messages=False) + ) + except Exception: return + + hello_message = ( + "Привет, {name}! " + "Для того чтобы удостоверится что ты наш игрок, мы попросим пройти верификацию!\n" + "У вас есть 2 минуты на прохождение верификации!" + ) + + me = await bot.get_me() + captcha_message = await bot.send_message( + message.chat.id, + hello_message.format(user_id=user_id, name=message.from_user.first_name), + reply_markup=verify_button(message.from_user, me.username) + ) + timer_manager.add_timer(user_id, captcha_message, KickTimer(int(config['Bot']['verify_timeout']), captcha_kick_user, + args=[captcha_message, chat_id, user_id])) + #Timer(int(config['Bot']['verify_timeout']), captcha_kick_user, args=[captcha_message, chat_id, user_id]) + await message.delete() diff --git a/handlers/group/main.py b/handlers/group/main.py new file mode 100644 index 0000000..4dba582 --- /dev/null +++ b/handlers/group/main.py @@ -0,0 +1,12 @@ +from aiogram import types + +from load import dp, bot, config +from database.worker import delete + + +@dp.message_handler(is_chat=True, content_types=['left_chat_member']) +async def on_left(message: types.Message): + user_id = message.from_user.id + delete(user_id) + await message.delete() + diff --git a/handlers/group/users.py b/handlers/group/users.py new file mode 100644 index 0000000..600e1a9 --- /dev/null +++ b/handlers/group/users.py @@ -0,0 +1,18 @@ +from aiogram import types + +from load import bot, dp +from database.worker import get_users + + +@dp.message_handler(is_chat=True, commands=['get']) +async def get_users_info(message: types.Message): + text = '' + users = get_users() + if users is not None: + for p in users: + text += ("{first_name} ({nickname} {tag})\n").format( + user_id=p['user_id'], first_name=p['first_name'], nickname=p['nickname'], tag=p['tag'] + ) + await bot.send_message(message.chat.id, text) + else: + await bot.send_message(message.chat.id, "Ничего не найдено!") diff --git a/handlers/private/__init__.py b/handlers/private/__init__.py new file mode 100644 index 0000000..502eb5e --- /dev/null +++ b/handlers/private/__init__.py @@ -0,0 +1 @@ +from . import register \ No newline at end of file diff --git a/handlers/private/register.py b/handlers/private/register.py new file mode 100644 index 0000000..d494bee --- /dev/null +++ b/handlers/private/register.py @@ -0,0 +1,114 @@ +from aiogram import types +from aiogram.types.chat_permissions import ChatPermissions +from aiogram.dispatcher import FSMContext +from aiogram.dispatcher.filters import ChatTypeFilter + +from load import dp, bot, coc_api, config +from fsm.base import Verifycation +from database.worker import register, check_register +from utils.timer import timer_manager + + +@dp.message_handler(ChatTypeFilter(['private']), commands=['start'], state='*') +async def verify(message: types.Message): + + status = await bot.get_chat_member(config['Bot']["chat_id"], message.from_user.id) + + if status["status"] == "left": + await bot.send_message(message.chat.id, "Пошёл нахуй! В чат заходить не учили?") + return + if check_register(message.from_user.id): + await bot.send_message(message.chat.id, "Вы уже прошли верификацию!") + return + + state = dp.current_state(chat=message.chat.id, user=message.from_user.id) + await state.finish() + + await bot.send_message(message.chat.id, "Отправь мне свой тег") + await Verifycation.tag.set() + +@dp.message_handler(state=Verifycation.tag) +async def tag_set(message: types.Message, state: FSMContext): + await state.update_data(tag=message.text) + + await bot.send_message( + message.chat.id, + ("Отправь мне свой токен авторизации\n" + "! Владелец или кто-либо не получает доступа к вашему акаунту !\n" + "" + "Если хотите убедиться в этом, можете посмотреть официальную API, которую использует данный бот") + ) + await Verifycation.token.set() + +@dp.message_handler(state=Verifycation.token) +async def token_set(message: types.Message, state: FSMContext): + chat_id = config['Bot']["chat_id"] + + await state.update_data(token=message.text) + info = await state.get_data() + await state.finish() + + data = coc_api.verify_token(**info) + #{'tag': '#TAG', 'token': 'TOKEN', 'status': 'ok'} + + members = coc_api.clan_members() + + clan_member = False + for i in members['items']: + if data['tag'] in i['tag']: + clan_member = True + break + + if clan_member == False: + await bot.send_message( + message.chat.id, + ("Вы не являетесь учасником нашего клана! Ввойдите в клан, чтобы иметь доступ к чату\n" + "Вы были кикнуты с чата, вы можете перезайти и пройти проверку повторно!") + ) + + try: + await bot.ban_chat_member(chat_id, message.from_user.id) + except Exception: pass + await bot.unban_chat_member(chat_id, message.from_user.id, only_if_banned=True) + return + + if (data['status'] == 'ok'): + await bot.send_message(message.chat.id, "Вы прошли проверку!") + register(message.from_user, coc_api.get_player(data['tag'])) + + + group = await bot.get_chat(chat_id) + group_permissions = group["permissions"] + permissions = ChatPermissions( + can_send_messages= group_permissions["can_send_messages"], + can_send_media_messages= group_permissions["can_send_media_messages"], + can_send_polls= group_permissions["can_send_polls"], + can_send_other_messages= group_permissions["can_send_other_messages"], + can_add_web_page_previews= group_permissions["can_add_web_page_previews"], + can_change_info= group_permissions["can_change_info"], + can_invite_users= group_permissions["can_invite_users"], + can_pin_messages= group_permissions["can_pin_messages"] + ) + try: + await bot.restrict_chat_member( + chat_id=chat_id, + user_id=message.from_user.id, + permissions=permissions + ) + except Exception: pass + + else: + await bot.send_message( + message.chat.id, + ("Вы ввели неверный токен или тег!\n" + "Вы были кикнуты с чата, вы можете перезайти и пройти проверку повторно!") + ) + + try: + await bot.ban_chat_member(chat_id, message.from_user.id) + except Exception: pass + await bot.unban_chat_member(chat_id, message.from_user.id, only_if_banned=True) + + m = timer_manager.cancel_timer(message.from_user.id) + if m is not None: + await m.delete() diff --git a/keyboards/__init__.py b/keyboards/__init__.py new file mode 100644 index 0000000..677f7cb --- /dev/null +++ b/keyboards/__init__.py @@ -0,0 +1 @@ +from . import inline \ No newline at end of file diff --git a/keyboards/inline/__init__.py b/keyboards/inline/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/keyboards/inline/keyboard.py b/keyboards/inline/keyboard.py new file mode 100644 index 0000000..27cf4a4 --- /dev/null +++ b/keyboards/inline/keyboard.py @@ -0,0 +1,18 @@ +from aiogram import types +from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup + + +def verify_button(user: types.User, bot_username: str) -> InlineKeyboardMarkup: + return InlineKeyboardMarkup( + inline_keyboard=[ + [ + InlineKeyboardButton( + "Подтвердить", + url="https://t.me/{bot}?start={user_id}".format( + bot=bot_username, + user_id=user.id + ) + ) + ] + ] + ) diff --git a/load.py b/load.py new file mode 100644 index 0000000..48b88a6 --- /dev/null +++ b/load.py @@ -0,0 +1,26 @@ +from aiogram import Dispatcher, Bot +from aiogram.bot.api import TelegramAPIServer +from aiogram.contrib.fsm_storage.memory import MemoryStorage + +from playhouse.db_url import connect + +from config import config +from coc import ClashOfClans + + +bot = Bot( + token=config["Bot"]["token"], + server=TelegramAPIServer.from_base(config["Bot"]["base_server"]), + parse_mode='HTML' +) +storage = MemoryStorage() +dp = Dispatcher(bot=bot, storage=storage) + + +db = connect(url=config["DataBase"]["link"]) + + +coc_api = ClashOfClans( + api_token=open(config["API"]["api_file"], "r").read(), + clan_tag=config["API"]["clan_tag"] +) \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..7625492 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,3 @@ +aiogram==2.19 +requests==2.27.1 +peewee==3.14.10 \ No newline at end of file diff --git a/utils/timer.py b/utils/timer.py new file mode 100644 index 0000000..30de344 --- /dev/null +++ b/utils/timer.py @@ -0,0 +1,60 @@ +import asyncio + + +class KickTimer: + def __init__(self, timeout, callback, args): + self._timeout = timeout + self._callback = callback + self._args = args + self._task = None + + async def _job(self): + await asyncio.sleep(self._timeout) + await self._callback(*self._args) + await asyncio.sleep(5) + self._task.cancel() + + async def _run_now(self): + await self._callback(*self._args) + await asyncio.sleep(5) + self._task.cancel() + + def start(self): + self._task = asyncio.ensure_future(self._job()) + + def execute_now(self): + self._task = asyncio.ensure_future(self._run_now()) + + def cancel(self): + self._task.cancel() + + + +class TimerManager: + def __init__(self): + self.timers = {} + + def add_timer(self, user_id: int, message, timer: KickTimer): + if user_id not in self.timers: + timer.start() + self.timers[user_id] = [timer, message] + + def cancel_timer(self, user_id): + if user_id in self.timers: + self.timers[user_id][0].cancel() + message = self.timers[user_id][1] + del self.timers[user_id] + print(f"Timer for user {user_id} canceled") + + return message + + def run_now(self, user_id): + if user_id in self.timers: + self.timers[user_id][0].execute_now() + message = self.timers[user_id][1] + del self.timers[user_id] + return message + + +timer_manager = TimerManager() +