#!/usr/bin/env python3
import paho.mqtt.client as mqtt
import re

from home.mqtt import MqttBase
from home.config import config
from home.mqtt.payload.sensors import Temperature
from home.api.types import TemperatureSensorLocation
from home.database import SensorsDatabase


def get_sensor_type(sensor: str) -> TemperatureSensorLocation:
    for item in TemperatureSensorLocation:
        if sensor == item.name.lower():
            return item
    raise ValueError(f'unexpected sensor value: {sensor}')


class MqttServer(MqttBase):
    def __init__(self):
        super().__init__(clean_session=False)
        self.database = SensorsDatabase()

    def on_connect(self, client: mqtt.Client, userdata, flags, rc):
        super().on_connect(client, userdata, flags, rc)
        self._logger.info("subscribing to hk/#")
        client.subscribe('hk/#', qos=1)

    def on_message(self, client: mqtt.Client, userdata, msg):
        super().on_message(client, userdata, msg)
        try:
            variants = '|'.join([s.name.lower() for s in TemperatureSensorLocation])
            match = re.match(rf'hk/(\d+)/si7021/({variants})', msg.topic)
            if not match:
                return

            # FIXME string home_id must be supported
            home_id = int(match.group(1))
            sensor = get_sensor_type(match.group(2))

            payload = Temperature.unpack(msg.payload)
            self.database.add_temperature(home_id, payload.time, sensor,
                                          temp=int(payload.temp*100),
                                          rh=int(payload.rh*100))
        except Exception as e:
            self._logger.exception(str(e))


if __name__ == '__main__':
    config.load('sensors_mqtt_receiver')

    server = MqttServer()
    server.connect_and_loop()
