jiachenlong/backend/app/routers/seek.py

# seek - 寻配号路由
# Version: 0.0.9 (2026-04-24)
# 稳定版：自有匹配+网络匹配缓存

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
from app.routers.information import match_collections_count, match_collections_count_from_coolbot
from app.utils.coolbot_matcher import (
    match_pattern,
    match_self_collections_count,
    match_collections_count_from_coolbot,
    match_collections_list_from_coolbot
)

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

class MatchConfirmRequest(BaseModel):
    info_id: str
    contact: Optional[str] = None
    collection_id: Optional[str] = None

# ============ 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()
    
    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 '匿名用户'
        
        # 构建响应（不实时计算网络匹配，网络匹配在点击时再查）
        # 网络匹配数量从数据库读取（不再实时计算，初始为0）
        network_matched_count = 0
        if hasattr(item, 'network_matched_count') and item.network_matched_count:
            network_matched_count = item.network_matched_count
        
        # 自有匹配数量（使用缓存方式，如为0则计算一次）
        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)
        
        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="请先登录")
    
    # 计算网络匹配数量（创建时初始化）
    network_matched_count = 0
    if data.expect_number and len(data.expect_number) == 10:
        network_matched_count = match_collections_count_from_coolbot(data.expect_number)
    
    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",
        view_count=0,
        contact_count=0,
        network_matched_count=network_matched_count  # 保存初始化值
    )
    db.add(seek)
    db.commit()
    db.refresh(seek)
    return seek

# ============ 匹配相关API ============

@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()
    
    # 按号码特征模式匹配
    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(100, ge=1, le=500),
    db: Session = Depends(get_db)
):
    """获取一尘数据库中匹配的藏品列表，点击时更新network_matched_count"""
    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)
    actual_count = len(matched)
    
    # 点击查看时更新network_matched_count
    info.network_matched_count = actual_count
    db.commit()
    
    return {"matched_count": actual_count, "collections": matched}

@router.post("/match-confirm")
def match_seek_confirm(
    request: MatchConfirmRequest,
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """确认匹配寻号 - 用户愿意交换联系方式给发布者"""
    info = db.query(SeekInfo).filter(
        SeekInfo.id == request.info_id,
        SeekInfo.status == "active"
    ).first()
    
    if not info:
        raise HTTPException(status_code=404, detail="寻配号信息不存在")
    
    # 检查是否已被匹配（一个寻号只能被一个用户匹配）
    if info.is_matched == "matched":
        raise HTTPException(status_code=400, detail="该寻号已被其他用户匹配")
    
    # 检查是否是自己发布的
    if info.user_id == current_user.f99_90_id:
        raise HTTPException(status_code=400, detail="不能匹配自己发布的寻号")
    
    # 更新匹配状态
    info.is_matched = "matched"
    info.matched_user_id = current_user.f99_90_id
    info.matched_contact = request.contact or ''
    
    # 更新发布寻号者的内容，显示有藏品被匹配
    original_content = info.content or ""
    match_info = f"\n藏品被{current_user.f01_01_name}匹配成功！联系方式：{request.collection_id or '已交换'}"
    info.content = original_content + match_info
    
    db.commit()
    
    return {"message": "匹配成功，已通知发布者", "is_matched": "matched"}

@router.get("/matched-user/{info_id}")
def get_matched_user(
    info_id: str,
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """获取寻号的匹配者信息（仅发布者可见）"""
    info = db.query(SeekInfo).filter(SeekInfo.id == info_id).first()
    
    if not info:
        raise HTTPException(status_code=404, detail="寻配号信息不存在")
    
    # 只有发布者可以看到匹配者信息
    if info.user_id != current_user.f99_90_id:
        raise HTTPException(status_code=403, detail="无权访问")
    
    if not info.matched_user_id:
        return {"message": "暂无匹配者"}
    
    # 获取匹配者信息
    matched_user = db.query(User).filter(User.f99_90_id == info.matched_user_id).first()
    if not matched_user:
        return {"message": "匹配者不存在"}
    
    return {
        "matched_user_id": info.matched_user_id,
        "user_name": matched_user.f01_01_name,
        "phone": matched_user.phone,
        "matched_contact": info.matched_contact,
    }

@router.get("/publisher/{info_id}")
def get_publisher_info(
    info_id: str,
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """获取寻号的发布者信息（仅匹配者可见）"""
    info = db.query(SeekInfo).filter(SeekInfo.id == info_id).first()
    
    if not info:
        raise HTTPException(status_code=404, detail="寻配号信息不存在")
    
    # 只有匹配者可以看到发布者信息
    if not info.matched_user_id or info.matched_user_id != current_user.f99_90_id:
        raise HTTPException(status_code=403, detail="无权访问")
    
    # 获取发布者信息
    publisher = db.query(User).filter(User.f99_90_id == info.user_id).first()
    if not publisher:
        return {"message": "发布者不存在"}
    
    # 从content中解析联系方式
    contact = ''
    if info.content:
        import re
        match = re.search(r'联系方式[:\s]*(.+?)(?:\n|$)', info.content)
        if match:
            contact = match.group(1).strip()
    
    return {
        "user_id": info.user_id,
        "user_name": publisher.f01_01_name,
        "phone": publisher.phone,
        "contact": contact,
    }


# ============ 通配路由（必须放最后，避免匹配冲突）============
@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).first()
    if not seek:
        raise HTTPException(status_code=404, detail="寻配号不存在")
    if seek.user_id != current_user.f99_90_id:
        raise HTTPException(status_code=403, detail="无权限")
    if data.title is not None:
        seek.title = data.title
    if data.content is not None:
        seek.content = data.content
    if data.expect_category is not None:
        seek.expect_category = data.expect_category
    if data.expect_version is not None:
        seek.expect_version = data.expect_version
    if data.expect_packaging is not None:
        seek.expect_packaging = data.expect_packaging
    if data.status is not None:
        seek.status = data.status
    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).first()
    if not seek:
        raise HTTPException(status_code=404, detail="寻配号不存在")
    if seek.user_id != current_user.f99_90_id:
        raise HTTPException(status_code=403, detail="无权限")
    seek.status = "deleted"
    db.commit()
    return {"message": "删除成功"}

# ============ 留言功能（从information.py迁移）============
from pydantic import BaseModel

class CommentRequest(BaseModel):
    information_id: str
    content: str

@router.post("/comment")
def add_comment(
    request: CommentRequest,
    current_user = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """添加留言 - 支持seek_info表"""
    if not current_user:
        raise HTTPException(status_code=401, detail="请先登录")
    
    # 查询seek_info表
    info = db.query(SeekInfo).filter(SeekInfo.id == request.information_id).first()
    if not info:
        raise HTTPException(status_code=404, detail="寻配号不存在")
    
    # 创建留言
    from app.models.models import InformationComment
    comment = InformationComment(
        information_id=request.information_id,
        user_id=current_user.f99_90_id,
        content=request.content
    )
    db.add(comment)
    db.commit()
    
    return {
        "message": "留言成功",
        "comment": {
            "id": comment.id,
            "content": comment.content,
            "user_name": current_user.f01_01_name,
            "user_avatar": current_user.avatar,
            "created_at": comment.created_at
        }
    }

@router.get("/comments/{information_id}")
def get_comments(
    information_id: str,
    db: Session = Depends(get_db)
):
    """获取寻配号的评论列表"""
    from app.models.models import InformationComment
    comments = db.query(InformationComment).filter(
        InformationComment.information_id == information_id
    ).order_by(InformationComment.created_at.desc()).all()
    
    return [
        {
            "id": c.id,
            "content": c.content,
            "user_name": c.user.f01_01_name if c.user else '匿名用户',
            "user_avatar": c.user.avatar if c.user else None,
            "created_at": c.created_at
        }
        for c in comments
    ]