from pydantic_settings import BaseSettings
from pathlib import Path
import os


class Settings(BaseSettings):
    # 应用
    APP_NAME: str = "AI 文档模板生成系统"
    APP_VERSION: str = "1.0.0"
    DEBUG: bool = True

    # === 数据库 MySQL ===
    DB_HOST: str = "localhost"
    DB_PORT: int = 3306
    DB_USER: str = "docforge"
    DB_PASSWORD: str = "docforge123"
    DB_NAME: str = "doc_forge"
    @property
    def DATABASE_URL(self) -> str:
        return f"mysql+asyncmy://{self.DB_USER}:{self.DB_PASSWORD}@{self.DB_HOST}:{self.DB_PORT}/{self.DB_NAME}?charset=utf8mb4"

    # === MinIO 文件存储 ===
    MINIO_ENDPOINT: str = "localhost:9000"
    MINIO_ACCESS_KEY: str = "docforge"
    MINIO_SECRET_KEY: str = "docforge123"
    MINIO_BUCKET_TEMPLATES: str = "doc-forge-templates"
    MINIO_BUCKET_UPLOADS: str = "doc-forge-uploads"
    MINIO_BUCKET_OUTPUTS: str = "doc-forge-outputs"
    MINIO_USE_SSL: bool = False

    # 本地缓存目录（MinIO 文件的本地临时缓存）
    LOCAL_CACHE_DIR: str = "local_cache"

    # 文件上传限制
    MAX_UPLOAD_SIZE: int = 50 * 1024 * 1024  # 50MB
    ALLOWED_EXTENSIONS: list = [".docx", ".doc", ".xlsx", ".xls", ".xlsm", ".csv", ".pdf", ".txt", ".md", ".json"]

    # 加密（用于 API Key 加密）
    ENCRYPTION_KEY: str = "change-this-to-a-32-byte-key-in-production!!"

    # AI 模型默认配置
    AI_REQUEST_TIMEOUT: int = 60
    AI_MAX_RETRIES: int = 3
    AI_MAX_CONCURRENT: int = 5
    AI_GLOBAL_CONCURRENT: int = 10

    # 服务端口
    HOST: str = "0.0.0.0"
    PORT: int = 8000

    class Config:
        env_file = ".env"
        env_file_encoding = "utf-8"


settings = Settings()

# 创建本地缓存目录
os.makedirs(settings.LOCAL_CACHE_DIR, exist_ok=True)