from __future__ import annotations

from typing import Any

from yuxi.channels.models import HealthStatus


async def probe_bridge(bridge) -> HealthStatus:
    return await bridge.health_check()


async def wait_for_connection(bridge, timeout: float = 30.0) -> dict[str, Any]:
    import asyncio

    deadline = asyncio.get_event_loop().time() + timeout
    while asyncio.get_event_loop().time() < deadline:
        health = await bridge.health_check()
        if health.status == "healthy":
            return {"connected": True, "jid": health.metadata.get("jid", "")}
        await asyncio.sleep(1.0)
    return {"connected": False, "error": "Connection timeout"}