from __future__ import annotations

import os
from pathlib import Path
from typing import Any

from dotenv import load_dotenv

from cooperbench.agents.swe_agent import REPO_ROOT
from cooperbench.agents.swe_agent.utils.log import get_logger

logger = get_logger("swea-config", emoji="🔧")


def _convert_path_relative_to_repo_root(path: Path | str, root: Path | None = None) -> Path | str:
    original_type = type(path)
    path = Path(path).resolve()
    root = Path(root or os.getenv("SWE_AGENT_CONFIG_ROOT", REPO_ROOT))
    relative_path = path.relative_to(root) if root in path.parents else path
    return relative_path if original_type is Path else str(relative_path)


def _could_be_a_path(v: Any) -> bool:
    try:
        return Path(v).exists()
    except Exception:
        return False


def _strip_abspath_from_dict(value: dict | list | str, root: Path | None = None) -> dict | list | str:
    root = Path(root or os.getenv("SWE_AGENT_CONFIG_ROOT", REPO_ROOT))
    if isinstance(value, dict):
        return {k: _strip_abspath_from_dict(v, root) for k, v in value.items()}
    elif isinstance(value, list):
        return [_strip_abspath_from_dict(v, root) for v in value]
    elif isinstance(value, str) and _could_be_a_path(value):
        return _convert_path_relative_to_repo_root(value, root)
    else:
        return value


def _convert_path_to_abspath(path: Path | str) -> Path:
    """If path is not absolute, convert it to an absolute path
    using the SWE_AGENT_CONFIG_ROOT environment variable (if set) or
    REPO_ROOT as base.
    """
    path = Path(path)
    root = Path(os.getenv("SWE_AGENT_CONFIG_ROOT", REPO_ROOT))
    assert root.is_dir()
    if not path.is_absolute():
        path = root / path
    assert path.is_absolute()
    return path.resolve()


def _convert_paths_to_abspath(paths: list[Path] | list[str]) -> list[Path]:
    return [_convert_path_to_abspath(p) for p in paths]


def load_environment_variables(path: Path | None = None):
    """Load environment variables from a .env file.
    If path is not provided, we first look for a .env file in the current working
    directory and then in the repository root.
    """
    if path is None:
        cwd_path = Path.cwd() / ".env"
        repo_path = REPO_ROOT / ".env"
        if cwd_path.exists():
            path = cwd_path
        elif repo_path.exists():
            path = REPO_ROOT / ".env"
        else:
            logger.debug("No .env file found")
            return
    if not path.is_file():
        msg = f"No .env file found at {path}"
        raise FileNotFoundError(msg)
    anything_loaded = load_dotenv(dotenv_path=path)
    if anything_loaded:
        logger.info(f"Loaded environment variables from {path}")
