from fastapi import APIRouter, Depends, Query, HTTPException
from sqlalchemy.orm import Session
from pydantic import BaseModel
from typing import Optional
from datetime import datetime
from app.core.database import get_db
from app.core.auth import get_current_user
from app.models.seek_info import SeekInfo
from app.models.models import User, Collection, Information
from sqlalchemy import text

router = APIRouter(prefix="/api/seek", tags=["寻配号"])

# ============ Schema ============
class SeekInfoCreate(BaseModel):
    title: str
    content: Optional[str] = None
    expect_category: Optional[str] = None
    expect_version: Optional[str] = None
    expect_packaging: Optional[str] = None
    expect_number: Optional[str] = None
    expect_price_min: Optional[float] = None
    expect_price_max: Optional[float] = None

class SeekInfoUpdate(BaseModel):
    title: Optional[str] = None
    content: Optional[str] = None
    expect_category: Optional[str] = None
    expect_version: Optional[str] = None
    expect_packaging: Optional[str] = None
    expect_number: Optional[str] = None
    expect_price_min: Optional[float] = None
    expect_price_max: Optional[float] = None
    status: Optional[str] = None

class SeekInfoResponse(BaseModel):
    id: str
    user_id: str
    title: str
    content: Optional[str]
    expect_category: Optional[str]
    expect_version: Optional[str]
    expect_packaging: Optional[str]
    expect_number: Optional[str]
    expect_price_min: Optional[float]
    expect_price_max: Optional[float]
    status: str
    is_matched: Optional[str]
    matched_user_id: Optional[str]
    matched_contact: Optional[str]
    view_count: int
    contact_count: int
    created_at: Optional[datetime]
    updated_at: Optional[datetime]
    user_name: Optional[str] = None  # 发布者用户名
    matched_count: Optional[int] = 0  # 自有匹配数量
    network_matched_count: Optional[int] = 0  # 网络匹配数量

    class Config:
        from_attributes = True

# ============ API ============
@router.get("/list", response_model=list[SeekInfoResponse])
def get_seek_list(
    status: str = Query("active"),
    page: int = Query(1, ge=1),
    page_size: int = Query(20, ge=1, le=1000),
    user_only: bool = Query(False),
    current_user: Optional = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """获取寻配号列表"""
    query = db.query(SeekInfo).filter(SeekInfo.status == status)
    
    # 我的寻配号：只查看自己的
    if user_only and current_user:
        query = query.filter(SeekInfo.user_id == current_user.f99_90_id)
    
    # 排序
    query = query.order_by(SeekInfo.created_at.desc())
    
    # 分页
    offset = (page - 1) * page_size
    items = query.offset(offset).limit(page_size).all()
    
    # 添加用户名和匹配数量
    from app.routers.information import match_collections_count, match_collections_count_from_coolbot
    result = []
    for item in items:
        user = db.query(User).filter(User.f99_90_id == item.user_id).first()
        user_name = user.f01_01_name if user else '匿名用户'
        
        # 计算匹配数量
        matched_count = 0
        network_matched_count = 0
        if item.expect_number and len(item.expect_number) == 10:
            if current_user and current_user.f99_90_id:
                matched_count = match_collections_count(db, current_user.f99_90_id, item.expect_number)
            network_matched_count = match_collections_count_from_coolbot(item.expect_number)
        
        # 构建响应
        result.append(SeekInfoResponse(
            id=item.id,
            user_id=item.user_id,
            title=item.title,
            content=item.content,
            expect_category=item.expect_category,
            expect_version=item.expect_version,
            expect_packaging=item.expect_packaging,
            expect_number=item.expect_number,
            expect_price_min=item.expect_price_min,
            expect_price_max=item.expect_price_max,
            status=item.status,
            is_matched=item.is_matched,
            matched_user_id=item.matched_user_id,
            matched_contact=item.matched_contact,
            view_count=item.view_count,
            contact_count=item.contact_count,
            created_at=item.created_at,
            updated_at=item.updated_at,
            user_name=user_name,
            matched_count=matched_count,
            network_matched_count=network_matched_count
        ))
    
    return result

@router.get("/stats")
def get_seek_stats(
    current_user = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """获取寻配号统计"""
    total = db.query(SeekInfo).filter(SeekInfo.status == "active").count()
    matched = db.query(SeekInfo).filter(SeekInfo.is_matched == "true").count()
    
    return {
        "total": total,
        "matched": matched,
        "unmatched": total - matched
    }

@router.post("", response_model=SeekInfoResponse)
def create_seek(
    data: SeekInfoCreate,
    current_user = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """创建寻配号"""
    if not current_user:
        raise HTTPException(status_code=401, detail="请先登录")
    
    seek = SeekInfo(
        user_id=current_user.f99_90_id,
        title=data.title,
        content=data.content,
        expect_category=data.expect_category,
        expect_version=data.expect_version,
        expect_packaging=data.expect_packaging,
        expect_number=data.expect_number,
        expect_price_min=data.expect_price_min,
        expect_price_max=data.expect_price_max,
        status="active"
    )
    db.add(seek)
    db.commit()
    db.refresh(seek)
    return seek

@router.get("/{seek_id}", response_model=SeekInfoResponse)
def get_seek(
    seek_id: str,
    current_user = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """获取寻配号详情"""
    seek = db.query(SeekInfo).filter(SeekInfo.id == seek_id).first()
    if not seek:
        raise HTTPException(status_code=404, detail="寻配号不存在")
    
    # 增加浏览数
    seek.view_count += 1
    db.commit()
    
    return seek

@router.put("/{seek_id}", response_model=SeekInfoResponse)
def update_seek(
    seek_id: str,
    data: SeekInfoUpdate,
    current_user = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """更新寻配号"""
    if not current_user:
        raise HTTPException(status_code=401, detail="请先登录")
    
    seek = db.query(SeekInfo).filter(
        SeekInfo.id == seek_id,
        SeekInfo.user_id == current_user.f99_90_id
    ).first()
    
    if not seek:
        raise HTTPException(status_code=404, detail="寻配号不存在")
    
    for key, value in data.model_dump(exclude_unset=True).items():
        setattr(seek, key, value)
    
    db.commit()
    db.refresh(seek)
    return seek

@router.delete("/{seek_id}")
def delete_seek(
    seek_id: str,
    current_user = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """删除寻配号"""
    if not current_user:
        raise HTTPException(status_code=401, detail="请先登录")
    
    seek = db.query(SeekInfo).filter(
        SeekInfo.id == seek_id,
        SeekInfo.user_id == current_user.f99_90_id
    ).first()
    
    if not seek:
        raise HTTPException(status_code=404, detail="寻配号不存在")
    
    seek.status = "deleted"
    db.commit()
    
    return {"message": "删除成功"}
# 获取自有藏品匹配列表
@router.get("/my-match")
def get_seek_match(
    info_id: str,
    current_user = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """获取符合条件的我的藏品推荐"""
    if not current_user:
        raise HTTPException(status_code=401, detail="请先登录")
    
    info = db.query(SeekInfo).filter(SeekInfo.id == info_id).first()
    if not info:
        raise HTTPException(status_code=404, detail="寻配号信息不存在")
    
    # 获取用户所有藏品
    collections = db.query(Collection).filter(
        Collection.f99_91_user_id == current_user.f99_90_id,
        Collection.f01_04_status == "in_collection"
    ).all()
    
    # 按号码特征模式匹配
    from app.routers.information import match_pattern
    matched = []
    if info.expect_number and len(info.expect_number) == 10:
        pattern = info.expect_number[2:]
        for c in collections:
            number = c.f02_10_prefix_serial or ''
            if len(number) >= 10 and number.startswith('J0'):
                col_pattern = number[2:10]
                if match_pattern(col_pattern, pattern):
                    matched.append(c)
            elif len(number) >= 8:
                col_pattern = number[:8]
                if match_pattern(col_pattern, pattern):
                    matched.append(c)
    else:
        matched = collections
    
    return {
        "matched_count": len(matched),
        "collections": [
            {"id": c.f99_90_id, "code": c.f01_02_code or '', "name": c.f01_01_name, 
             "number": c.f02_10_prefix_serial, "status": c.f01_04_status}
            for c in matched[:20]
        ]
    }

# 获取网络数据匹配列表
@router.get("/network-match/{info_id}")
def get_network_match(
    info_id: str,
    limit: int = Query(20, ge=1, le=100),
    db: Session = Depends(get_db)
):
    """获取一尘数据库中匹配的藏品列表"""
    from app.routers.information import match_collections_list_from_coolbot
    
    info = db.query(SeekInfo).filter(SeekInfo.id == info_id).first()
    if not info:
        raise HTTPException(status_code=404, detail="寻配号信息不存在")
    
    if not info.expect_number:
        return {"matched_count": 0, "collections": []}
    
    matched = match_collections_list_from_coolbot(info.expect_number, limit=limit)
    return {"matched_count": len(matched), "collections": matched}