mirror of
https://github.com/zadam/trilium.git
synced 2025-02-19 20:44:03 +08:00
187 lines
No EOL
4.7 KiB
JavaScript
187 lines
No EOL
4.7 KiB
JavaScript
"use strict";
|
|
|
|
const utils = require('./utils');
|
|
const log = require('./log');
|
|
const dataDir = require('./data_dir');
|
|
|
|
const dbReady = (() => {
|
|
const db = require('sqlite');
|
|
|
|
return db.open(dataDir.DOCUMENT_PATH, {Promise}).then(() => db);
|
|
})();
|
|
|
|
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 replace(table_name, rec) {
|
|
return await insert(table_name, rec, true);
|
|
}
|
|
|
|
async function beginTransaction() {
|
|
return await wrap(async db => db.run("BEGIN"));
|
|
}
|
|
|
|
async function commit() {
|
|
return await wrap(async db => db.run("COMMIT"));
|
|
}
|
|
|
|
async function rollback() {
|
|
return await wrap(async db => db.run("ROLLBACK"));
|
|
}
|
|
|
|
async function getSingleResult(query, params = []) {
|
|
return await wrap(async db => db.get(query, ...params));
|
|
}
|
|
|
|
async function getSingleResultOrNull(query, params = []) {
|
|
const all = await wrap(async db => db.all(query, ...params));
|
|
|
|
return all.length > 0 ? all[0] : null;
|
|
}
|
|
|
|
async function getSingleValue(query, params = []) {
|
|
const row = await getSingleResultOrNull(query, params);
|
|
|
|
if (!row) {
|
|
return null;
|
|
}
|
|
|
|
return row[Object.keys(row)[0]];
|
|
}
|
|
|
|
async function getResults(query, params = []) {
|
|
return await wrap(async db => db.all(query, ...params));
|
|
}
|
|
|
|
async function getMap(query, params = []) {
|
|
const map = {};
|
|
const results = await getResults(query, params);
|
|
|
|
for (const row of results) {
|
|
const keys = Object.keys(row);
|
|
|
|
map[row[keys[0]]] = row[keys[1]];
|
|
}
|
|
|
|
return map;
|
|
}
|
|
|
|
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 wrap(async db => db.run(query, ...params));
|
|
}
|
|
|
|
async function executeScript(query) {
|
|
return await wrap(async db => db.exec(query));
|
|
}
|
|
|
|
async function remove(tableName, noteId) {
|
|
return await execute("DELETE FROM " + tableName + " WHERE note_id = ?", [noteId]);
|
|
}
|
|
|
|
async function addAudit(category, browserId=null, noteId=null, changeFrom=null, changeTo=null, comment=null) {
|
|
const now = utils.nowTimestamp();
|
|
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, browserId, noteId) {
|
|
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 wrap(func) {
|
|
const db = await dbReady;
|
|
|
|
try {
|
|
return await func(db);
|
|
}
|
|
catch (e) {
|
|
log.error("Error executing query. Inner exception: " + e.stack + error.stack);
|
|
|
|
throw new Error();
|
|
}
|
|
}
|
|
|
|
async function doInTransaction(func) {
|
|
const error = new Error(); // to capture correct stack trace in case of exception
|
|
|
|
try {
|
|
|
|
await beginTransaction();
|
|
|
|
await func();
|
|
|
|
await commit();
|
|
}
|
|
catch (e) {
|
|
log.error("Error executing transaction, executing rollback. Inner exception: " + e.stack + error.stack);
|
|
|
|
await rollback();
|
|
|
|
throw e;
|
|
}
|
|
}
|
|
|
|
dbReady
|
|
.then(async () => {
|
|
const tableResults = await getResults("SELECT name FROM sqlite_master WHERE type='table' AND name='notes'");
|
|
|
|
if (tableResults.length !== 1) {
|
|
console.log("No connection to initialized DB.");
|
|
process.exit(1);
|
|
}
|
|
})
|
|
.catch(e => {
|
|
console.log("Error connecting to DB.", e);
|
|
process.exit(1);
|
|
});
|
|
|
|
module.exports = {
|
|
dbReady,
|
|
insert,
|
|
replace,
|
|
getSingleValue,
|
|
getSingleResult,
|
|
getSingleResultOrNull,
|
|
getResults,
|
|
getMap,
|
|
getFlattenedResults,
|
|
execute,
|
|
executeScript,
|
|
addAudit,
|
|
deleteRecentAudits,
|
|
remove,
|
|
doInTransaction
|
|
}; |