From 948d5762cd2c73179a77835bd22bb7fae76458d6 Mon Sep 17 00:00:00 2001
From: Matteo Pagliazzi <matteopagliazzi@gmail.com>
Date: Fri, 12 Dec 2014 11:39:22 +0100
Subject: [PATCH] fix(flag-message): small refactor

---
 src/controllers/groups.js | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/controllers/groups.js b/src/controllers/groups.js
index caa36809cc..06f87e82a5 100644
--- a/src/controllers/groups.js
+++ b/src/controllers/groups.js
@@ -292,14 +292,18 @@ api.clearFlagCount = function(req, res, next){
 
   if(!message) return res.json(404, {err: "Message not found!"});
 
-  if(user.contributor.admin)
+  if(user.contributor.admin){
     message.flagCount = 0;
+
+    group.markModified('chat');
+    group.save(function(err,_saved){
+      if(err) return next(err);
+      return res.send(204);
+    });
+  }else{
+    return res.json(401, {err: "Only an admin can clear the flag count!"})
+  }
   
-  group.markModified('chat');
-  group.save(function(err,_saved){
-    if(err) return next(err);
-    return res.send(204);
-  });
 }
 
 api.seenMessage = function(req,res,next){