const fs = require("fs"); const { R } = require("redbean-node"); const { setSetting, setting } = require("./util-server"); class Database { static templatePath = "./db/kuma.db" static dataDir; static path; static latestVersion = 9; static noReject = true; static sqliteInstance = null; static async connect() { const acquireConnectionTimeout = 120 * 1000; R.setup("sqlite", { filename: Database.path, useNullAsDefault: true, acquireConnectionTimeout: acquireConnectionTimeout, }, { min: 1, max: 1, idleTimeoutMillis: 120 * 1000, propagateCreateError: false, acquireTimeoutMillis: acquireConnectionTimeout, }); if (process.env.SQL_LOG === "1") { R.debug(true); } // Auto map the model to a bean object R.freeze(true) await R.autoloadModels("./server/model"); // Change to WAL await R.exec("PRAGMA journal_mode = WAL"); console.log(await R.getAll("PRAGMA journal_mode")); } static async patch() { let version = parseInt(await setting("database_version")); if (! version) { version = 0; } console.info("Your database version: " + version); console.info("Latest database version: " + this.latestVersion); if (version === this.latestVersion) { console.info("Database no need to patch"); } else if (version > this.latestVersion) { console.info("Warning: Database version is newer than expected"); } else { console.info("Database patch is needed") console.info("Backup the db") const backupPath = this.dataDir + "kuma.db.bak" + version; fs.copyFileSync(Database.path, backupPath); const shmPath = Database.path + "-shm"; if (fs.existsSync(shmPath)) { fs.copyFileSync(shmPath, shmPath + ".bak" + version); } const walPath = Database.path + "-wal"; if (fs.existsSync(walPath)) { fs.copyFileSync(walPath, walPath + ".bak" + version); } // Try catch anything here, if gone wrong, restore the backup try { for (let i = version + 1; i <= this.latestVersion; i++) { const sqlFile = `./db/patch${i}.sql`; console.info(`Patching ${sqlFile}`); await Database.importSQLFile(sqlFile); console.info(`Patched ${sqlFile}`); await setSetting("database_version", i); } console.log("Database Patched Successfully"); } catch (ex) { await Database.close(); console.error("Patch db failed!!! Restoring the backup") fs.copyFileSync(backupPath, Database.path); console.error(ex) console.error("Start Uptime-Kuma failed due to patch db failed") console.error("Please submit the bug report if you still encounter the problem after restart: https://github.com/louislam/uptime-kuma/issues") process.exit(1); } } } /** * Sadly, multi sql statements is not supported by many sqlite libraries, I have to implement it myself * @param filename * @returns {Promise} */ static async importSQLFile(filename) { await R.getCell("SELECT 1"); let text = fs.readFileSync(filename).toString(); // Remove all comments (--) let lines = text.split("\n"); lines = lines.filter((line) => { return ! line.startsWith("--") }); // Split statements by semicolon // Filter out empty line text = lines.join("\n") let statements = text.split(";") .map((statement) => { return statement.trim(); }) .filter((statement) => { return statement !== ""; }) for (let statement of statements) { await R.exec(statement); } } static getBetterSQLite3Database() { return R.knex.client.acquireConnection(); } /** * Special handle, because tarn.js throw a promise reject that cannot be caught * @returns {Promise} */ static async close() { if (this.sqliteInstance) { this.sqliteInstance.close(); } console.log("Stopped database"); } } module.exports = Database;