from nonebot import on_command, get_plugin_config,logger
from nonebot.permission import SUPERUSER
from nonebot.rule import to_me
from nonebot.adapters.onebot.v11 import PrivateMessageEvent, GroupMessageEvent, MessageSegment
from nonebot.params import Depends
from .config import Config
from .utils import post, get_classes, post_vcode, get_log
import random
import asyncio
import time

plugin_config = get_plugin_config(Config)


help = on_command("咸鸭蛋",rule=to_me(),aliases={"apihelp", "sudhelp"},permission=SUPERUSER, priority=0, block=True)
@help.handle()
async def _():
    await asyncio.sleep(random.uniform(2, 3))
    await help.finish(plugin_config.HelpStr)

ddonline = on_command("在线人数",rule=to_me(),aliases={"ddonline", "ddop"}, priority=0, block=True)
@ddonline.handle()
async def _(event:PrivateMessageEvent):
    id:str = str(event.user_id)
    msg:str = await post("在线人数",id)
    await asyncio.sleep(random.uniform(2, 3))
    await ddonline.finish(msg)

addkami = on_command("添加卡密",rule=to_me(),aliases={"addkami", "akm"},permission=SUPERUSER, priority=0, block=True)
@addkami.handle()
async def _(event:PrivateMessageEvent):
    id:str = str(event.user_id) 
    msg:str = event.get_plaintext()
    if len(msg.split(' ')) != 3:
        await asyncio.sleep(random.uniform(2, 3))
        await ddonline.finish("参数不正确！")

    classes:str = msg.split(' ')[1]
    classes = get_classes(classes)
    if classes == '':
        await ddonline.finish("卡密类型不正确！")

    kami:str = msg.split(' ')[2]
    msg:str = await post("添加卡密",id,{
        "classes":classes,
        "kami":kami
    })
    await asyncio.sleep(random.uniform(2, 3))
    await ddonline.finish(msg)

createkami = on_command("生成卡密",rule=to_me(),aliases={"createkami", "ckm"},permission=SUPERUSER, priority=0, block=True)
@createkami.handle()
async def _(event:PrivateMessageEvent):
    id:str = str(event.user_id) 
    msg:str = event.get_plaintext()
    if len(msg.split(' ')) != 2:
        await asyncio.sleep(random.uniform(2, 3))
        await ddonline.finish("参数不正确！")

    classes:str = msg.split(' ')[1]
    classes = get_classes(classes)
    if classes == '':
        await ddonline.finish("卡密类型不正确！")

    msg:str = await post("生成卡密",id,{
        "classes":classes
    })
    await asyncio.sleep(random.uniform(2, 3))
    await ddonline.finish(msg)

addviptime = on_command("用户加时",rule=to_me(),aliases={"addviptime", "avt"},permission=SUPERUSER, priority=0, block=True)
@addviptime.handle()
async def _(event:PrivateMessageEvent):
    id:str = str(event.user_id) 
    msg:str = event.get_plaintext()
    if len(msg.split(' ')) != 3:
        await asyncio.sleep(random.uniform(2, 3))
        await ddonline.finish("参数不正确！")
        
    username:str = msg.split(' ')[1]
    classes:str = msg.split(' ')[2]
    classes = get_classes(classes)
    if classes == '':
        await ddonline.finish("卡密类型不正确！")

    msg:str = await post("用户加时",id,{
        "username":username,
        "classes":classes
    })
    await asyncio.sleep(random.uniform(2, 3))
    await ddonline.finish(msg)

generate_qq_vcode = on_command("绑定QQ",aliases={"bindqq", "绑定qq"}, priority=0, block=True)

# 添加用户使用时间记录字典
user_last_use_time = {}

@generate_qq_vcode.handle()
async def _(event: GroupMessageEvent): # GroupMessageEvent PrivateMessageEvent
    # 检查是否来自指定群组
    if event.group_id != 621016172:
        return
    # if event.user_id != 1424473282:
    #     return
        
    id:str = str(event.user_id)
    
    # 限流检查：检查用户上次使用时间
    current_time = time.time()
    if id in user_last_use_time:
        time_diff = current_time - user_last_use_time[id]
        if time_diff < 60:  # 60秒内已使用过
            await generate_qq_vcode.finish(f"请求过于频繁，请在{int(60 - time_diff)}秒后再试")
            return
    
    # 更新用户最后使用时间
    user_last_use_time[id] = current_time
    
    msg:str = await post_vcode(id)
    await asyncio.sleep(random.uniform(2, 3))
    # 在消息前添加@用户
    at_user = MessageSegment.at(event.user_id)
    await generate_qq_vcode.finish(at_user + " " + msg)




view_logs = on_command("查看日志",aliases={"logs", "查询日志"}, priority=0, block=True)
@view_logs.handle()
async def _(event:GroupMessageEvent): # GroupMessageEvent PrivateMessageEvent
    # 检查是否来自指定群组
    if event.group_id != 621016172:
        return
    # if event.user_id != 1424473282:
    #     return
        
    id:str = str(event.user_id) 
    msg:str = await get_log(id)
    await asyncio.sleep(random.uniform(2, 3))
    # 在消息前添加@用户
    at_user = MessageSegment.at(event.user_id)
    await view_logs.finish(at_user + " " + msg)