#10 Fix unaudited filter on auxSrv when entity is root entity (withou…

…t parent)
TheHive-Project · Feb 22, 2017 · 5c5b702 · 5c5b702
1 parent 3ad86fc
commit 5c5b702
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/app/org/elastic4play/services/AuxSrv.scala b/app/org/elastic4play/services/AuxSrv.scala
@@ -50,15 +50,15 @@ class AuxSrv @Inject() (
             log.warn(s"Child entity (${childModel.name} ${entity.id}) has no parent !")
             JsObject(Nil)
           })
-      case _ ⇒ Future.successful(Json.toJson(entity).as[JsObject])
+      case _ if removeUnaudited ⇒ Future.successful(removeUnauditedAttributes(entity))
+      case _                    ⇒ Future.successful(Json.toJson(entity).as[JsObject])
     }
     if (withStats) {
       for {
         e ← entityWithParent
         s ← entity.model.getStats(entity)
       } yield e + ("stats" → s)
-    }
-    else entityWithParent
+    } else entityWithParent
   }
 
   def apply[A](entities: Source[BaseEntity, A], nparent: Int, withStats: Boolean, removeUnaudited: Boolean): Source[JsObject, A] = {