from builtins import staticmethod

from interruptingcow import timeout
from mqtt import MqttMessage
from workers.base import BaseWorker

REQUIREMENTS = ['python-eq3bt']

monitoredAttrs = ["low_battery", "valve_state", "target_temperature"]

STATE_BOOST = 'boost'
STATE_AWAY = 'away'
STATE_MANUAL = 'manual'

class ThermostatWorker(BaseWorker):
  class ModesMapper():
    def __init__(self):
      from eq3bt import Mode

      self._mapped_modes = {
        Mode.Closed: 'off',
        Mode.Open: 'on',
        Mode.Auto: 'auto',
        Mode.Manual: STATE_MANUAL,
        Mode.Away: STATE_AWAY,
        Mode.Boost: STATE_BOOST,
      }

      self._reverse_modes = {v: k for k, v in self._mapped_modes.items()}

    def get_mapping(self, mode):
      if mode < 0:
        return None
      return self._mapped_modes[mode]

    def get_reverse_mapping(self, mode):
      return self._reverse_modes[mode]

    @staticmethod
    def away_mode_on_off(mode):
      if mode == STATE_AWAY:
        return 'ON'
      else:
        return 'OFF'

    @staticmethod
    def on_off_to_mode(on_off):
      if on_off == 'ON':
        return STATE_AWAY
      else:
        return 'auto'


  def _setup(self):
    from eq3bt import Thermostat

    for name, mac in self.devices.items():
      self.devices[name] = Thermostat(mac)

    self._modes_mapper = self.ModesMapper()

  def status_update(self):
    ret = []
    for name, thermostat in self.devices.items():
      try:
        ret += self.update_device_state(name, thermostat)
      except RuntimeError:
        pass

    return ret

  def on_command(self, topic, value):
    _, device_name, method, _ = topic.split('/')
    thermostat = self.devices[device_name]

    value = value.decode('utf-8')

    if method == STATE_AWAY:
      method = "mode"
      value = ModesMapper.on_off_to_mode(value)

    # It needs to be on separate if because first if can change method
    if method == "mode":
      value = self._modes_mapper.get_reverse_mapping(value)
    elif method == "target_temperature":
      value = float(value)

    setattr(thermostat, method, value)
    return self.update_device_state(device_name, thermostat)

  @timeout(8.0)
  def update_device_state(self, name, thermostat):
    thermostat.update()

    ret = []
    for attr in monitoredAttrs:
      ret.append(MqttMessage(topic=self.format_topic(name, attr), payload=getattr(thermostat, attr)))

    ret.append(MqttMessage(topic=self.format_topic(name, 'mode'), payload=self._modes_mapper.get_mapping(thermostat.mode)))
    ret.append(MqttMessage(topic=self.format_topic(name, 'away'), payload=self.ModesMapper.away_mode_on_off(thermostat.mode)))

    return ret

  def device_for(self, mac):
    return eq3bt.Thermostat(mac)