from fastapi import APIRouter, Depends
from app.models.database import get_db
from app.api.deps import get_admin_user
from app.models.user import UserDB, UserRole
from app.models.order import ShippingOrderDB, OrderStatus
from app.models.community import CommunityDB, CommunityStatus
from sqlalchemy.orm import Session
from datetime import datetime, timedelta
from app.core.response import success_response, error_response, ResponseModel
from typing import Optional
from app.models.user_auth import UserAuthDB
router = APIRouter()

# 获取基础信息
# 总用户数， 今日新增用户数， 总配送订单数， 今日新增配送订单数
# 小区数量， 今日新增小区数

@router.get("")
async def get_dashboard_info(
    db: Session = Depends(get_db)
):
    # 获取接口包括：
    # 总用户数， 今日新增用户数，昨日新增用户数， 总配送订单数， 今日新增配送订单数，昨日新增配送订单数
    # 小区数量， 今日新增小区数，昨日新增小区数

    # 小区数量
    total_community_count = db.query(CommunityDB).filter(CommunityDB.status == CommunityStatus.OPENING).count()

    # 总用户数
    total_user_count = db.query(UserDB).count()

    # 查询已下单用户数
    orders = db.query(ShippingOrderDB).filter(ShippingOrderDB.status.in_(['COMPLETED', 'UNPAID'])).all()
    has_order_user_count = len(set([order.userid for order in orders]))
    has_paid_user_count = len(set([order.userid for order in orders if order.status == OrderStatus.COMPLETED and order.final_amount > 0]))
    order_count = len(orders)
    order_pay_count = len([order for order in orders if order.status == OrderStatus.COMPLETED and order.final_amount > 0])
    pay_rate = order_pay_count / order_count if order_count > 0 else 0
    order_amount = sum([order.original_amount_with_additional_fee for order in orders])
    pay_amount = sum([order.final_amount for order in orders if order.status == OrderStatus.COMPLETED and order.final_amount > 0])


    return success_response(data={
        "total_community_count": total_community_count,
        "total_user_count": total_user_count,
        "has_order_user_count": has_order_user_count,
        "has_paid_user_count": has_paid_user_count,
        "order_count": order_count,
        "order_pay_count": order_pay_count,
        "pay_rate": pay_rate,
        "order_amount": order_amount,
        "pay_amount": pay_amount
    })

@router.get("/deliveryman")
async def get_deliveryman_dashboard_info(
    community_id: Optional[int] = None,
    db: Session = Depends(get_db)
):
    # 获取配送员信息
    query = db.query(UserDB, UserAuthDB.name.label('auth_name')).join(UserAuthDB, UserDB.userid == UserAuthDB.user_id).filter(UserDB.roles.contains(UserRole.DELIVERYMAN))
    if community_id:
        query = query.filter(UserDB.community_id == community_id)
    deliverymans = query.all()
    
    result = []
    for deliveryman in deliverymans:
        orders = db.query(ShippingOrderDB).filter(ShippingOrderDB.deliveryman_user_id == deliveryman.UserDB.userid, ShippingOrderDB.status.in_([OrderStatus.COMPLETED, OrderStatus.UNPAID])).all()
        today_orders = db.query(ShippingOrderDB).filter(ShippingOrderDB.deliveryman_user_id == deliveryman.UserDB.userid, ShippingOrderDB.status.in_([OrderStatus.COMPLETED, OrderStatus.UNPAID]), ShippingOrderDB.completed_time >= datetime.now().date()).all()
        result.append({
            "deliveryman_id": deliveryman.UserDB.userid,
            "deliveryman_name": deliveryman.UserDB.nickname,
            "deliveryman_auth_name": deliveryman.auth_name,
            "deliveryman_community_name": deliveryman.UserDB.community.name,
            "order_count": len(orders),
            "order_amount": sum([order.original_amount_with_additional_fee for order in orders]),
            "today_order_count": len(today_orders),
            "today_order_amount": sum([order.original_amount_with_additional_fee for order in today_orders])
        })

        # 按照今日订单量排序
        result.sort(key=lambda x: x["today_order_count"], reverse=True)
    return success_response(data=result)