"""Interrumpe la generacion en curso de ComfyUI y devuelve el estado de la cola.

Funcion impura: hace red (HTTP POST /interrupt + GET /queue). Solo stdlib.

POST /interrupt corta el prompt que ComfyUI esta ejecutando ahora mismo (no vacia
la cola: los prompts pendientes siguen). GET /queue devuelve queue_running (lo que
se ejecuta) y queue_pending (lo encolado). Esta funcion combina ambos en un dict
honesto que NO lanza excepcion en fallo de red: devuelve {ok: False, error}.
"""
import json
import urllib.error
import urllib.request


def comfyui_interrupt_queue(server: str = "127.0.0.1:8188") -> dict:
    """Interrumpe la generacion en curso y devuelve el estado de la cola.

    Args:
        server: host:port del servidor ComfyUI sin esquema (default
            "127.0.0.1:8188").

    Returns:
        dict con:
        - ok (bool): True si tanto el interrupt como la lectura de la cola
          tuvieron exito.
        - interrupted (bool): True si el POST /interrupt respondio sin error.
        - queue_running (int): numero de prompts ejecutandose ahora mismo.
        - queue_pending (int): numero de prompts encolados pendientes.
        - error (str): mensaje de error si algo fallo; cadena vacia si todo OK.
    """
    out = {
        "ok": False,
        "interrupted": False,
        "queue_running": 0,
        "queue_pending": 0,
        "error": "",
    }
    base = f"http://{server}"

    # 1. POST /interrupt (cuerpo vacio): corta el prompt en ejecucion.
    try:
        req = urllib.request.Request(f"{base}/interrupt", data=b"", method="POST")
        with urllib.request.urlopen(req, timeout=10.0):
            out["interrupted"] = True
    except urllib.error.URLError as exc:
        reason = getattr(exc, "reason", exc)
        out["error"] = f"interrupt fallo: no se pudo conectar a {base}/interrupt: {reason}"
        return out

    # 2. GET /queue: estado actual de la cola tras el interrupt.
    try:
        with urllib.request.urlopen(f"{base}/queue", timeout=10.0) as resp:
            data = json.loads(resp.read())
        out["queue_running"] = len(data.get("queue_running", []))
        out["queue_pending"] = len(data.get("queue_pending", []))
        out["ok"] = True
    except urllib.error.URLError as exc:
        reason = getattr(exc, "reason", exc)
        out["error"] = f"queue fallo: no se pudo conectar a {base}/queue: {reason}"
    except json.JSONDecodeError as exc:
        out["error"] = f"queue fallo: respuesta no es JSON valido: {exc}"
    return out


if __name__ == "__main__":
    res = comfyui_interrupt_queue()
    print(
        f"ok={res['ok']} interrupted={res['interrupted']} "
        f"running={res['queue_running']} pending={res['queue_pending']} "
        f"error={res['error']!r}"
    )