Browse Source

Merge branch 'master' of http://smartcost.f3322.net:3000/SmartCost/ConstructionCost

TonyKang 8 years ago
parent
commit
f7e019ba18
1 changed files with 5 additions and 1 deletions
  1. 5 1
      server.js

+ 5 - 1
server.js

@@ -39,7 +39,11 @@ app.use(session({
 app.use(function (req, res, next) {
     var url = req.originalUrl;
     if (!/^\/login/.test(req.originalUrl) && !req.session.userAccount) {
-        return res.redirect("/login" + '?referer=' + req.originalUrl);
+        if (req.headers.referer) {
+            return res.redirect('/login' + '?referer=' + req.headers.referer);
+        } else {
+            return res.redirect("/login" + '?referer=' + req.originalUrl);
+        }
     }
     next();
 });