Browse Source

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

TonyKang 8 năm trước cách đây
mục cha
commit
f7e019ba18
1 tập tin đã thay đổi với 5 bổ sung1 xóa
  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();
 });