import time

from bluepy.btle import Scanner, DefaultDelegate
from mqtt import MqttMessage

from workers.base import BaseWorker
import logger

REQUIREMENTS = ['bluepy']
_LOGGER = logger.get(__name__)


class ScanDelegate(DefaultDelegate):
  def __init__(self):
    DefaultDelegate.__init__(self)

  def handleDiscovery(self, dev, isNewDev, isNewData):
    if isNewDev:
      _LOGGER.debug("Discovered new device: %s" % dev.addr)

class ToothbrushWorker(BaseWorker):
  def searchmac(self, devices, mac):
    for dev in devices:
      if dev.addr == mac.lower():
         return dev

    return None

  def status_update(self):
    scanner = Scanner().withDelegate(ScanDelegate())
    devices = scanner.scan(5.0)
    ret = []

    for name, mac in self.devices.items():
      device = self.searchmac(devices, mac)
      if device is None:
        ret.append(MqttMessage(topic=self.format_topic(name+'/presence'), payload="0"))
      else:
        ret.append(MqttMessage(topic=self.format_topic(name+'/presence/rssi'), payload=device.rssi))
        ret.append(MqttMessage(topic=self.format_topic(name+'/presence'), payload="1"))
        _LOGGER.debug("text: %s" % device.getValueText(255) )
        bytes_ = bytearray(bytes.fromhex(device.getValueText(255)))
        ret.append(MqttMessage(topic=self.format_topic(name+'/running'), payload=bytes_[5] ))
        ret.append(MqttMessage(topic=self.format_topic(name+'/pressure'), payload=bytes_[6] ))
        ret.append(MqttMessage(topic=self.format_topic(name+'/time'), payload=bytes_[7]*60 + bytes_[8] ))
        ret.append(MqttMessage(topic=self.format_topic(name+'/mode'), payload=bytes_[9] ))
        ret.append(MqttMessage(topic=self.format_topic(name+'/quadrant'), payload=bytes_[10] ))

    return ret