From 06c6cebf1debf93f08dc51d8c8eabf7f0a1d721f Mon Sep 17 00:00:00 2001
From: Korina Cordero <cordero.korina@gmail.com>
Date: Thu, 10 Feb 2022 09:16:26 +0000
Subject: [PATCH] Add checking for mqtt server. #646

---
 utils/server_status/check_server_status.py | 46 ++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/utils/server_status/check_server_status.py b/utils/server_status/check_server_status.py
index 3eebd85c..85748e47 100644
--- a/utils/server_status/check_server_status.py
+++ b/utils/server_status/check_server_status.py
@@ -1,5 +1,8 @@
 import sys
 import socket
+import time
+import ssl
+import paho.mqtt.client as mqtt
 
 # TODO: make port a constant? 80 for http
 # 443 for https
@@ -23,8 +26,51 @@ def servertest(argv):
             s.close()
             return True
 
+def on_log(client, userdata, level, buf):
+    print("log: ", buf)
+
+def on_connect(client, userdata, flags, rc):
+    if rc == 0:
+        client.connected_flag = True
+        print("connected to mqtt server")
+    else:
+        client.bad_connection_flag = True
+        if rc == 5:
+            print("broker requires authentication")
+
+def on_message(client, userdata, message):
+    print("message received " ,str(message.payload.decode("utf-8")))
+    print("message topic=",message.topic)
+    print("message qos=",message.qos)
+    print("message retain flag=",message.retain)
+
 if __name__ == "__main__":
     if servertest(sys.argv):
         print("Server is up")
     else:
         print("Server is down")
+
+broker_address = "205.185.116.243"
+print ("creating new mqtt client instance")
+client = mqtt.Client("dev-client")
+
+# initialize flags
+client.connected_flag = False
+client.bad_connection_flag = False
+
+
+client.on_log = on_log
+client.on_connect = on_connect
+client.tls_set(ca_certs="/root/mqtt_certs/chain.pem", certfile="/root/mqtt_certs/cert.pem", tls_version=ssl.PROTOCOL_TLSv1_2)
+client.on_message=on_message        #attach function to callback
+print ("connecting to broker")
+
+client.connect(broker_address, 8883)
+client.loop_start()
+#print("Subscribing to topic", "/rider/+/location")
+#client.subscribe("/rider/+/location")
+
+time.sleep(4)
+client.loop_stop()
+
+