WFL/agent.py

import asyncio
import json
import time
import socket
import uuid
from urllib.parse import urlencode
from xml.etree import ElementTree as ET

import requests
import websockets

VMIX_API = "http://127.0.0.1:8088/api"
WS_BASE = "wss://wfl.tvstart.ru/ws/vmix-client"
# WS_BASE = "ws://127.0.0.1:8000/ws/vmix-client"

CLIENT_ID = f"vmix-{socket.gethostname().lower()}-{uuid.uuid4().hex[:6]}"
POLL_INTERVAL = 3
REQUEST_TIMEOUT = 3
RECONNECT_DELAY = 5

def read_vmix_dynamic_values():
    try:
        resp = requests.get(VMIX_API, timeout=REQUEST_TIMEOUT)
        resp.raise_for_status()
        xml_text = resp.text

        root = ET.fromstring(xml_text)

        def get_text(path: str) -> str | None:
            node = root.find(path)
            if node is None or node.text is None:
                return None
            value = node.text.strip()
            return value or None

        session_token = get_text("./dynamic/value1")
        match_id = get_text("./dynamic/value2")
        group_name = get_text("./dynamic/value3")

        return {
            "ok": True,
            "session_token": session_token,
            "match_id": int(match_id) if match_id and match_id.isdigit() else None,
            "group_name": group_name,
        }
    except Exception as e:
        return {
            "ok": False,
            "error": str(e),
            "session_token": None,
            "match_id": None,
            "group_name": None,
        }

def execute_vmix_command(path: str):
    path = str(path or "").strip()
    if not path.startswith("/api/"):
        return {"ok": False, "error": "invalid command", "path": path}

    url = f"http://127.0.0.1:8088{path}"
    try:
        resp = requests.get(url, timeout=REQUEST_TIMEOUT)
        return {
            "ok": resp.ok,
            "status_code": resp.status_code,
            "path": path,
            "response": resp.text[:300],
        }
    except Exception as e:
        return {
            "ok": False,
            "path": path,
            "error": str(e),
        }

async def wait_for_session():
    while True:
        data = read_vmix_dynamic_values()
        if data["ok"] and data["session_token"]:
            return data

        print("[agent] waiting for vMix session...", data)
        await asyncio.sleep(POLL_INTERVAL)

def build_ws_url(session_token: str, match_id: int | None, group_name: str | None):
    params = {
        "client_id": CLIENT_ID,
        "session_token": session_token,
    }
    if match_id is not None:
        params["match_id"] = match_id
    if group_name:
        params["group_name"] = group_name
    return f"{WS_BASE}?{urlencode(params)}"

async def ping_loop(ws):
    while True:
        await asyncio.sleep(15)
        await ws.send(json.dumps({"type": "ping"}))

async def run_agent():
    current_session = None

    while True:
        vmix_data = await wait_for_session()

        session_token = vmix_data["session_token"]
        match_id = vmix_data["match_id"]
        group_name = vmix_data["group_name"]

        if current_session != session_token:
            print(f"[agent] found session: {session_token}")
            current_session = session_token

        ws_url = build_ws_url(session_token, match_id, group_name)

        try:
            async with websockets.connect(ws_url, ping_interval=None, max_size=2**20) as ws:
                print("[agent] connected to server")

                await ws.send(json.dumps({
                    "type": "register",
                    "client_id": CLIENT_ID,
                    "session_token": session_token,
                    "match_id": match_id,
                    "group_name": group_name,
                }))

                pinger = asyncio.create_task(ping_loop(ws))

                try:
                    while True:
                        # следим, не сменился ли проект/session в vMix
                        latest = read_vmix_dynamic_values()
                        if latest["ok"] and latest["session_token"] and latest["session_token"] != current_session:
                            print("[agent] session changed in vMix, reconnecting")
                            break

                        try:
                            raw = await asyncio.wait_for(ws.recv(), timeout=3)
                        except asyncio.TimeoutError:
                            continue

                        msg = json.loads(raw)
                        if msg.get("type") == "vmix_command":
                            print("[agent] got commands:", msg.get("commands", []))
                            results = [execute_vmix_command(cmd) for cmd in msg.get("commands", [])]
                            print("[agent] results:", results)
                            await ws.send(json.dumps({
                                "type": "vmix_result",
                                "client_id": CLIENT_ID,
                                "session_token": session_token,
                                "results": results,
                                "sent_at": time.time(),
                            }))
                finally:
                    pinger.cancel()

        except Exception as e:
            print("[agent] websocket error:", e)

        await asyncio.sleep(RECONNECT_DELAY)

if __name__ == "__main__":
    asyncio.run(run_agent())