from __future__ import annotations

from pathlib import Path

import pytest

from host_agent.config import (
    HostAgentConfigurationError,
    HostAgentConfig,
    load_host_agent_config,
)


BASE_ENV = {
    "HOST_AGENT_HOST_ID": "host-a",
    "HOST_AGENT_TOKEN": "secret",
}


def test_load_host_agent_config_uses_local_network_defaults() -> None:
    assert load_host_agent_config(BASE_ENV) == HostAgentConfig(
        control_plane_url="http://127.0.0.1:8001",
        host_id="host-a",
        token="secret",
    )


def test_load_host_agent_config_parses_poll_and_retry_values() -> None:
    config = load_host_agent_config(
        {
            **BASE_ENV,
            "HOST_AGENT_CONTROL_PLANE_URL": "https://cloud.example/v1/",
            "HOST_AGENT_HEARTBEAT_INTERVAL_SECONDS": "10",
            "HOST_AGENT_POLL_TIMEOUT_SECONDS": "15",
            "HOST_AGENT_RETRY_BACKOFF_SECONDS": "2",
            "HOST_AGENT_MAX_RETRY_BACKOFF_SECONDS": "20",
        }
    )

    assert config.control_plane_url == "https://cloud.example/v1"
    assert config.poll_timeout_seconds == 15
    assert config.max_retry_backoff_seconds == 20


def test_load_host_agent_config_supports_managed_enrollment(tmp_path) -> None:
    identity_path = tmp_path / "host_identity.json"
    config = load_host_agent_config(
        {
            "HOST_AGENT_CONTROL_PLANE_URL": "https://cloud.example",
            "HOST_AGENT_ENROLLMENT_TOKEN": "one-time-token",
            "HOST_AGENT_IDENTITY_PATH": str(identity_path),
            "HOST_AGENT_DISPLAY_NAME": "Edge Mac",
        }
    )

    assert config.host_id == ""
    assert config.token == ""
    assert config.enrollment_token == "one-time-token"
    assert config.identity_path == identity_path
    assert config.enrollment_managed is True
    assert config.display_name == "Edge Mac"
    assert "one-time-token" not in repr(config)


def test_existing_identity_state_allows_restart_without_enrollment_token(
    tmp_path,
) -> None:
    identity_path = tmp_path / "host_identity.json"
    identity_path.write_text("{}", encoding="utf-8")

    config = load_host_agent_config({"HOST_AGENT_IDENTITY_PATH": str(identity_path)})

    assert config.identity_path == Path(identity_path)
    assert config.enrollment_managed is True


@pytest.mark.parametrize(
    "overrides",
    [
        {"HOST_AGENT_HOST_ID": ""},
        {"HOST_AGENT_TOKEN": ""},
        {"HOST_AGENT_HOST_ID": "host-a", "HOST_AGENT_TOKEN": ""},
        {"HOST_AGENT_CONTROL_PLANE_URL": "ftp://cloud.example"},
        {"HOST_AGENT_POLL_TIMEOUT_SECONDS": "0"},
        {
            "HOST_AGENT_RETRY_BACKOFF_SECONDS": "10",
            "HOST_AGENT_MAX_RETRY_BACKOFF_SECONDS": "5",
        },
    ],
)
def test_load_host_agent_config_rejects_invalid_values(
    overrides: dict[str, str],
) -> None:
    with pytest.raises(HostAgentConfigurationError):
        load_host_agent_config({**BASE_ENV, **overrides})