#1733 Add alert import date in extraData

TheHive-Project · Jan 7, 2021 · 20e7add · 20e7add
1 parent abfb9c0
commit 20e7add
Showing 1 changed file with 15 additions and 8 deletions.
diff --git a/thehive/app/org/thp/thehive/controllers/v1/AlertRenderer.scala b/thehive/app/org/thp/thehive/controllers/v1/AlertRenderer.scala
@@ -1,7 +1,6 @@
 package org.thp.thehive.controllers.v1
 
-import java.util.{List => JList, Map => JMap}
-
+import java.util.{Date, List => JList, Map => JMap}
 import org.thp.scalligraph.auth.AuthContext
 import org.thp.scalligraph.traversal.TraversalOps._
 import org.thp.scalligraph.traversal.{Converter, Traversal}
@@ -39,12 +38,20 @@ trait AlertRenderer extends BaseRenderer[Alert] {
     _.similarCases(None).fold.domainMap(sc => JsArray(sc.sorted.map(Json.toJson(_))))
   }
 
-  def alertStatsRenderer(extraData: Set[String])(
-    implicit authContext: AuthContext
+  def importDate: Traversal.V[Alert] => Traversal[JsValue, JList[Date], Converter[JsValue, JList[Date]]] =
+    _.importDate.fold.domainMap(_.headOption.fold[JsValue](JsNull)(d => JsNumber(d.getTime)))
+
+  def alertStatsRenderer(extraData: Set[String])(implicit
+      authContext: AuthContext
   ): Traversal.V[Alert] => JsTraversal = { implicit traversal =>
-    baseRenderer(extraData, traversal, {
-      case (f, "similarCases") => addData("similarCases", f)(similarCasesStats)
-      case (f, _)              => f
-    })
+    baseRenderer(
+      extraData,
+      traversal,
+      {
+        case (f, "similarCases") => addData("similarCases", f)(similarCasesStats)
+        case (f, "importDate")   => addData("importDate", f)(importDate)
+        case (f, _)              => f
+      }
+    )
   }
 }