Trilium/src/routes/api/attachments.js

const becca = require("../../becca/becca");
const NotFoundError = require("../../errors/not_found_error");
const utils = require("../../services/utils");
const blobService = require("../../services/blob.js");

function getAttachmentBlob(req) {
    const full = req.query.full === 'true';

    return blobService.getBlobPojo('attachments', req.params.attachmentId, { full });
}

function getAttachments(req) {
    const includeContent = req.query.includeContent === 'true';
    const {noteId} = req.params;

    const note = becca.getNote(noteId);

    if (!note) {
        throw new NotFoundError(`Note '${noteId}' doesn't exist.`);
    }

    return note.getAttachments()
        .map(attachment => processAttachment(attachment, includeContent));
}

function getAttachment(req) {
    const includeContent = req.query.includeContent === 'true';
    const {attachmentId} = req.params;

    const attachment = becca.getAttachment(attachmentId);

    if (!attachment) {
        throw new NotFoundError(`Attachment '${attachmentId}' doesn't exist.`);
    }

    return processAttachment(attachment, includeContent);
}

function processAttachment(attachment, includeContent) {
    const pojo = attachment.getPojo();

    if (includeContent) {
        if (utils.isStringNote(null, attachment.mime)) {
            pojo.content = attachment.getContent()?.toString();
            pojo.contentLength = pojo.content.length;

            const MAX_ATTACHMENT_LENGTH = 1_000_000;

            if (pojo.content.length > MAX_ATTACHMENT_LENGTH) {
                pojo.content = pojo.content.substring(0, MAX_ATTACHMENT_LENGTH);
            }
        } else {
            const content = attachment.getContent();
            pojo.contentLength = content?.length;
        }
    }

    return pojo;
}

function saveAttachment(req) {
    const {noteId} = req.params;
    const {attachmentId, role, mime, title, content} = req.body;

    const note = becca.getNote(noteId);

    if (!note) {
        throw new NotFoundError(`Note '${noteId}' doesn't exist.`);
    }

    note.saveAttachment({attachmentId, role, mime, title, content});
}

function deleteAttachment(req) {
    const {attachmentId} = req.params;

    const attachment = becca.getAttachment(attachmentId);

    if (attachment) {
        attachment.markAsDeleted();
    }
}

function convertAttachmentToNote(req) {
    const {attachmentId} = req.params;

    const attachment = becca.getAttachment(attachmentId);

    if (!attachment) {
        throw new NotFoundError(`Attachment '${attachmentId}' doesn't exist.`);
    }

    return attachment.convertToNote();
}

module.exports = {
    getAttachmentBlob,
    getAttachments,
    getAttachment,
    saveAttachment,
    deleteAttachment,
    convertAttachmentToNote
};