import os.path
import paho.mqtt.client as mqtt
import ssl
import logging

from typing import Tuple
from ..config import config

logger = logging.getLogger(__name__)


def username_and_password() -> Tuple[str, str]:
    username = config['mqtt']['username'] if 'username' in config['mqtt'] else None
    password = config['mqtt']['password'] if 'password' in config['mqtt'] else None
    return username, password


class MQTTBase:
    def __init__(self, clean_session=True):
        self.client = mqtt.Client(client_id=config['mqtt']['client_id'],
                                  protocol=mqtt.MQTTv311,
                                  clean_session=clean_session)
        self.client.on_connect = self.on_connect
        self.client.on_disconnect = self.on_disconnect
        self.client.on_message = self.on_message

        self.home_id = 1

        username, password = username_and_password()
        if username and password:
            self.client.username_pw_set(username, password)

    def configure_tls(self):
        ca_certs = os.path.realpath(os.path.join(
            os.path.dirname(os.path.realpath(__file__)),
            '..',
            '..',
            '..',
            'assets',
            'mqtt_ca.crt'
        ))
        self.client.tls_set(ca_certs=ca_certs, cert_reqs=ssl.CERT_REQUIRED, tls_version=ssl.PROTOCOL_TLSv1_2)

    def connect_and_loop(self, loop_forever=True):
        host = config['mqtt']['host']
        port = config['mqtt']['port']

        self.client.connect(host, port, 60)
        if loop_forever:
            self.client.loop_forever()
        else:
            self.client.loop_start()

    def on_connect(self, client: mqtt.Client, userdata, flags, rc):
        logger.info("Connected with result code " + str(rc))

    def on_disconnect(self, client: mqtt.Client, userdata, rc):
        logger.info("Disconnected with result code " + str(rc))

    def on_message(self, client: mqtt.Client, userdata, msg):
        logger.info(msg.topic + ": " + str(msg.payload))