from __future__ import annotations

import io
import os
from pathlib import Path
import subprocess
import sys

import pexpect
import pytest

from tests import TESTS_ROOT
from tests.e2e.common import (
    ansi_tolerant_pattern,
    send_ctrl_c_until_quit_confirmation,
    wait_for_main_screen,
    wait_for_request_count,
)
from tests.e2e.mock_server import StreamingMockServer


def _venv_executable(venv_path: Path, name: str) -> Path:
    if os.name == "nt":
        return venv_path / "Scripts" / f"{name}.exe"
    return venv_path / "bin" / name


def _build_wheel(dist_dir: Path) -> Path:
    subprocess.run(
        ["uv", "build", "--wheel", "--out-dir", str(dist_dir)],
        cwd=TESTS_ROOT.parent,
        check=True,
    )
    wheels = sorted(dist_dir.glob("mistral_vibe-*.whl"))
    assert len(wheels) == 1
    return wheels[0]


def _install_fresh_wheel(tmp_path: Path, wheel_path: Path) -> Path:
    venv_path = tmp_path / "fresh-install-venv"
    subprocess.run(
        ["uv", "venv", "--no-config", "--python", sys.executable, str(venv_path)],
        cwd=tmp_path,
        check=True,
    )

    python_path = _venv_executable(venv_path, "python")
    subprocess.run(
        [
            "uv",
            "pip",
            "install",
            "--no-config",
            "--refresh",
            "--python",
            str(python_path),
            str(wheel_path),
        ],
        cwd=tmp_path,
        check=True,
    )
    return _venv_executable(venv_path, "vibe")


@pytest.mark.timeout(90)
def test_fresh_wheel_install_can_spawn_cli_and_complete_happy_path(
    streaming_mock_server: StreamingMockServer,
    setup_e2e_env: None,
    e2e_workdir: Path,
    tmp_path: Path,
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    wheel_path = _build_wheel(tmp_path / "dist")
    vibe_executable = _install_fresh_wheel(tmp_path, wheel_path)

    monkeypatch.delenv("PYTHONPATH", raising=False)

    captured = io.StringIO()
    child = pexpect.spawn(
        str(vibe_executable),
        ["--workdir", str(e2e_workdir)],
        cwd=str(tmp_path),
        env=os.environ,
        encoding="utf-8",
        timeout=30,
        dimensions=(36, 120),
    )
    child.logfile_read = captured

    try:
        wait_for_main_screen(child, timeout=20)
        child.send("Greet")
        child.send("\r")

        wait_for_request_count(
            lambda: len(streaming_mock_server.requests), expected_count=1, timeout=10
        )
        child.expect(ansi_tolerant_pattern("Hello from mock server"), timeout=10)

        send_ctrl_c_until_quit_confirmation(child, captured, timeout=5)
        child.expect(pexpect.EOF, timeout=10)
    finally:
        if child.isalive():
            child.terminate(force=True)
        if not child.closed:
            child.close()

    output = captured.getvalue()
    assert "Welcome to Mistral Vibe" not in output
    assert streaming_mock_server.requests[-1].get("model") == "mock-model"