1 files changed, 61 insertions, 0 deletions
diff --git a/src/home/mqtt/mqtt.py b/src/home/mqtt/mqtt.py
new file mode 100644
index 0000000..b360d22
--- /dev/null
+++ b/src/home/mqtt/mqtt.py
@@ -0,0 +1,61 @@
+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))