import logging
from time import sleep

import paho.mqtt.client as mqtt
from rich.align import Align
from rich.layout import Layout
from rich.live import Live
from rich.panel import Panel
from rich.table import Table

MQTT_HOST = "nas.home.kunbox.net"
MQTT_SUB = "sdm630/"
TITLE = "SDM630"
TABLE_LAYOUT = {
    "row1": {
        "voltage": {
            "name": "Voltage",
            "suffix": "V",
        },
        "current": {
            "name": "Current",
            "suffix": "A",
        },
        "current-demand": {
            "name": "Current Demand",
            "suffix": "A",
        },
    },
    "row2": {
        "apparent-power": {
            "name": "Apparent Power",
            "suffix": "W",
        },
        "reactive-power": {
            "name": "Reactive Power",
            "suffix": "VAr",
        },
        "power-factor": {
            "name": "Power Factor",
            "suffix": "",
        },
    },
}

# logging.basicConfig(level=logging.DEBUG)
LOG = logging.getLogger(__name__)

mqtt_data = {}


def on_connect(client, userdata, flags, rc):
    LOG.info(f"Connected to mqtt server")
    mqtt.subscribe(f"{MQTT_SUB}#")


def on_disconnect(client, userdata, rc):
    LOG.info(f"Disconnected from mqtt server")


def on_message(client, userdata, msg):
    value = msg.payload.decode()
    try:
        value = round(float(value), 2)
    except Exception:
        pass
    mqtt_data[msg.topic[len(MQTT_SUB) :]] = value


def generate_layout():
    layout = Layout()

    layout.split_column(
        Layout(Panel(Align.center(TITLE)), name="header"),
        Layout(name="main"),
    )
    layout["header"].size = 3

    layout["main"].split_column(*[Layout(name=k) for k in TABLE_LAYOUT.keys()])
    for k, i in TABLE_LAYOUT.items():
        layout["main"][k].split_row(*[Layout(name=v) for v in i.keys()])

    for k, i in TABLE_LAYOUT.items():
        for k2, v in i.items():
            table = Table(
                title=v["name"],
                show_header=False,
                box=None,
            )
            for topic, value in sorted(mqtt_data.items()):
                if topic.startswith(f"{k2}/"):
                    table.add_row(topic[len(f"{k2}/") :], f"{value} {v['suffix']}")
            layout["main"][k][k2].update(Panel(Align.center(table)))
    return layout


try:
    mqtt = mqtt.Client()
    mqtt.on_connect = on_connect
    mqtt.on_disconnect = on_disconnect
    mqtt.on_message = on_message
    mqtt.connect(MQTT_HOST, 1883, 10)
    mqtt.loop_start()

    with Live(generate_layout(), screen=True) as live:
        while True:
            live.update(generate_layout())
            sleep(0.5)
except Exception:
    LOG.exception("oops")