trilium/src/routes/api/export.js

"use strict";

const express = require('express');
const router = express.Router();
const sql = require('../../services/sql');
const html = require('html');
const auth = require('../../services/auth');
const wrap = require('express-promise-wrap').wrap;
const tar = require('tar-stream');
const sanitize = require("sanitize-filename");
const Repository = require("../../services/repository");

router.get('/:noteId/', auth.checkApiAuthOrElectron, wrap(async (req, res, next) => {
    const noteId = req.params.noteId;
    const repo = new Repository(req);

    const noteTreeId = await sql.getValue('SELECT noteTreeId FROM note_tree WHERE noteId = ?', [noteId]);

    const pack = tar.pack();

    const name = await exportNote(noteTreeId, '', pack, repo);

    pack.finalize();

    res.setHeader('Content-Disposition', 'attachment; filename="' + name + '.tar"');
    res.setHeader('Content-Type', 'application/tar');

    pack.pipe(res);
}));

async function exportNote(noteTreeId, directory, pack, repo) {
    const noteTree = await sql.getRow("SELECT * FROM note_tree WHERE noteTreeId = ?", [noteTreeId]);
    const note = await repo.getEntity("SELECT notes.* FROM notes WHERE noteId = ?", [noteTree.noteId]);

    if (note.isProtected) {
        return;
    }

    const metadata = await getMetadata(note);

    if ('exclude_from_export' in metadata.attributes) {
        return;
    }

    const metadataJson = JSON.stringify(metadata, null, '\t');
    const childFileName = directory + sanitize(note.title);

    pack.entry({ name: childFileName + ".meta", size: metadataJson.length }, metadataJson);

    const content = note.type === 'text' ? html.prettyPrint(note.content, {indent_size: 2}) : note.content;

    pack.entry({ name: childFileName + ".dat", size: content.length }, content);

    const children = await sql.getRows("SELECT * FROM note_tree WHERE parentNoteId = ? AND isDeleted = 0", [note.noteId]);

    if (children.length > 0) {
        for (const child of children) {
            await exportNote(child.noteTreeId, childFileName + "/", pack, repo);
        }
    }

    return childFileName;
}

async function getMetadata(note) {
    return {
        title: note.title,
        type: note.type,
        mime: note.mime,
        attributes: (await note.getAttributes()).map(attr => {
            return {
                name: attr.name,
                value: attr.value
            };
        })
    };
}

module.exports = router;