from typing import Callable

from telegram import Update
from telegram.error import BadRequest, Forbidden
from telegram.ext import CallbackContext

from logger import Log
from plugins.base import BasePlugins
from service import BaseService


def bot_admins_only(func: Callable) -> Callable:  # noqa
    async def decorator(self, update: Update, context: CallbackContext):
        admin_list = await self.service.admin.get_admin_list()
        if update.message.from_user.id in admin_list:
            return await func(self, update, context)
        else:
            await update.message.reply_text("权限不足")
            return None

    return decorator


class Admin(BasePlugins):
    def __init__(self, service: BaseService):
        super().__init__(service)

    @bot_admins_only
    async def add_admin(self, update: Update, _: CallbackContext):
        message = update.message
        reply_to_message = message.reply_to_message
        admin_list = await self.service.admin.get_admin_list()
        if reply_to_message is None:
            await message.reply_text("请回复对应消息")
        else:
            if reply_to_message.from_user.id in admin_list:
                await message.reply_text("该用户已经存在管理员列表")
            else:
                await self.service.admin.add_admin(reply_to_message.from_user.id)
                await message.reply_text("添加成功")

    @bot_admins_only
    async def del_admin(self, update: Update, _: CallbackContext):
        message = update.message
        reply_to_message = message.reply_to_message
        admin_list = await self.service.admin.get_admin_list()
        if reply_to_message is None:
            await message.reply_text("请回复对应消息")
        else:
            if reply_to_message.from_user.id in admin_list:
                await self.service.admin.delete_admin(reply_to_message.from_user.id)
                await message.reply_text("删除成功")
            else:
                await message.reply_text("该用户不存在管理员列表")

    @bot_admins_only
    async def leave_chat(self, update: Update, context: CallbackContext):
        message = update.message
        try:
            args = message.text.split()
            if len(args) >= 2:
                char_id = int(args[1])
            else:
                await message.reply_text("输入错误")
                return
        except ValueError as error:
            Log.error("获取 char_id 发生错误！ 错误信息为 \n", error)
            await message.reply_text("输入错误")
            return
        try:
            try:
                char = await context.bot.get_chat(char_id)
                await message.reply_text(f"正在尝试退出群 {char.title}[{char.id}]")
            except (BadRequest, Forbidden):
                pass
            await context.bot.leave_chat(char_id)
        except (BadRequest, Forbidden) as error:
            Log.error(f"退出 char_id[{char_id}] 发生错误！ 错误信息为 \n", error)
            await message.reply_text(f"退出 char_id[{char_id}] 发生错误！ 错误信息为 {str(error)}")
            return
        await message.reply_text(f"退出 char_id[{char_id}] 成功！")