uptime-kuma/server/model/maintenance.js

216 lines
6.7 KiB
JavaScript
Raw Normal View History

const { BeanModel } = require("redbean-node/dist/bean-model");
const { parseTimeObject, parseTimeFromTimeObject, utcToLocal, localToUTC, log } = require("../../src/util");
const { timeObjectToUTC, timeObjectToLocal } = require("../util-server");
2022-10-10 13:45:30 -04:00
const { R } = require("redbean-node");
const dayjs = require("dayjs");
class Maintenance extends BeanModel {
/**
2022-04-30 09:50:05 -04:00
* Return an object that ready to parse to JSON for public
* Only show necessary data to public
2022-04-30 09:50:05 -04:00
* @returns {Object}
*/
2022-10-10 08:48:11 -04:00
async toPublicJSON() {
let dateRange = [];
if (this.start_date) {
dateRange.push(utcToLocal(this.start_date));
if (this.end_date) {
dateRange.push(utcToLocal(this.end_date));
}
}
let timeRange = [];
2022-10-10 08:48:11 -04:00
let startTime = timeObjectToLocal(parseTimeObject(this.start_time));
timeRange.push(startTime);
2022-10-10 08:48:11 -04:00
let endTime = timeObjectToLocal(parseTimeObject(this.end_time));
timeRange.push(endTime);
let obj = {
id: this.id,
title: this.title,
description: this.description,
2022-09-23 14:33:29 -04:00
strategy: this.strategy,
intervalDay: this.interval_day,
2022-09-23 14:33:29 -04:00
active: !!this.active,
dateRange: dateRange,
timeRange: timeRange,
weekdays: (this.weekdays) ? JSON.parse(this.weekdays) : [],
daysOfMonth: (this.days_of_month) ? JSON.parse(this.days_of_month) : [],
2022-10-11 06:23:17 -04:00
timeslotList: [],
};
2022-10-11 06:23:17 -04:00
const timeslotList = await this.getTimeslotList();
for (let timeslot of timeslotList) {
obj.timeslotList.push(await timeslot.toPublicJSON());
}
2022-10-12 10:13:07 -04:00
if (!Array.isArray(obj.weekdays)) {
obj.weekdays = [];
}
2022-10-12 10:13:07 -04:00
if (!Array.isArray(obj.daysOfMonth)) {
obj.daysOfMonth = [];
}
2022-10-10 13:45:30 -04:00
// Maintenance Status
if (!obj.active) {
obj.status = "inactive";
2022-10-11 06:23:17 -04:00
} else if (obj.strategy === "manual") {
obj.status = "under-maintenance";
} else if (obj.timeslotList.length > 0) {
let currentTimestamp = dayjs().unix();
2022-10-10 13:45:30 -04:00
for (let timeslot of obj.timeslotList) {
if (dayjs.utc(timeslot.startDate).unix() <= currentTimestamp && dayjs.utc(timeslot.endDate).unix() >= currentTimestamp) {
log.debug("timeslot", "Timeslot ID: " + timeslot.id);
log.debug("timeslot", "currentTimestamp:" + currentTimestamp);
log.debug("timeslot", "timeslot.start_date:" + dayjs.utc(timeslot.startDate).unix());
log.debug("timeslot", "timeslot.end_date:" + dayjs.utc(timeslot.endDate).unix());
2022-10-10 13:45:30 -04:00
obj.status = "under-maintenance";
break;
}
}
if (!obj.status) {
obj.status = "scheduled";
}
} else if (obj.timeslotList.length === 0) {
obj.status = "ended";
} else {
obj.status = "unknown";
}
return obj;
}
2022-10-10 13:45:30 -04:00
/**
* Only get future or current timeslots only
* @returns {Promise<[]>}
*/
async getTimeslotList() {
2022-10-11 06:23:17 -04:00
return R.convertToBeans("maintenance_timeslot", await R.getAll(`
2022-10-10 13:45:30 -04:00
SELECT maintenance_timeslot.*
FROM maintenance_timeslot, maintenance
WHERE maintenance_timeslot.maintenance_id = maintenance.id
AND maintenance.id = ?
AND ${Maintenance.getActiveAndFutureMaintenanceSQLCondition()}
`, [
this.id
2022-10-11 06:23:17 -04:00
]));
2022-10-10 13:45:30 -04:00
}
/**
2022-04-30 09:50:05 -04:00
* Return an object that ready to parse to JSON
* @param {string} timezone If not specified, the timeRange will be in UTC
2022-04-30 09:50:05 -04:00
* @returns {Object}
*/
async toJSON(timezone = null) {
return this.toPublicJSON(timezone);
}
getDayOfWeekList() {
log.debug("timeslot", "List: " + this.weekdays);
return JSON.parse(this.weekdays).sort(function (a, b) {
return a - b;
});
}
getDayOfMonthList() {
return JSON.parse(this.days_of_month).sort(function (a, b) {
return a - b;
});
}
getStartDateTime() {
let startOfTheDay = dayjs.utc(this.start_date).format("HH:mm");
log.debug("timeslot", "startOfTheDay: " + startOfTheDay);
// Start Time
let startTimeSecond = dayjs.utc(this.start_time, "HH:mm").diff(dayjs.utc(startOfTheDay, "HH:mm"), "second");
log.debug("timeslot", "startTime: " + startTimeSecond);
// Bake StartDate + StartTime = Start DateTime
return dayjs.utc(this.start_date).add(startTimeSecond, "second");
}
getDuration() {
let duration = dayjs.utc(this.end_time, "HH:mm").diff(dayjs.utc(this.start_time, "HH:mm"), "second");
// Add 24hours if it is across day
if (duration < 0) {
duration += 24 * 3600;
}
return duration;
}
2022-10-10 08:48:11 -04:00
static jsonToBean(bean, obj) {
if (obj.id) {
bean.id = obj.id;
}
// Apply timezone offset to timeRange, as it cannot apply automatically.
2022-10-10 08:48:11 -04:00
if (obj.timeRange[0]) {
timeObjectToUTC(obj.timeRange[0]);
if (obj.timeRange[1]) {
timeObjectToUTC(obj.timeRange[1]);
}
}
bean.title = obj.title;
bean.description = obj.description;
bean.strategy = obj.strategy;
bean.interval_day = obj.intervalDay;
bean.active = obj.active;
if (obj.dateRange[0]) {
bean.start_date = localToUTC(obj.dateRange[0]);
if (obj.dateRange[1]) {
bean.end_date = localToUTC(obj.dateRange[1]);
}
}
bean.start_time = parseTimeFromTimeObject(obj.timeRange[0]);
bean.end_time = parseTimeFromTimeObject(obj.timeRange[1]);
bean.weekdays = JSON.stringify(obj.weekdays);
bean.days_of_month = JSON.stringify(obj.daysOfMonth);
return bean;
}
/**
* SQL conditions for active maintenance
* @returns {string}
*/
static getActiveMaintenanceSQLCondition() {
return `
(maintenance_timeslot.start_date <= DATETIME('now')
AND maintenance_timeslot.end_date >= DATETIME('now')
AND maintenance.active = 1)
2022-10-10 08:48:11 -04:00
OR
(maintenance.strategy = 'manual' AND active = 1)
`;
}
2022-10-10 13:45:30 -04:00
/**
* SQL conditions for active and future maintenance
* @returns {string}
*/
static getActiveAndFutureMaintenanceSQLCondition() {
return `
2022-10-11 06:23:17 -04:00
((maintenance_timeslot.end_date >= DATETIME('now')
2022-10-10 13:45:30 -04:00
AND maintenance.active = 1)
OR
2022-10-11 06:23:17 -04:00
(maintenance.strategy = 'manual' AND active = 1))
2022-10-10 13:45:30 -04:00
`;
}
}
module.exports = Maintenance;