const { BeanModel } = require("redbean-node/dist/bean-model"); const { R } = require("redbean-node"); const cheerio = require("cheerio"); const { UptimeKumaServer } = require("../uptime-kuma-server"); class StatusPage extends BeanModel { /** * Like this: { "test-uptime.kuma.pet": "default" } * @type {{}} */ static domainMappingList = { }; /** * * @param {Response} response * @param {string} indexHTML * @param {string} slug */ static async handleStatusPageResponse(response, indexHTML, slug) { let statusPage = await R.findOne("status_page", " slug = ? ", [ slug ]); if (statusPage) { response.send(StatusPage.renderHTML(indexHTML, statusPage)); } else { response.status(404).send(UptimeKumaServer.getInstance().indexHTML); } } /** * SSR for status pages * @param {string} indexHTML * @param {StatusPage} statusPage */ static renderHTML(indexHTML, statusPage) { const $ = cheerio.load(indexHTML); const description155 = statusPage.description.substring(0, 155); $("title").text(statusPage.title); $("meta[name=description]").attr("content", description155); if (statusPage.icon) { $("link[rel=icon]") .attr("href", statusPage.icon) .removeAttr("type"); } const head = $("head"); // OG Meta Tags head.append(``); head.append(``); return $.root().html(); } /** * Loads domain mapping from DB * Return object like this: { "test-uptime.kuma.pet": "default" } * @returns {Promise} */ static async loadDomainMappingList() { StatusPage.domainMappingList = await R.getAssoc(` SELECT domain, slug FROM status_page, status_page_cname WHERE status_page.id = status_page_cname.status_page_id `); } /** * Send status page list to client * @param {Server} io io Socket server instance * @param {Socket} socket Socket.io instance * @returns {Promise} */ static async sendStatusPageList(io, socket) { let result = {}; let list = await R.findAll("status_page", " ORDER BY title "); for (let item of list) { result[item.id] = await item.toJSON(); } io.to(socket.userID).emit("statusPageList", result); return list; } /** * Update list of domain names * @param {string[]} domainNameList * @returns {Promise} */ async updateDomainNameList(domainNameList) { if (!Array.isArray(domainNameList)) { throw new Error("Invalid array"); } let trx = await R.begin(); await trx.exec("DELETE FROM status_page_cname WHERE status_page_id = ?", [ this.id, ]); try { for (let domain of domainNameList) { if (typeof domain !== "string") { throw new Error("Invalid domain"); } if (domain.trim() === "") { continue; } // If the domain name is used in another status page, delete it await trx.exec("DELETE FROM status_page_cname WHERE domain = ?", [ domain, ]); let mapping = trx.dispense("status_page_cname"); mapping.status_page_id = this.id; mapping.domain = domain; await trx.store(mapping); } await trx.commit(); } catch (error) { await trx.rollback(); throw error; } } /** * Get list of domain names * @returns {Object[]} */ getDomainNameList() { let domainList = []; for (let domain in StatusPage.domainMappingList) { let s = StatusPage.domainMappingList[domain]; if (this.slug === s) { domainList.push(domain); } } return domainList; } /** * Return an object that ready to parse to JSON * @returns {Object} */ async toJSON() { return { id: this.id, slug: this.slug, title: this.title, description: this.description, icon: this.getIcon(), theme: this.theme, published: !!this.published, showTags: !!this.show_tags, domainNameList: this.getDomainNameList(), customCSS: this.custom_css, footerText: this.footer_text, showPoweredBy: !!this.show_powered_by, }; } /** * Return an object that ready to parse to JSON for public * Only show necessary data to public * @returns {Object} */ async toPublicJSON() { return { slug: this.slug, title: this.title, description: this.description, icon: this.getIcon(), theme: this.theme, published: !!this.published, showTags: !!this.show_tags, customCSS: this.custom_css, footerText: this.footer_text, showPoweredBy: !!this.show_powered_by, }; } /** * Convert slug to status page ID * @param {string} slug */ static async slugToID(slug) { return await R.getCell("SELECT id FROM status_page WHERE slug = ? ", [ slug ]); } /** * Get path to the icon for the page * @returns {string} */ getIcon() { if (!this.icon) { return "/icon.svg"; } else { return this.icon; } } } module.exports = StatusPage;