1 files changed, 15 insertions, 1 deletions
diff --git a/server/lib/middleware.js b/server/lib/middleware.js
index 35c520c..9d09a10 100644
--- a/server/lib/middleware.js
+++ b/server/lib/middleware.js
@@ -7,6 +7,7 @@ var passport = require('passport'),
 	User = require('./schemas/User'),
 	Collaborator = require('./schemas/Collaborator'),
 	Project = require('./schemas/Project'),
+	Layout = require('./schemas/Layout'),
 	Plan = require('./schemas/Plan');
 
 
@@ -63,7 +64,20 @@ var middleware = {
 		res.locals.opt = {}
 		next()
 	},
-  
+
+  ensureUserProjectsCount: function(req, res, next){
+    Project.count({ user_id: req.user.id }, function(err, count){
+      res.locals.projectCount = count || 0
+      next()
+    })
+  },
+  ensureUserLayoutsCount: function(req, res, next){
+    Layout.count({ user_id: req.user.id }, function(err, count){
+      res.locals.layoutCount = count || 0
+      next()
+    })
+  },
+
 	ensureProject: function (req, res, next) {
 		if (req.params.slug) {
 			Project.findOne({ slug: req.params.slug }, function(err, project){