import subprocess
import time

import pytest
from fastembed import TextEmbedding
from qdrant_client import QdrantClient
from qdrant_client.models import Distance, PointStruct, VectorParams


@pytest.fixture(scope="session")
def embedding_model():
    return TextEmbedding()


@pytest.fixture(scope="session")
def qdrant_client(embedding_model: TextEmbedding):
    client = QdrantClient(":memory:")

    documents: list[str] = [
        "Rick es el mas guapo",
        "Los pulpos tienen tres corazones y sangre azul",
        "Las cucarachas pueden vivir hasta una semana sin cabeza",
        "Los koalas tienen huellas dactilares casi idénticas a las humanas",
        "La miel nunca se echa a perder, incluso después de miles de años",
    ]
    embeddings = list(embedding_model.embed(documents))
    size = len(embeddings[0])

    _ = client.recreate_collection(
        collection_name="dummy_collection",
        vectors_config=VectorParams(distance=Distance.COSINE, size=size),
    )

    for idx, (emb, document) in enumerate(zip(embeddings, documents)):
        _ = client.upsert(
            collection_name="dummy_collection",
            points=[
                PointStruct(id=idx, vector=emb.tolist(), payload={"text": document})
            ],
        )

    yield client


@pytest.fixture(scope="session", autouse=True)
def run_mcp():
    # Start the MCP server in the background
    process = subprocess.Popen(
        ["uv", "run", "searchbox-mcp"],
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        text=True,
    )

    # Give the server more time to start up properly
    time.sleep(5)

    # Check if process is still running
    if process.poll() is not None:
        stdout, stderr = process.communicate()
        pytest.fail(f"MCP server failed to start. stdout: {stdout}, stderr: {stderr}")

    try:
        yield "http://localhost:8000/sse"
    finally:
        # Clean up the process when tests are done
        process.terminate()
        try:
            _ = process.wait(timeout=5)
        except subprocess.TimeoutExpired:
            process.kill()
            _ = process.wait()