uptime-kuma/server/socket-handlers/status-page-socket-handler.js

323 lines
9.7 KiB
JavaScript
Raw Normal View History

2021-09-16 07:48:28 -07:00
const { R } = require("redbean-node");
2022-03-18 02:57:08 -07:00
const { checkLogin, setSettings, setSetting } = require("../util-server");
2021-09-16 07:48:28 -07:00
const dayjs = require("dayjs");
const { debug } = require("../../src/util");
const ImageDataURI = require("../image-data-uri");
const Database = require("../database");
const apicache = require("../modules/apicache");
const StatusPage = require("../model/status_page");
2022-03-18 00:19:52 -07:00
const server = require("../server");
2021-09-16 07:48:28 -07:00
module.exports.statusPageSocketHandler = (socket) => {
// Post or edit incident
socket.on("postIncident", async (slug, incident, callback) => {
2021-09-16 07:48:28 -07:00
try {
checkLogin(socket);
let statusPageID = await StatusPage.slugToID(slug);
if (!statusPageID) {
throw new Error("slug is not found");
}
await R.exec("UPDATE incident SET pin = 0 WHERE status_page_id = ? ", [
statusPageID
]);
2021-09-16 07:48:28 -07:00
let incidentBean;
if (incident.id) {
incidentBean = await R.findOne("incident", " id = ? AND status_page_id = ? ", [
incident.id,
statusPageID
2021-09-16 07:48:28 -07:00
]);
}
if (incidentBean == null) {
incidentBean = R.dispense("incident");
}
incidentBean.title = incident.title;
incidentBean.content = incident.content;
incidentBean.style = incident.style;
incidentBean.pin = true;
incidentBean.status_page_id = statusPageID;
if (incident.id) {
incidentBean.lastUpdatedDate = R.isoDateTime(dayjs.utc());
} else {
incidentBean.createdDate = R.isoDateTime(dayjs.utc());
}
2021-09-16 07:48:28 -07:00
await R.store(incidentBean);
callback({
ok: true,
incident: incidentBean.toPublicJSON(),
});
2021-09-16 07:48:28 -07:00
} catch (error) {
callback({
ok: false,
msg: error.message,
});
2021-09-16 07:48:28 -07:00
}
});
socket.on("unpinIncident", async (slug, callback) => {
2021-09-16 07:48:28 -07:00
try {
checkLogin(socket);
let statusPageID = await StatusPage.slugToID(slug);
await R.exec("UPDATE incident SET pin = 0 WHERE pin = 1 AND status_page_id = ? ", [
statusPageID
]);
2021-09-16 07:48:28 -07:00
callback({
ok: true,
});
2021-09-16 07:48:28 -07:00
} catch (error) {
callback({
ok: false,
msg: error.message,
});
2021-09-16 07:48:28 -07:00
}
});
// Save Status Page
// imgDataUrl Only Accept PNG!
2022-03-14 21:00:29 -07:00
socket.on("saveStatusPage", async (slug, config, imgDataUrl, publicGroupList, callback) => {
try {
2022-03-24 08:43:07 -07:00
checkSlug(config.slug);
checkLogin(socket);
apicache.clear();
// Save Config
let statusPage = await R.findOne("status_page", " slug = ? ", [
slug
]);
if (!statusPage) {
throw new Error("No slug?");
}
const header = "data:image/png;base64,";
// Check logo format
// If is image data url, convert to png file
// Else assume it is a url, nothing to do
if (imgDataUrl.startsWith("data:")) {
if (! imgDataUrl.startsWith(header)) {
throw new Error("Only allowed PNG logo.");
}
const filename = `logo${statusPage.id}.png`;
// Convert to file
await ImageDataURI.outputFile(imgDataUrl, Database.uploadDir + filename);
config.logo = `/upload/${filename}?t=` + Date.now();
} else {
config.icon = imgDataUrl;
}
2022-03-14 21:00:29 -07:00
statusPage.slug = config.slug;
statusPage.title = config.title;
2022-03-15 23:14:47 -07:00
statusPage.description = config.description;
2022-03-14 21:00:29 -07:00
statusPage.icon = config.logo;
statusPage.theme = config.theme;
//statusPage.published = ;
//statusPage.search_engine_index = ;
statusPage.show_tags = config.showTags;
//statusPage.password = null;
2022-03-15 23:14:47 -07:00
statusPage.modified_date = R.isoDateTime();
2022-03-14 21:00:29 -07:00
await R.store(statusPage);
2021-09-20 05:44:07 -07:00
2021-09-22 00:10:08 -07:00
// Save Public Group List
const groupIDList = [];
let groupOrder = 1;
for (let group of publicGroupList) {
let groupBean;
if (group.id) {
groupBean = await R.findOne("group", " id = ? AND public = 1 AND status_page_id = ? ", [
group.id,
statusPage.id
]);
2021-09-22 00:10:08 -07:00
} else {
groupBean = R.dispense("group");
}
groupBean.status_page_id = statusPage.id;
2021-09-22 00:10:08 -07:00
groupBean.name = group.name;
groupBean.public = true;
groupBean.weight = groupOrder++;
await R.store(groupBean);
await R.exec("DELETE FROM monitor_group WHERE group_id = ? ", [
groupBean.id
]);
2021-09-22 00:10:08 -07:00
let monitorOrder = 1;
2021-09-22 00:23:58 -07:00
2021-09-22 00:10:08 -07:00
for (let monitor of group.monitorList) {
let relationBean = R.dispense("monitor_group");
relationBean.weight = monitorOrder++;
relationBean.group_id = groupBean.id;
relationBean.monitor_id = monitor.id;
await R.store(relationBean);
}
2021-09-22 00:10:08 -07:00
groupIDList.push(groupBean.id);
group.id = groupBean.id;
}
// Delete groups that not in the list
debug("Delete groups that not in the list");
const slots = groupIDList.map(() => "?").join(",");
const data = [
...groupIDList,
statusPage.id
];
await R.exec(`DELETE FROM \`group\` WHERE id NOT IN (${slots}) AND status_page_id = ?`, data);
2022-03-18 02:57:08 -07:00
// Also change entry page to new slug if it is the default one, and slug is changed.
if (server.entryPage === "statusPage-" + slug && statusPage.slug !== slug) {
server.entryPage = "statusPage-" + statusPage.slug;
await setSetting("entryPage", server.entryPage, "general");
}
2021-09-22 00:10:08 -07:00
callback({
ok: true,
publicGroupList,
});
2021-09-22 00:10:08 -07:00
} catch (error) {
2022-03-17 08:38:43 -07:00
console.error(error);
callback({
ok: false,
msg: error.message,
});
}
});
2022-03-17 08:38:43 -07:00
// Add a new status page
socket.on("addStatusPage", async (title, slug, callback) => {
try {
checkLogin(socket);
title = title?.trim();
slug = slug?.trim();
// Check empty
if (!title || !slug) {
throw new Error("Please input all fields");
}
// Make sure slug is string
if (typeof slug !== "string") {
throw new Error("Slug -Accept string only");
}
// lower case only
slug = slug.toLowerCase();
2022-03-24 08:43:07 -07:00
checkSlug(slug);
2022-03-17 08:38:43 -07:00
let statusPage = R.dispense("status_page");
statusPage.slug = slug;
statusPage.title = title;
statusPage.theme = "light";
statusPage.icon = "";
await R.store(statusPage);
callback({
ok: true,
msg: "OK!"
});
} catch (error) {
console.error(error);
callback({
ok: false,
msg: error.message,
});
}
});
2022-03-18 00:19:52 -07:00
// Delete a status page
socket.on("deleteStatusPage", async (slug, callback) => {
try {
checkLogin(socket);
let statusPageID = await StatusPage.slugToID(slug);
if (statusPageID) {
// Reset entry page if it is the default one.
if (server.entryPage === "statusPage-" + slug) {
server.entryPage = "dashboard";
2022-03-18 02:57:08 -07:00
await setSetting("entryPage", server.entryPage, "general");
2022-03-18 00:19:52 -07:00
}
// No need to delete records from `status_page_cname`, because it has cascade foreign key.
// But for incident & group, it is hard to add cascade foreign key during migration, so they have to be deleted manually.
// Delete incident
await R.exec("DELETE FROM incident WHERE status_page_id = ? ", [
statusPageID
]);
// Delete group
await R.exec("DELETE FROM `group` WHERE status_page_id = ? ", [
statusPageID
]);
// Delete status_page
await R.exec("DELETE FROM status_page WHERE id = ? ", [
statusPageID
]);
} else {
throw new Error("Status Page is not found");
}
callback({
ok: true,
});
} catch (error) {
callback({
ok: false,
msg: error.message,
});
}
});
};
2022-03-24 08:43:07 -07:00
/**
* Check slug a-z, 0-9, - only
* Regex from: https://stackoverflow.com/questions/22454258/js-regex-string-validation-for-slug
*/
function checkSlug(slug) {
if (typeof slug !== "string") {
throw new Error("Slug must be string");
}
slug = slug.trim();
if (!slug) {
throw new Error("Slug cannot be empty");
}
if (!slug.match(/^[A-Za-z0-9]+(?:-[A-Za-z0-9]+)*$/)) {
throw new Error("Invalid Slug");
}
}