uptime-kuma/server/model/maintenance.js

const dayjs = require("dayjs");
const { BeanModel } = require("redbean-node/dist/bean-model");
const { parseTimeObject, parseTimeFromTimeObject } = require("../../src/util");
const { isArray } = require("chart.js/helpers");
const { timeObjectToUTC, timeObjectToLocal } = require("../util-server");
const { R } = require("redbean-node");

class Maintenance extends BeanModel {

    /**
     * Return an object that ready to parse to JSON for public
     * Only show necessary data to public
     * @param {string} timezone If not specified, the timeRange will be in UTC
     * @returns {Object}
     */
    async toPublicJSON(timezone = null) {

        let dateTimeRange = [];
        if (this.start_datetime) {

            dateTimeRange.push(dayjs.utc(this.start_datetime).toISOString());
            if (this.end_datetime) {
                dateTimeRange.push(dayjs.utc(this.end_datetime).toISOString());
            }
        }

        let dateRange = [];
        if (this.start_date) {
            dateRange.push(dayjs.utc(this.start_date).toISOString());
            if (this.end_date) {
                dateRange.push(dayjs.utc(this.end_date).toISOString());
            }
        }

        let timeRange = [];
        let startTime = parseTimeObject(this.start_time);
        timeRange.push(startTime);
        let endTime = parseTimeObject(this.end_time);
        timeRange.push(endTime);

        // Apply timezone offset
        if (timezone) {
            if (this.start_time) {
                timeObjectToLocal(startTime, timezone);
            }
            if (this.end_time) {
                timeObjectToLocal(endTime, timezone);
            }
        }

        let obj = {
            id: this.id,
            title: this.title,
            description: this.description,
            strategy: this.strategy,
            intervalDay: this.interval_day,
            active: !!this.active,
            dateTimeRange: dateTimeRange,
            dateRange: dateRange,
            timeRange: timeRange,
            weekdays: (this.weekdays) ? JSON.parse(this.weekdays) : [],
            daysOfMonth: (this.days_of_month) ? JSON.parse(this.days_of_month) : [],
        };

        if (!isArray(obj.weekdays)) {
            obj.weekdays = [];
        }

        if (!isArray(obj.daysOfMonth)) {
            obj.daysOfMonth = [];
        }

        return obj;
    }

    /**
     * Return an object that ready to parse to JSON
     * @param {string} timezone If not specified, the timeRange will be in UTC
     * @returns {Object}
     */
    async toJSON(timezone = null) {
        return this.toPublicJSON(timezone);
    }

    static jsonToBean(bean, obj, timezone) {
        if (obj.id) {
            bean.id = obj.id;
        }

        // Apply timezone offset to timeRange, as it cannot apply automatically.
        if (timezone) {
            if (obj.timeRange[0]) {
                timeObjectToUTC(obj.timeRange[0], timezone);
                if (obj.timeRange[1]) {
                    timeObjectToUTC(obj.timeRange[1], timezone);
                }
            }
        }

        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 = R.isoDate(dayjs(obj.dateRange[0]).utc());

            if (obj.dateRange[1]) {
                bean.end_date = R.isoDate(dayjs(obj.dateRange[1]).utc());
            }
        }

        if (obj.dateTimeRange[0]) {
            bean.start_datetime = R.isoDateTime(dayjs(obj.dateTimeRange[0]).utc());

            if (obj.dateTimeRange[1]) {
                bean.end_datetime = R.isoDateTime(dayjs(obj.dateTimeRange[1]).utc());
            }
        }

        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)
            AND
            (maintenance.strategy = 'manual' AND active = 1)

        `;
    }
}

module.exports = Maintenance;