trilium/src/services/script_context.js

const log = require('./log');
const protected_session = require('./protected_session');
const notes = require('./notes');
const attributes = require('./attributes');
const date_notes = require('./date_notes');
const Repository = require('./repository');

function ScriptContext(noteId, dataKey) {
    this.dataKey = protected_session.getDataKey(dataKey);
    this.repository = new Repository(dataKey);

    this.getNoteById = async function(noteId) {
        return this.repository.getNote(noteId);
    };

    this.getNotesWithAttribute = async function (attrName, attrValue) {
        return await attributes.getNotesWithAttribute(this.dataKey, attrName, attrValue);
    };

    this.getNoteWithAttribute = async function (attrName, attrValue) {
        const notes = await this.getNotesWithAttribute(attrName, attrValue);

        return notes.length > 0 ? notes[0] : null;
    };

    this.createNote = async function (parentNoteId, name, jsonContent, extraOptions = {}) {
        const note = {
            title: name,
            content: extraOptions.json ? JSON.stringify(jsonContent, null, '\t') : jsonContent,
            target: 'into',
            isProtected: extraOptions.isProtected !== undefined ? extraOptions.isProtected : false,
            type: extraOptions.type,
            mime: extraOptions.mime
        };

        if (extraOptions.json) {
            note.type = "code";
            note.mime = "application/json";
        }

        if (!note.type) {
            note.type = "text";
            note.mime = "text/html";
        }

        const noteId = (await notes.createNewNote(parentNoteId, note, this.dataKey)).noteId;

        if (extraOptions.attributes) {
            for (const attrName in extraOptions.attributes) {
                await attributes.createAttribute(noteId, attrName, extraOptions.attributes[attrName]);
            }
        }

        return noteId;
    };

    this.createAttribute = attributes.createAttribute;

    this.updateEntity = this.repository.updateEntity;

    this.log = function(message) {
        log.info(`Script: ${message}`);
    };

    this.getDateNoteId = date_notes.getDateNoteId;
}

module.exports = ScriptContext;