From 692a11e51e8f93672659d6d066c9ccbdda97234d Mon Sep 17 00:00:00 2001
From: LouisLam <louislam@users.noreply.github.com>
Date: Tue, 10 Aug 2021 17:51:30 +0800
Subject: [PATCH] pass tls info to prometheus.update

---
 server/model/monitor.js | 12 +++++++++---
 server/prometheus.js    | 36 ++++++++++++++++++++----------------
 2 files changed, 29 insertions(+), 19 deletions(-)

diff --git a/server/model/monitor.js b/server/model/monitor.js
index 2520c282..ce8a249c 100644
--- a/server/model/monitor.js
+++ b/server/model/monitor.js
@@ -79,6 +79,10 @@ class Monitor extends BeanModel {
 
         const beat = async () => {
 
+            // Expose here for prometheus update
+            // undefined if not https
+            let tlsInfo = undefined;
+
             if (! previousBeat) {
                 previousBeat = await R.findOne("heartbeat", " monitor_id = ? ORDER BY time DESC", [
                     this.id,
@@ -132,7 +136,7 @@ class Monitor extends BeanModel {
                     let certInfoStartTime = dayjs().valueOf();
                     if (this.getUrl()?.protocol === "https:") {
                         try {
-                            await this.updateTlsInfo(checkCertificate(res));
+                            tlsInfo = await this.updateTlsInfo(checkCertificate(res));
                         } catch (e) {
                             if (e.message !== "No TLS certificate in response") {
                                 console.error(e.message)
@@ -254,7 +258,7 @@ class Monitor extends BeanModel {
                 console.warn(`Monitor #${this.id} '${this.name}': Failing: ${bean.msg} | Type: ${this.type}`)
             }
 
-            prometheus.update(bean)
+            prometheus.update(bean, tlsInfo)
 
             io.to(this.user_id).emit("heartbeat", bean.toJSON());
 
@@ -289,7 +293,7 @@ class Monitor extends BeanModel {
     /**
      * Store TLS info to database
      * @param checkCertificateResult
-     * @returns {Promise<void>}
+     * @returns {Promise<object>}
      */
     async updateTlsInfo(checkCertificateResult) {
         let tls_info_bean = await R.findOne("monitor_tls_info", "monitor_id = ?", [
@@ -301,6 +305,8 @@ class Monitor extends BeanModel {
         }
         tls_info_bean.info_json = JSON.stringify(checkCertificateResult);
         await R.store(tls_info_bean);
+
+        return checkCertificateResult;
     }
 
     static async sendStats(io, monitorID, userID) {
diff --git a/server/prometheus.js b/server/prometheus.js
index a6b4d7aa..21b83b34 100644
--- a/server/prometheus.js
+++ b/server/prometheus.js
@@ -1,33 +1,33 @@
-const PrometheusClient = require('prom-client');
+const PrometheusClient = require("prom-client");
 
 const commonLabels = [
-    'monitor_name',
-    'monitor_type',
-    'monitor_url',
-    'monitor_hostname',
-    'monitor_port',
+    "monitor_name",
+    "monitor_type",
+    "monitor_url",
+    "monitor_hostname",
+    "monitor_port",
 ]
 
 const monitor_cert_days_remaining = new PrometheusClient.Gauge({
-    name: 'monitor_cert_days_remaining',
-    help: 'The number of days remaining until the certificate expires',
+    name: "monitor_cert_days_remaining",
+    help: "The number of days remaining until the certificate expires",
     labelNames: commonLabels
 });
 
 const monitor_cert_is_valid = new PrometheusClient.Gauge({
-    name: 'monitor_cert_is_valid',
-    help: 'Is the certificate still valid? (1 = Yes, 0= No)',
+    name: "monitor_cert_is_valid",
+    help: "Is the certificate still valid? (1 = Yes, 0= No)",
     labelNames: commonLabels
 });
 const monitor_response_time = new PrometheusClient.Gauge({
-    name: 'monitor_response_time',
-    help: 'Monitor Response Time (ms)',
+    name: "monitor_response_time",
+    help: "Monitor Response Time (ms)",
     labelNames: commonLabels
 });
 
 const monitor_status = new PrometheusClient.Gauge({
-    name: 'monitor_status',
-    help: 'Monitor Status (1 = UP, 0= DOWN)',
+    name: "monitor_status",
+    help: "Monitor Status (1 = UP, 0= DOWN)",
     labelNames: commonLabels
 });
 
@@ -44,7 +44,11 @@ class Prometheus {
         }
     }
 
-    update(heartbeat) {
+    update(heartbeat, tlsInfo) {
+
+        // TODO: TLS Info here
+        console.log(tlsInfo)
+
         try {
             monitor_status.set(this.monitorLabelValues, heartbeat.status)
         } catch (e) {
@@ -52,7 +56,7 @@ class Prometheus {
         }
 
         try {
-            if (typeof heartbeat.ping === 'number') {
+            if (typeof heartbeat.ping === "number") {
                 monitor_response_time.set(this.monitorLabelValues, heartbeat.ping)
             } else {
                 // Is it good?