import netifaces
import ipaddress
import asyncio

PING_RETRIES: int = 3
PING_TIMEOUT: int = 1


class MinerNetwork:
    def __init__(self, ip_addr=None):
        self.network = None
        self.ip_addr = ip_addr
        self.connected_miners = {}

    def get_network(self):
        if self.network:
            return self.network
        gateways = netifaces.gateways()
        if not self.ip_addr:
            default_gateway = gateways['default'][netifaces.AF_INET][0]
        else:
            default_gateway = self.ip_addr
        subnet_mask = netifaces.ifaddresses(gateways['default'][netifaces.AF_INET][1])[netifaces.AF_INET][0]['netmask']
        return ipaddress.ip_network(f"{default_gateway}/{subnet_mask}", strict=False)

    async def scan_network_for_miners(self):
        local_network = self.get_network()
        print(f"Scanning {local_network} for miners...")
        scan_tasks = []
        for host in local_network.hosts():
            scan_tasks.append(self.ping_miner(host))
        await asyncio.gather(*scan_tasks)
        print(f"Found {len(self.connected_miners)} connected miners...")

    async def ping_miner(self, ip: ipaddress.ip_address):
        for i in range(PING_RETRIES):
            connection_fut = asyncio.open_connection(str(ip), 4028)
            try:
                # get the read and write streams from the connection
                reader, writer = await asyncio.wait_for(connection_fut, timeout=PING_TIMEOUT)
                # immediately close connection, we know connection happened
                writer.close()
                # make sure the writer is closed
                await writer.wait_closed()
                # add miner to dict
                self.connected_miners[ip] = {}
                # ping was successful
                return True
            except asyncio.exceptions.TimeoutError:
                # ping failed if we time out
                continue
            except ConnectionRefusedError:
                # handle for other connection errors
                print("Unknown error...")
            # ping failed, likely with an exception
            except Exception as e:
                print(e)
            continue
        return False