Forráskód Böngészése

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

TonyKang 8 éve
szülő
commit
f7e019ba18
1 módosított fájl, 5 hozzáadás és 1 törlés
  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();
 });