from prometheus_client import start_http_server, Summary, Gauge
import paho.mqtt.publish as publish
import paho.mqtt.client as mqtt
import random
import time
from node_api import NodeAPI

client_id = "exporter"
topics = [
        ("$SYS/broker/clients/connected", 0),
        ("$CONNECTED/#", 0)
        ]

node_api = NodeAPI()

MQTT_CONNECTED = Gauge('mqtt_node_connected', 'MQTT node connection status', ['node', 'description'])
MQTT_BROKER_CLIENTS = Gauge('mqtt_broker_clients', 'MQTT broker clients connected')

def on_connect(client, userdata, flags, rc):
    print("Connected with result code " + str(rc))

    client.subscribe(topics)
    if rc==0:
        client.connected_flag=True
        client.publish("$CONNECTED/"+client_id, 1, retain=True)
    else:
        client.bad_connection_flag=True


#def on_disconnect(client, userdata, rc):
#    client.publish(connection_topic,0,retain=True)

def on_message(client, userdata, msg):
    topic = msg.topic.split('/')

    if topic[0] == "$CONNECTED":
        MQTT_CONNECTED.labels(topic[1], node_api.get(topic[1])).set(int(msg.payload))

    if msg.topic == "$SYS/broker/clients/connected":
        MQTT_BROKER_CLIENTS.set(int(msg.payload))

    print(msg.topic, int(msg.payload))


client = mqtt.Client(client_id)

client.on_connect = on_connect
#client.on_disconnect = on_disconnect
client.on_message = on_message
client.will_set("$CONNECTED/"+client_id, 0, qos=0, retain=True)
client.connect('localhost')
client.loop_start()


if __name__ == '__main__':
    # Start up the server to expose the metrics.
    start_http_server(8000)

    while True:
        time.sleep(1)