trilium/src/becca/entity_constructor.js
2021-05-17 22:09:49 +02:00

59 lines
1.5 KiB
JavaScript

const Note = require('./entities/note.js');
const NoteRevision = require('./entities/note_revision.js');
const Branch = require('./entities/branch.js');
const Attribute = require('./entities/attribute.js');
const RecentNote = require('./entities/recent_note.js');
const ApiToken = require('./entities/api_token.js');
const ENTITY_NAME_TO_ENTITY = {
"attributes": Attribute,
"branches": Branch,
"notes": Note,
"note_contents": Note,
"note_revisions": NoteRevision,
"note_revision_contents": NoteRevision,
"recent_notes": RecentNote,
"api_tokens": ApiToken,
};
function getEntityFromEntityName(entityName) {
if (!(entityName in ENTITY_NAME_TO_ENTITY)) {
throw new Error(`Entity for table ${entityName} not found!`);
}
return ENTITY_NAME_TO_ENTITY[entityName];
}
function createEntityFromRow(row) {
let entity;
if (row.attributeId) {
entity = new Attribute(row);
}
else if (row.noteRevisionId) {
entity = new NoteRevision(row);
}
else if (row.branchId && row.notePath) {
entity = new RecentNote(row);
}
else if (row.apiTokenId) {
entity = new ApiToken(row);
}
else if (row.branchId) {
entity = new Branch(row);
}
else if (row.noteId) {
entity = new Note(row);
}
else {
throw new Error('Unknown entity type for row: ' + JSON.stringify(row));
}
return entity;
}
module.exports = {
createEntityFromRow,
getEntityFromEntityName
};