from nonebot import on_command
from nonebot.adapters.cqhttp import Bot, GroupMessageEvent
from nonebot.typing import T_State
from nonebot.adapters.cqhttp.permission import GROUP
from util.data_utils import init_rank
from models.bag_user import BagUser


my_gold = on_command("我的金币", priority=5, block=True, permission=GROUP)

gold_rank = on_command('金币排行', priority=5, block=True, permission=GROUP)


@my_gold.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
    await my_gold.finish(await BagUser.get_my_total_gold(event.user_id, event.group_id))


@gold_rank.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
    users = await BagUser.get_user_all(event.group_id)
    all_user_data = [user.gold for user in users]
    await gold_rank.finish(await init_rank(users, all_user_data, event.group_id))