# -*- coding: utf-8 -*-
"""
存储数据库连接信息和API密钥
"""

from pydantic_settings import BaseSettings
from typing import Optional
from pydantic import Field
from pathlib import Path

# 计算 .env 优先级：优先当前工作目录，其次项目根目录（MindSpider 的上级目录）
PROJECT_ROOT: Path = Path(__file__).resolve().parents[1]
CWD_ENV: Path = Path.cwd() / ".env"
ENV_FILE: str = str(CWD_ENV if CWD_ENV.exists() else (PROJECT_ROOT / ".env"))

class Settings(BaseSettings):
    """全局配置管理，优先从环境变量和.env加载。支持MySQL/PostgreSQL统一数据库参数命名。"""
    DB_DIALECT: str = Field("mysql", description="数据库类型，支持'mysql'或'postgresql'")
    DB_HOST: str = Field("your_host", description="数据库主机名或IP地址")
    DB_PORT: int = Field(3306, description="数据库端口号")
    DB_USER: str = Field("your_username", description="数据库用户名")
    DB_PASSWORD: str = Field("your_password", description="数据库密码")
    DB_NAME: str = Field("mindspider", description="数据库名称")
    DB_CHARSET: str = Field("utf8mb4", description="数据库字符集")
    CUSTOM_KEYWORDS_FILE: Optional[str] = Field(
        None,
        description="自定义关键词文件路径（可为绝对路径或相对MindSpider目录的路径）"
    )
    USE_DEFAULT_KEYWORDS_ONLY: bool = Field(
        True,
        description="为True时忽略数据库/自定义结果，直接使用默认关键词"
    )
    ENABLE_KEYWORD_SEARCH: bool = Field(
        True,
        description="开启后运行基于关键词的爬取流程，关闭则完全跳过关键词搜索"
    )
    MINDSPIDER_API_KEY: Optional[str] = Field(None, description="MINDSPIDER API密钥")
    MINDSPIDER_BASE_URL: Optional[str] = Field("https://api.deepseek.com", description="MINDSPIDER API基础URL，推荐deepseek-chat模型使用https://api.deepseek.com")
    MINDSPIDER_MODEL_NAME: Optional[str] = Field("deepseek-chat", description="MINDSPIDER API模型名称, 推荐deepseek-chat")

    class Config:
        env_file = ENV_FILE
        env_prefix = ""
        case_sensitive = False
        extra = "allow"

settings = Settings()