From 96c60dd94aa94c09ddd3039e61ea8eed433a082b Mon Sep 17 00:00:00 2001
From: Nelson Chan <chakhong.chan@aecom.com>
Date: Thu, 22 Jul 2021 16:04:32 +0800
Subject: [PATCH] Feat: Add database storage for TLS info

---
 db/patch2.sql           |  9 +++++++++
 server/database.js      |  2 +-
 server/model/monitor.js | 44 +++++++++++++++++++++++++++++++++--------
 src/mixins/socket.js    | 14 +++++++++++--
 src/pages/Details.vue   |  2 +-
 5 files changed, 59 insertions(+), 12 deletions(-)
 create mode 100644 db/patch2.sql

diff --git a/db/patch2.sql b/db/patch2.sql
new file mode 100644
index 00000000..012d0150
--- /dev/null
+++ b/db/patch2.sql
@@ -0,0 +1,9 @@
+BEGIN TRANSACTION;
+
+CREATE TABLE monitor_tls_info (
+	id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
+	monitor_id INTEGER NOT NULL,
+	info_json TEXT
+);
+
+COMMIT;
diff --git a/server/database.js b/server/database.js
index 49659e61..eef3587e 100644
--- a/server/database.js
+++ b/server/database.js
@@ -8,7 +8,7 @@ class Database {
 
     static templatePath = "./db/kuma.db"
     static path =  './data/kuma.db';
-    static latestVersion = 1;
+    static latestVersion = 2;
     static noReject = true;
 
     static async patch() {
diff --git a/server/model/monitor.js b/server/model/monitor.js
index 6c5f86f6..7afb82b6 100644
--- a/server/model/monitor.js
+++ b/server/model/monitor.js
@@ -74,8 +74,10 @@ class Monitor extends BeanModel {
                     })
                     bean.msg = `${res.status} - ${res.statusText}`
                     bean.ping = dayjs().valueOf() - startTime;
-                    if (this.url.startsWith("https")) {
-                        Monitor.sendCertInfo(checkCertificate(res), io, this.id, this.user_id);
+
+                    // Check certificate if https is used
+                    if (this.getUrl().protocol === "https:") {
+                        await this.updateTlsInfo(checkCertificate(res));
                     }
 
                     if (this.type === "http") {
@@ -168,10 +170,35 @@ class Monitor extends BeanModel {
         clearInterval(this.heartbeatInterval)
     }
 
+    // Helper Method: 
+    // returns URL object for further usage
+    // returns null if url is invalid
+    getUrl() {
+        try {
+            return new URL(this.url);
+        } catch (_) {
+            return null;
+        }
+    }
+
+    // Store TLS info to database
+    async updateTlsInfo(checkCertificateResult) {
+        let tls_info_bean = await R.findOne("monitor_tls_info", "monitor_id = ?", [
+            this.id
+        ]);
+        if (tls_info_bean == null) {
+            tls_info_bean = R.dispense("monitor_tls_info");
+            tls_info_bean.monitor_id = this.id;
+        }
+        tls_info_bean.info_json = JSON.stringify(checkCertificateResult);
+        R.store(tls_info_bean);
+    }
+
     static async sendStats(io, monitorID, userID) {
         Monitor.sendAvgPing(24, io, monitorID, userID);
         Monitor.sendUptime(24, io, monitorID, userID);
         Monitor.sendUptime(24 * 30, io, monitorID, userID);
+        Monitor.sendCertInfo(io, monitorID, userID);
     }
 
     /**
@@ -192,12 +219,13 @@ class Monitor extends BeanModel {
         io.to(userID).emit("avgPing", monitorID, avgPing);
     }
 
-    /**
-     *
-     * @param checkCertificateResult : Object return result of checkCertificate
-     */
-     static async sendCertInfo(checkCertificateResult, io, monitorID, userID) {
-        io.to(userID).emit("certInfo", monitorID, checkCertificateResult);
+    static async sendCertInfo(io, monitorID, userID) {
+         let tls_info = await R.findOne("monitor_tls_info", "monitor_id = ?", [
+            monitorID
+        ]);
+        if (tls_info != null) {
+            io.to(userID).emit("certInfo", monitorID, tls_info.info_json);
+        }
     }
 
     /**
diff --git a/src/mixins/socket.js b/src/mixins/socket.js
index 14f78c87..26a4611e 100644
--- a/src/mixins/socket.js
+++ b/src/mixins/socket.js
@@ -59,7 +59,17 @@ export default {
             this.$router.push("/setup")
         });
 
-        socket.on('monitorList', (data) => {
+        socket.on("monitorList", (data) => {
+            // Add Helper function
+            Object.entries(data).forEach(([monitorID, monitor]) => {
+                monitor.getUrl = () => {
+                    try {
+                        return new URL(monitor.url);
+                    } catch (_) {
+                        return null;
+                    }
+                };
+            });
             this.monitorList = data;
         });
 
@@ -116,7 +126,7 @@ export default {
         });
 
         socket.on('certInfo', (monitorID, data) => {
-            this.certInfoList[monitorID] = data
+            this.certInfoList[monitorID] = JSON.parse(data)
         });
 
         socket.on('importantHeartbeatList', (monitorID, data) => {
diff --git a/src/pages/Details.vue b/src/pages/Details.vue
index 727f0aab..a2952aa0 100644
--- a/src/pages/Details.vue
+++ b/src/pages/Details.vue
@@ -54,7 +54,7 @@
         </div>
     </div>
 
-    <div class="shadow-box big-padding text-center stats" v-if="monitor.type === 'http' && monitor.url.startsWith('https') && certInfo != null">
+    <div class="shadow-box big-padding text-center stats" v-if="monitor.type === 'http' && monitor.getUrl().protocol === 'https:' && certInfo != null">
         <div class="row">
             <div class="col">
                 <h4>Certificate Info</h4>