config.py

"""应用程序配置模块.

该模块定义了应用程序的所有配置项，包括数据库、Redis、Celery、邮件、认证等设置. 使用 pydantic_settings.BaseSettings 进行环境变量管理.
"""

from functools import lru_cache
from typing import Literal, Optional

from pydantic_settings import BaseSettings


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

    管理所有应用程序配置项，支持从环境变量和.env文件加载配置. 包含数据库、缓存、任务队列、邮件、认证、AI模型等相关配置.
    """

    # 数据库设置
    DATABASE_TYPE: str = "sqlite"  # 默认使用 SQLite
    DATABASE_HOST: Optional[str] = None
    DATABASE_PORT: Optional[int] = None
    DATABASE_USER: Optional[str] = None
    DATABASE_PASSWORD: Optional[str] = None
    DATABASE_NAME: str = "app.db"
    RAG_DATABASE_NAME: str = "rag.db"

    # Redis设置
    REDIS_HOST: str = "localhost"
    REDIS_PORT: int = 6379
    REDIS_DB: int = 0
    REDIS_PASSWORD: Optional[str] = None

    # Celery设置
    CELERY_BROKER_URL: Optional[str] = None
    CELERY_RESULT_BACKEND: Optional[str] = None

    # 任务处理设置
    TASK_PROCESSING_INTERVAL: int = 300  # 5分钟处理一次
    MAX_CONCURRENT_TASKS: int = 3  # 最大并发任务数

    # 邮件设置
    SMTP_SERVER: str = "smtp.gmail.com"
    SMTP_PORT: int = 587
    SMTP_USERNAME: str
    SMTP_PASSWORD: str
    SMTP_FROM_EMAIL: str
    SMTP_FROM_NAME: str = "AI文档系统"

    # 验证码设置
    VERIFICATION_CODE_LENGTH: int = 6
    VERIFICATION_CODE_EXPIRE_MINUTES: int = 10

    # 会话设置
    SESSION_COOKIE_NAME: str = "session_id"
    SESSION_EXPIRE_DAYS: int = 7

    # OAuth2 设置
    GITHUB_CLIENT_ID: Optional[str] = None
    GITHUB_CLIENT_SECRET: Optional[str] = None
    GITHUB_REDIRECT_URI: Optional[str] = None
    GOOGLE_CLIENT_ID: Optional[str] = None
    GOOGLE_CLIENT_SECRET: Optional[str] = None
    GOOGLE_REDIRECT_URI: Optional[str] = None

    # JWT设置（保留用于API访问）
    JWT_SECRET_KEY: str
    JWT_ALGORITHM: str = "HS256"
    ACCESS_TOKEN_EXPIRE_MINUTES: int = 30
    REFRESH_TOKEN_EXPIRE_DAYS: int = 7

    # OpenAI设置
    OPENAI_API_KEY: str
    OPENAI_BASE_URL: str
    LLM_MODEL: str
    EMBEDDING_MODEL: str
    EMBEDDING_API_KEY: str
    EMBEDDING_BASE_URL: str
    EMB_DIMENSIONS: int = 1024
    LLM_STANDARD_MODEL: str
    LLM_ADVANCED_MODEL: str

    # 全局模式
    GLOBAL_MODE: Literal["public", "private"] = "public"
    GLOBAL_LLM: Literal["public", "private"] = "public"
    DISABLE_KB_INDEXING: bool = False

    # 注册控制：内网域名白名单
    # INTERNAL_REGISTRATION_HOSTS 为空 -> 允许任何 Host 访问注册接口（public 注册）
    # 非空 -> 只有 Host 在该列表中的请求才允许注册
    INTERNAL_REGISTRATION_HOSTS: str = ""

    # Loki设置
    LOKI_URL: Optional[str] = None
    LOKI_TAG: Optional[str] = None

    @property
    def DATABASE_URL(self) -> str:
        """获取数据库连接URL.

        根据配置的数据库类型，生成对应的数据库连接URL.
        目前支持SQLite和PostgreSQL两种数据库类型.

        Returns:
            str: 数据库连接URL

        Raises:
            ValueError: 当配置了不支持的数据库类型时抛出
        """
        if self.DATABASE_TYPE == "sqlite":
            return f"sqlite:///./{self.DATABASE_NAME}"
        elif self.DATABASE_TYPE == "postgresql":
            return (
                f"postgresql://{self.DATABASE_USER}:{self.DATABASE_PASSWORD}"
                f"@{self.DATABASE_HOST}:{self.DATABASE_PORT}/{self.DATABASE_NAME}"
            )
        else:
            raise ValueError(f"不支持的数据库类型: {self.DATABASE_TYPE}")

    @property
    def RAG_DATABASE_URL(self) -> str:
        """获取RAG（检索增强生成）数据库连接URL.

        根据配置的数据库类型，生成对应的RAG数据库连接URL.
        目前支持SQLite和PostgreSQL两种数据库类型.

        Returns:
            str: RAG数据库连接URL

        Raises:
            ValueError: 当配置了不支持的数据库类型时抛出
        """
        if self.DATABASE_TYPE == "sqlite":
            return f"sqlite:///./{self.RAG_DATABASE_NAME}"
        elif self.DATABASE_TYPE == "postgresql":
            return (
                f"postgresql://{self.DATABASE_USER}:{self.DATABASE_PASSWORD}"
                f"@{self.DATABASE_HOST}:{self.DATABASE_PORT}/{self.RAG_DATABASE_NAME}"
            )
        else:
            raise ValueError(f"不支持的数据库类型: {self.DATABASE_TYPE}")

    @property
    def REDIS_URL(self) -> str:
        """获取Redis连接URL.

        根据配置生成Redis连接URL，支持带密码和不带密码两种形式.

        Returns:
            str: Redis连接URL
        """
        if self.REDIS_PASSWORD:
            return f"redis://:{self.REDIS_PASSWORD}@{self.REDIS_HOST}:{self.REDIS_PORT}/{self.REDIS_DB}"
        return f"redis://{self.REDIS_HOST}:{self.REDIS_PORT}/{self.REDIS_DB}"

    def __init__(self, **kwargs):
        """初始化Settings实例.

        初始化配置对象，设置默认的Celery broker和backend URL，
        并配置OpenAI客户端.

        Args:
            **kwargs: 配置参数字典
        """
        super().__init__(**kwargs)
        # 如果没有显式设置，使用Redis URL作为Celery的broker和backend
        if not self.CELERY_BROKER_URL:
            self.CELERY_BROKER_URL = self.REDIS_URL
        if not self.CELERY_RESULT_BACKEND:
            self.CELERY_RESULT_BACKEND = self.REDIS_URL

        # 配置OpenAI
        import openai

        openai.api_key = self.OPENAI_API_KEY

    class Config:
        """Pydantic配置类.

        定义配置文件的加载行为，指定默认的环境变量文件路径.
        """

        env_file = ".env"


@lru_cache()
def get_settings() -> Settings:
    """获取应用程序配置单例实例.

    使用lru_cache装饰器确保配置对象只被创建一次.

    Returns:
        Settings: 配置对象实例
    """
    return Settings()