import paho.mqtt.publish as publish
import paho.mqtt.client as mqtt

class MqttClient:

  #print({client_id})

  def __init__(self, config):
    self._config = config
    self._mqttc = mqtt.Client(client_id=self.client_id, clean_session=False)
    
    if self.username is not None and self.password is not None:
      self._auth = {'username': self.username, 'password': self.password}
    else:
      self._auth = None

  def publish(self, messages):
    if messages is None or len(messages) == 0:
      return
    
    if self.topic_prefix is not None:
      for m in messages:
        m.topic = "{}/{}".format(self.topic_prefix, m.topic)
    
    publish.multiple(list(map(lambda m: m.as_dict, messages)),
                     hostname=self.hostname,
                     port=self.port,
                     auth=self._auth)

  @property
  def client_id(self):
    return self._config['client_id'] if 'client_id' in self._config else 'bt-mqtt-gateway'

  @property
  def hostname(self):
    return self._config['host']

  @property
  def port(self):
    return self._config['port'] if 'port' in self._config else 1883

  @property
  def username(self):
    return self._config['username'] if 'username' in self._config else None

  @property
  def password(self):
    return self._config['password'] if 'password' in self._config else None

  @property
  def topic_prefix(self):
    return self._config['topic_prefix'] if 'topic_prefix' in self._config else None

  @property
  def mqttc(self):
    return self._mqttc

  def callbacks_subscription(self, callbacks):
    if self.username is not None and self.password is not None:
      self.mqttc.username_pw_set(self.username, self.password)
    self.mqttc.connect(self.hostname, port=self.port)

    for topic, callback in callbacks:
      self.mqttc.message_callback_add(topic, callback)
      self.mqttc.subscribe(topic)

    self.mqttc.loop_start()


class MqttMessage:
  def __init__(self, topic=None, payload=None):
    self._topic = topic
    self._payload = payload

  @property
  def topic(self):
    return self._topic

  @topic.setter
  def topic(self, new_topic):
    self._topic = new_topic

  @property
  def payload(self):
    return self._payload

  @property
  def as_dict(self):
    return {
      'topic': self.topic,
      'payload': self.payload
    }

  def __repr__(self):
    return self.as_dict.__str__()

  def __str__(self):
    return self.__repr__()