"""
客户管理路由模块

提供客户信息的管理功能，包括：
- 客户列表查询（支持分页、搜索）
- 创建新客户（自动生成客户编码）

路由前缀: /api/customers
"""
from fastapi import APIRouter, Depends, Query
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select
from typing import Optional, List
from datetime import datetime

from database.database import get_db_session
from services.auth_service import get_current_active_user
from models.database import User, Customer
from .schemas import CustomerCreate, CustomerResponse

router = APIRouter(prefix="/customers", tags=["客户管理"])


@router.get("", response_model=List[CustomerResponse])
async def list_customers(
    skip: int = Query(0, ge=0),
    limit: int = Query(20, ge=1, le=100),
    search: Optional[str] = None,
    db_session: AsyncSession = Depends(get_db_session),
    current_user: User = Depends(get_current_active_user)
):
    query = select(Customer).where(Customer.is_active == True)
    if search:
        query = query.where(Customer.name.ilike(f"%{search}%"))
    query = query.offset(skip).limit(limit).order_by(Customer.created_at.desc())
    result = await db_session.execute(query)
    return [CustomerResponse.from_orm(c) for c in result.scalars().all()]


@router.post("", response_model=CustomerResponse, status_code=201)
async def create_customer(
    customer_data: CustomerCreate,
    db_session: AsyncSession = Depends(get_db_session),
    current_user: User = Depends(get_current_active_user)
):
    data = customer_data.dict()
    if not data.get("code"):
        data["code"] = f"C{datetime.now().strftime('%Y%m%d%H%M%S')}"
    
    customer = Customer(**data)
    db_session.add(customer)
    await db_session.commit()
    await db_session.refresh(customer)
    return CustomerResponse.from_orm(customer)