From cfb6bf1f4d011dd0c3a94c1e1c43a003da7429ec Mon Sep 17 00:00:00 2001
From: liamcottle <liam@liamcottle.com>
Date: Sun, 31 Mar 2024 18:05:31 +1300
Subject: [PATCH] collect and show mqtt state for nodes

---
 .../migration.sql                             |  3 ++
 prisma/schema.prisma                          |  3 ++
 src/mqtt.js                                   | 30 +++++++++++++++++++
 src/public/index.html                         | 10 +++++++
 4 files changed, 46 insertions(+)
 create mode 100644 prisma/migrations/20240331045118_add_mqtt_connection_state_columns_to_nodes_table/migration.sql

diff --git a/prisma/migrations/20240331045118_add_mqtt_connection_state_columns_to_nodes_table/migration.sql b/prisma/migrations/20240331045118_add_mqtt_connection_state_columns_to_nodes_table/migration.sql
new file mode 100644
index 0000000..9ca74de
--- /dev/null
+++ b/prisma/migrations/20240331045118_add_mqtt_connection_state_columns_to_nodes_table/migration.sql
@@ -0,0 +1,3 @@
+-- AlterTable
+ALTER TABLE `nodes` ADD COLUMN `mqtt_connection_state` VARCHAR(191) NULL,
+    ADD COLUMN `mqtt_connection_state_updated_at` DATETIME(3) NULL;
diff --git a/prisma/schema.prisma b/prisma/schema.prisma
index 437247d..754457f 100644
--- a/prisma/schema.prisma
+++ b/prisma/schema.prisma
@@ -36,6 +36,9 @@ model Node {
   neighbours                        Json?
   neighbours_updated_at             DateTime?
 
+  mqtt_connection_state            String?
+  mqtt_connection_state_updated_at DateTime?
+
   created_at DateTime @default(now())
   updated_at DateTime @default(now()) @updatedAt
 
diff --git a/src/mqtt.js b/src/mqtt.js
index b84d60d..f330c10 100644
--- a/src/mqtt.js
+++ b/src/mqtt.js
@@ -225,6 +225,36 @@ client.on("connect", () => {
 client.on("message", async (topic, message) => {
     try {
 
+        // handle node status
+        if(topic.includes("/stat/!")){
+            try {
+
+                // get node id and status
+                const nodeIdHex = topic.split("/").pop();
+                const mqttConnectionState = message.toString();
+
+                // convert node id hex to int value
+                const nodeId = BigInt('0x' + nodeIdHex.replaceAll("!", ""));
+
+                // update mqtt connection state for node
+                await prisma.node.updateMany({
+                    where: {
+                        node_id: nodeId,
+                    },
+                    data: {
+                        mqtt_connection_state: mqttConnectionState,
+                        mqtt_connection_state_updated_at: new Date(),
+                    },
+                });
+
+                // no need to continue with this mqtt message
+                return;
+
+            } catch(e) {
+                console.error(e);
+            }
+        }
+
         // decode service envelope
         const envelope = ServiceEnvelope.decode(message);
         if(!envelope.packet){
diff --git a/src/public/index.html b/src/public/index.html
index bf72746..f40edde 100644
--- a/src/public/index.html
+++ b/src/public/index.html
@@ -2175,9 +2175,19 @@
 
     function getTooltipContentForNode(node) {
 
+        // human friendly connection state
+        var mqttStatus = "???";
+        var mqttStatusLastUpdated = node.mqtt_connection_state_updated_at ? `(${moment(new Date(node.mqtt_connection_state_updated_at)).fromNow()})` : "";
+        if(node.mqtt_connection_state === "online"){
+            mqttStatus = `<span class="text-green-700">Online</span> ${mqttStatusLastUpdated}`;
+        } else if(node.mqtt_connection_state === "offline"){
+            mqttStatus = `<span class="text-red-700">Offline</span> ${mqttStatusLastUpdated}`;
+        }
+
         var tooltip = `<img class="mb-4 w-40 mx-auto" src="/images/devices/${node.hardware_model_name}.png" onerror="this.classList.add('hidden')"/>` +
             `<b>${node.long_name}</b>` +
             `<br/>Short Name: ${node.short_name}` +
+            `<br/>MQTT Status: ${mqttStatus}` +
             `<br/><br/>Role: ${node.role_name}` +
             `<br/>Hardware: ${node.hardware_model_name}`;