from fastapi import APIRouter, Depends, HTTPException, Query
from sqlalchemy.orm import Session
from sqlalchemy import and_, desc, func
from typing import Optional, List
from app.core.response import success_response
from app.api.deps import get_db, get_current_user, get_admin_user
from app.models.merchant_complaint import MerchantComplaintDB, MerchantComplaintCreate, MerchantComplaintInfo, MerchantComplaintUpdate, ComplaintStatus
from app.models.user import UserDB
from app.models.merchant import MerchantDB
from datetime import datetime
from app.core.wecombot import WecomBot

router = APIRouter()

@router.get("/get_reason_list")
async def get_reason_list(
    db: Session = Depends(get_db)
):
    """
    获取投诉原因列表
    """
    # 获取投诉原因列表
    reason_list = [
        "商家不让使用快捷买单",
        "商家引导用户使用其他支付方式",
        "商家存在食品安全问题",
        "商家涉及违法违规行为"
    ]

    return success_response(data={"items": reason_list}, message="投诉原因列表获取成功")

@router.post("/")
async def create_complaint(
    complaint: MerchantComplaintCreate,
    db: Session = Depends(get_db),
    current_user: UserDB = Depends(get_current_user)
):
    """
    创建商家投诉
    """
    try:
        # 检查商家是否存在
        merchant = db.query(MerchantDB).filter(MerchantDB.id == complaint.merchant_id).first()
        if not merchant:
            raise HTTPException(status_code=404, detail="商家不存在")
    
        # 创建投诉记录
        db_complaint = MerchantComplaintDB(
            user_id=current_user.userid,  # 使用当前用户ID，而不是请求中的user_id
            merchant_id=complaint.merchant_id,
            content=complaint.content,
            image_urls=complaint.image_urls,
            status=ComplaintStatus.PENDING  # 默认状态为等待处理
        )
        
        db.add(db_complaint)
        db.commit()
        db.refresh(db_complaint)

        # 发送企业微信消息
        await WecomBot().send_merchant_complaint(db_complaint)

        return success_response(data={"id": db_complaint.id}, message="投诉提交成功")
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))

@router.get("/")
async def get_complaints(
    status: Optional[ComplaintStatus] = None,
    merchant_id: Optional[int] = None,
    start_time: Optional[datetime] = None,
    end_time: Optional[datetime] = None,
    skip: int = Query(0, ge=0),
    limit: int = Query(10, ge=1, le=100),
    db: Session = Depends(get_db),
    current_user: UserDB = Depends(get_current_user)
):
    """
    获取当前用户的投诉列表
    """
    # 构建查询条件
    conditions = [MerchantComplaintDB.user_id == current_user.userid]
    
    if status:
        conditions.append(MerchantComplaintDB.status == status)
    
    if merchant_id:
        conditions.append(MerchantComplaintDB.merchant_id == merchant_id)
    
    if start_time:
        conditions.append(MerchantComplaintDB.create_time >= start_time)
    
    if end_time:
        conditions.append(MerchantComplaintDB.create_time <= end_time)
    
    # 查询总数
    total = db.query(func.count(MerchantComplaintDB.id)).filter(and_(*conditions)).scalar()
    
    # 查询数据
    complaints = db.query(MerchantComplaintDB).filter(and_(*conditions)).order_by(
        desc(MerchantComplaintDB.create_time)
    ).offset(skip).limit(limit).all()
    
    # 获取商家名称
    result = []
    for complaint in complaints:        
        complaint_info = MerchantComplaintInfo.model_validate(complaint)
        complaint_info.merchant_name = complaint.merchant.name
        complaint_info.user_name = complaint.user.nickname
        
        result.append(complaint_info)
    
    return success_response(data={
        "total": total,
        "items": result
    })

@router.put("/{complaint_id}")
async def update_complaint_status(
    complaint_id: int,
    complaint_update: MerchantComplaintUpdate,
    db: Session = Depends(get_db),
    _: UserDB = Depends(get_admin_user)  # 确保只有管理员可以修改状态
):
    """
    更新投诉状态（仅管理员可用）
    """
    # 查询投诉记录
    complaint = db.query(MerchantComplaintDB).filter(MerchantComplaintDB.id == complaint_id).first()
    if not complaint:
        raise HTTPException(status_code=404, detail="投诉记录不存在")
    
    # 更新状态
    complaint.status = complaint_update.status
    db.commit()
    db.refresh(complaint)
    
    return success_response(message="投诉状态更新成功")