# SPDX-License-Identifier: 0BSD

"""Optional integration checks for Argos CLI (skipped when not installed)."""

import shutil
import urllib.error
import urllib.request

import pytest

from meshchatx.src.backend.translator_handler import (
    ARGOS_CLI_EXECUTABLE_NAMES,
    TranslatorHandler,
    _find_argos_cli_executable,
)


def _argos_cli_on_path() -> bool:
    return _find_argos_cli_executable() is not None


_STANZA_RESOURCES_URL = (
    "https://raw.githubusercontent.com/stanfordnlp/stanza-resources/"
    "main/resources_1.10.0.json"
)


def _argos_cli_needs_network() -> bool:
    """Argos sentence splitting can trigger Stanza to fetch resources over HTTPS."""
    req = urllib.request.Request(
        _STANZA_RESOURCES_URL,
        method="HEAD",
        headers={"User-Agent": "MeshChatX-Tests/1"},
    )
    try:
        with urllib.request.urlopen(req, timeout=5) as resp:
            code = getattr(resp, "status", resp.getcode())
            return code is not None and int(code) < 500
    except urllib.error.HTTPError as e:
        return e.code < 500
    except OSError:
        return False


@pytest.mark.integration
def test_find_argos_cli_matches_shutil_which():
    expected = None
    for name in ARGOS_CLI_EXECUTABLE_NAMES:
        expected = shutil.which(name)
        if expected:
            break
    assert _find_argos_cli_executable() == expected


@pytest.mark.integration
@pytest.mark.skipif(not _argos_cli_on_path(), reason="Argos CLI not on PATH")
@pytest.mark.skipif(
    not _argos_cli_needs_network(),
    reason="Network unreachable (Argos CLI may download Stanza resources)",
)
def test_translate_en_es_via_cli_round_trip():
    handler = TranslatorHandler(
        translator_argos_enabled=True,
        translator_libretranslate_enabled=True,
    )
    assert handler.has_argos_cli
    assert not handler.has_argos_lib

    try:
        result = handler.translate_text("Hello", "en", "es", use_argos=True)
    except RuntimeError as exc:
        err = str(exc)
        if any(
            needle in err
            for needle in (
                "Network is unreachable",
                "Failed to establish a new connection",
                "Max retries exceeded",
            )
        ):
            pytest.skip(
                "Argos CLI could not reach Stanza resources over HTTPS "
                "(offline, sandbox, or subprocess has no route)"
            )
        raise
    assert result["source"] == "argos"
    assert result["source_lang"] == "en"
    assert result["target_lang"] == "es"
    assert len(result["translated_text"].strip()) > 0


@pytest.mark.integration
@pytest.mark.skipif(not _argos_cli_on_path(), reason="Argos CLI not on PATH")
def test_get_supported_languages_includes_argos_when_libretranslate_down():
    handler = TranslatorHandler(
        translator_argos_enabled=True,
        translator_libretranslate_enabled=True,
    )
    langs = handler.get_supported_languages()
    argos = [x for x in langs if x.get("source") == "argos"]
    assert len(argos) >= 1