2017-12-03 10:48:22 +08:00
|
|
|
"use strict";
|
|
|
|
|
|
|
|
const sql = require('../../services/sql');
|
2017-12-03 22:19:48 +08:00
|
|
|
const html = require('html');
|
2018-02-25 23:55:21 +08:00
|
|
|
const tar = require('tar-stream');
|
|
|
|
const sanitize = require("sanitize-filename");
|
2018-03-03 22:30:18 +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-03 22:30:18 +08:00
|
|
|
const repo = new Repository(req);
|
2017-12-03 10:48:22 +08:00
|
|
|
|
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 03:34:07 +08:00
|
|
|
const name = await exportNoteInner(branchId, '', pack, repo);
|
2017-12-03 10:48:22 +08:00
|
|
|
|
2018-02-25 23:55:21 +08:00
|
|
|
pack.finalize();
|
2017-12-03 13:10:43 +08:00
|
|
|
|
2018-03-28 10:11:06 +08:00
|
|
|
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 03:34:07 +08:00
|
|
|
async function exportNoteInner(branchId, directory, pack, repo) {
|
2018-03-25 09:39:15 +08:00
|
|
|
const branch = await sql.getRow("SELECT * FROM branches WHERE branchId = ?", [branchId]);
|
|
|
|
const note = await repo.getEntity("SELECT notes.* FROM notes WHERE noteId = ?", [branch.noteId]);
|
2017-12-03 13:10:43 +08:00
|
|
|
|
2018-02-26 13:07:43 +08:00
|
|
|
if (note.isProtected) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2018-02-27 09:47:34 +08:00
|
|
|
const metadata = await getMetadata(note);
|
|
|
|
|
2018-03-25 10:02:26 +08:00
|
|
|
if (metadata.labels.find(attr => attr.name === 'exclude_from_export')) {
|
2018-02-27 09:47:34 +08:00
|
|
|
return;
|
|
|
|
}
|
2017-12-03 10:48:22 +08:00
|
|
|
|
2018-02-27 09:47:34 +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
|
|
|
|
2018-02-27 09:47:34 +08:00
|
|
|
pack.entry({ name: childFileName + ".meta", size: metadataJson.length }, metadataJson);
|
2017-12-03 10:48:22 +08:00
|
|
|
|
2018-02-27 09:47:34 +08:00
|
|
|
const content = note.type === 'text' ? html.prettyPrint(note.content, {indent_size: 2}) : note.content;
|
2017-12-03 10:48:22 +08:00
|
|
|
|
2018-02-27 09:47:34 +08:00
|
|
|
pack.entry({ name: childFileName + ".dat", size: content.length }, content);
|
2017-12-03 10:48:22 +08:00
|
|
|
|
2018-03-25 09:39:15 +08:00
|
|
|
const children = await sql.getRows("SELECT * FROM branches WHERE parentNoteId = ? AND isDeleted = 0", [note.noteId]);
|
2018-02-25 23:55:21 +08:00
|
|
|
|
|
|
|
if (children.length > 0) {
|
2017-12-03 10:48:22 +08:00
|
|
|
for (const child of children) {
|
2018-03-31 03:34:07 +08:00
|
|
|
await exportNoteInner(child.branchId, childFileName + "/", pack, repo);
|
2017-12-03 10:48:22 +08:00
|
|
|
}
|
|
|
|
}
|
2018-02-25 23:55:21 +08:00
|
|
|
|
|
|
|
return childFileName;
|
|
|
|
}
|
|
|
|
|
|
|
|
async function getMetadata(note) {
|
2018-02-27 09:47:34 +08:00
|
|
|
return {
|
2018-03-03 22:32:21 +08:00
|
|
|
version: 1,
|
2018-02-25 23:55:21 +08:00
|
|
|
title: note.title,
|
|
|
|
type: note.type,
|
|
|
|
mime: note.mime,
|
2018-03-25 10:02:26 +08:00
|
|
|
labels: (await note.getLabels()).map(attr => {
|
2018-03-03 22:30:18 +08:00
|
|
|
return {
|
|
|
|
name: attr.name,
|
|
|
|
value: attr.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
|
|
|
|
};
|