from .group_user_checkin import group_user_check_in, group_user_check, group_impression_rank, impression_rank
from nonebot.typing import T_State
from nonebot.adapters.cqhttp import Bot, GroupMessageEvent
from nonebot.adapters.cqhttp.permission import GROUP
from utils.init_result import image
from nonebot import on_command
from utils.utils import get_message_text
from pathlib import Path
from configs.path_config import DATA_PATH
try:
    import ujson as json
except ModuleNotFoundError:
    import json

__plugin_name__ = '签到'
__plugin_usage__ = (
    '用法：\n'
    '对我说 “签到” 来签到\n'
    '“我的签到” 来获取历史签到信息\n'
    '“好感度排行” 来查看当前好感度前十的伙伴\n'
    '/ 签到时有 3% 概率 * 2 /'
)

_file = Path(f'{DATA_PATH}/not_show_sign_rank_user.json')
try:
    data = json.load(open(_file, 'r', encoding='utf8'))
except (FileNotFoundError, ValueError, TypeError):
    data = {
        '0': []
    }


sign = on_command("签到", priority=5, permission=GROUP, block=True)
my_sign = on_command(cmd="我的签到", aliases={'好感度'}, priority=5, permission=GROUP, block=True)
sign_rank = on_command(cmd="积分排行", aliases={'好感度排行', '签到排行', '积分排行', '好感排行',
                                            '好感度排名，签到排名，积分排名'},
                       priority=5, permission=GROUP, block=True)
total_sign_rank = on_command('签到总排行', aliases={'好感度总排行', '好感度总榜', '签到总榜'}, priority=5, block=True)


@sign.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
    await sign.send(
        await group_user_check_in(event.user_id, event.group_id),
        at_sender=True,
    )


@my_sign.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
    await my_sign.send(
        await group_user_check(event.user_id, event.group_id),
        at_sender=True,
    )


@sign_rank.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
    await sign_rank.send(
        await group_impression_rank(event.group_id)
    )


@total_sign_rank.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
    msg = get_message_text(event.json())
    if not msg:
        await total_sign_rank.send('请稍等..正在整理数据...')
        await total_sign_rank.send(image(b64=await impression_rank(0, data)))
    elif msg in ['屏蔽我']:
        if event.user_id in data['0']:
            await total_sign_rank.finish('您已经在屏蔽名单中了，请勿重复添加！', at_sender=True)
        data['0'].append(event.user_id)
        await total_sign_rank.send('设置成功，您不会出现在签到总榜中！', at_sender=True)
    elif msg in ['显示我']:
        if event.user_id not in data['0']:
            await total_sign_rank.finish('您不在屏蔽名单中！', at_sender=True)
        data['0'].remove(event.user_id)
        await total_sign_rank.send('设置成功，签到总榜将会显示您的头像名称以及好感度！', at_sender=True)
    with open(_file, 'w', encoding='utf8') as f:
        json.dump(data, f, ensure_ascii=False, indent=4)