"""
群组统计API接口
"""

from typing import Optional
from fastapi import APIRouter, HTTPException
from pydantic import BaseModel
from loguru import logger

from app.services.group_stats_service import group_stats_service


router = APIRouter()


class GroupStatsResponse(BaseModel):
    """群组统计响应模型"""
    success: bool
    message: str
    data: dict


class ClearStatsRequest(BaseModel):
    """清空统计请求模型"""
    group_id: str


@router.get("/group-stats/{group_id}", response_model=GroupStatsResponse)
async def get_group_stats(group_id: str):
    """
    获取指定群组的统计信息

    Args:
        group_id: 群组ID

    Returns:
        群组统计信息
    """
    try:
        stats_summary = group_stats_service.get_group_stats_summary(group_id)
        
        return GroupStatsResponse(
            success=True,
            message=f"获取群组统计信息成功: {group_id}",
            data=stats_summary
        )

    except Exception as e:
        logger.error(f"获取群组统计信息失败: group_id={group_id}, error={str(e)}")
        raise HTTPException(status_code=500, detail=f"获取群组统计信息失败: {str(e)}")


@router.get("/group-stats/{group_id}/most-active", response_model=GroupStatsResponse)
async def get_most_active_user(group_id: str):
    """
    获取指定群组中发言次数最多的用户昵称

    Args:
        group_id: 群组ID

    Returns:
        最活跃用户信息
    """
    try:
        most_active_nickname = group_stats_service.get_most_active_user_nickname(group_id)
        message_stats = group_stats_service.get_group_message_stats(group_id)
        
        # 找到最活跃用户的详细信息
        most_active_user_id = None
        max_count = 0
        
        for user_id, count in message_stats.items():
            if count > max_count:
                max_count = count
                most_active_user_id = user_id
        
        data = {
            "group_id": group_id,
            "most_active_nickname": most_active_nickname,
            "most_active_user_id": most_active_user_id,
            "message_count": max_count,
            "total_users": len([count for count in message_stats.values() if count > 0])
        }
        
        return GroupStatsResponse(
            success=True,
            message=f"获取最活跃用户成功: {group_id}",
            data=data
        )

    except Exception as e:
        logger.error(f"获取最活跃用户失败: group_id={group_id}, error={str(e)}")
        raise HTTPException(status_code=500, detail=f"获取最活跃用户失败: {str(e)}")


@router.post("/group-stats/clear", response_model=GroupStatsResponse)
async def clear_group_stats(request: ClearStatsRequest):
    """
    清空指定群组的统计数据

    Args:
        request: 清空统计请求

    Returns:
        操作结果
    """
    try:
        success = group_stats_service.clear_group_stats(request.group_id)
        
        if success:
            return GroupStatsResponse(
                success=True,
                message=f"群组统计数据已清空: {request.group_id}",
                data={"group_id": request.group_id, "cleared": True}
            )
        else:
            raise HTTPException(status_code=400, detail=f"清空群组统计数据失败: {request.group_id}")

    except HTTPException:
        raise
    except Exception as e:
        logger.error(f"清空群组统计数据失败: group_id={request.group_id}, error={str(e)}")
        raise HTTPException(status_code=500, detail=f"清空群组统计数据失败: {str(e)}")


@router.get("/group-stats", response_model=GroupStatsResponse)
async def get_all_groups_overview():
    """
    获取所有群组的统计概览

    Returns:
        所有群组的统计概览
    """
    try:
        # 这里可以扩展为获取所有群组的统计信息
        # 目前返回基本信息
        data = {
            "message": "群组统计功能已启用",
            "endpoints": {
                "get_group_stats": "/api/v1/group-stats/{group_id}",
                "get_most_active": "/api/v1/group-stats/{group_id}/most-active",
                "clear_stats": "/api/v1/group-stats/clear"
            }
        }
        
        return GroupStatsResponse(
            success=True,
            message="获取群组统计概览成功",
            data=data
        )

    except Exception as e:
        logger.error(f"获取群组统计概览失败: error={str(e)}")
        raise HTTPException(status_code=500, detail=f"获取群组统计概览失败: {str(e)}")