2021-06-25 13:55:49 +00:00
|
|
|
const express = require('express');
|
|
|
|
const app = express();
|
|
|
|
const http = require('http');
|
|
|
|
const server = http.createServer(app);
|
|
|
|
const { Server } = require("socket.io");
|
|
|
|
const io = new Server(server);
|
|
|
|
const dayjs = require("dayjs");
|
|
|
|
const {R} = require("redbean-node");
|
|
|
|
const passwordHash = require('password-hash');
|
|
|
|
const jwt = require('jsonwebtoken');
|
|
|
|
const Monitor = require("./model/monitor");
|
|
|
|
|
|
|
|
let stop = false;
|
|
|
|
let interval = 6000;
|
|
|
|
let totalClient = 0;
|
|
|
|
let jwtSecret = null;
|
|
|
|
let loadFromDatabase = true;
|
|
|
|
let monitorList = {};
|
|
|
|
|
|
|
|
(async () => {
|
|
|
|
R.setup('sqlite', {
|
2021-06-27 08:10:55 +00:00
|
|
|
filename: './data/kuma.db'
|
2021-06-25 13:55:49 +00:00
|
|
|
});
|
|
|
|
R.freeze(true)
|
2021-06-27 08:10:55 +00:00
|
|
|
await R.autoloadModels("./server/model");
|
2021-06-25 13:55:49 +00:00
|
|
|
|
|
|
|
await initDatabase();
|
|
|
|
|
|
|
|
app.use('/', express.static("public"));
|
|
|
|
|
|
|
|
io.on('connection', async (socket) => {
|
|
|
|
console.log('a user connected');
|
|
|
|
totalClient++;
|
|
|
|
|
|
|
|
socket.on('disconnect', () => {
|
|
|
|
console.log('user disconnected');
|
|
|
|
totalClient--;
|
|
|
|
});
|
|
|
|
|
|
|
|
// Public API
|
|
|
|
|
2021-06-29 08:06:20 +00:00
|
|
|
/*
|
|
|
|
firstConnect - true = send monitor list + heartbeat list history
|
|
|
|
false = do not send
|
|
|
|
*/
|
2021-06-25 13:55:49 +00:00
|
|
|
socket.on("loginByToken", async (token, callback) => {
|
|
|
|
|
|
|
|
try {
|
|
|
|
let decoded = jwt.verify(token, jwtSecret);
|
|
|
|
|
|
|
|
console.log("Username from JWT: " + decoded.username)
|
|
|
|
|
|
|
|
let user = await R.findOne("user", " username = ? AND active = 1 ", [
|
|
|
|
decoded.username
|
|
|
|
])
|
|
|
|
|
|
|
|
if (user) {
|
|
|
|
await afterLogin(socket, user)
|
|
|
|
|
|
|
|
callback({
|
|
|
|
ok: true,
|
|
|
|
})
|
|
|
|
} else {
|
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: "The user is inactive or deleted."
|
|
|
|
})
|
|
|
|
}
|
|
|
|
} catch (error) {
|
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: "Invalid token."
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
});
|
|
|
|
|
|
|
|
socket.on("login", async (data, callback) => {
|
|
|
|
console.log("Login")
|
|
|
|
|
|
|
|
let user = await R.findOne("user", " username = ? AND active = 1 ", [
|
|
|
|
data.username
|
|
|
|
])
|
|
|
|
|
|
|
|
if (user && passwordHash.verify(data.password, user.password)) {
|
|
|
|
|
|
|
|
await afterLogin(socket, user)
|
|
|
|
|
|
|
|
callback({
|
|
|
|
ok: true,
|
|
|
|
token: jwt.sign({
|
|
|
|
username: data.username
|
|
|
|
}, jwtSecret)
|
|
|
|
})
|
|
|
|
} else {
|
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: "Incorrect username or password."
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
});
|
|
|
|
|
|
|
|
socket.on("logout", async (callback) => {
|
|
|
|
socket.leave(socket.userID)
|
|
|
|
socket.userID = null;
|
|
|
|
callback();
|
|
|
|
});
|
|
|
|
|
|
|
|
// Auth Only API
|
|
|
|
|
|
|
|
socket.on("add", async (monitor, callback) => {
|
|
|
|
try {
|
|
|
|
checkLogin(socket)
|
|
|
|
|
|
|
|
let bean = R.dispense("monitor")
|
|
|
|
bean.import(monitor)
|
|
|
|
bean.user_id = socket.userID
|
|
|
|
await R.store(bean)
|
|
|
|
|
2021-06-27 08:10:55 +00:00
|
|
|
await startMonitor(socket.userID, bean.id);
|
|
|
|
await sendMonitorList(socket);
|
|
|
|
|
2021-06-25 13:55:49 +00:00
|
|
|
callback({
|
|
|
|
ok: true,
|
|
|
|
msg: "Added Successfully.",
|
|
|
|
monitorID: bean.id
|
|
|
|
});
|
|
|
|
|
2021-06-27 08:10:55 +00:00
|
|
|
} 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)
|
|
|
|
}
|
|
|
|
|
2021-06-25 13:55:49 +00:00
|
|
|
await sendMonitorList(socket);
|
|
|
|
|
2021-06-27 08:10:55 +00:00
|
|
|
callback({
|
|
|
|
ok: true,
|
|
|
|
msg: "Saved.",
|
|
|
|
monitorID: bean.id
|
|
|
|
});
|
|
|
|
|
2021-06-25 13:55:49 +00:00
|
|
|
} catch (e) {
|
2021-06-27 08:10:55 +00:00
|
|
|
console.log(e)
|
2021-06-25 13:55:49 +00:00
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: e.message
|
|
|
|
});
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
socket.on("getMonitor", async (monitorID, callback) => {
|
|
|
|
try {
|
|
|
|
checkLogin(socket)
|
|
|
|
|
|
|
|
console.log(`Get Monitor: ${monitorID} User ID: ${socket.userID}`)
|
|
|
|
|
|
|
|
let bean = await R.findOne("monitor", " id = ? AND user_id = ? ", [
|
|
|
|
monitorID,
|
|
|
|
socket.userID,
|
|
|
|
])
|
|
|
|
|
|
|
|
callback({
|
|
|
|
ok: true,
|
|
|
|
monitor: bean.toJSON(),
|
|
|
|
});
|
|
|
|
|
|
|
|
} catch (e) {
|
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: e.message
|
|
|
|
});
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
// Start or Resume the monitor
|
|
|
|
socket.on("resumeMonitor", async (monitorID, callback) => {
|
|
|
|
try {
|
|
|
|
checkLogin(socket)
|
|
|
|
await startMonitor(socket.userID, monitorID);
|
|
|
|
await sendMonitorList(socket);
|
|
|
|
|
|
|
|
callback({
|
|
|
|
ok: true,
|
2021-06-25 19:03:06 +00:00
|
|
|
msg: "Resumed Successfully."
|
2021-06-25 13:55:49 +00:00
|
|
|
});
|
|
|
|
|
|
|
|
} catch (e) {
|
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: e.message
|
|
|
|
});
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
socket.on("pauseMonitor", async (monitorID, callback) => {
|
|
|
|
try {
|
|
|
|
checkLogin(socket)
|
|
|
|
await pauseMonitor(socket.userID, monitorID)
|
|
|
|
await sendMonitorList(socket);
|
|
|
|
|
|
|
|
callback({
|
|
|
|
ok: true,
|
|
|
|
msg: "Paused Successfully."
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
|
|
} catch (e) {
|
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: e.message
|
|
|
|
});
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
socket.on("deleteMonitor", async (monitorID, callback) => {
|
|
|
|
try {
|
|
|
|
checkLogin(socket)
|
|
|
|
|
|
|
|
console.log(`Delete Monitor: ${monitorID} User ID: ${socket.userID}`)
|
|
|
|
|
|
|
|
if (monitorID in monitorList) {
|
|
|
|
monitorList[monitorID].stop();
|
|
|
|
delete monitorList[monitorID]
|
|
|
|
}
|
|
|
|
|
|
|
|
await R.exec("DELETE FROM monitor WHERE id = ? AND user_id = ? ", [
|
|
|
|
monitorID,
|
|
|
|
socket.userID
|
|
|
|
]);
|
|
|
|
|
|
|
|
callback({
|
|
|
|
ok: true,
|
|
|
|
msg: "Deleted Successfully."
|
|
|
|
});
|
|
|
|
|
|
|
|
await sendMonitorList(socket);
|
|
|
|
|
|
|
|
} catch (e) {
|
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: e.message
|
|
|
|
});
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
socket.on("changePassword", async (password, callback) => {
|
|
|
|
try {
|
|
|
|
checkLogin(socket)
|
|
|
|
|
|
|
|
if (! password.currentPassword) {
|
|
|
|
throw new Error("Invalid new password")
|
|
|
|
}
|
|
|
|
|
|
|
|
let user = await R.findOne("user", " id = ? AND active = 1 ", [
|
|
|
|
socket.userID
|
|
|
|
])
|
|
|
|
|
|
|
|
if (user && passwordHash.verify(password.currentPassword, user.password)) {
|
|
|
|
|
|
|
|
await R.exec("UPDATE `user` SET password = ? WHERE id = ? ", [
|
|
|
|
passwordHash.generate(password.newPassword),
|
|
|
|
socket.userID
|
|
|
|
]);
|
|
|
|
|
|
|
|
callback({
|
|
|
|
ok: true,
|
|
|
|
msg: "Password has been updated successfully."
|
|
|
|
})
|
|
|
|
} else {
|
|
|
|
throw new Error("Incorrect current password")
|
|
|
|
}
|
|
|
|
|
|
|
|
} catch (e) {
|
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: e.message
|
|
|
|
});
|
|
|
|
}
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
server.listen(3001, () => {
|
|
|
|
console.log('Listening on 3001');
|
|
|
|
startMonitors();
|
|
|
|
});
|
|
|
|
|
|
|
|
})();
|
|
|
|
|
|
|
|
async function checkOwner(userID, monitorID) {
|
|
|
|
let row = await R.getRow("SELECT id FROM monitor WHERE id = ? AND user_id = ? ", [
|
|
|
|
monitorID,
|
|
|
|
userID,
|
|
|
|
])
|
|
|
|
|
|
|
|
if (! row) {
|
|
|
|
throw new Error("You do not own this monitor.");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
async function sendMonitorList(socket) {
|
2021-06-29 08:06:20 +00:00
|
|
|
let list = await getMonitorJSONList(socket.userID);
|
|
|
|
io.to(socket.userID).emit("monitorList", list)
|
|
|
|
return list;
|
2021-06-25 13:55:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
async function afterLogin(socket, user) {
|
|
|
|
socket.userID = user.id;
|
|
|
|
socket.join(user.id)
|
2021-06-29 08:06:20 +00:00
|
|
|
|
|
|
|
let monitorList = await sendMonitorList(socket)
|
|
|
|
|
|
|
|
for (let monitorID in monitorList) {
|
|
|
|
await sendHeartbeatList(socket, monitorID);
|
|
|
|
}
|
2021-06-25 13:55:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
async function getMonitorJSONList(userID) {
|
2021-06-27 08:10:55 +00:00
|
|
|
let result = {};
|
2021-06-25 13:55:49 +00:00
|
|
|
|
2021-06-27 08:10:55 +00:00
|
|
|
let monitorList = await R.find("monitor", " user_id = ? ORDER BY active DESC, name ASC ", [
|
2021-06-25 13:55:49 +00:00
|
|
|
userID
|
|
|
|
])
|
|
|
|
|
|
|
|
for (let monitor of monitorList) {
|
2021-06-27 08:10:55 +00:00
|
|
|
result[monitor.id] = monitor.toJSON();
|
2021-06-29 08:06:20 +00:00
|
|
|
|
2021-06-25 13:55:49 +00:00
|
|
|
}
|
|
|
|
|
2021-06-29 08:06:20 +00:00
|
|
|
|
|
|
|
|
2021-06-25 13:55:49 +00:00
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
function checkLogin(socket) {
|
|
|
|
if (! socket.userID) {
|
|
|
|
throw new Error("You are not logged in.");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
async function initDatabase() {
|
|
|
|
let jwtSecretBean = await R.findOne("setting", " `key` = ? ", [
|
|
|
|
"jwtSecret"
|
|
|
|
]);
|
|
|
|
|
|
|
|
if (! jwtSecretBean) {
|
|
|
|
console.log("JWT secret is not found, generate one.")
|
|
|
|
jwtSecretBean = R.dispense("setting")
|
|
|
|
jwtSecretBean.key = "jwtSecret"
|
|
|
|
|
|
|
|
jwtSecretBean.value = passwordHash.generate(dayjs() + "")
|
|
|
|
await R.store(jwtSecretBean)
|
|
|
|
} else {
|
|
|
|
console.log("Load JWT secret from database.")
|
|
|
|
}
|
|
|
|
|
|
|
|
jwtSecret = jwtSecretBean.value;
|
|
|
|
}
|
|
|
|
|
|
|
|
async function startMonitor(userID, monitorID) {
|
|
|
|
await checkOwner(userID, monitorID)
|
|
|
|
|
|
|
|
console.log(`Resume Monitor: ${monitorID} User ID: ${userID}`)
|
|
|
|
|
|
|
|
await R.exec("UPDATE monitor SET active = 1 WHERE id = ? AND user_id = ? ", [
|
|
|
|
monitorID,
|
|
|
|
userID
|
|
|
|
]);
|
|
|
|
|
|
|
|
let monitor = await R.findOne("monitor", " id = ? ", [
|
|
|
|
monitorID
|
|
|
|
])
|
|
|
|
|
2021-06-27 08:10:55 +00:00
|
|
|
if (monitor.id in monitorList) {
|
|
|
|
monitorList[monitor.id].stop();
|
|
|
|
}
|
|
|
|
|
2021-06-25 13:55:49 +00:00
|
|
|
monitorList[monitor.id] = monitor;
|
|
|
|
monitor.start(io)
|
|
|
|
}
|
|
|
|
|
2021-06-27 08:10:55 +00:00
|
|
|
async function restartMonitor(userID, monitorID) {
|
|
|
|
return await startMonitor(userID, monitorID)
|
|
|
|
}
|
|
|
|
|
2021-06-25 13:55:49 +00:00
|
|
|
async function pauseMonitor(userID, monitorID) {
|
|
|
|
await checkOwner(userID, monitorID)
|
|
|
|
|
|
|
|
console.log(`Pause Monitor: ${monitorID} User ID: ${userID}`)
|
|
|
|
|
|
|
|
await R.exec("UPDATE monitor SET active = 0 WHERE id = ? AND user_id = ? ", [
|
|
|
|
monitorID,
|
|
|
|
userID
|
|
|
|
]);
|
|
|
|
|
|
|
|
if (monitorID in monitorList) {
|
|
|
|
monitorList[monitorID].stop();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Resume active monitors
|
|
|
|
*/
|
|
|
|
async function startMonitors() {
|
|
|
|
let list = await R.find("monitor", " active = 1 ")
|
|
|
|
|
|
|
|
for (let monitor of list) {
|
|
|
|
monitor.start(io)
|
|
|
|
monitorList[monitor.id] = monitor;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-06-29 08:06:20 +00:00
|
|
|
/**
|
|
|
|
* Send Heartbeat History list to socket
|
|
|
|
*/
|
|
|
|
async function sendHeartbeatList(socket, monitorID) {
|
|
|
|
let list = await R.find("heartbeat", `
|
|
|
|
monitor_id = ?
|
|
|
|
ORDER BY time DESC
|
|
|
|
LIMIT 100
|
|
|
|
`, [
|
|
|
|
monitorID
|
|
|
|
])
|
|
|
|
|
|
|
|
let result = [];
|
|
|
|
|
|
|
|
for (let bean of list) {
|
|
|
|
result.unshift(bean.toJSON())
|
|
|
|
}
|
|
|
|
|
|
|
|
socket.emit("heartbeatList", monitorID, result)
|
|
|
|
}
|
|
|
|
|