1 files changed, 11 insertions, 0 deletions
diff --git a/lib/bucky.js b/lib/bucky.js
index c003103..bfc4b6f 100644
--- a/lib/bucky.js
+++ b/lib/bucky.js
@@ -175,4 +175,15 @@ var bucky = module.exports = {
       next()
     })
   },
+  ensureMessage: function(req, res, next){
+    db.getMessage(req.params.id).then(function(message){
+      var username = req.user.get('username')
+      if (username !== message.get('recipient') && username !== message.get('sender')) {
+        res.sendStatus(404)
+        return
+      }
+      res.message = message
+      next()
+    })
+  }
 }
 \ No newline at end of file