from nonebot import on_request, logger
from nonebot.adapters.cqhttp import Bot, Event
from nonebot.permission import SUPERUSER
from nonebot.typing import T_State
from nonebot.adapters.cqhttp.permission import PRIVATE_FRIEND
from nonebot.adapters.cqhttp.event import FriendRequestEvent, GroupRequestEvent, RequestEvent

friend_req = on_request(priority=5)

@friend_req.handle()
async def add_superuser(bot: Bot, event: RequestEvent, state: T_State):
    if str(event.user_id) in bot.config.superusers and event.request_type == 'private':
        await event.approve(bot)
        logger.info('add user {}'.format(event.user_id))
    elif event.sub_type == 'invite' and str(event.user_id) in bot.config.superusers and event.request_type == 'group':
        await event.approve(bot)
        logger.info('add group {}'.format(event.group_id))