Added redirectIfNotLogged decorator.

Fixes #5

controllers/auth/views.js

@@ -5,6 +5,19 @@ const mail = require('mail');
 const pug = require('pug');
 const config = require('settings/config');
 
+module.exports.redirectIfNotLogged = function(redirectionPage, toDecorate) {
+    return function(req, res, render) {
+        if (req.session.user === undefined) {
+            req.session.redirectAfterLog = redirectionPage;
+            req.session.save();
+            res.redirect(getUrl('login'));
+            return;
+        }
+
+        return toDecorate(...arguments);
+    };
+}
+
 module.exports.login = function(req, res, render) {
     render('login.pug');
 }
@@ -26,8 +39,12 @@ module.exports.loginTarget = function(req, res, render) {
         } else {
             if (testPassword(req.body.password, user.password)) {
                 req.session.user = user;
+
+                let redirection = req.session.redirectAfterLog || 'index';
+                req.session.redirectAfterLog = undefined;
                 req.session.save();
-                res.redirect(getUrl('index'));
+
+                res.redirect(getUrl(redirection));
             } else {
                 res.redirect(getUrl('login'));
             }

controllers/settings/views.js

@@ -1,9 +1,11 @@
 const getUrl = require('create-url').getUrl;
 const User = require('auth/models');
 
-module.exports.settings = function(req, res, render) {
+const redirectIfNotLogged = require('auth/views').redirectIfNotLogged;
+
+module.exports.settings = redirectIfNotLogged('settings', function(req, res, render) {
     render('settings.pug');
-}
+});
 
 module.exports.settingsTarget = function(req, res, render) {
     let projectId = parseInt(req.body.projectId, 10);

controllers/total/views.js

@@ -1,4 +1,5 @@
 const cal = require('calendar');
+const redirectIfNotLogged = require('auth/views').redirectIfNotLogged;
 
 function computeUserTable(user, callback) {
 
@@ -33,9 +34,9 @@ function computeUserTable(user, callback) {
     });
 }
 
-module.exports.total = function(req, res, render) {
+module.exports.total = redirectIfNotLogged('total', function(req, res, render) {
     render('total.pug');
-}
+});
 
 module.exports.totalTable = function(req, res, render) {
     computeUserTable(req.session.user, (result) => {