"""Discovery and registration tests for generic adapter registry."""

from __future__ import annotations

import pytest
from lfx.services.adapters import registry as adapter_registry_mod
from lfx.services.adapters.schema import AdapterType

from tests.unit.services.adapter_test_helpers import DeploymentAdapterStub, make_deployment_adapter_registry


class DummyDecoratorAdapter(DeploymentAdapterStub):
    pass


class DummyEntryPointAdapter(DeploymentAdapterStub):
    pass


class DummyConfigAdapter(DeploymentAdapterStub):
    pass


class DummyAlternateAdapter(DeploymentAdapterStub):
    pass


class DummyEntryPoint:
    """Simple entry point stub for importlib.metadata.entry_points()."""

    def __init__(self, name: str, obj: type):
        self.name = name
        self._obj = obj

    def load(self):
        return self._obj


pytestmark = pytest.mark.usefixtures("clean_adapter_globals")


def test_get_adapter_registry_singleton():
    first = make_deployment_adapter_registry()
    second = make_deployment_adapter_registry()

    assert first is second


def test_get_adapter_registry_raises_on_parameter_mismatch_for_same_adapter_type():
    make_deployment_adapter_registry()

    with pytest.raises(adapter_registry_mod.AdapterRegistryConflictError):
        adapter_registry_mod.get_adapter_registry(
            adapter_type=AdapterType.DEPLOYMENT,
            entry_point_group="lfx.OTHER.group",
            config_section_path=("other", "path"),
        )


def test_get_adapter_registry_convention_defaults():
    """Convention-derived params match explicit equivalents."""
    registry = adapter_registry_mod.get_adapter_registry(adapter_type=AdapterType.DEPLOYMENT)

    assert registry.entry_point_group == "lfx.deployment.adapters"
    assert registry.config_section_path == ("deployment", "adapters")


def test_discovery_precedence_entrypoint_decorator_config(tmp_path, monkeypatch):
    registry = make_deployment_adapter_registry()

    monkeypatch.setattr(
        "importlib.metadata.entry_points",
        lambda group: [DummyEntryPoint("local", DummyEntryPointAdapter)] if group == "lfx.deployment.adapters" else [],
    )

    adapter_registry_mod.register_adapter(AdapterType.DEPLOYMENT, "local")(DummyDecoratorAdapter)

    (tmp_path / "lfx.toml").write_text(
        f"""
[deployment.adapters]
local = "{__name__}:DummyConfigAdapter"
"""
    )

    registry.discover(config_dir=tmp_path)

    assert registry.get_class("local") is DummyConfigAdapter


def test_lfx_toml_takes_precedence_over_pyproject(tmp_path):
    registry = make_deployment_adapter_registry()

    (tmp_path / "lfx.toml").write_text(
        f"""
[deployment.adapters]
local = "{__name__}:DummyConfigAdapter"
"""
    )
    (tmp_path / "pyproject.toml").write_text(
        f"""
[tool.lfx.deployment.adapters]
local = "{__name__}:DummyAlternateAdapter"
"""
    )

    registry.discover(config_dir=tmp_path)

    assert registry.get_class("local") is DummyConfigAdapter


def test_discover_only_once(tmp_path):
    registry = make_deployment_adapter_registry()

    (tmp_path / "lfx.toml").write_text(
        f"""
[deployment.adapters]
local = "{__name__}:DummyConfigAdapter"
"""
    )
    registry.discover(config_dir=tmp_path)
    assert registry.get_class("local") is DummyConfigAdapter

    (tmp_path / "lfx.toml").write_text(
        f"""
[deployment.adapters]
local = "{__name__}:DummyAlternateAdapter"
"""
    )
    registry.discover(config_dir=tmp_path)

    assert registry.get_class("local") is DummyConfigAdapter


def test_invalid_import_path_is_ignored(tmp_path):
    registry = make_deployment_adapter_registry()

    (tmp_path / "lfx.toml").write_text(
        """
[deployment.adapters]
local = "invalid_without_colon"
"""
    )

    registry.discover(config_dir=tmp_path)

    assert registry.get_class("local") is None


def test_register_class_override_false_preserves_existing():
    registry = make_deployment_adapter_registry()
    registry.register_class("local", DummyEntryPointAdapter, override=True)
    registry.register_class("local", DummyDecoratorAdapter, override=False)

    assert registry.get_class("local") is DummyEntryPointAdapter


def test_decorator_override_false_wins_over_entrypoint(tmp_path, monkeypatch):
    """Decorator registers at import time (before discover), so it wins.

    Both sources use override=False.  The decorator fires first, and the
    entry point can't overwrite it.  This mirrors ``@register_service``
    behavior for top-level services.
    """
    registry = make_deployment_adapter_registry()
    monkeypatch.setattr(
        "importlib.metadata.entry_points",
        lambda group: [DummyEntryPoint("local", DummyEntryPointAdapter)] if group == "lfx.deployment.adapters" else [],
    )
    adapter_registry_mod.register_adapter(
        AdapterType.DEPLOYMENT,
        "local",
        override=False,
    )(DummyDecoratorAdapter)

    registry.discover(config_dir=tmp_path)

    assert registry.get_class("local") is DummyDecoratorAdapter


def test_decorator_override_false_skips_when_same_key_already_decorated(tmp_path):
    adapter_registry_mod.register_adapter(AdapterType.DEPLOYMENT, "local")(DummyDecoratorAdapter)
    adapter_registry_mod.register_adapter(AdapterType.DEPLOYMENT, "local", override=False)(DummyAlternateAdapter)

    registry = make_deployment_adapter_registry()
    registry.discover(config_dir=tmp_path)

    assert registry.get_class("local") is DummyDecoratorAdapter


def test_list_keys_is_sorted():
    registry = make_deployment_adapter_registry()
    registry.register_class("zeta", DummyEntryPointAdapter)
    registry.register_class("alpha", DummyDecoratorAdapter)

    assert registry.list_keys() == ["alpha", "zeta"]


def test_repr_before_and_after_discovery(tmp_path):
    registry = make_deployment_adapter_registry()
    registry.register_class("local", DummyEntryPointAdapter)

    r = repr(registry)
    assert "adapter_type=<AdapterType.DEPLOYMENT: 'deployment'>" in r
    assert "keys=['local']" in r
    assert "discovered=False" in r

    registry.discover(config_dir=tmp_path)
    assert "discovered=True" in repr(registry)


def test_pyproject_only_discovery(tmp_path):
    registry = make_deployment_adapter_registry()
    (tmp_path / "pyproject.toml").write_text(
        f"""
[tool.lfx.deployment.adapters]
local = "{__name__}:DummyConfigAdapter"
"""
    )

    registry.discover(config_dir=tmp_path)

    assert registry.get_class("local") is DummyConfigAdapter


def test_discovery_handles_malformed_toml(tmp_path):
    registry = make_deployment_adapter_registry()
    (tmp_path / "lfx.toml").write_text(
        """
[deployment.adapters
local = "pathlib:Path"
"""
    )

    registry.discover(config_dir=tmp_path)
    assert registry.get_class("local") is None


def test_discovery_ignores_missing_or_empty_section(tmp_path):
    registry = make_deployment_adapter_registry()
    (tmp_path / "lfx.toml").write_text(
        """
[other]
key = "value"
"""
    )

    registry.discover(config_dir=tmp_path)
    assert registry.list_keys() == []


def test_entry_point_load_failure_does_not_abort_discovery(tmp_path, monkeypatch):
    registry = make_deployment_adapter_registry()

    class BrokenEntryPoint:
        name = "broken"

        def load(self):
            msg = "boom"
            raise RuntimeError(msg)

    monkeypatch.setattr(
        "importlib.metadata.entry_points",
        lambda group: [BrokenEntryPoint()] if group == "lfx.deployment.adapters" else [],
    )
    (tmp_path / "lfx.toml").write_text(
        f"""
[deployment.adapters]
local = "{__name__}:DummyConfigAdapter"
"""
    )

    registry.discover(config_dir=tmp_path)
    assert registry.get_class("local") is DummyConfigAdapter


def test_get_nested_section_returns_none_for_non_dict_path():
    from lfx.services.config_discovery import get_nested_section

    nested = {"tool": {"lfx": {"deployment": "not-a-dict"}}}

    section = get_nested_section(nested, ("tool", "lfx", "deployment", "adapters"))
    assert section is None


def test_config_with_nonexistent_module_is_ignored(tmp_path):
    """Config referencing a module that cannot be imported is skipped."""
    registry = make_deployment_adapter_registry()

    (tmp_path / "lfx.toml").write_text(
        """
[deployment.adapters]
local = "nonexistent_pkg_xyz.adapters:Adapter"
"""
    )

    registry.discover(config_dir=tmp_path)
    assert registry.get_class("local") is None


def test_config_with_nonexistent_class_in_valid_module_is_ignored(tmp_path):
    """Config referencing a real module but missing class is skipped."""
    registry = make_deployment_adapter_registry()

    (tmp_path / "lfx.toml").write_text(
        """
[deployment.adapters]
local = "pathlib:TotallyMadeUpClassName"
"""
    )

    registry.discover(config_dir=tmp_path)
    assert registry.get_class("local") is None


def test_config_with_non_string_value_is_ignored(tmp_path):
    registry = make_deployment_adapter_registry()

    (tmp_path / "lfx.toml").write_text(
        """
[deployment.adapters]
local = 42
"""
    )

    registry.discover(config_dir=tmp_path)
    assert registry.get_class("local") is None


def test_multi_key_discovery_across_sources(tmp_path, monkeypatch):
    registry = make_deployment_adapter_registry()

    monkeypatch.setattr(
        "importlib.metadata.entry_points",
        lambda group: (
            [DummyEntryPoint("ep_only", DummyEntryPointAdapter)] if group == "lfx.deployment.adapters" else []
        ),
    )

    adapter_registry_mod.register_adapter(AdapterType.DEPLOYMENT, "dec_only")(DummyDecoratorAdapter)

    (tmp_path / "lfx.toml").write_text(
        f"""
[deployment.adapters]
cfg_only = "{__name__}:DummyConfigAdapter"
"""
    )

    registry.discover(config_dir=tmp_path)

    assert registry.get_class("ep_only") is DummyEntryPointAdapter
    assert registry.get_class("dec_only") is DummyDecoratorAdapter
    assert registry.get_class("cfg_only") is DummyConfigAdapter
    assert registry.list_keys() == ["cfg_only", "dec_only", "ep_only"]