from pathlib import Path
from stats import AppStats
import errno
import json
import traceback
import socket
import stats
from stats import WorkerAppStats
from stats import AppStats

from rich.console import Console

from stats import RouterSubscription
import time
from rich.live import Live
from rich.table import Table
import pydantic

class RouterStats(pydantic.BaseModel):
    model_config = pydantic.ConfigDict(strict=True)
    version: str
    pid: int = pydantic.Field(ge=0)
    uid: int = pydantic.Field(ge=0)
    gid: int = pydantic.Field(ge=0)
    cwd: str
    active_sessions: int = pydantic.Field(ge=0)
    http: list[str]  # ['0.0.0.0:8034', '127.0.0.1:5700'],
    subscriptions: list[RouterSubscription]
    cheap: int = pydantic.Field(ge=0)

def run(stats_address):
    if not all([stats_address.exists(), stats_address.is_socket()]):
        raise Exception(f"unable to read stats from {(stats_address)}")

    def unix_addr(arg):
        sfamily = socket.AF_UNIX
        addr = arg
        return sfamily, addr, socket.gethostname()

    js = ""
    sfamily, addr, host = unix_addr(stats_address)
    # console.info(f"{sfamily=} {str(addr)=} {host=}")

    try:
        s = None
        s = socket.socket(sfamily, socket.SOCK_STREAM)
        s.connect(str(addr))
        while True:
            data = s.recv(4096)
            if len(data) < 1:
                break
            js += data.decode('utf8', 'ignore')
        if s:
            s.close()
    except ConnectionRefusedError as e:
        print('connection refused')
    except FileNotFoundError as e:
        print(f"socket @ {addr} not available")
    except IOError as e:
        if e.errno != errno.EINTR:
            #uwsgi.log(f"socket @ {addr} not available")
            pass
    except Exception:
        print(traceback.format_exc())
    else:
        try:
            return json.loads(js)
        except json.JSONDecodeError:
            print(traceback.format_exc())
            print(js)

if __name__ == "__main__":
    stats_address = Path("run/sub1-stats.sock")
    stats = run(stats_address)
    print(stats)

class SocketStats(pydantic.BaseModel):
    name:str
    queue: int

class WorkerStats(pydantic.BaseModel):
    id:int
    accepting:int

class AppStats(pydantic.BaseModel):
    model_config = pydantic.ConfigDict(strict=True)
    version: str
    listen_queue: int = 0
    listen_queue_errors: int = 0
    signal_queue: int
    load: int
    pid: int
    uid: int
    gid: int
    cwd: str
    locks: list[dict[str, int]]
    sockets: list[SocketStats]
    workers: list[WorkerStats]

for sub in stats["subscriptions"]:
    print(sub)
    for node in sub["nodes"]:
        print(node)
        node_name = node["name"]
        print(f"{node_name}")
        port = node_name.split(":")[-1]
        print(f"{port}")

# stats_address2 = f"{port}_stats.sock"
stats_address2 = Path(f"run/{port}_stats.sock")
stats2 = run(stats_address2)
print(stats2)

for sub2 in stats2["workers"]:
    print(sub2)

app_stats = AppStats(**stats2)
print(app_stats)

worker_amount = app_stats.workers
print(len(app_stats.workers)) # print amount workers

router_stats = RouterStats(**stats)
print(router_stats)

     # import ipdb;ipdb.set_trace()
for sub in router_stats.subscriptions:
    print(f"{sub.key=}")

table = Table()
table.add_column("Virtual Hosts", justify="right", style="cyan", no_wrap=True)
table.add_column("Nodes", style="magenta")
table.add_column("Total Workers", style="magenta")
table.add_row(sub.key, f"{len(sub.nodes)}", f"{len(app_stats.workers)}")
console = Console()
console.print(table)