import os
import tempfile
from pathlib import Path
from pydantic_settings import BaseSettings
from typing import List


def _default_data_dir() -> str:
    configured = os.environ.get("CHATLAB_DATA_DIR")
    if configured:
        return str(Path(configured).expanduser())
    appdata = os.environ.get("APPDATA")
    if appdata:
        return str(Path(appdata) / "ChatLab")
    return str(Path.home() / ".chatlab")


def _default_static_dir() -> str:
    configured = os.environ.get("CHATLAB_STATIC_DIR")
    if configured:
        return str(Path(configured).expanduser())
    return str((Path(__file__).resolve().parents[1] / "chatlab-web" / "frontend" / "dist"))


class Settings(BaseSettings):
    chatlog_base_url: str = "http://127.0.0.1:5030"
    ai_base_url: str = "https://dashscope.aliyuncs.com/compatible-mode/v1"
    ai_api_key: str = ""
    ai_model: str = ""        # 不设默认值，必须由用户在设置页配置
    summary_model: str = ""   # 不设默认值，必须由用户在设置页配置
    voice_model: str = ""     # 不设默认值，必须由用户在设置页配置
    vision_model: str = ""    # 不设默认值，必须由用户在设置页配置
    data_dir: str = _default_data_dir()
    static_dir: str = _default_static_dir()
    db_path: str = str(Path(_default_data_dir()) / "data" / "knowledge.db")
    cors_origins: List[str] = [
        "http://127.0.0.1:5173",
        "http://localhost:5173",
        "http://localhost:3000",
    ]

    class Config:
        env_file = ".env"

settings = Settings()

try:
    Path(settings.data_dir).mkdir(parents=True, exist_ok=True)
    Path(settings.db_path).parent.mkdir(parents=True, exist_ok=True)
except PermissionError:
    fallback_dir = Path(tempfile.gettempdir()) / "ChatLab"
    fallback_dir.mkdir(parents=True, exist_ok=True)
    settings.data_dir = str(fallback_dir)
    settings.db_path = str(fallback_dir / "data" / "knowledge.db")
    Path(settings.db_path).parent.mkdir(parents=True, exist_ok=True)