From 9aa923e90ee9cde5def3c8aec7c58b9a52d4b50e Mon Sep 17 00:00:00 2001
From: Precious Jahlom Agboado <japrag@outlook.com>
Date: Tue, 4 Mar 2014 10:37:19 +0000
Subject: [PATCH] Added a redirection to referrer page

---
 controllers/user.js | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/controllers/user.js b/controllers/user.js
index 97bbb5ec30..c707209130 100644
--- a/controllers/user.js
+++ b/controllers/user.js
@@ -9,6 +9,15 @@ var User = require('../models/User');
 
 exports.getLogin = function(req, res) {
   if (req.user) return res.redirect('/');
+  
+  // If the host in the header of the User Agent making the request is
+  // the same as the server host, keep a copy of the referrer url and 
+  // redirect to that later
+  if(req.header('host') == 'localhost:3000'){
+    req.session.referrer = req.header('Referrer');
+    console.log(req.session.referrer);
+  }
+  
   res.render('account/login', {
     title: 'Login'
   });
@@ -43,7 +52,9 @@ exports.postLogin = function(req, res, next) {
     req.logIn(user, function(err) {
       if (err) return next(err);
       req.flash('success', { msg: 'Success! You are logged in.' });
-      return res.redirect('/');
+      // redirecting toa referrer page else to home page
+      if(req.session.referrer) return res.redirect(req.session.referrer);
+      return res.redirect('/')
     });
   })(req, res, next);
 };