import asyncio

from fastapi import WebSocket

from network import MinerNetwork
from tools.web_monitor.func import get_current_miner_list
from miners.miner_factory import MinerFactory


async def do_websocket_scan(websocket: WebSocket, network_ip: str):
    cur_miners = get_current_miner_list()
    try:
        if "/" in network_ip:
            network_ip, network_subnet = network_ip.split("/")
            network = MinerNetwork(network_ip, mask=network_subnet)
        else:
            network = MinerNetwork(network_ip)
        miner_generator = network.scan_network_generator()
        miners = []
        async for miner_ip in miner_generator:
            if miner_ip and str(miner_ip) not in cur_miners:
                miners.append(miner_ip)

        get_miner_generator = MinerFactory().get_miner_generator(miners)
        all_miners = []
        async for found_miner in get_miner_generator:
            all_miners.append(
                {"ip": found_miner.ip, "model": await found_miner.get_model()}
            )
            all_miners.sort(key=lambda x: x["ip"])
            send_miners = []
            for miner_ip in all_miners:
                send_miners.append(
                    {"ip": str(miner_ip["ip"]), "model": miner_ip["model"]}
                )
            await websocket.send_json(send_miners)
        await websocket.send_text("Done")
    except asyncio.CancelledError:
        raise