From 8f44e7d13a6a190277597ee416131573f3be5401 Mon Sep 17 00:00:00 2001
From: marcsello <punkosdmarcell@rocketmail.com>
Date: Wed, 30 Sep 2020 04:08:36 +0200
Subject: [PATCH] Implemented mqtt stuff

---
 requirements.txt         |  2 ++
 src/actuator/__init__.py |  1 +
 src/app.py               | 36 ++++++++++++++++++++++++++++++++++++
 src/utils/config.py      |  6 ++++++
 4 files changed, 45 insertions(+)

diff --git a/requirements.txt b/requirements.txt
index 9ca9f42..d809b2c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,6 +3,8 @@ requests
 
 cython
 
+paho-mqtt
+
 cycler==0.10.0
 deprecation==2.0.7
 eyeD3==0.9.5
diff --git a/src/actuator/__init__.py b/src/actuator/__init__.py
index e69de29..6a3722d 100644
--- a/src/actuator/__init__.py
+++ b/src/actuator/__init__.py
@@ -0,0 +1 @@
+from .loudspeaker import Loudspeaker
\ No newline at end of file
diff --git a/src/app.py b/src/app.py
index ae9ac16..f1b19a1 100644
--- a/src/app.py
+++ b/src/app.py
@@ -5,6 +5,9 @@ from sentry_sdk.integrations.logging import LoggingIntegration
 from utils import config, LoopingTimer
 from signal_processor import SoundSignalProcessor
 from birbnetes_iot_platform_raspberry import BirbnetesIoTPlatformStatusDriver
+from actuator import Loudspeaker
+import paho.mqtt.client
+import json
 
 """
 Main Entrypoint
@@ -42,6 +45,28 @@ def timer_tick(*args) -> None:
         abcsignaprocessor.processcurrentsignal()
 
 
+def mqtt_on_connect(client, userdata, flags, rc):
+    BirbnetesIoTPlatformStatusDriver.enqueue_pattern('green', [1])
+    client.subscribe(f"command/{config.DEVICE_ID}")
+    logging.error("MQTT Connected!")
+
+
+def mqtt_on_disconnect(client, userdata, rc):
+    BirbnetesIoTPlatformStatusDriver.enqueue_pattern('red', [1])
+    logging.error("Unexpected MQTT Disconnect")
+
+
+def mqtt_on_command(client, userdata, message):
+    try:
+        msg = json.loads(message.payload.decode())
+    except (UnicodeError, json.JSONDecodeError) as e:
+        logging.error(f"MQTT Invalid message recieved: {e}")
+        return
+
+    if msg.get("command") == 'doAlert':
+        userdata.act()
+
+
 def main() -> None:
     """
     Main function
@@ -52,6 +77,17 @@ def main() -> None:
     loopingtimer = LoopingTimer(function=timer_tick, args=[listofabcsignaprocessors], interval=config.TICK_INTERVAL)
     loopingtimer.start()
 
+    client = paho.mqtt.client.Client(userdata=Loudspeaker(config.ENEMY_SOUNDS), client_id=config.DEVICE_ID)
+    client.on_connect = mqtt_on_connect
+    client.on_disconnect = mqtt_on_disconnect
+    client.on_message = mqtt_on_command
+
+    if config.MQTT_USERNAME:
+        client.username_pw_set(config.MQTT_USERNAME, config.MQTT_PASSWORD)
+
+    client.connect(config.MQTT_HOSTNAME, config.MQTT_PORT, 60)
+    client.loop_forever()
+
 
 if __name__ == "__main__":
     main()
diff --git a/src/utils/config.py b/src/utils/config.py
index 7ac632e..8556deb 100644
--- a/src/utils/config.py
+++ b/src/utils/config.py
@@ -19,6 +19,12 @@ DEVICE_ID = os.environ.get("DEVICE_ID", "devraspi")
 TICK_INTERVAL = float(os.environ.get("TICK_INTERVAL", 0.4))
 SAMPLE_LENGTH = float(os.environ.get("SAMPLE_LENGTH", 1))
 
+ENEMY_SOUNDS = os.environ.get("ENEMY_SOUNDS", '/var/lib/birbnetes/enemy_sounds')
+
+MQTT_HOSTNAME = os.getenv("GUARD_MQTT_HOSTNAME", "localhost")
+MQTT_PORT = int(os.getenv("GUARD_MQTT_PORT", 1883))
+MQTT_USERNAME = os.getenv("GUARD_MQTT_USERNAME", None)
+MQTT_PASSWORD = os.getenv("GUARD_MQTT_PASSWORD", None)
 
 TARGET_NAME = os.environ.get("TARGET_CLASS_NAME")
 MODEL_ID = os.environ.get("MODEL_ID")