//response //external modules var ejs = require('ejs'); var fs = require('fs'); var path = require('path'); var uuid = require('node-uuid'); var markdownpdf = require("markdown-pdf"); var LZString = require('lz-string'); var S = require('string'); var shortId = require('shortid'); //core var config = require("../config.js"); //others var db = require("./db.js"); var Note = require("./note.js"); //public var response = { errorForbidden: function (res) { res.status(403).send("Forbidden, oh no."); }, errorNotFound: function (res) { responseError(res, "404", "Not Found", "oops."); }, errorInternalError: function (res) { responseError(res, "500", "Internal Error", "wtf."); }, errorServiceUnavailable: function (res) { res.status(503).send("I'm busy right now, try again later."); }, newNote: newNote, showFeatures: showFeatures, showNote: showNote, showShareNote: showShareNote, noteActions: noteActions, shareNoteActions: shareNoteActions }; function responseError(res, code, detail, msg) { res.writeHead(code, { 'Content-Type': 'text/html' }); var template = config.errorpath; var content = ejs.render(fs.readFileSync(template, 'utf8'), { title: code + ' ' + detail + ' ' + msg, cache: !config.debug, filename: template, code: code, detail: detail, msg: msg }); res.write(content); res.end(); } function responseHackMD(res, noteId) { if (noteId != config.featuresnotename) { if (!Note.checkNoteIdValid(noteId)) { responseError(res, "404", "Not Found", "oops."); return; } noteId = LZString.decompressFromBase64(noteId); if (!noteId) { responseError(res, "404", "Not Found", "oops."); return; } } db.readFromDB(noteId, function (err, data) { if (err) { responseError(res, "404", "Not Found", "oops."); return; } var title = data.rows[0].title; var decodedTitle = LZString.decompressFromBase64(title); if (decodedTitle) title = decodedTitle; title = Note.generateWebTitle(title); var template = config.hackmdpath; var options = { cache: !config.debug, filename: template }; var compiled = ejs.compile(fs.readFileSync(template, 'utf8'), options); var html = compiled({ title: title }); var buf = html; res.writeHead(200, { 'Content-Type': 'text/html; charset=UTF-8', 'Cache-Control': 'private', 'Content-Length': buf.length }); res.end(buf); }); } function newNote(req, res, next) { var newId = uuid.v4(); var body = fs.readFileSync(config.defaultnotepath, 'utf8'); body = LZString.compressToBase64(body); var owner = null; if (req.isAuthenticated()) { owner = req.session.passport.user; } db.newToDB(newId, owner, body, function (err, result) { if (err) { responseError(res, "500", "Internal Error", "wtf."); return; } res.redirect("/" + LZString.compressToBase64(newId)); }); } function showFeatures(req, res, next) { db.readFromDB(config.featuresnotename, function (err, data) { if (err) { var body = fs.readFileSync(config.defaultfeaturespath, 'utf8'); body = LZString.compressToBase64(body); db.newToDB(config.featuresnotename, null, body, function (err, result) { if (err) { responseError(res, "500", "Internal Error", "wtf."); return; } responseHackMD(res, config.featuresnotename); }); } else { responseHackMD(res, config.featuresnotename); } }); } function showNote(req, res, next) { var noteId = req.params.noteId; if (!Note.checkNoteIdValid(noteId)) { responseError(res, "404", "Not Found", "oops."); return; } responseHackMD(res, noteId); } function showShareNote(req, res, next) { var shortid = req.params.shortid; if (shortId.isValid(shortid)) { Note.findNote(shortid, function (err, note) { if (err || !note) { responseError(res, "404", "Not Found", "oops."); return; } //increase note viewcount Note.increaseViewCount(note, function (err, note) { if (err || !note) { responseError(res, "404", "Not Found", "oops."); return; } db.readFromDB(note.id, function (err, data) { if (err) { responseError(res, "404", "Not Found", "oops."); return; } var body = LZString.decompressFromBase64(data.rows[0].content); var updatetime = data.rows[0].update_time; var text = S(body).escapeHTML().s; var title = data.rows[0].title; var decodedTitle = LZString.decompressFromBase64(title); if (decodedTitle) title = decodedTitle; title = Note.generateWebTitle(title); var template = config.prettypath; var options = { cache: !config.debug, filename: template }; var compiled = ejs.compile(fs.readFileSync(template, 'utf8'), options); var origin = "//" + req.headers.host; var html = compiled({ title: title, viewcount: note.viewcount, updatetime: updatetime, url: origin, body: text }); var buf = html; res.writeHead(200, { 'Content-Type': 'text/html; charset=UTF-8', 'Cache-Control': 'private', 'Content-Length': buf.length }); res.end(buf); }); }); }); } else { responseError(res, "404", "Not Found", "oops."); } } function actionShare(req, res, noteId) { db.readFromDB(noteId, function (err, data) { if (err) { responseError(res, "404", "Not Found", "oops."); return; } var owner = data.rows[0].owner; var permission = "freely"; if (owner && owner != "null") { permission = "editable"; } Note.findOrNewNote(noteId, permission, function (err, note) { if (err) { responseError(res, "404", "Not Found", "oops."); return; } res.redirect("/s/" + note.shortid); }); }); } //pretty api is deprecated function actionPretty(req, res, noteId) { db.readFromDB(noteId, function (err, data) { if (err) { responseError(res, "404", "Not Found", "oops."); return; } var body = LZString.decompressFromBase64(data.rows[0].content); var text = S(body).escapeHTML().s; var title = data.rows[0].title; var decodedTitle = LZString.decompressFromBase64(title); if (decodedTitle) title = decodedTitle; title = Note.generateWebTitle(title); var template = config.prettypath; var compiled = ejs.compile(fs.readFileSync(template, 'utf8')); var origin = "//" + req.headers.host; var html = compiled({ title: title, url: origin, body: text }); var buf = html; res.writeHead(200, { 'Content-Type': 'text/html; charset=UTF-8', 'Cache-Control': 'private', 'Content-Length': buf.length }); res.end(buf); }); } function actionDownload(req, res, noteId) { db.readFromDB(noteId, function (err, data) { if (err) { responseError(res, "404", "Not Found", "oops."); return; } var body = LZString.decompressFromBase64(data.rows[0].content); var title = Note.getNoteTitle(body); res.writeHead(200, { 'Content-Type': 'text/markdown; charset=UTF-8', 'Cache-Control': 'private', 'Content-disposition': 'attachment; filename=' + title + '.md', 'Content-Length': body.length }); res.end(body); }); } function actionPDF(req, res, noteId) { db.readFromDB(noteId, function (err, data) { if (err) { responseError(res, "404", "Not Found", "oops."); return; } var body = LZString.decompressFromBase64(data.rows[0].content); var title = Note.getNoteTitle(body); if (!fs.existsSync(config.tmppath)) { fs.mkdirSync(config.tmppath); } var path = config.tmppath + Date.now() + '.pdf'; markdownpdf().from.string(body).to(path, function () { var stream = fs.createReadStream(path); var filename = title; // Be careful of special characters filename = encodeURIComponent(filename); // Ideally this should strip them res.setHeader('Content-disposition', 'attachment; filename="' + filename + '.pdf"'); res.setHeader('Cache-Control', 'private'); res.setHeader('Content-Type', 'application/pdf; charset=UTF-8'); stream.pipe(res); fs.unlink(path); }); }); } function noteActions(req, res, next) { var noteId = req.params.noteId; if (noteId != config.featuresnotename) { if (!Note.checkNoteIdValid(noteId)) { responseError(res, "404", "Not Found", "oops."); return; } noteId = LZString.decompressFromBase64(noteId); if (!noteId) { responseError(res, "404", "Not Found", "oops."); return; } } var action = req.params.action; switch (action) { case "share": case "pretty": //pretty deprecated actionShare(req, res, noteId); break; case "download": actionDownload(req, res, noteId); break; case "pdf": actionPDF(req, res, noteId); break; default: if (noteId != config.featuresnotename) res.redirect('/' + LZString.compressToBase64(noteId)); else res.redirect('/' + noteId); break; } } function shareNoteActions(req, res, next) { var action = req.params.action; switch (action) { case "edit": var shortid = req.params.shortid; if (shortId.isValid(shortid)) { Note.findNote(shortid, function (err, note) { if (err || !note) { responseError(res, "404", "Not Found", "oops."); return; } if (note.id != config.featuresnotename) res.redirect('/' + LZString.compressToBase64(note.id)); else res.redirect('/' + note.id); }); } break; } } module.exports = response;