mirror of
https://github.com/laurent22/joplin.git
synced 2024-12-24 10:27:10 +02:00
353 lines
8.8 KiB
JavaScript
353 lines
8.8 KiB
JavaScript
import { Log } from 'lib/log.js';
|
|
import { Database } from 'lib/database.js';
|
|
import { uuid } from 'lib/uuid.js';
|
|
import { time } from 'lib/time-utils.js';
|
|
|
|
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 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) {
|
|
let fields = this.fields();
|
|
for (let i = 0; i < fields.length; i++) {
|
|
if (fields[i].name == name) return fields[i].type;
|
|
}
|
|
throw new Error('Unknown field: ' + name);
|
|
}
|
|
|
|
static fields() {
|
|
return this.db().tableFields(this.tableName());
|
|
}
|
|
|
|
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() {
|
|
return this.db().selectOne('SELECT count(*) as total FROM `' + this.tableName() + '`').then((r) => {
|
|
return r ? r['total'] : 0;
|
|
});
|
|
}
|
|
|
|
static load(id) {
|
|
return this.loadByField('id', id);
|
|
}
|
|
|
|
static shortId(id) {
|
|
return id.substr(0, 5);
|
|
}
|
|
|
|
static loadByPartialId(partialId) {
|
|
return this.modelSelectOne('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) {
|
|
let q = this.applySqlOptions(options, 'SELECT * FROM `' + this.tableName() + '`');
|
|
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) {
|
|
return this.modelSelectOne('SELECT * FROM `' + this.tableName() + '` WHERE `' + fieldName + '` = ?', [fieldValue]);
|
|
}
|
|
|
|
static loadByTitle(fieldValue) {
|
|
return this.modelSelectOne('SELECT * FROM `' + this.tableName() + '` WHERE `title` = ?', [fieldValue]);
|
|
}
|
|
|
|
static diffObjects(oldModel, newModel) {
|
|
let output = {};
|
|
let type = null;
|
|
for (let n in newModel) {
|
|
if (n == 'type_') {
|
|
type = n;
|
|
continue;
|
|
}
|
|
if (!newModel.hasOwnProperty(n)) continue;
|
|
if (!(n in oldModel) || newModel[n] !== oldModel[n]) {
|
|
output[n] = newModel[n];
|
|
}
|
|
}
|
|
if (type !== null) output.type_ = type;
|
|
return output;
|
|
}
|
|
|
|
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];
|
|
}
|
|
o = temp;
|
|
|
|
let query = {};
|
|
let modelId = o.id;
|
|
|
|
const timeNow = time.unixMs();
|
|
|
|
if (options.autoTimestamp && this.hasField('updated_time')) {
|
|
o.updated_time = timeNow;
|
|
}
|
|
|
|
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 = 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 save(o, options = null) {
|
|
options = this.modOptions(options);
|
|
options.isNew = this.isNew(o, options);
|
|
|
|
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);
|
|
}
|
|
|
|
return this.db().transactionExecBatch(queries).then(() => {
|
|
o = Object.assign({}, o);
|
|
o.id = modelId;
|
|
if ('updated_time' in saveQuery.modObject) o.updated_time = saveQuery.modObject.updated_time;
|
|
if ('created_time' in saveQuery.modObject) o.created_time = saveQuery.modObject.created_time;
|
|
if ('user_updated_time' in saveQuery.modObject) o.user_updated_time = saveQuery.modObject.user_updated_time;
|
|
if ('user_created_time' in saveQuery.modObject) o.user_created_time = saveQuery.modObject.user_created_time;
|
|
o = this.addModelMd(o);
|
|
return this.filter(o);
|
|
}).catch((error) => {
|
|
Log.error('Cannot save model', error);
|
|
});
|
|
}
|
|
|
|
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;
|
|
if (output[n] === false) output[n] = 0;
|
|
}
|
|
|
|
return output;
|
|
}
|
|
|
|
static delete(id, options = null) {
|
|
options = this.modOptions(options);
|
|
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) {
|
|
options = this.modOptions(options);
|
|
if (!ids.length) throw new Error('Cannot delete object without an ID');
|
|
return this.db().exec('DELETE FROM ' + this.tableName() + ' WHERE id IN ("' + ids.join('","') + '")');
|
|
}
|
|
|
|
static db() {
|
|
if (!this.db_) throw new Error('Accessing database before it has been initialised');
|
|
return this.db_;
|
|
}
|
|
|
|
static isReady() {
|
|
return !!this.db_;
|
|
}
|
|
|
|
}
|
|
|
|
BaseModel.TYPE_NOTE = 1;
|
|
BaseModel.TYPE_FOLDER = 2;
|
|
BaseModel.TYPE_SETTING = 3;
|
|
BaseModel.TYPE_RESOURCE = 4;
|
|
BaseModel.TYPE_TAG = 5;
|
|
BaseModel.TYPE_NOTE_TAG = 6;
|
|
|
|
BaseModel.db_ = null;
|
|
BaseModel.dispatch = function(o) {};
|
|
|
|
export { BaseModel }; |