from aiogram.types import Message
import json

from config import config


async def handle_stats_command(message: Message):
    """处理统计命令"""
    if not config.is_feature_enabled('stats', message.chat.id):
        return
    if message.chat.type not in ['group', 'supergroup']:
        await message.reply("此命令仅在群组中可用")
        return
    chat_id = str(message.chat.id)
    try:
        with open('message_stats.json', 'r', encoding='utf-8') as f:
            stats = json.load(f).get(chat_id)
    except (FileNotFoundError, json.JSONDecodeError):
        stats = {}.get(chat_id)

    print(stats)

    if not stats:
        await message.reply("暂无统计数据")
        return

    # 按消息数量排序用户
    sorted_users = sorted(
        stats['users'].items(),
        key=lambda x: x[1]['message_count'],
        reverse=True
    )

    # 构建统计消息
    text = f"📊 群组统计\n\n"
    text += f"总消息数: {stats['total_messages']}\n"
    text += f"活跃用户数: {len(stats['users'])}\n\n"
    text += "🏆 发言排行榜:\n"

    for i, (user_id, user_data) in enumerate(sorted_users[:10], 1):
        name = user_data['first_name'] or user_data['username'] or str(user_id)
        text += f"{i}. {name}: {user_data['message_count']} 条\n"

    await message.reply(text)