#1946 Use indexQuery for tag usage and observable count in case extra…

…Data
TheHive-Project · Apr 11, 2021 · 3df647a · 3df647a
1 parent 9fe06fb
commit 3df647a
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 20 deletions.
diff --git a/thehive/app/org/thp/thehive/controllers/v1/CaseCtrl.scala b/thehive/app/org/thp/thehive/controllers/v1/CaseCtrl.scala
@@ -65,7 +65,7 @@ class CaseCtrl @Inject() (
     Query.init[Long](
       "countCase",
       (graph, authContext) =>
-        graph.indexCountQuery(s"""v."_label":Case AND v.organisationIds:${organisationSrv.currentId(graph, authContext).value} """)
+        graph.indexCountQuery(s"""v."_label":Case AND v.organisationIds:${organisationSrv.currentId(graph, authContext).value}""")
     ),
     Query[Traversal.V[Case], Traversal.V[Observable]](
       "observables",

diff --git a/thehive/app/org/thp/thehive/controllers/v1/CaseRenderer.scala b/thehive/app/org/thp/thehive/controllers/v1/CaseRenderer.scala
@@ -17,11 +17,15 @@ import java.util.{Collection => JCollection, List => JList, Map => JMap}
 trait CaseRenderer extends BaseRenderer[Case] {
   val limitedCountThreshold: Long
 
-  def observableStats(implicit authContext: AuthContext): Traversal.V[Case] => Traversal[JsValue, JLong, Converter[JsValue, JLong]] =
-    _.share
-      .observables
-      .limitedCount(limitedCountThreshold)
-      .domainMap(count => Json.obj("total" -> count))
+  def observableStats(implicit authContext: AuthContext): Traversal.V[Case] => Traversal[JsObject, AnyRef, Converter[JsObject, AnyRef]] =
+    t =>
+      t._id.domainMap { caseId =>
+        Json.obj("total" -> t.graph.indexCountQuery(s"""v."_label":Case AND v.relatedId:${caseId.value}"""))
+      }
+//    _.share
+//      .observables
+//      .limitedCount(limitedCountThreshold)
+//      .domainMap(count => Json.obj("total" -> count))
 
   def taskStats(implicit
       authContext: AuthContext

diff --git a/thehive/app/org/thp/thehive/controllers/v1/TagRenderer.scala b/thehive/app/org/thp/thehive/controllers/v1/TagRenderer.scala
@@ -15,20 +15,36 @@ trait TagRenderer extends BaseRenderer[Tag] {
   def usageStats(implicit
       authContext: AuthContext
   ): Traversal.V[Tag] => Traversal[JsObject, JMap[String, Any], Converter[JsObject, JMap[String, Any]]] =
-    _.project(
-      _.by(_.`case`.limitedCount(limitedCountThreshold))
-        .by(_.alert.limitedCount(limitedCountThreshold))
-        .by(_.observable.limitedCount(limitedCountThreshold))
-        .by(_.caseTemplate.limitedCount(limitedCountThreshold))
-    ).domainMap {
-      case (caseCount, alertCount, observableCount, caseTemplateCount) =>
-        Json.obj(
-          "case"         -> caseCount,
-          "alert"        -> alertCount,
-          "observable"   -> observableCount,
-          "caseTemplate" -> caseTemplateCount
-        )
-    }
+    t =>
+      t.project(
+        _.by(_.value(_.predicate))
+          .by(_.caseTemplate.limitedCount(limitedCountThreshold))
+      ).domainMap {
+        case (tag, caseTemplateCount) =>
+          Json.obj(
+            "case"         -> t.graph.indexCountQuery(s"""v."_label":Case AND v.tags:"${tag.replaceAllLiterally("\"", "\\\"")}""""),
+            "alert"        -> t.graph.indexCountQuery(s"""v."_label":Alert AND v.tags:"${tag.replaceAllLiterally("\"", "\\\"")}""""),
+            "observable"   -> t.graph.indexCountQuery(s"""v."_label":Observable AND v.tags:"${tag.replaceAllLiterally("\"", "\\\"")}""""),
+            "caseTemplate" -> caseTemplateCount
+          )
+      }
+//  def usageStats(implicit
+//      authContext: AuthContext
+//  ): Traversal.V[Tag] => Traversal[JsObject, JMap[String, Any], Converter[JsObject, JMap[String, Any]]] =
+//    _.project(
+//      _.by(_.`case`.limitedCount(limitedCountThreshold))
+//        .by(_.alert.limitedCount(limitedCountThreshold))
+//        .by(_.observable.limitedCount(limitedCountThreshold))
+//        .by(_.caseTemplate.limitedCount(limitedCountThreshold))
+//    ).domainMap {
+//      case (caseCount, alertCount, observableCount, caseTemplateCount) =>
+//        Json.obj(
+//          "case"         -> caseCount,
+//          "alert"        -> alertCount,
+//          "observable"   -> observableCount,
+//          "caseTemplate" -> caseTemplateCount
+//        )
+//    }
 
   def tagStatsRenderer(extraData: Set[String])(implicit
       authContext: AuthContext