power_consumption = Gauge(

    "power_consumption_watt", "The current power used", labelnames=["location"]

)

co2_emission = Gauge(

    "co2_emission_kilogramm", "The CO2 emission in kilogramm", labelnames=["location"]

)

energy_consumption = Gauge(

    "energy_consumption_kilowatthour",

    "The energy consumption in kilowatthour",

    labelnames=["location"],

)

def on_connect(client, userdata, flags, rc):

    client.publish("availability/mqtt2prometheus", "online", 2, True)

    topics_handler = {

        "availability/#": handle_availability,

        "space/status": handle_space_status,

        "space/member_count": handle_member_count,

        "sensors/people_now_present": handle_people_now_present,

    }

    for topic, handler in topics_handler.items():

        client.message_callback_add(topic, handler)

        client.subscribe(topic)

def handle_people_now_present(client, userdata, msg):

    people_now_present.set(json.loads(msg.payload))

def handle_space_status(client, userdata, msg):

    state = json.loads(msg.payload)

    space_state.set(int(state["open"]))

    space_lastchange.set(state["lastchange"])

def handle_member_count(client, userdata, msg):

    count = json.loads(msg.payload)

    member_count.labels(count["name"]).set(count["value"])

def handle_availability(client, userdata, msg):

    t = msg.topic.split("/")

    if len(t) == 2:

        c = t[-1]

    else:

        c = "_".join(t[1:])

    v = 1 if msg.payload == b"online" else 0

    availability.labels(c).set(v)

def handle_sensor(client, userdata, msg):

    location = "-".join(t[2:])

    if t[1] == "people_now_present":

        return

    metric = globals().get(t[1], None)

    if metric is not None:

        try:

            metric.labels(location).set(float(msg.payload))

        except:

            pass

def main():

    client = mqtt.Client("mqtt2prometheus")

    client.on_connect = on_connect

    client.username_pw_set(CONFIG["mqtt"]["username"], CONFIG["mqtt"]["password"])

    client.will_set("availability/mqtt2prometheus", "offline", 2, True)

    client.connect(CONFIG["mqtt"]["host"])

    client.loop_start()

    start_http_server(CONFIG["prometheus"]["port"])

    while True:

        time.sleep(1)