2018-02-15 12:31:20 +08:00
|
|
|
"use strict";
|
|
|
|
|
2018-04-02 09:27:46 +08:00
|
|
|
const noteService = require('../../services/notes');
|
2018-08-07 19:33:10 +08:00
|
|
|
const attributeService = require('../../services/attributes');
|
2018-04-02 09:27:46 +08:00
|
|
|
const protectedSessionService = require('../../services/protected_session');
|
2018-04-01 23:42:12 +08:00
|
|
|
const repository = require('../../services/repository');
|
2018-02-15 12:31:20 +08:00
|
|
|
|
2018-03-31 05:29:13 +08:00
|
|
|
async function uploadFile(req) {
|
2018-02-15 12:31:20 +08:00
|
|
|
const parentNoteId = req.params.parentNoteId;
|
|
|
|
const file = req.file;
|
2018-02-19 10:28:24 +08:00
|
|
|
const originalName = file.originalname;
|
|
|
|
const size = file.size;
|
2018-02-15 12:31:20 +08:00
|
|
|
|
2018-04-01 23:42:12 +08:00
|
|
|
const parentNote = await repository.getNote(parentNoteId);
|
2018-02-15 12:31:20 +08:00
|
|
|
|
2018-04-01 23:42:12 +08:00
|
|
|
if (!parentNote) {
|
2018-03-31 05:29:13 +08:00
|
|
|
return [404, `Note ${parentNoteId} doesn't exist.`];
|
2018-02-15 12:31:20 +08:00
|
|
|
}
|
|
|
|
|
2018-04-02 09:27:46 +08:00
|
|
|
const {note} = await noteService.createNewNote(parentNoteId, {
|
2018-03-31 05:29:13 +08:00
|
|
|
title: originalName,
|
|
|
|
content: file.buffer,
|
|
|
|
target: 'into',
|
|
|
|
isProtected: false,
|
|
|
|
type: 'file',
|
|
|
|
mime: file.mimetype
|
2018-04-01 23:42:12 +08:00
|
|
|
});
|
2018-03-31 05:29:13 +08:00
|
|
|
|
2018-08-07 19:33:10 +08:00
|
|
|
await attributeService.createLabel(note.noteId, "originalFileName", originalName);
|
|
|
|
await attributeService.createLabel(note.noteId, "fileSize", size);
|
2018-03-31 05:29:13 +08:00
|
|
|
|
|
|
|
return {
|
2018-04-01 23:42:12 +08:00
|
|
|
noteId: note.noteId
|
2018-03-31 05:29:13 +08:00
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
async function downloadFile(req, res) {
|
2018-02-19 10:28:24 +08:00
|
|
|
const noteId = req.params.noteId;
|
2018-04-01 23:42:12 +08:00
|
|
|
const note = await repository.getNote(noteId);
|
2018-02-19 10:28:24 +08:00
|
|
|
|
|
|
|
if (!note) {
|
2018-02-24 11:58:24 +08:00
|
|
|
return res.status(404).send(`Note ${noteId} doesn't exist.`);
|
|
|
|
}
|
|
|
|
|
2018-04-02 09:27:46 +08:00
|
|
|
if (note.isProtected && !protectedSessionService.isProtectedSessionAvailable()) {
|
2018-04-01 23:42:12 +08:00
|
|
|
res.status(401).send("Protected session not available");
|
|
|
|
return;
|
2018-02-19 10:28:24 +08:00
|
|
|
}
|
|
|
|
|
2018-08-07 19:33:10 +08:00
|
|
|
const originalFileName = await note.getLabel('originalFileName');
|
|
|
|
const fileName = originalFileName.value || note.title;
|
2018-02-19 10:28:24 +08:00
|
|
|
|
2018-04-01 23:42:12 +08:00
|
|
|
res.setHeader('Content-Disposition', 'file; filename="' + fileName + '"');
|
2018-02-19 10:28:24 +08:00
|
|
|
res.setHeader('Content-Type', note.mime);
|
|
|
|
|
|
|
|
res.send(note.content);
|
2018-03-31 05:29:13 +08:00
|
|
|
}
|
2018-02-19 10:28:24 +08:00
|
|
|
|
2018-03-31 05:29:13 +08:00
|
|
|
module.exports = {
|
|
|
|
uploadFile,
|
|
|
|
downloadFile
|
|
|
|
};
|