many update

server/model/monitor.js

@@ -1,6 +1,18 @@
 const dayjs = require("dayjs");
+const utc = require('dayjs/plugin/utc')
+var timezone = require('dayjs/plugin/timezone')
+dayjs.extend(utc)
+dayjs.extend(timezone)
+const axios = require("axios");
+const {R} = require("redbean-node");
 const {BeanModel} = require("redbean-node/dist/bean-model");
 
+
+/**
+ * status:
+ *      0 = DOWN
+ *      1 = UP
+ */
 class Monitor extends BeanModel {
 
     toJSON() {
@@ -16,9 +28,36 @@ class Monitor extends BeanModel {
     }
 
     start(io) {
-        const beat = () => {
+        const beat = async () => {
             console.log(`Monitor ${this.id}: Heartbeat`)
-            io.to(this.user_id).emit("heartbeat", dayjs().unix());
+
+            let bean = R.dispense("heartbeat")
+            bean.monitor_id = this.id;
+            bean.time = R.isoDateTime(dayjs.utc());
+            bean.status = 0;
+
+            try {
+                if (this.type === "http") {
+                    let startTime = dayjs().valueOf();
+                    let res = await axios.get(this.url)
+                    bean.msg = `${res.status} - ${res.statusText}`
+                    bean.ping = dayjs().valueOf() - startTime;
+                    bean.status = 1;
+                }
+
+            } catch (error) {
+                bean.msg = error.message;
+            }
+
+            io.to(this.user_id).emit("heartbeat", {
+                monitorID: this.id,
+                status: bean.status,
+                time: bean.time,
+                msg: bean.msg,
+                ping: bean.ping,
+            });
+
+            await R.store(bean)
         }
 
         beat();

server/server.js

@@ -4,7 +4,6 @@ const http = require('http');
 const server = http.createServer(app);
 const { Server } = require("socket.io");
 const io = new Server(server);
-const axios = require('axios');
 const dayjs = require("dayjs");
 const {R} = require("redbean-node");
 const passwordHash = require('password-hash');
@@ -12,7 +11,6 @@ const jwt = require('jsonwebtoken');
 const Monitor = require("./model/monitor");
 const {sleep} = require("./util");
 
-
 let stop = false;
 let interval = 6000;
 let totalClient = 0;
@@ -21,12 +19,11 @@ let loadFromDatabase = true;
 let monitorList = {};
 
 (async () => {
-
     R.setup('sqlite', {
-        filename: '../data/kuma.db'
+        filename: './data/kuma.db'
     });
     R.freeze(true)
-    await R.autoloadModels("./model");
+    await R.autoloadModels("./server/model");
 
     await initDatabase();
 
@@ -118,15 +115,54 @@ let monitorList = {};
                 bean.user_id = socket.userID
                 await R.store(bean)
 
+                await startMonitor(socket.userID, bean.id);
+                await sendMonitorList(socket);
+
                 callback({
                     ok: true,
                     msg: "Added Successfully.",
                     monitorID: bean.id
                 });
 
+            } catch (e) {
+                callback({
+                    ok: false,
+                    msg: e.message
+                });
+            }
+        });
+
+        socket.on("editMonitor", async (monitor, callback) => {
+            try {
+                checkLogin(socket)
+
+                let bean = await R.findOne("monitor", " id = ? ", [ monitor.id ])
+
+                if (bean.user_id !== socket.userID) {
+                    throw new Error("Permission denied.")
+                }
+
+                bean.name = monitor.name
+                bean.type = monitor.type
+                bean.url = monitor.url
+                bean.interval = monitor.interval
+
+                await R.store(bean)
+
+                if (bean.active) {
+                    await restartMonitor(socket.userID, bean.id)
+                }
+
                 await sendMonitorList(socket);
 
+                callback({
+                    ok: true,
+                    msg: "Saved.",
+                    monitorID: bean.id
+                });
+
             } catch (e) {
+                console.log(e)
                 callback({
                     ok: false,
                     msg: e.message
@@ -294,14 +330,14 @@ async function afterLogin(socket, user) {
 }
 
 async function getMonitorJSONList(userID) {
-    let result = [];
+    let result = {};
 
-    let monitorList = await R.find("monitor", " user_id = ? ORDER BY weight DESC ", [
+    let monitorList = await R.find("monitor", " user_id = ? ORDER BY active DESC, name ASC ", [
         userID
     ])
 
     for (let monitor of monitorList) {
-        result.push(monitor.toJSON())
+        result[monitor.id] = monitor.toJSON();
     }
 
     return result;
@@ -346,10 +382,18 @@ async function startMonitor(userID, monitorID) {
         monitorID
     ])
 
+    if (monitor.id in monitorList) {
+        monitorList[monitor.id].stop();
+    }
+
     monitorList[monitor.id] = monitor;
     monitor.start(io)
 }
 
+async function restartMonitor(userID, monitorID) {
+    return await startMonitor(userID, monitorID)
+}
+
 async function pauseMonitor(userID, monitorID) {
     await checkOwner(userID, monitorID)