trilium/src/routes/custom.js

const log = require('../services/log');
const fileUploadService = require('./api/files');
const scriptService = require('../services/script');
const cls = require('../services/cls');
const sql = require("../services/sql");
const becca = require("../becca/becca");

function handleRequest(req, res) {
    // express puts content after first slash into 0 index element

    const path = req.params.path + req.params[0];

    const attributeIds = sql.getColumn("SELECT attributeId FROM attributes WHERE isDeleted = 0 AND type = 'label' AND name IN ('customRequestHandler', 'customResourceProvider')");

    const attrs = attributeIds.map(attrId => becca.getAttribute(attrId));

    for (const attr of attrs) {
        if (!attr.value.trim()) {
            continue;
        }

        const regex = new RegExp(`^${attr.value}$`);
        let match;

        try {
            match = path.match(regex);
        }
        catch (e) {
            log.error(`Testing path for label ${attr.attributeId}, regex=${attr.value} failed with error ${e.stack}`);
            continue;
        }

        if (!match) {
            continue;
        }

        if (attr.name === 'customRequestHandler') {
            const note = attr.getNote();

            log.info(`Handling custom request "${path}" with note ${note.noteId}`);

            try {
                scriptService.executeNote(note, {
                    pathParams: match.slice(1),
                    req,
                    res
                });
            }
            catch (e) {
                log.error(`Custom handler ${note.noteId} failed with ${e.message}`);

                res.setHeader("Content-Type", "text/plain")
                    .status(500)
                    .send(e.message);
            }
        }
        else if (attr.name === 'customResourceProvider') {
            fileUploadService.downloadNoteFile(attr.noteId, res);
        }
        else {
            throw new Error(`Unrecognized attribute name ${attr.name}`);
        }

        return; // only first handler is executed
    }

    const message = `No handler matched for custom ${path} request.`;

    log.info(message);
    res.setHeader("Content-Type", "text/plain")
        .status(404)
        .send(message);
}

function register(router) {
    // explicitly no CSRF middleware since it's meant to allow integration from external services

    router.all('/custom/:path*', (req, res, next) => {
        cls.namespace.bindEmitter(req);
        cls.namespace.bindEmitter(res);

        cls.init(() => handleRequest(req, res));
    });
}

module.exports = {
    register
};