"""Agent-Loader: scannt agents/ Ordner und lädt alle BaseAgent-Subklassen."""
import importlib
import inspect
import logging
from pathlib import Path
from typing import Dict, List

from agents.base import BaseAgent, AgentConfig


logger = logging.getLogger("agents.loader")


def discover_agents() -> Dict[str, BaseAgent]:
    """Scannt das agents/ Verzeichnis und gibt alle Agent-Instanzen zurück."""
    agents: Dict[str, BaseAgent] = {}
    agents_dir = Path(__file__).parent

    for py_file in agents_dir.glob("*.py"):
        if py_file.name.startswith("_") or py_file.name in ("base.py", "__init__.py"):
            continue
        module_name = f"agents.{py_file.stem}"
        try:
            module = importlib.import_module(module_name)
        except Exception as e:
            logger.warning(f"Konnte {module_name} nicht laden: {e}")
            continue

        for name, obj in inspect.getmembers(module, inspect.isclass):
            if obj is BaseAgent:
                continue
            if issubclass(obj, BaseAgent) and obj.__module__ == module_name:
                try:
                    instance = obj()
                    if hasattr(instance, "config"):
                        agents[instance.config.id] = instance
                        logger.info(f"Agent geladen: {instance.config.id} ({instance.config.name})")
                except Exception as e:
                    logger.warning(f"Konnte Agent {name} aus {module_name} nicht instanziieren: {e}")

    return agents


def get_agent(agent_id: str) -> BaseAgent:
    """Holt einen Agent by ID."""
    agents = discover_agents()
    if agent_id not in agents:
        raise KeyError(f"Agent '{agent_id}' nicht gefunden. Verfügbar: {list(agents.keys())}")
    return agents[agent_id]


def list_agents() -> List[AgentConfig]:
    """Gibt Configs aller Agents zurück."""
    return [a.config for a in discover_agents().values()]