'''
Note that these functions need to parse the message['payload'] into a float
'''
import json
import logging

log = logging.getLogger('conv')
converters = []


def f_from_c(c):
    return (c * 9 / 5) + 32


def topic(topic_pattern: str):

    def decorator(func):
        func.topic_pattern = topic_pattern
        converters.append(func)
        return func

    return decorator


@topic(r'([^-]+)-air-quality/sensor/particulate_matter__10_0__m_concentration/state')
def pm(message, topicGroups: tuple[str]) -> list[dict]:
    return [{
        'name': 'air_quality_pm',
        'labels': [{
            'labelName': 'location',
            'labelValue': topicGroups[0],
        }, {
            'labelName': 'size',
            'labelValue': '10',
        }],
        'value': float(message['payload']),
    }]


@topic(r'([^-]+)-air-quality/sensor/air_temperature_c/state')
def air_temp(message, topicGroups: tuple[str]) -> list[dict]:
    return [{
        'name': 'air_temperature_f',
        'labels': [{
            'labelName': 'location',
            'labelValue': topicGroups[0],
        }],
        'value': f_from_c(float(message['payload'])),
    }]


@topic(r'(tele/st-wall-power/SENSOR|tele/ki-fridge/SENSOR|tt-console-power/SENSOR)')
def poorly_named_sonoff_powermeter(message, topicGroups: tuple[str]) -> list[dict]:
    energy = json.loads(message['payload'])['ENERGY']
    sensor = {
        'tele/st-wall-power/SENSOR': 'st_wall',
        'tele/ki-fridge/SENSOR': 'ki_fridge',
        'tt-console-power/SENSOR': 'tt_console',
    }[topicGroups[0]]
    return [{ 'name': 'powermeter_w',                  'labels': [{'labelName': 'sensor', 'labelValue': sensor}], 'value': energy['Power'] },
            { 'name': 'powermeter_apparent_power_va',  'labels': [{'labelName': 'sensor', 'labelValue': sensor}], 'value': energy['ApparentPower'] },
            { 'name': 'powermeter_reactive_power_var', 'labels': [{'labelName': 'sensor', 'labelValue': sensor}], 'value': energy['ReactivePower'] },
            { 'name': 'powermeter_factor',             'labels': [{'labelName': 'sensor', 'labelValue': sensor}], 'value': energy['Factor'] },
            { 'name': 'powermeter_voltage_v',          'labels': [{'labelName': 'sensor', 'labelValue': sensor}], 'value': energy['Voltage'] },
            { 'name': 'powermeter_current_a',          'labels': [{'labelName': 'sensor', 'labelValue': sensor}], 'value': energy['Current'] } ] # yapf: disable

@topic(r'(.*)-power/status/switch:0')
def shelly_powermeter(message, topicGroups: tuple[str]) -> list[dict]:
    sensor = topicGroups[0].replace('-', '_')
    msg = json.loads(message['payload'])
    return [
        { 'name': 'powermeter_w',                 'labels': [{'labelName': 'sensor', 'labelValue': sensor}], 'value': msg['apower'] },
        { 'name': 'powermeter_voltage_v',         'labels': [{'labelName': 'sensor', 'labelValue': sensor}], 'value': msg['voltage'] },
        { 'name': 'powermeter_current_a',         'labels': [{'labelName': 'sensor', 'labelValue': sensor}], 'value': msg['current'] },
        { 'name': 'powermeter_apparent_power_va', 'labels': [{'labelName': 'sensor', 'labelValue': sensor}], 'value': msg['apower'] },
        { 'name': 'powermeter_temp_f',            'labels': [{'labelName': 'sensor', 'labelValue': sensor}], 'value': f_from_c(msg['temperature']['tC']) },
    ] # yapf: disable

@topic(r'^zigbee/([^/]+)$')
def zigbee(message, topicGroups: tuple[str]) -> list[dict]:
    dev = topicGroups[0]
    body = json.loads(message['payload'])
    ret = []
    for v in ['update_available', 'link_quality', 'battery', 'occupancy', 'brightness', 'contact']:
        body_key = 'linkquality' if v == 'link_quality' else v
        if body_key not in body:
            continue
        value = body[body_key]
        if v in ['update_available', 'contact', 'occupancy']:
            value = int(bool(value))
        if v == 'brightness' and body.get('state', None) != 'ON':
            value = 0
        if v == 'link_quality' and value is None:
            value = -1
        if value is None:
            logging.warning(f'zigbee {v=} {value=}')
            continue
        ret.append({'name': 'zigbee_' + v, 'labels': [{'labelName': 'dev', 'labelValue': dev}], 'value': value})
    return ret


@topic(r'^zigbee/([^/]+)/availability$')
def zigbee_availability(message, topicGroups: tuple[str]) -> list[dict]:
    dev = topicGroups[0]
    online = message['payload'] == "online"
    ret = [{'name': 'zigbee_online', 'labels': [{'labelName': 'dev', 'labelValue': dev}], 'value': int(online)}]
    if online and dev.startswith('0x'):
        ret.append({'name': 'zigbee_unnamed_dev', 'labels': [{'labelName': 'dev', 'labelValue': dev}], 'value': 1})
    return ret