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 collections import Counter
from rich.console import Console
import sys
import glob

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

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]

# import ipdb;ipdb.set_trace()



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 read_stats(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__":
    # ini_directory = Path("/home/jvved/dev/testpks/")
    # ini_file = ini_directory.glob("sub*.ini")
    # print(ini_file)
    stats_list = ["run/sub1-stats.sock", "run/sub2-stats.sock"]
    for stats in stats_list:
        stats_address = Path(stats_list)
        stats = read_stats(stats_address)
        print(stats)
        router_stats = RouterStats(**stats)
        all_app_stats = {}
        total_workers = 0

    for sub in router_stats.subscriptions:
        print(sub)
        for node in sub.nodes:
            print(node.name)
            port = node.name.split(":")[-1]
            print(port)
            stats_address2 = Path(f"run/{port}_stats.sock")
            stats2 = read_stats(stats_address2)
            app_stats = AppStats(**stats2)
            print(f"App Stats for {port}:{app_stats}")
            print(app_stats.workers)
            print(len(app_stats.workers))
            worker_amount = len(app_stats.workers)
            total_workers = total_workers + worker_amount
            print(total_workers)


    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"{total_workers}")
    console = Console()
    console.print(table)