"""

This file provides:

- Path settings for global config file & relative directories

- Version numbering

- Protocols for the core components of mini-swe-agent.

  By the magic of protocols & duck typing, you can pretty much ignore them,

  unless you want the static type checking.

"""

__version__ = "2.0.0a2"

import os

from pathlib import Path

from typing import Any, Protocol

import dotenv

from platformdirs import user_config_dir

from rich.console import Console

from minisweagent.utils.log import logger

package_dir = Path(__file__).resolve().parent

global_config_dir = Path(os.getenv("MSWEA_GLOBAL_CONFIG_DIR") or user_config_dir("mini-swe-agent"))

global_config_dir.mkdir(parents=True, exist_ok=True)

global_config_file = Path(global_config_dir) / ".env"

if not os.getenv("MSWEA_SILENT_STARTUP"):

    Console().print(

        f"👋 This is [bold green]mini-swe-agent[/bold green] version [bold green]{__version__}[/bold green].\n"

        f"Check the [bold red]v2 migration guide[/] at [bold red]https://klieret.short.gy/mini-v2-migration[/]\n",

        f"Loading global config from [bold green]'{global_config_file}'[/bold green]",

    )

dotenv.load_dotenv(dotenv_path=global_config_file)

# === Protocols ===

# You can ignore them unless you want static type checking.

class Model(Protocol):

    """Protocol for language models."""

    config: Any

    def query(self, messages: list[dict[str, str]], **kwargs) -> dict: ...

    def format_message(self, **kwargs) -> dict: ...

    def format_observation_messages(

        self, message: dict, outputs: list[dict], template_vars: dict | None = None

    ) -> list[dict]: ...

    def get_template_vars(self, **kwargs) -> dict[str, Any]: ...

    def serialize(self) -> dict: ...

class Environment(Protocol):

    """Protocol for execution environments."""

    config: Any

    def execute(self, action: dict, cwd: str = "") -> dict[str, Any]: ...

    def get_template_vars(self, **kwargs) -> dict[str, Any]: ...

    def serialize(self) -> dict: ...

class Agent(Protocol):

    """Protocol for agents."""

    config: Any

    def run(self, task: str, **kwargs) -> dict: ...

    def save(self, path: Path | None, *extra_dicts) -> dict: ...

__all__ = [

    "Agent",

    "Model",

    "Environment",

    "package_dir",

    "__version__",

    "global_config_file",

    "global_config_dir",

    "logger",

]