trilium/src/routes/api/export.js

73 lines
1.9 KiB
JavaScript
Raw Normal View History

2017-12-03 10:48:22 +08:00
"use strict";
const sql = require('../../services/sql');
const html = require('html');
2018-02-25 23:55:21 +08:00
const tar = require('tar-stream');
const sanitize = require("sanitize-filename");
2018-03-31 21:07:58 +08:00
const repository = require("../../services/repository");
2017-12-03 10:48:22 +08:00
2018-03-31 03:34:07 +08:00
async function exportNote(req, res) {
2017-12-03 10:48:22 +08:00
const noteId = req.params.noteId;
2018-03-25 09:39:15 +08:00
const branchId = await sql.getValue('SELECT branchId FROM branches WHERE noteId = ?', [noteId]);
2017-12-03 10:48:22 +08:00
2018-02-25 23:55:21 +08:00
const pack = tar.pack();
2017-12-03 10:48:22 +08:00
2018-03-31 21:07:58 +08:00
const name = await exportNoteInner(branchId, '', pack);
2017-12-03 10:48:22 +08:00
2018-02-25 23:55:21 +08:00
pack.finalize();
res.setHeader('Content-Disposition', 'file; filename="' + name + '.tar"');
2018-02-25 23:55:21 +08:00
res.setHeader('Content-Type', 'application/tar');
2017-12-03 10:48:22 +08:00
2018-02-25 23:55:21 +08:00
pack.pipe(res);
2018-03-31 03:34:07 +08:00
}
2017-12-03 10:48:22 +08:00
2018-03-31 21:07:58 +08:00
async function exportNoteInner(branchId, directory, pack) {
const branch = await repository.getBranch(branchId);
const note = await branch.getNote();
if (note.isProtected) {
return;
}
const metadata = await getMetadata(note);
2018-04-03 09:56:55 +08:00
if (metadata.labels.find(label => label.name === 'excludeFromExport')) {
return;
}
2017-12-03 10:48:22 +08:00
const metadataJson = JSON.stringify(metadata, null, '\t');
2018-02-25 23:55:21 +08:00
const childFileName = directory + sanitize(note.title);
2017-12-03 10:48:22 +08:00
pack.entry({ name: childFileName + ".meta", size: metadataJson.length }, metadataJson);
2017-12-03 10:48:22 +08:00
const content = note.type === 'text' ? html.prettyPrint(note.content, {indent_size: 2}) : note.content;
2017-12-03 10:48:22 +08:00
pack.entry({ name: childFileName + ".dat", size: content.length }, content);
2017-12-03 10:48:22 +08:00
for (const child of await note.getChildBranches()) {
await exportNoteInner(child.branchId, childFileName + "/", pack);
2017-12-03 10:48:22 +08:00
}
2018-02-25 23:55:21 +08:00
return childFileName;
}
async function getMetadata(note) {
return {
version: 1,
2018-02-25 23:55:21 +08:00
title: note.title,
type: note.type,
mime: note.mime,
labels: (await note.getLabels()).map(label => {
return {
name: label.name,
value: label.value
};
})
2018-02-25 23:55:21 +08:00
};
2017-12-03 10:48:22 +08:00
}
2018-03-31 03:34:07 +08:00
module.exports = {
exportNote
};