mirror of
https://github.com/laurent22/joplin.git
synced 2024-11-24 08:12:24 +02:00
551 lines
15 KiB
JavaScript
551 lines
15 KiB
JavaScript
const { Database } = require('lib/database.js');
|
|
const { uuid } = require('lib/uuid.js');
|
|
const { time } = require('lib/time-utils.js');
|
|
const Mutex = require('async-mutex').Mutex;
|
|
|
|
class BaseModel {
|
|
static modelType() {
|
|
throw new Error('Must be overriden');
|
|
}
|
|
|
|
static tableName() {
|
|
throw new Error('Must be overriden');
|
|
}
|
|
|
|
static addModelMd(model) {
|
|
if (!model) return model;
|
|
|
|
if (Array.isArray(model)) {
|
|
let output = [];
|
|
for (let i = 0; i < model.length; i++) {
|
|
output.push(this.addModelMd(model[i]));
|
|
}
|
|
return output;
|
|
} else {
|
|
model = Object.assign({}, model);
|
|
model.type_ = this.modelType();
|
|
return model;
|
|
}
|
|
}
|
|
|
|
static logger() {
|
|
return this.db().logger();
|
|
}
|
|
|
|
static useUuid() {
|
|
return false;
|
|
}
|
|
|
|
static byId(items, id) {
|
|
for (let i = 0; i < items.length; i++) {
|
|
if (items[i].id == id) return items[i];
|
|
}
|
|
return null;
|
|
}
|
|
|
|
static modelIndexById(items, id) {
|
|
for (let i = 0; i < items.length; i++) {
|
|
if (items[i].id == id) return i;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
static modelsByIds(items, ids) {
|
|
const output = [];
|
|
for (let i = 0; i < items.length; i++) {
|
|
if (ids.indexOf(items[i].id) >= 0) {
|
|
output.push(items[i]);
|
|
}
|
|
}
|
|
return output;
|
|
}
|
|
|
|
// Prefer the use of this function to compare IDs as it handles the case where
|
|
// one ID is null and the other is "", in which case they are actually considered to be the same.
|
|
static idsEqual(id1, id2) {
|
|
if (!id1 && !id2) return true;
|
|
if (!id1 && !!id2) return false;
|
|
if (!!id1 && !id2) return false;
|
|
return id1 === id2;
|
|
}
|
|
|
|
static modelTypeToName(type) {
|
|
for (let i = 0; i < BaseModel.typeEnum_.length; i++) {
|
|
const e = BaseModel.typeEnum_[i];
|
|
if (e[1] === type) return e[0].substr(5).toLowerCase();
|
|
}
|
|
throw new Error(`Unknown model type: ${type}`);
|
|
}
|
|
|
|
static hasField(name) {
|
|
let fields = this.fieldNames();
|
|
return fields.indexOf(name) >= 0;
|
|
}
|
|
|
|
static fieldNames(withPrefix = false) {
|
|
let output = this.db().tableFieldNames(this.tableName());
|
|
if (!withPrefix) return output;
|
|
|
|
let p = withPrefix === true ? this.tableName() : withPrefix;
|
|
let temp = [];
|
|
for (let i = 0; i < output.length; i++) {
|
|
temp.push(`${p}.${output[i]}`);
|
|
}
|
|
|
|
return temp;
|
|
}
|
|
|
|
static fieldType(name, defaultValue = null) {
|
|
let fields = this.fields();
|
|
for (let i = 0; i < fields.length; i++) {
|
|
if (fields[i].name == name) return fields[i].type;
|
|
}
|
|
if (defaultValue !== null) return defaultValue;
|
|
throw new Error(`Unknown field: ${name}`);
|
|
}
|
|
|
|
static fields() {
|
|
return this.db().tableFields(this.tableName());
|
|
}
|
|
|
|
static removeUnknownFields(model) {
|
|
const newModel = {};
|
|
for (let n in model) {
|
|
if (!model.hasOwnProperty(n)) continue;
|
|
if (!this.hasField(n) && n !== 'type_') continue;
|
|
newModel[n] = model[n];
|
|
}
|
|
return newModel;
|
|
}
|
|
|
|
static new() {
|
|
let fields = this.fields();
|
|
let output = {};
|
|
for (let i = 0; i < fields.length; i++) {
|
|
let f = fields[i];
|
|
output[f.name] = f.default;
|
|
}
|
|
return output;
|
|
}
|
|
|
|
static modOptions(options) {
|
|
if (!options) {
|
|
options = {};
|
|
} else {
|
|
options = Object.assign({}, options);
|
|
}
|
|
if (!('isNew' in options)) options.isNew = 'auto';
|
|
if (!('autoTimestamp' in options)) options.autoTimestamp = true;
|
|
return options;
|
|
}
|
|
|
|
static count(options = null) {
|
|
if (!options) options = {};
|
|
let sql = `SELECT count(*) as total FROM \`${this.tableName()}\``;
|
|
if (options.where) sql += ` WHERE ${options.where}`;
|
|
return this.db()
|
|
.selectOne(sql)
|
|
.then(r => {
|
|
return r ? r['total'] : 0;
|
|
});
|
|
}
|
|
|
|
static load(id, options = null) {
|
|
return this.loadByField('id', id, options);
|
|
}
|
|
|
|
static shortId(id) {
|
|
return id.substr(0, 5);
|
|
}
|
|
|
|
static loadByPartialId(partialId) {
|
|
return this.modelSelectAll(`SELECT * FROM \`${this.tableName()}\` WHERE \`id\` LIKE ?`, [`${partialId}%`]);
|
|
}
|
|
|
|
static applySqlOptions(options, sql, params = null) {
|
|
if (!options) options = {};
|
|
|
|
if (options.order && options.order.length) {
|
|
let items = [];
|
|
for (let i = 0; i < options.order.length; i++) {
|
|
const o = options.order[i];
|
|
let item = o.by;
|
|
if (options.caseInsensitive === true) item += ' COLLATE NOCASE';
|
|
if (o.dir) item += ` ${o.dir}`;
|
|
items.push(item);
|
|
}
|
|
sql += ` ORDER BY ${items.join(', ')}`;
|
|
}
|
|
|
|
if (options.limit) sql += ` LIMIT ${options.limit}`;
|
|
|
|
return { sql: sql, params: params };
|
|
}
|
|
|
|
static async allIds(options = null) {
|
|
let q = this.applySqlOptions(options, `SELECT id FROM \`${this.tableName()}\``);
|
|
const rows = await this.db().selectAll(q.sql, q.params);
|
|
return rows.map(r => r.id);
|
|
}
|
|
|
|
static async all(options = null) {
|
|
if (!options) options = {};
|
|
if (!options.fields) options.fields = '*';
|
|
|
|
let q = this.applySqlOptions(options, `SELECT ${this.db().escapeFields(options.fields)} FROM \`${this.tableName()}\``);
|
|
return this.modelSelectAll(q.sql);
|
|
}
|
|
|
|
static async byIds(ids, options = null) {
|
|
if (!ids.length) return [];
|
|
if (!options) options = {};
|
|
if (!options.fields) options.fields = '*';
|
|
|
|
let sql = `SELECT ${this.db().escapeFields(options.fields)} FROM \`${this.tableName()}\``;
|
|
sql += ` WHERE id IN ("${ids.join('","')}")`;
|
|
let q = this.applySqlOptions(options, sql);
|
|
return this.modelSelectAll(q.sql);
|
|
}
|
|
|
|
static async search(options = null) {
|
|
if (!options) options = {};
|
|
if (!options.fields) options.fields = '*';
|
|
|
|
let conditions = options.conditions ? options.conditions.slice(0) : [];
|
|
let params = options.conditionsParams ? options.conditionsParams.slice(0) : [];
|
|
|
|
if (options.titlePattern) {
|
|
let pattern = options.titlePattern.replace(/\*/g, '%');
|
|
conditions.push('title LIKE ?');
|
|
params.push(pattern);
|
|
}
|
|
|
|
if ('limit' in options && options.limit <= 0) return [];
|
|
|
|
let sql = `SELECT ${this.db().escapeFields(options.fields)} FROM \`${this.tableName()}\``;
|
|
if (conditions.length) sql += ` WHERE ${conditions.join(' AND ')}`;
|
|
|
|
let query = this.applySqlOptions(options, sql, params);
|
|
return this.modelSelectAll(query.sql, query.params);
|
|
}
|
|
|
|
static modelSelectOne(sql, params = null) {
|
|
if (params === null) params = [];
|
|
return this.db()
|
|
.selectOne(sql, params)
|
|
.then(model => {
|
|
return this.filter(this.addModelMd(model));
|
|
});
|
|
}
|
|
|
|
static modelSelectAll(sql, params = null) {
|
|
if (params === null) params = [];
|
|
return this.db()
|
|
.selectAll(sql, params)
|
|
.then(models => {
|
|
return this.filterArray(this.addModelMd(models));
|
|
});
|
|
}
|
|
|
|
static loadByField(fieldName, fieldValue, options = null) {
|
|
if (!options) options = {};
|
|
if (!('caseInsensitive' in options)) options.caseInsensitive = false;
|
|
if (!options.fields) options.fields = '*';
|
|
let sql = `SELECT ${this.db().escapeFields(options.fields)} FROM \`${this.tableName()}\` WHERE \`${fieldName}\` = ?`;
|
|
if (options.caseInsensitive) sql += ' COLLATE NOCASE';
|
|
return this.modelSelectOne(sql, [fieldValue]);
|
|
}
|
|
|
|
static loadByTitle(fieldValue) {
|
|
return this.modelSelectOne(`SELECT * FROM \`${this.tableName()}\` WHERE \`title\` = ?`, [fieldValue]);
|
|
}
|
|
|
|
static diffObjects(oldModel, newModel) {
|
|
let output = {};
|
|
const fields = this.diffObjectsFields(oldModel, newModel);
|
|
for (let i = 0; i < fields.length; i++) {
|
|
output[fields[i]] = newModel[fields[i]];
|
|
}
|
|
if ('type_' in newModel) output.type_ = newModel.type_;
|
|
return output;
|
|
}
|
|
|
|
static diffObjectsFields(oldModel, newModel) {
|
|
let output = [];
|
|
for (let n in newModel) {
|
|
if (!newModel.hasOwnProperty(n)) continue;
|
|
if (n == 'type_') continue;
|
|
if (!(n in oldModel) || newModel[n] !== oldModel[n]) {
|
|
output.push(n);
|
|
}
|
|
}
|
|
return output;
|
|
}
|
|
|
|
static modelsAreSame(oldModel, newModel) {
|
|
const diff = this.diffObjects(oldModel, newModel);
|
|
delete diff.type_;
|
|
return !Object.getOwnPropertyNames(diff).length;
|
|
}
|
|
|
|
static saveMutex(modelOrId) {
|
|
const noLockMutex = {
|
|
acquire: function() {
|
|
return null;
|
|
},
|
|
};
|
|
|
|
if (!modelOrId) return noLockMutex;
|
|
|
|
let modelId = typeof modelOrId === 'string' ? modelOrId : modelOrId.id;
|
|
|
|
if (!modelId) return noLockMutex;
|
|
|
|
let mutex = BaseModel.saveMutexes_[modelId];
|
|
if (mutex) return mutex;
|
|
|
|
mutex = new Mutex();
|
|
BaseModel.saveMutexes_[modelId] = mutex;
|
|
return mutex;
|
|
}
|
|
|
|
static releaseSaveMutex(modelOrId, release) {
|
|
if (!release) return;
|
|
if (!modelOrId) return release();
|
|
|
|
let modelId = typeof modelOrId === 'string' ? modelOrId : modelOrId.id;
|
|
|
|
if (!modelId) return release();
|
|
|
|
let mutex = BaseModel.saveMutexes_[modelId];
|
|
if (!mutex) return release();
|
|
|
|
delete BaseModel.saveMutexes_[modelId];
|
|
release();
|
|
}
|
|
|
|
static saveQuery(o, options) {
|
|
let temp = {};
|
|
let fieldNames = this.fieldNames();
|
|
for (let i = 0; i < fieldNames.length; i++) {
|
|
let n = fieldNames[i];
|
|
if (n in o) temp[n] = o[n];
|
|
}
|
|
|
|
// Remove fields that are not in the `fields` list, if provided.
|
|
// Note that things like update_time, user_updated_time will still
|
|
// be part of the final list of fields if autoTimestamp is on.
|
|
// id also will stay.
|
|
if (!options.isNew && options.fields) {
|
|
const filtered = {};
|
|
for (let k in temp) {
|
|
if (!temp.hasOwnProperty(k)) continue;
|
|
if (k !== 'id' && options.fields.indexOf(k) < 0) continue;
|
|
filtered[k] = temp[k];
|
|
}
|
|
temp = filtered;
|
|
}
|
|
|
|
o = temp;
|
|
|
|
let modelId = temp.id;
|
|
let query = {};
|
|
|
|
const timeNow = time.unixMs();
|
|
|
|
if (options.autoTimestamp && this.hasField('updated_time')) {
|
|
o.updated_time = timeNow;
|
|
}
|
|
|
|
// The purpose of user_updated_time is to allow the user to manually set the time of a note (in which case
|
|
// options.autoTimestamp will be `false`). However note that if the item is later changed, this timestamp
|
|
// will be set again to the current time.
|
|
//
|
|
// The technique to modify user_updated_time while keeping updated_time current (so that sync can happen) is to
|
|
// manually set updated_time when saving and to set autoTimestamp to false, for example:
|
|
// Note.save({ id: "...", updated_time: Date.now(), user_updated_time: 1436342618000 }, { autoTimestamp: false })
|
|
if (options.autoTimestamp && this.hasField('user_updated_time')) {
|
|
o.user_updated_time = timeNow;
|
|
}
|
|
|
|
if (options.isNew) {
|
|
if (this.useUuid() && !o.id) {
|
|
modelId = uuid.create();
|
|
o.id = modelId;
|
|
}
|
|
|
|
if (!o.created_time && this.hasField('created_time')) {
|
|
o.created_time = timeNow;
|
|
}
|
|
|
|
if (!o.user_created_time && this.hasField('user_created_time')) {
|
|
o.user_created_time = o.created_time ? o.created_time : timeNow;
|
|
}
|
|
|
|
if (!o.user_updated_time && this.hasField('user_updated_time')) {
|
|
o.user_updated_time = o.updated_time ? o.updated_time : timeNow;
|
|
}
|
|
|
|
query = Database.insertQuery(this.tableName(), o);
|
|
} else {
|
|
let where = { id: o.id };
|
|
let temp = Object.assign({}, o);
|
|
delete temp.id;
|
|
|
|
query = Database.updateQuery(this.tableName(), temp, where);
|
|
}
|
|
|
|
query.id = modelId;
|
|
query.modObject = o;
|
|
|
|
return query;
|
|
}
|
|
|
|
static async save(o, options = null) {
|
|
// When saving, there's a mutex per model ID. This is because the model returned from this function
|
|
// is basically its input `o` (instead of being read from the database, for performance reasons).
|
|
// This works well in general except if that model is saved simultaneously in two places. In that
|
|
// case, the output won't be up-to-date and would cause for example display issues with out-dated
|
|
// notes being displayed. This was an issue when notes were being synchronised while being decrypted
|
|
// at the same time.
|
|
|
|
const mutexRelease = await this.saveMutex(o).acquire();
|
|
|
|
options = this.modOptions(options);
|
|
options.isNew = this.isNew(o, options);
|
|
|
|
// Diff saving is an optimisation which takes a new version of the item and an old one,
|
|
// do a diff and save only this diff. IMPORTANT: When using this make sure that both
|
|
// models have been normalised using ItemClass.filter()
|
|
const isDiffSaving = options && options.oldItem && !options.isNew;
|
|
|
|
if (isDiffSaving) {
|
|
const newObject = BaseModel.diffObjects(options.oldItem, o);
|
|
newObject.type_ = o.type_;
|
|
newObject.id = o.id;
|
|
o = newObject;
|
|
}
|
|
|
|
o = this.filter(o);
|
|
|
|
let queries = [];
|
|
let saveQuery = this.saveQuery(o, options);
|
|
let modelId = saveQuery.id;
|
|
|
|
queries.push(saveQuery);
|
|
|
|
if (options.nextQueries && options.nextQueries.length) {
|
|
queries = queries.concat(options.nextQueries);
|
|
}
|
|
|
|
let output = null;
|
|
|
|
try {
|
|
await this.db().transactionExecBatch(queries);
|
|
|
|
o = Object.assign({}, o);
|
|
// eslint-disable-next-line require-atomic-updates
|
|
if (modelId) o.id = modelId;
|
|
// eslint-disable-next-line require-atomic-updates
|
|
if ('updated_time' in saveQuery.modObject) o.updated_time = saveQuery.modObject.updated_time;
|
|
// eslint-disable-next-line require-atomic-updates
|
|
if ('created_time' in saveQuery.modObject) o.created_time = saveQuery.modObject.created_time;
|
|
// eslint-disable-next-line require-atomic-updates
|
|
if ('user_updated_time' in saveQuery.modObject) o.user_updated_time = saveQuery.modObject.user_updated_time;
|
|
// eslint-disable-next-line require-atomic-updates
|
|
if ('user_created_time' in saveQuery.modObject) o.user_created_time = saveQuery.modObject.user_created_time;
|
|
o = this.addModelMd(o);
|
|
|
|
if (isDiffSaving) {
|
|
for (let n in options.oldItem) {
|
|
if (!options.oldItem.hasOwnProperty(n)) continue;
|
|
if (n in o) continue;
|
|
o[n] = options.oldItem[n];
|
|
}
|
|
}
|
|
|
|
output = this.filter(o);
|
|
} finally {
|
|
this.releaseSaveMutex(o, mutexRelease);
|
|
}
|
|
|
|
return output;
|
|
}
|
|
|
|
static isNew(object, options) {
|
|
if (options && 'isNew' in options) {
|
|
// options.isNew can be "auto" too
|
|
if (options.isNew === true) return true;
|
|
if (options.isNew === false) return false;
|
|
}
|
|
|
|
return !object.id;
|
|
}
|
|
|
|
static filterArray(models) {
|
|
let output = [];
|
|
for (let i = 0; i < models.length; i++) {
|
|
output.push(this.filter(models[i]));
|
|
}
|
|
return output;
|
|
}
|
|
|
|
static filter(model) {
|
|
if (!model) return model;
|
|
|
|
let output = Object.assign({}, model);
|
|
for (let n in output) {
|
|
if (!output.hasOwnProperty(n)) continue;
|
|
|
|
// The SQLite database doesn't have booleans so cast everything to int
|
|
if (output[n] === true) {
|
|
output[n] = 1;
|
|
} else if (output[n] === false) {
|
|
output[n] = 0;
|
|
} else {
|
|
const t = this.fieldType(n, Database.TYPE_UNKNOWN);
|
|
if (t === Database.TYPE_INT) {
|
|
output[n] = !n ? 0 : parseInt(output[n], 10);
|
|
}
|
|
}
|
|
}
|
|
|
|
return output;
|
|
}
|
|
|
|
static delete(id) {
|
|
if (!id) throw new Error('Cannot delete object without an ID');
|
|
return this.db().exec(`DELETE FROM ${this.tableName()} WHERE id = ?`, [id]);
|
|
}
|
|
|
|
static batchDelete(ids, options = null) {
|
|
if (!ids.length) return;
|
|
options = this.modOptions(options);
|
|
const idFieldName = options.idFieldName ? options.idFieldName : 'id';
|
|
const sql = `DELETE FROM ${this.tableName()} WHERE ${idFieldName} IN ("${ids.join('","')}")`;
|
|
return this.db().exec(sql);
|
|
}
|
|
|
|
static db() {
|
|
if (!this.db_) throw new Error('Accessing database before it has been initialised');
|
|
return this.db_;
|
|
}
|
|
|
|
static isReady() {
|
|
return !!this.db_;
|
|
}
|
|
}
|
|
|
|
BaseModel.typeEnum_ = [['TYPE_NOTE', 1], ['TYPE_FOLDER', 2], ['TYPE_SETTING', 3], ['TYPE_RESOURCE', 4], ['TYPE_TAG', 5], ['TYPE_NOTE_TAG', 6], ['TYPE_SEARCH', 7], ['TYPE_ALARM', 8], ['TYPE_MASTER_KEY', 9], ['TYPE_ITEM_CHANGE', 10], ['TYPE_NOTE_RESOURCE', 11], ['TYPE_RESOURCE_LOCAL_STATE', 12], ['TYPE_REVISION', 13], ['TYPE_MIGRATION', 14]];
|
|
|
|
for (let i = 0; i < BaseModel.typeEnum_.length; i++) {
|
|
const e = BaseModel.typeEnum_[i];
|
|
BaseModel[e[0]] = e[1];
|
|
}
|
|
|
|
BaseModel.db_ = null;
|
|
BaseModel.dispatch = function() {};
|
|
BaseModel.saveMutexes_ = {};
|
|
|
|
module.exports = BaseModel;
|