"use strict"; const db = require('sqlite'); const utils = require('./utils'); const log = require('./log'); async function insert(table_name, rec, replace = false) { const keys = Object.keys(rec); if (keys.length === 0) { log.error("Can't insert empty object into table " + table_name); return; } const columns = keys.join(", "); const questionMarks = keys.map(p => "?").join(", "); const query = "INSERT " + (replace ? "OR REPLACE" : "") + " INTO " + table_name + "(" + columns + ") VALUES (" + questionMarks + ")"; const res = await execute(query, Object.values(rec)); return res.lastID; } async function beginTransaction() { return await db.run("BEGIN"); } async function commit() { return await db.run("COMMIT"); } async function rollback() { return await db.run("ROLLBACK"); } async function getOption(optName) { const row = await getSingleResultOrNull("SELECT opt_value FROM options WHERE opt_name = ?", [optName]); if (!row) { throw new Error("Option " + optName + " doesn't exist"); } return row['opt_value']; } async function setOption(optName, optValue) { const now = utils.nowTimestamp(); await execute("UPDATE options SET opt_value = ?, date_modified = ? WHERE opt_name = ?", [optValue, now, optName]); } async function getSingleResult(query, params = []) { return await db.get(query, ...params); } async function getSingleResultOrNull(query, params = []) { const all = await db.all(query, ...params); return all ? all[0] : null; } async function getResults(query, params = []) { return await db.all(query, ...params); } async function getFlattenedResults(key, query, params = []) { const list = []; const result = await getResults(query, params); for (const row of result) { list.push(row[key]); } return list; } async function execute(query, params = []) { return await db.run(query, ...params); } async function executeScript(query) { return await db.exec(query); } async function remove(tableName, noteId) { return await execute("DELETE FROM " + tableName + " WHERE note_id = ?", [noteId]); } async function addAudit(category, req=null, noteId=null, changeFrom=null, changeTo=null, comment=null) { const now = utils.nowTimestamp(); const browserId = req == null ? null : req.get('x-browser-id'); log.info("audit: " + category + ", browserId=" + browserId + ", noteId=" + noteId + ", from=" + changeFrom + ", to=" + changeTo + ", comment=" + comment); const id = utils.randomString(14); await execute("INSERT INTO audit_log (id, date_modified, category, browser_id, note_id, change_from, change_to, comment)" + " VALUES (?, ?, ?, ?, ?, ?, ?, ?)", [id, now, category, browserId, noteId, changeFrom, changeTo, comment]); } async function deleteRecentAudits(category, req, noteId) { const browserId = req.get('x-browser-id'); const deleteCutoff = utils.nowTimestamp() - 10 * 60; await execute("DELETE FROM audit_log WHERE category = ? AND browser_id = ? AND note_id = ? AND date_modified > ?", [category, browserId, noteId, deleteCutoff]) } async function doInTransaction(func) { try { await beginTransaction(); await func(); await commit(); } catch (e) { log.error("Error executing transaction, executing rollback. Inner exception: " + e.stack); await rollback(); } } module.exports = { insert, getSingleResult, getResults, getFlattenedResults, execute, executeScript, getOption, setOption, addAudit, deleteRecentAudits, remove, doInTransaction };