from telegram import Update
from telegram.ext import CallbackContext, filters

from core.admin import BotAdminService
from core.plugin import Plugin, handler
from utils.log import logger


class BotJoiningGroupsVerification(Plugin):

    def __init__(self, bot_admin_service: BotAdminService = None):
        self.bot_admin_service = bot_admin_service

    @handler.message.new_chat_members(priority=1)
    async def new_member(self, update: Update, context: CallbackContext) -> None:
        message = update.effective_message
        chat = message.chat
        from_user = message.from_user
        quit_status = False
        if filters.ChatType.GROUPS.filter(message):
            for user in message.new_chat_members:
                if user.id == context.bot.id:
                    if from_user is not None:
                        logger.info(f"用户 {from_user.full_name}[{from_user.id}] 在群 {chat.title}[{chat.id}] 邀请BOT")
                        admin_list = await self.bot_admin_service.get_admin_list()
                        if from_user.id in admin_list:
                            await context.bot.send_message(message.chat_id,
                                                           '感谢邀请小派蒙到本群！请使用 /help 查看咱已经学会的功能。')
                        else:
                            quit_status = True
                    else:
                        logger.info(f"未知用户 在群 {chat.title}[{chat.id}] 邀请BOT")
                        quit_status = True
        if quit_status:
            logger.warning("不是管理员邀请！退出群聊。")
            await context.bot.send_message(message.chat_id, "派蒙不想进去！不是旅行者的邀请！")
            await context.bot.leave_chat(chat.id)