import json
import asyncio
import uvicorn
from fastapi import FastAPI
from fastapi import Request
from fastapi import WebSocket, WebSocketDisconnect
from fastapi.templating import Jinja2Templates
from fastapi.staticfiles import StaticFiles
from tools.web_monitor.miner_factory import miner_factory
from network import MinerNetwork

app = FastAPI()

app.mount("/static", StaticFiles(directory="static"), name="static")
templates = Jinja2Templates(directory="templates")


@app.get("/")
def dashboard(request: Request):
    return templates.TemplateResponse("index.html", {"request": request})


@app.get("/scan")
def scan(request: Request):
    return templates.TemplateResponse("scan.html", {"request": request})


@app.websocket("/scan/ws")
async def websocket_scan(websocket: WebSocket):
    await websocket.accept()
    try:
        while True:
            network_ip = await websocket.receive_text()
            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 in miner_generator:
                if miner:
                    miners.append(miner)

            get_miner_genenerator = miner_factory.get_miner_generator(miners)
            all_miners = []
            async for found_miner in get_miner_genenerator:
                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 in all_miners:
                    send_miners.append({"ip": str(miner["ip"]), "model": miner["model"]})
                await websocket.send_json(send_miners)

            await websocket.send_text("Done")
    except WebSocketDisconnect:
        pass


if __name__ == "__main__":
    uvicorn.run("app:app", host="127.0.0.1", port=80)