523 lines
16 KiB
JavaScript
523 lines
16 KiB
JavaScript
//response
|
|
//external modules
|
|
var ejs = require('ejs');
|
|
var fs = require('fs');
|
|
var path = require('path');
|
|
var markdownpdf = require("markdown-pdf");
|
|
var LZString = require('lz-string');
|
|
var S = require('string');
|
|
var shortId = require('shortid');
|
|
var metaMarked = require('meta-marked');
|
|
var querystring = require('querystring');
|
|
var request = require('request');
|
|
|
|
//core
|
|
var config = require("./config.js");
|
|
var logger = require("./logger.js");
|
|
var models = require("./models");
|
|
|
|
//slides
|
|
var md = require('reveal.js/plugin/markdown/markdown');
|
|
var Mustache = require('mustache');
|
|
|
|
//reveal.js
|
|
var opts = {
|
|
template: fs.readFileSync(config.slidepath).toString(),
|
|
theme: 'css/theme/black.css',
|
|
highlightTheme: 'zenburn',
|
|
separator: '^(\r\n?|\n)---(\r\n?|\n)$',
|
|
verticalSeparator: '^(\r\n?|\n)----(\r\n?|\n)$',
|
|
revealOptions: {}
|
|
};
|
|
|
|
//public
|
|
var response = {
|
|
errorForbidden: function (res) {
|
|
responseError(res, "403", "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,
|
|
showNote: showNote,
|
|
showPublishNote: showPublishNote,
|
|
showPublishSlide: showPublishSlide,
|
|
showIndex: showIndex,
|
|
noteActions: noteActions,
|
|
publishNoteActions: publishNoteActions,
|
|
githubActions: githubActions,
|
|
gitlabActions: gitlabActions
|
|
};
|
|
|
|
function responseError(res, code, detail, msg) {
|
|
res.writeHead(code, {
|
|
'Content-Type': 'text/html'
|
|
});
|
|
var template = config.errorpath;
|
|
var options = {
|
|
cache: !config.debug,
|
|
filename: template
|
|
};
|
|
var compiled = ejs.compile(fs.readFileSync(template, 'utf8'), options);
|
|
var content = compiled({
|
|
url: config.serverurl,
|
|
title: code + ' ' + detail + ' ' + msg,
|
|
cache: !config.debug,
|
|
filename: template,
|
|
code: code,
|
|
detail: detail,
|
|
msg: msg,
|
|
useCDN: config.usecdn
|
|
});
|
|
res.write(content);
|
|
res.end();
|
|
}
|
|
|
|
function showIndex(req, res, next) {
|
|
res.writeHead(200, {
|
|
'Content-Type': 'text/html'
|
|
});
|
|
var template = config.indexpath;
|
|
var options = {
|
|
cache: !config.debug,
|
|
filename: template
|
|
};
|
|
var compiled = ejs.compile(fs.readFileSync(template, 'utf8'), options);
|
|
var content = compiled({
|
|
url: config.serverurl,
|
|
useCDN: config.usecdn,
|
|
facebook: config.facebook,
|
|
twitter: config.twitter,
|
|
github: config.github,
|
|
gitlab: config.gitlab,
|
|
dropbox: config.dropbox
|
|
});
|
|
res.write(content);
|
|
res.end();
|
|
}
|
|
|
|
function responseHackMD(res, note) {
|
|
var body = LZString.decompressFromBase64(note.content);
|
|
var meta = null;
|
|
try {
|
|
meta = metaMarked(body).meta;
|
|
} catch(err) {
|
|
//na
|
|
}
|
|
var title = models.Note.decodeTitle(note.title);
|
|
title = models.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({
|
|
url: config.serverurl,
|
|
title: title,
|
|
useCDN: config.usecdn,
|
|
robots: (meta && meta.robots) || false, //default allow robots
|
|
facebook: config.facebook,
|
|
twitter: config.twitter,
|
|
github: config.github,
|
|
gitlab: config.gitlab,
|
|
dropbox: config.dropbox
|
|
});
|
|
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 owner = null;
|
|
if (req.isAuthenticated()) {
|
|
owner = req.user.id;
|
|
}
|
|
models.Note.create({
|
|
ownerId: owner
|
|
}).then(function (note) {
|
|
return res.redirect(config.serverurl + "/" + LZString.compressToBase64(note.id));
|
|
}).catch(function (err) {
|
|
logger.error(err);
|
|
return response.errorInternalError(res);
|
|
});
|
|
}
|
|
|
|
function checkViewPermission(req, note) {
|
|
if (note.permission == 'private') {
|
|
if (!req.isAuthenticated() || note.ownerId != req.user.id)
|
|
return false;
|
|
else
|
|
return true;
|
|
} else {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
function findNote(req, res, callback, include) {
|
|
var id = req.params.noteId || req.params.shortid;
|
|
models.Note.parseNoteId(id, function (err, _id) {
|
|
models.Note.findOne({
|
|
where: {
|
|
id: _id
|
|
},
|
|
include: include || null
|
|
}).then(function (note) {
|
|
if (!note) {
|
|
return response.errorNotFound(res);
|
|
}
|
|
if (!checkViewPermission(req, note)) {
|
|
return response.errorForbidden(res);
|
|
} else {
|
|
return callback(note);
|
|
}
|
|
}).catch(function (err) {
|
|
logger.error(err);
|
|
return response.errorInternalError(res);
|
|
});
|
|
});
|
|
}
|
|
|
|
function showNote(req, res, next) {
|
|
findNote(req, res, function (note) {
|
|
return responseHackMD(res, note);
|
|
});
|
|
}
|
|
|
|
function showPublishNote(req, res, next) {
|
|
var include = [{
|
|
model: models.User,
|
|
as: "owner"
|
|
}, {
|
|
model: models.User,
|
|
as: "lastchangeuser"
|
|
}];
|
|
findNote(req, res, function (note) {
|
|
note.increment('viewcount').then(function (note) {
|
|
if (!note) {
|
|
return response.errorNotFound(res);
|
|
}
|
|
var body = LZString.decompressFromBase64(note.content);
|
|
var meta = null;
|
|
try {
|
|
meta = metaMarked(body).meta;
|
|
} catch(err) {
|
|
//na
|
|
}
|
|
var createtime = note.createdAt;
|
|
var updatetime = note.lastchangeAt;
|
|
var text = S(body).escapeHTML().s;
|
|
var title = models.Note.decodeTitle(note.title);
|
|
title = models.Note.generateWebTitle(title);
|
|
var origin = config.serverurl;
|
|
var data = {
|
|
title: title,
|
|
viewcount: note.viewcount,
|
|
createtime: createtime,
|
|
updatetime: updatetime,
|
|
url: origin,
|
|
body: text,
|
|
useCDN: config.usecdn,
|
|
lastchangeuserprofile: note.lastchangeuser ? models.User.parseProfile(note.lastchangeuser.profile) : null,
|
|
robots: (meta && meta.robots) || false //default allow robots
|
|
};
|
|
return renderPublish(data, res);
|
|
}).catch(function (err) {
|
|
logger.error(err);
|
|
return response.errorInternalError(res);
|
|
});
|
|
}, include);
|
|
}
|
|
|
|
function renderPublish(data, res) {
|
|
var template = config.prettypath;
|
|
var options = {
|
|
url: config.serverurl,
|
|
cache: !config.debug,
|
|
filename: template
|
|
};
|
|
var compiled = ejs.compile(fs.readFileSync(template, 'utf8'), options);
|
|
var html = compiled(data);
|
|
var buf = html;
|
|
res.writeHead(200, {
|
|
'Content-Type': 'text/html; charset=UTF-8',
|
|
'Cache-Control': 'private',
|
|
'Content-Length': buf.length
|
|
});
|
|
res.end(buf);
|
|
}
|
|
|
|
function actionPublish(req, res, note) {
|
|
res.redirect(config.serverurl + "/s/" + (note.alias || note.shortid));
|
|
}
|
|
|
|
function actionSlide(req, res, note) {
|
|
res.redirect(config.serverurl + "/p/" + (note.alias || note.shortid));
|
|
}
|
|
|
|
function actionDownload(req, res, note) {
|
|
var body = LZString.decompressFromBase64(note.content);
|
|
var title = models.Note.decodeTitle(note.title);
|
|
var filename = title;
|
|
filename = encodeURIComponent(filename);
|
|
res.writeHead(200, {
|
|
'Access-Control-Allow-Origin': '*', //allow CORS as API
|
|
'Access-Control-Allow-Headers': 'Range',
|
|
'Access-Control-Expose-Headers': 'Cache-Control, Content-Encoding, Content-Range',
|
|
'Content-Type': 'text/markdown; charset=UTF-8',
|
|
'Cache-Control': 'private',
|
|
'Content-disposition': 'attachment; filename=' + filename + '.md',
|
|
'Content-Length': body.length,
|
|
'X-Robots-Tag': 'noindex, nofollow' // prevent crawling
|
|
});
|
|
res.end(body);
|
|
}
|
|
|
|
function actionPDF(req, res, note) {
|
|
var body = LZString.decompressFromBase64(note.content);
|
|
try {
|
|
body = metaMarked(body).markdown;
|
|
} catch(err) {
|
|
//na
|
|
}
|
|
var title = models.Note.decodeTitle(note.title);
|
|
|
|
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');
|
|
res.setHeader('X-Robots-Tag', 'noindex, nofollow'); // prevent crawling
|
|
stream.pipe(res);
|
|
fs.unlink(path);
|
|
});
|
|
}
|
|
|
|
function actionGist(req, res, note) {
|
|
var data = {
|
|
client_id: config.github.clientID,
|
|
redirect_uri: config.serverurl + '/auth/github/callback/' + LZString.compressToBase64(note.id) + '/gist',
|
|
scope: "gist",
|
|
state: shortId.generate()
|
|
};
|
|
var query = querystring.stringify(data);
|
|
res.redirect("https://github.com/login/oauth/authorize?" + query);
|
|
}
|
|
|
|
function noteActions(req, res, next) {
|
|
var noteId = req.params.noteId;
|
|
findNote(req, res, function (note) {
|
|
var action = req.params.action;
|
|
switch (action) {
|
|
case "publish":
|
|
case "pretty": //pretty deprecated
|
|
actionPublish(req, res, note);
|
|
break;
|
|
case "slide":
|
|
actionSlide(req, res, note);
|
|
break;
|
|
case "download":
|
|
actionDownload(req, res, note);
|
|
break;
|
|
case "pdf":
|
|
actionPDF(req, res, note);
|
|
break;
|
|
case "gist":
|
|
actionGist(req, res, note);
|
|
break;
|
|
default:
|
|
return res.redirect(config.serverurl + '/' + noteId);
|
|
break;
|
|
}
|
|
});
|
|
}
|
|
|
|
function publishNoteActions(req, res, next) {
|
|
findNote(req, res, function (note) {
|
|
var action = req.params.action;
|
|
switch (action) {
|
|
case "edit":
|
|
res.redirect(config.serverurl + '/' + (note.alias ? note.alias : LZString.compressToBase64(note.id)));
|
|
break;
|
|
default:
|
|
res.redirect(config.serverurl + '/s/' + note.shortid);
|
|
break;
|
|
}
|
|
});
|
|
}
|
|
|
|
function githubActions(req, res, next) {
|
|
var noteId = req.params.noteId;
|
|
findNote(req, res, function (note) {
|
|
var action = req.params.action;
|
|
switch (action) {
|
|
case "gist":
|
|
githubActionGist(req, res, note);
|
|
break;
|
|
default:
|
|
res.redirect(config.serverurl + '/' + noteId);
|
|
break;
|
|
}
|
|
});
|
|
}
|
|
|
|
function githubActionGist(req, res, note) {
|
|
var code = req.query.code;
|
|
var state = req.query.state;
|
|
if (!code || !state) {
|
|
return response.errorForbidden(res);
|
|
} else {
|
|
var data = {
|
|
client_id: config.github.clientID,
|
|
client_secret: config.github.clientSecret,
|
|
code: code,
|
|
state: state
|
|
}
|
|
var auth_url = 'https://github.com/login/oauth/access_token';
|
|
request({
|
|
url: auth_url,
|
|
method: "POST",
|
|
json: data
|
|
}, function (error, httpResponse, body) {
|
|
if (!error && httpResponse.statusCode == 200) {
|
|
var access_token = body.access_token;
|
|
if (access_token) {
|
|
var content = LZString.decompressFromBase64(note.content);
|
|
var title = models.Note.decodeTitle(note.title);
|
|
var filename = title.replace('/', ' ') + '.md';
|
|
var gist = {
|
|
"files": {}
|
|
};
|
|
gist.files[filename] = {
|
|
"content": content
|
|
};
|
|
var gist_url = "https://api.github.com/gists";
|
|
request({
|
|
url: gist_url,
|
|
headers: {
|
|
'User-Agent': 'HackMD',
|
|
'Authorization': 'token ' + access_token
|
|
},
|
|
method: "POST",
|
|
json: gist
|
|
}, function (error, httpResponse, body) {
|
|
if (!error && httpResponse.statusCode == 201) {
|
|
res.setHeader('referer', '');
|
|
res.redirect(body.html_url);
|
|
} else {
|
|
return response.errorForbidden(res);
|
|
}
|
|
});
|
|
} else {
|
|
return response.errorForbidden(res);
|
|
}
|
|
} else {
|
|
return response.errorForbidden(res);
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
function gitlabActions(req, res, next) {
|
|
var noteId = req.params.noteId;
|
|
findNote(req, res, function (note) {
|
|
var action = req.params.action;
|
|
switch (action) {
|
|
case "projects":
|
|
gitlabActionProjects(req, res, note);
|
|
break;
|
|
default:
|
|
res.redirect(config.serverurl + '/' + noteId);
|
|
break;
|
|
}
|
|
});
|
|
}
|
|
|
|
function gitlabActionProjects(req, res, note) {
|
|
if (req.isAuthenticated()) {
|
|
models.User.findOne({
|
|
where: {
|
|
id: req.user.id
|
|
}
|
|
}).then(function (user) {
|
|
if (!user)
|
|
return response.errorNotFound(res);
|
|
var ret = { baseURL: config.gitlab.baseURL };
|
|
ret.accesstoken = user.accessToken;
|
|
ret.profileid = user.profileid;
|
|
request(
|
|
config.gitlab.baseURL + '/api/v3/projects?access_token=' + user.accessToken,
|
|
function(error, httpResponse, body) {
|
|
if (!error && httpResponse.statusCode == 200) {
|
|
ret.projects = JSON.parse(body);
|
|
return res.send(ret);
|
|
} else {
|
|
return res.send(ret);
|
|
}
|
|
}
|
|
);
|
|
}).catch(function (err) {
|
|
logger.error('gitlab action projects failed: ' + err);
|
|
return response.errorInternalError(res);
|
|
});
|
|
} else {
|
|
return response.errorForbidden(res);
|
|
}
|
|
}
|
|
|
|
function showPublishSlide(req, res, next) {
|
|
findNote(req, res, function (note) {
|
|
note.increment('viewcount').then(function (note) {
|
|
if (!note) {
|
|
return response.errorNotFound(res);
|
|
}
|
|
var body = LZString.decompressFromBase64(note.content);
|
|
try {
|
|
body = metaMarked(body).markdown;
|
|
} catch(err) {
|
|
//na
|
|
}
|
|
var title = models.Note.decodeTitle(note.title);
|
|
title = models.Note.generateWebTitle(title);
|
|
var text = S(body).escapeHTML().s;
|
|
render(res, title, text);
|
|
}).catch(function (err) {
|
|
logger.error(err);
|
|
return response.errorInternalError(res);
|
|
});
|
|
});
|
|
}
|
|
|
|
//reveal.js render
|
|
var render = function (res, title, markdown) {
|
|
var slides = md.slidify(markdown, opts);
|
|
|
|
res.end(Mustache.to_html(opts.template, {
|
|
url: config.serverurl,
|
|
title: title,
|
|
theme: opts.theme,
|
|
highlightTheme: opts.highlightTheme,
|
|
slides: slides,
|
|
options: JSON.stringify(opts.revealOptions, null, 2)
|
|
}));
|
|
};
|
|
|
|
module.exports = response;
|