Merge branch 'master' of github.com:okfocus/vvalls

1 files changed, 11 insertions, 0 deletions

diff --git a/server/lib/middleware.js b/server/lib/middleware.js
index 7b37658..4848ab0 100644
--- a/server/lib/middleware.js
+++ b/server/lib/middleware.js
@@ -33,6 +33,17 @@ var middleware = {
 		}
 		next();
 	},
+	
+	ensureIP: function (req, res, next) {
+    if (req.session.ip) {
+      req.ip = req.session.ip || "127.0.0.1"
+    }
+    else if (req.ips) {
+      req.ip = req.session.ip = req.ips[0]
+      req.session.save()
+    }
+    next()
+	},
 
 	ensureLocals: function (req, res, next) {
 		res.locals.token = req.csrfToken();