"""
数据库配置模块
使用pydantic进行配置验证和管理
"""

from typing import Optional, List
from pydantic_settings import BaseSettings
from functools import lru_cache
from pydantic import PostgresDsn, field_validator
from pydantic_core.core_schema import FieldValidationInfo


class Settings(BaseSettings):
    """应用配置类"""

    RTSP_ENABLED: bool = True

    # API配置
    API_V1_PREFIX: str = "/api/v1"
    PROJECT_NAME: str = "algorithm-service"
    PROJECT_VERSION: str = "1.0.0"
    PROJECT_DESCRIPTION: str = "algorithm-service"
    BACKEND_CORS_ORIGINS: List[str] = ["http://localhost:3000", "http://localhost:8000"]


    # 数据库配置
    DATABASE_HOST: str = "localhost"
    DATABASE_PORT: int = 5432
    DATABASE_USER: str = "postgres"
    DATABASE_PASSWORD: str = "yipai123"
    DATABASE_NAME: str = "pmms"
    DATABASE_SCHEMA: str = "public"

    # 连接池配置
    DATABASE_POOL_SIZE: int = 10
    DATABASE_MAX_OVERFLOW: int = 20
    DATABASE_POOL_RECYCLE: int = 3600  # 连接回收时间（秒）
    DATABASE_ECHO: bool = False  # SQL日志，生产环境设为False

    # 应用配置
    APP_NAME: str = "SurFaceFeature API"
    APP_VERSION: str = "1.0.0"
    DEBUG: bool = False

    # 日志配置
    LOG_LEVEL: str = "INFO"
    LOG_FILE: Optional[str] = None

    # 异步配置
    ASYNC_MODE: bool = False

    # 资源文件夹配置
    FACE_REGISTER_IMAGE_RESOURCE_DIR: str = "D:/ruoyi/uploadPath/face"
    VIDEO_RESOURCE_DIR: str = "D:/ruoyi/uploadPath/video"
    FACE_CAL_FEATURE_TIMEOUT_HOURS: int = 10
    FACE_MODEL_VERSION: int = 0 #insight_face_buffalo_l
    FACE_USE_GPU: bool = True
    FACE_USE_NPU: bool = False
    SUR_CONFIG_TYPE_FACE: int = 0
    SUR_CONFIG_SCOPE_GLOBAL: int = 0

    # JWT配置（预留）
    SECRET_KEY: str = "your-secret-key-here-change-in-production"
    ALGORITHM: str = "HS256"
    ACCESS_TOKEN_EXPIRE_MINUTES: int = 30

    @property
    def DATABASE_URL(self) -> str:
        """构建数据库连接URL"""
        return f"postgresql://{self.DATABASE_USER}:{self.DATABASE_PASSWORD}@{self.DATABASE_HOST}:{self.DATABASE_PORT}/{self.DATABASE_NAME}"

    @property
    def ASYNC_DATABASE_URL(self) -> str:
        """构建异步数据库连接URL"""
        return f"postgresql+asyncpg://{self.DATABASE_USER}:{self.DATABASE_PASSWORD}@{self.DATABASE_HOST}:{self.DATABASE_PORT}/{self.DATABASE_NAME}"

    @field_validator("DATABASE_POOL_SIZE")
    def validate_pool_size(cls, v):
        """验证连接池大小"""
        if v < 1:
            raise ValueError("DATABASE_POOL_SIZE must be at least 1")
        if v > 100:
            raise ValueError("DATABASE_POOL_SIZE cannot exceed 100")
        return v

    class Config:
        env_file = ".env"
        env_file_encoding = "utf-8"
        case_sensitive = False
        extra = "ignore"


@lru_cache()
def get_settings() -> Settings:
    """
    获取配置单例
    使用lru_cache避免重复加载.env文件
    """
    return Settings()


# 导出配置实例
settings = get_settings()