from datetime import datetime, timezone, timedelta

from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy import select
from sqlalchemy.exc import IntegrityError
from sqlalchemy.ext.asyncio import AsyncSession

from app.dependencies import get_db, get_current_user
from app.models.contact_request import ContactRequest
from app.models.user import User
from app.schemas.contact import ContactRequestCreate
from app.utils.id_gen import generate_id

router = APIRouter(prefix="/contact", tags=["contact"])


@router.post("/request", summary="提交联系请求", status_code=status.HTTP_201_CREATED)
async def create_contact_request(
    request: ContactRequestCreate,
    db: AsyncSession = Depends(get_db),
    user: User = Depends(get_current_user),
):
    today_str = datetime.now(timezone.utc).strftime("%Y-%m-%d")
    
    async with db.begin_nested():
        user_result = await db.execute(
            select(User).where(User.id == user.id).with_for_update().limit(1)
        )
        locked_user = user_result.scalar_one_or_none()
        if not locked_user:
            raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="用户不存在")
        
        existing = await db.execute(
            select(ContactRequest.id)
            .where(ContactRequest.user_id == user.id)
            .where(ContactRequest.submit_date == today_str)
            .limit(1)
        )
        if existing.scalar_one_or_none():
            raise HTTPException(
                status_code=status.HTTP_429_TOO_MANY_REQUESTS,
                detail="每个账号每天只能提交一次联系我们"
            )
    
    try:
        contact_request = ContactRequest(
            id=generate_id(),
            user_id=user.id,
            phone=request.phone,
            company_name=request.company_name,
            industry=request.industry,
            name=request.name,
            message=request.message,
            submit_date=today_str,
        )
        db.add(contact_request)
        await db.commit()
    except IntegrityError:
        await db.rollback()
        raise HTTPException(
            status_code=status.HTTP_429_TOO_MANY_REQUESTS,
            detail="每个账号每天只能提交一次联系我们"
        )
    
    return {"message": "提交成功，我们会尽快与您联系"}