import os
from pathlib import Path

from dotenv import load_dotenv

# Load .env so that TRADINGAGENTS_* variables are available before
# DEFAULT_CONFIG is evaluated.  CWD is checked first, then the project
# root (two levels up from this file).  load_dotenv never overwrites
# variables that are already present in the environment.
load_dotenv()
load_dotenv(Path(__file__).resolve().parent.parent / ".env")


def _env(key: str, default=None):
    """Read ``TRADINGAGENTS_<KEY>`` from the environment.

    Returns *default* when the variable is unset **or** empty, so that
    ``TRADINGAGENTS_MID_THINK_LLM=`` in a ``.env`` file is treated the
    same as not setting it at all (preserving the ``None`` semantics for
    "fall back to the parent setting").

    This is the single source of truth for config env-var reading.
    Import and reuse this helper instead of duplicating it elsewhere.
    """
    val = os.getenv(f"TRADINGAGENTS_{key.upper()}")
    if not val:  # None or ""
        return default
    return val


def _env_int(key: str, default=None):
    """Like :func:`_env` but coerces the value to ``int``."""
    val = _env(key)
    if val is None:
        return default
    try:
        return int(val)
    except (ValueError, TypeError):
        return default


def _env_float(key: str, default=None):
    """Like :func:`_env` but coerces the value to ``float``."""
    val = _env(key)
    if val is None:
        return default
    try:
        return float(val)
    except (ValueError, TypeError):
        return default


DEFAULT_CONFIG = {
    "project_dir": os.path.abspath(os.path.join(os.path.dirname(__file__), ".")),
    "results_dir": _env("RESULTS_DIR", "./reports"),
    "data_cache_dir": os.path.join(
        os.path.abspath(os.path.join(os.path.dirname(__file__), ".")),
        "dataflows/data_cache",
    ),
    # LLM settings — all overridable via TRADINGAGENTS_<KEY> env vars
    "llm_provider": _env("LLM_PROVIDER", "openai"),
    "deep_think_llm": _env("DEEP_THINK_LLM", "gpt-5.2"),
    "mid_think_llm": _env("MID_THINK_LLM"),              # falls back to quick_think_llm when None
    "quick_think_llm": _env("QUICK_THINK_LLM", "gpt-5-mini"),
    "backend_url": _env("BACKEND_URL", "https://api.openai.com/v1"),
    # Per-role provider overrides (fall back to llm_provider / backend_url when None)
    "deep_think_llm_provider": _env("DEEP_THINK_LLM_PROVIDER"),    # e.g. "google", "anthropic", "openrouter"
    "deep_think_backend_url": _env("DEEP_THINK_BACKEND_URL"),       # override backend URL for deep-think model
    "mid_think_llm_provider": _env("MID_THINK_LLM_PROVIDER"),      # e.g. "ollama"
    "mid_think_backend_url": _env("MID_THINK_BACKEND_URL"),         # override backend URL for mid-think model
    "quick_think_llm_provider": _env("QUICK_THINK_LLM_PROVIDER"),  # e.g. "openai", "ollama"
    "quick_think_backend_url": _env("QUICK_THINK_BACKEND_URL"),     # override backend URL for quick-think model
    # Provider-specific thinking configuration (applies to all roles unless overridden)
    "google_thinking_level": _env("GOOGLE_THINKING_LEVEL"),      # "high", "minimal", etc.
    "openai_reasoning_effort": _env("OPENAI_REASONING_EFFORT"),  # "medium", "high", "low"
    "anthropic_effort": _env("ANTHROPIC_EFFORT"),                # "high", "medium", "low"
    # Per-role provider-specific thinking configuration
    "deep_think_google_thinking_level": _env("DEEP_THINK_GOOGLE_THINKING_LEVEL"),
    "deep_think_openai_reasoning_effort": _env("DEEP_THINK_OPENAI_REASONING_EFFORT"),
    "mid_think_google_thinking_level": _env("MID_THINK_GOOGLE_THINKING_LEVEL"),
    "mid_think_openai_reasoning_effort": _env("MID_THINK_OPENAI_REASONING_EFFORT"),
    "quick_think_google_thinking_level": _env("QUICK_THINK_GOOGLE_THINKING_LEVEL"),
    "quick_think_openai_reasoning_effort": _env("QUICK_THINK_OPENAI_REASONING_EFFORT"),
    # Debate and discussion settings
    "max_debate_rounds": _env_int("MAX_DEBATE_ROUNDS", 2),
    "max_risk_discuss_rounds": _env_int("MAX_RISK_DISCUSS_ROUNDS", 2),
    "max_recur_limit": _env_int("MAX_RECUR_LIMIT", 100),
    # Data vendor configuration
    # Category-level configuration (default for all tools in category)
    "data_vendors": {
        "core_stock_apis": _env("VENDOR_CORE_STOCK_APIS", "yfinance"),
        "technical_indicators": _env("VENDOR_TECHNICAL_INDICATORS", "yfinance"),
        "fundamental_data": _env("VENDOR_FUNDAMENTAL_DATA", "yfinance"),
        "news_data": _env("VENDOR_NEWS_DATA", "yfinance"),
        "scanner_data": _env("VENDOR_SCANNER_DATA", "yfinance"),
        "calendar_data": _env("VENDOR_CALENDAR_DATA", "finnhub"),
    },
    # Tool-level configuration (takes precedence over category-level)
    "tool_vendors": {
        # Finnhub free tier provides same data + MSPR aggregate bonus signal
        "get_insider_transactions": "finnhub",
    },
}