Browse Source

minor refactorings

master
Christian Müller 9 years ago
parent
commit
dbcde233c6
  1. 74
      server.js
  2. 0
      storage.js
  3. 6
      view.js

74
server.js

@ -1,6 +1,6 @@
var express = require('express'); var express = require('express');
var view = require('./src/view'); var view = require('./view');
var storage = require('./src/storage'); var storage = require('./storage');
var md5 = require('md5'); var md5 = require('md5');
var LRU = require("lru-cache") var LRU = require("lru-cache")
var bodyParser = require('body-parser'); var bodyParser = require('body-parser');
@ -40,16 +40,14 @@ var log = function() {
console.log.apply(console, message); console.log.apply(console, message);
} }
app.get('/TOS', function(req, res) { app.get('/TOS', (req, res) => res.send(view.renderTOS()));
res.send(view.renderTOS());
});
app.get('/new', function(req, res) { app.get('/new', (req, res) => {
log(req.ip, "opens /new"); log(req.ip, "opens /new");
res.send(view.newNotePage(getTimeStamp() + md5(Math.random()))); res.send(view.newNotePage(getTimeStamp() + md5(Math.random())));
}); });
app.post('/note', function(req, res) { app.post('/note', (req, res) => {
var body = req.body, var body = req.body,
session = body.session, session = body.session,
note = body.note, note = body.note,
@ -62,8 +60,7 @@ app.post('/note', function(req, res) {
return sendResponse(res, 400, "Bad request"); return sendResponse(res, 400, "Bad request");
if (session.indexOf(getTimeStamp()) != 0) if (session.indexOf(getTimeStamp()) != 0)
return sendResponse(res, 400, "Session expired"); return sendResponse(res, 400, "Session expired");
var expectedSignature = md5(session + note.replace(/[\n\r]/g, "")); if (body.signature != md5(session + note.replace(/[\n\r]/g, "")))
if (expectedSignature != body.signature)
return sendResponse(res, 400, "Signature mismatch"); return sendResponse(res, 400, "Signature mismatch");
if (action == "POST") if (action == "POST")
storage.addNote(note, password).then(goToNote); storage.addNote(note, password).then(goToNote);
@ -82,7 +79,7 @@ app.post('/note', function(req, res) {
} }
}); });
app.get("/:year/:month/:day/:title", function(req, res) { app.get("/:year/:month/:day/:title", (req, res) => {
var P = req.params, url = P.year + "/" + P.month + "/" + P.day + "/" + P.title; var P = req.params, url = P.year + "/" + P.month + "/" + P.day + "/" + P.title;
log(req.ip, "resolves deprecated id", url); log(req.ip, "resolves deprecated id", url);
if (CACHE.has(url)) { if (CACHE.has(url)) {
@ -102,52 +99,52 @@ app.get("/:year/:month/:day/:title", function(req, res) {
}); });
}); });
app.get(/\/([a-z0-9]+\/edit)/, function(req, res) { app.get(/\/([a-z0-9]+)\/edit/, (req, res) => {
var link = req.params["0"].replace("/edit", ""); var id = req.params["0"];
log(req.ip, "calls /edit on", link); log(req.ip, "calls /edit on", id);
storage.getNote(link).then(note => res.send(note storage.getNote(id).then(note => res.send(note
? view.editNotePage(getTimeStamp() + md5(Math.random()), note) ? view.editNotePage(getTimeStamp() + md5(Math.random()), note)
: notFound(res))); : notFound(res)));
}); });
app.get(/\/([a-z0-9]+\/export)/, function(req, res) { app.get(/\/([a-z0-9]+)\/export/, (req, res) => {
var link = req.params["0"].replace("/export", ""); var id = req.params["0"];
log(req.ip, "calls /export on", link); log(req.ip, "calls /export on", id);
res.set({ 'Content-Type': 'text/plain', 'Charset': 'utf-8' }); res.set({ 'Content-Type': 'text/plain', 'Charset': 'utf-8' });
storage.getNote(link).then(note => note storage.getNote(id).then(note => note
? res.send(note.text) ? res.send(note.text)
: notFound(res)); : notFound(res));
}); });
app.get(/\/([a-z0-9]+\/stats)/, function(req, res) { app.get(/\/([a-z0-9]+)\/stats/, (req, res) => {
var link = req.params["0"].replace("/stats", ""); var id = req.params["0"];
log(req.ip, "calls /stats on", link); log(req.ip, "calls /stats on", id);
var promise = link in MODELS var promise = id in MODELS
? new Promise(resolve => resolve(MODELS[link])) ? new Promise(resolve => resolve(MODELS[id]))
: storage.getNote(link); : storage.getNote(id);
promise.then(note => note promise.then(note => note
? res.send(view.renderStats(note)) ? res.send(view.renderStats(note))
: notFound(res)); : notFound(res));
}); });
app.get(/\/([a-z0-9]+)/, function(req, res) { app.get(/\/([a-z0-9]+)/, (req, res) => {
var link = req.params["0"]; var id = req.params["0"];
log(req.ip, "open note", link, "from", req.get("Referer")); log(req.ip, "open note", id, "from", req.get("Referer"));
if (CACHE.has(link)) { if (CACHE.has(id)) {
log(link, "is cached!"); log(id, "is cached!");
var note = MODELS[link]; var note = MODELS[id];
if (!note) return notFound(res); if (!note) return notFound(res);
note.views++; note.views++;
res.send(CACHE.get(link)); res.send(CACHE.get(id));
} else storage.getNote(link).then(note => { } else storage.getNote(id).then(note => {
log(link, "is not cached, resolving..."); log(id, "is not cached, resolving...");
if (!note) { if (!note) {
CACHE.set(link, null); CACHE.set(id, null);
return notFound(res); return notFound(res);
} }
var content = view.renderNote(note, blackList); var content = view.renderNote(note, blackList);
CACHE.set(link, content); CACHE.set(id, content);
MODELS[link] = note; MODELS[id] = note;
note.views++; note.views++;
res.send(content); res.send(content);
}); });
@ -160,9 +157,8 @@ var sendResponse = (res, code, message) => {
var notFound = res => sendResponse(res, 404, "Not found"); var notFound = res => sendResponse(res, 404, "Not found");
var server = app.listen(process.env.PORT || 3000, function() { var server = app.listen(process.env.PORT || 3000,
log('NoteHub server listening on port', server.address().port); () => log('NoteHub server listening on port', server.address().port));
});
setInterval(() => { setInterval(() => {
var keys = Object.keys(MODELS); var keys = Object.keys(MODELS);

0
src/storage.js → storage.js

6
src/view.js → view.js

@ -7,14 +7,13 @@ var pageTemplate = fs.readFileSync("resources/template.html", "utf-8");
var footerTemplate = fs.readFileSync("resources/footer.html", "utf-8"); var footerTemplate = fs.readFileSync("resources/footer.html", "utf-8");
var editTemplate = fs.readFileSync("resources/edit.html", "utf-8"); var editTemplate = fs.readFileSync("resources/edit.html", "utf-8");
var header = fs.readFileSync(process.env.HEADER || "/dev/null", "utf-8"); var header = fs.readFileSync(process.env.HEADER || "/dev/null", "utf-8");
var defaultBL = new Set();
var deriveTitle = text => text var deriveTitle = text => text
.split(/[\n\r]/)[0].slice(0,25) .split(/[\n\r]/)[0].slice(0,25)
.replace(/[^a-zA-Z0-9\s]/g, ""); .replace(/[^a-zA-Z0-9\s]/g, "");
var renderPage = (id, title, content, footer, blackList) => pageTemplate var renderPage = (id, title, content, footer, blackList) => pageTemplate
.replace("%HEADER%", (blackList || defaultBL).has(id) ? header : "") .replace("%HEADER%", blackList && blackList.has(id) ? header : "")
.replace("%TITLE%", title) .replace("%TITLE%", title)
.replace("%CONTENT%", content.replace(/<meta.*?>/gi, "").replace(/<script[\s\S.]*?\/script>/gi, "")) .replace("%CONTENT%", content.replace(/<meta.*?>/gi, "").replace(/<script[\s\S.]*?\/script>/gi, ""))
.replace("%FOOTER%", footer || ""); .replace("%FOOTER%", footer || "");
@ -34,8 +33,7 @@ module.exports.renderStats = note => renderPage(note.id, deriveTitle(note.text),
<tr><td>Views</td><td>${note.views}</td></tr> <tr><td>Views</td><td>${note.views}</td></tr>
</table>`); </table>`);
module.exports.renderTOS = () => module.exports.renderTOS = () => renderPage("tos", "Terms of Service", marked(TOS));
renderPage("tos", "Terms of Service", marked(TOS));
module.exports.renderNote = (note, blackList) => renderPage(note.id, module.exports.renderNote = (note, blackList) => renderPage(note.id,
deriveTitle(note.text), deriveTitle(note.text),
Loading…
Cancel
Save