import shutil
import subprocess
import sys

import pytest

from proxyscraper import cli, completion


def generate(shell, capsys):
    with pytest.raises(SystemExit) as exit_info:
        cli.parse_args(["--completion", shell])
    assert exit_info.value.code == 0
    return capsys.readouterr().out


@pytest.mark.parametrize("shell", completion.SHELLS)
def test_every_option_is_in_the_script(shell, capsys):
    out = generate(shell, capsys)
    for flag in ("--types", "--rotate", "--no-datacenter", "--serve-host", "--recheck"):
        assert flag.lstrip("-") in out
    assert "round-robin" in out and "socks5" in out  # the choices come along


def test_short_help_keeps_abbreviations():
    assert completion.short_help("max. time for the connect (default: 3)") == "max. time for the connect"
    assert completion.short_help("only these countries, e.g. DE,AT,CH") == "only these countries"
    assert completion.short_help("x" * 80).endswith("…")


def test_zsh_braces_stay_outside_the_quotes(capsys):
    out = generate("zsh", capsys)
    assert "'(-c --concurrency)'{-c,--concurrency}'[" in out
    assert "{_files; compadd live}" in out
    assert "--types[which protocols]:*-*:types:(http socks4 socks5)" in out


def test_help_texts_are_short(capsys):
    out = generate("fish", capsys)
    assert "show help" in out and "show this help message" not in out


# on Windows "bash" is usually WSL and doesn't see the Windows paths – it isn't needed there either
needs_bash = pytest.mark.skipif(not shutil.which("bash") or sys.platform == "win32", reason="no bash")


def complete_bash(script, cwd, *words):
    """Call _proxy_scraper with COMP_WORDS, one line per suggestion."""
    quoted = " ".join("'" + w + "'" for w in words)
    code = (script + f'\nCOMP_WORDS=({quoted}); COMP_CWORD={len(words) - 1}; _proxy_scraper; '
            'printf "%s\\n" "${COMPREPLY[@]}"\n')
    out = subprocess.run(["bash", "-s"], input=code, capture_output=True, text=True, check=True, cwd=cwd).stdout
    return [line for line in out.splitlines() if line]


@needs_bash
def test_bash_completes_values(capsys, tmp_path):
    script = generate("bash", capsys)
    assert complete_bash(script, tmp_path, "proxy-scraper", "--rotate", "r") == ["random", "round-robin"]
    no_d = ["--no-dnsbl", "--no-datacenter", "--no-discover"]  # in the order of --help
    assert complete_bash(script, tmp_path, "proxy-scraper", "--no-d") == no_d


@needs_bash
def test_bash_offers_more_types_after_the_first(capsys, tmp_path):
    script = generate("bash", capsys)
    assert complete_bash(script, tmp_path, "proxy-scraper", "--types", "http", "s") == ["socks4", "socks5"]
    assert complete_bash(script, tmp_path, "proxy-scraper", "--types", "http", "--fa") == ["--fast"]


@needs_bash
def test_bash_keeps_paths_with_spaces_together(capsys, tmp_path):
    (tmp_path / "my list.txt").write_text("")
    script = generate("bash", capsys)
    assert complete_bash(script, tmp_path, "proxy-scraper", "--recheck", "m") == ["my list.txt"]
    assert complete_bash(script, tmp_path, "proxy-scraper", "--recheck", "l") == ["live"]


@pytest.mark.skipif(not shutil.which("zsh"), reason="no zsh")
def test_zsh_script_parses(capsys, tmp_path):
    script = tmp_path / "comp.zsh"
    script.write_text(generate("zsh", capsys))
    subprocess.run(["zsh", "-n", str(script)], check=True)


# PowerShell: runs the real completer through TabExpansion2 wherever pwsh is installed (PROXY_SCRAPER_PWSH to point
# at a portable one)
PWSH = __import__("os").environ.get("PROXY_SCRAPER_PWSH") or shutil.which("pwsh")
needs_pwsh = pytest.mark.skipif(not PWSH, reason="no pwsh")


def complete_pwsh(script_path, line):
    code = (f". '{script_path}'; (TabExpansion2 -inputScript '{line}' -cursorColumn {len(line)})"
            ".CompletionMatches | ForEach-Object { $_.CompletionText }")
    out = subprocess.run([PWSH, "-NoProfile", "-NonInteractive", "-Command", code], capture_output=True, text=True,
                         check=True).stdout
    return [line for line in out.splitlines() if line]


def test_powershell_script_registers_a_native_completer(capsys):
    out = generate("powershell", capsys)
    assert "Register-ArgumentCompleter -Native -CommandName 'proxy-scraper'" in out
    assert "'--no-datacenter'" in out and "'round-robin'" in out


@needs_pwsh
def test_powershell_completes_options_and_values(capsys, tmp_path):
    script = tmp_path / "comp.ps1"
    script.write_text(generate("powershell", capsys))
    assert "--types" in complete_pwsh(script, "proxy-scraper --ty")
    from proxyscraper.server.pool import STRATEGIES
    assert set(complete_pwsh(script, "proxy-scraper --rotate ")) == set(STRATEGIES)
    assert complete_pwsh(script, "proxy-scraper --types http so") == ["socks4", "socks5"]  # several values
    assert "-c" in complete_pwsh(script, "proxy-scraper -")
    assert "--json" in complete_pwsh(script, "proxy-scraper --list-sources ")  # optional value: options too
    (tmp_path / "hits.txt").write_text("")
    assert {"live", str(tmp_path / "hits.txt")} <= set(complete_pwsh(script, f"proxy-scraper --recheck {tmp_path}/")
                                                    + complete_pwsh(script, "proxy-scraper --recheck "))