From a6d54686a153806c75f51801d2087c07a29e7e0e Mon Sep 17 00:00:00 2001 From: Robin Riclet Date: Tue, 16 Feb 2021 11:17:12 +0100 Subject: [PATCH] #1766 Added patternParent to procedure --- .../thp/thehive/controllers/v1/ProcedureRenderer.scala | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/thehive/app/org/thp/thehive/controllers/v1/ProcedureRenderer.scala b/thehive/app/org/thp/thehive/controllers/v1/ProcedureRenderer.scala index 943488c375..41758ac6f9 100644 --- a/thehive/app/org/thp/thehive/controllers/v1/ProcedureRenderer.scala +++ b/thehive/app/org/thp/thehive/controllers/v1/ProcedureRenderer.scala @@ -13,13 +13,17 @@ trait ProcedureRenderer extends BaseRenderer[Procedure] { def patternStats: Traversal.V[Procedure] => Traversal[JsValue, JMap[String, Any], Converter[JsValue, JMap[String, Any]]] = _.pattern.richPattern.domainMap(_.toJson) + def patternParentStats: Traversal.V[Procedure] => Traversal[JsValue, JMap[String, Any], Converter[JsValue, JMap[String, Any]]] = + _.pattern.parent.richPattern.domainMap(_.toJson) + def procedureStatsRenderer(extraData: Set[String]): Traversal.V[Procedure] => JsTraversal = { implicit traversal => baseRenderer( extraData, traversal, { - case (f, "pattern") => addData("pattern", f)(patternStats) - case (f, _) => f + case (f, "pattern") => addData("pattern", f)(patternStats) + case (f, "patternParent") => addData("patternParent", f)(patternParentStats) + case (f, _) => f } ) }