import uuid
from datetime import datetime, timezone, timedelta, date
from decimal import Decimal
from typing import List

from fastapi import FastAPI, APIRouter, HTTPException
from pydantic import BaseModel, ConfigDict
from sqlmodel import SQLModel, Field, select

from app.utils.model_utils import to_camel, format_datetime_to_string, format_date_to_string, FormattedDatetime, FormattedDate, current_datetime
from app.utils.mysql_utils import AsyncSessionDep


class LlmDemoModel(SQLModel, table=True):
  # Pydantic V2 的模型配置
  model_config = ConfigDict(
    alias_generator=to_camel,  # 使用 to_camel 函数生成别名
    populate_by_name=True,  # 允许通过原始字段名（snake_case）赋值
    extra='ignore',  # 忽略模型中未定义的额外字段，避免验证失败
    json_encoders={
      datetime: format_datetime_to_string,  # 为 datetime 类型指定自定义的 JSON 编码器
      date: format_date_to_string  # 为 date 类型指定自定义的 JSON 编码器
    }
  )

  __tablename__ = "llm_demo"

  id: str = Field(default=None, primary_key=True, description="唯一标识，编号")
  created_at: FormattedDatetime = Field(default_factory=current_datetime, description="创建时间")
  updated_at: FormattedDatetime = Field(default_factory=current_datetime, description="更新时间")
  created_by: str | None = Field(default=None, description="创建人id")
  updated_by: str | None = Field(default=None, description="更新人id")

  full_name: str = Field(default=None, description="用户名称")
  datetime_start: FormattedDatetime = Field(default=None, description="开通会员时间")
  datetime_end: FormattedDatetime = Field(default=None, description="会员截止到期时间")
  birthday: FormattedDate = Field(default=None, description="生日")
  amount: Decimal = Field(default=0, description="金额")


def add_llm_demo_route(app: FastAPI):
  route_prefix = "/llm_demo"
  router = APIRouter(prefix=route_prefix, tags=[route_prefix])

  @router.post('/list')
  async def _list(
    session: AsyncSessionDep,
    body: ModelQuerySchema = ModelQuerySchema(),
  ):
    """
    获取所有用户列表
    """

    # 构建SQL查询执行对象
    query = select(LlmDemoModel).order_by(
      LlmDemoModel.created_at.desc()
    )
    # 计算偏移量（跳过前N条），并查询比一页多1条的记录（用于判断是否有下一页）
    query = query.offset(body.page * body.page_size).limit(body.page_size + 1)
    result = await session.execute(query)
    query_cls_list: List[LlmDemoModel] = result.scalars().all()
    has_next = len(query_cls_list) == body.page_size + 1
    # 若有下一页，移除多查询的那一条记录
    if has_next:
      query_cls_list.pop()
    return {
      "list": query_cls_list,
      "has_next": has_next
    }

  @router.post('/insert')
  async def _insert(body: LlmDemoModel, session: AsyncSessionDep):
    """
    插入用户
    """
    if not body.id:
      body.id = str(uuid.uuid4())
    session.add(body)
    await session.commit()
    return body

  @router.post('/update')
  async def _update(body: LlmDemoModel, session: AsyncSessionDep):
    """
    更新用户
    """
    query = select(LlmDemoModel).where(LlmDemoModel.id == body.id)
    result = await session.execute(query)
    query_cls = result.scalars().first()
    if not query_cls:
      raise HTTPException(status_code=500, detail=f"Update row with id:{body.id} not found")

    update_data_exclude_true = body.model_dump(exclude_unset=True, exclude={'id'})
    update_data_exclude_false = body.model_dump(exclude={'id'})

    print(f"update_data_exclude_true: {update_data_exclude_true}")
    print(f"update_data_exclude_false: {update_data_exclude_false}")

    update_data = body.model_dump(exclude_unset=True, exclude={'id'})
    for field, value in update_data.items():
      setattr(query_cls, field, value)
    session.add(query_cls)
    await session.commit()
    return query_cls

  @router.post('/delete')
  async def _delete(body: LlmDemoModel, session: AsyncSessionDep):
    """
    更新用户
    """
    query = select(LlmDemoModel).where(LlmDemoModel.id == body.id)
    result = await session.execute(query)
    query_cls = result.scalars().first()
    if not query_cls:
      return {"affect_row_count": 0}
    await session.delete(query_cls)
    await session.commit()
    return {"affect_row_count": 1}

  app.include_router(router)


class ModelQuerySchema(BaseModel):
  page: int = Field(default=0, description="页码")
  page_size: int = Field(default=5, description="每页数量")