#1454 Add tasks in case template output

TheHive-Project · Jul 29, 2020 · 834c74b · 834c74b
1 parent 5176b1f
commit 834c74b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/dto/src/main/scala/org/thp/thehive/dto/v1/CaseTemplate.scala b/dto/src/main/scala/org/thp/thehive/dto/v1/CaseTemplate.scala
@@ -42,7 +42,8 @@ case class OutputCaseTemplate(
     tlp: Option[Int],
     pap: Option[Int],
     summary: Option[String],
-    customFields: Set[OutputCustomFieldValue] = Set.empty
+    customFields: Set[OutputCustomFieldValue] = Set.empty,
+    tasks: Seq[OutputTask]
 )
 
 object OutputCaseTemplate {

diff --git a/thehive/app/org/thp/thehive/controllers/v1/Conversion.scala b/thehive/app/org/thp/thehive/controllers/v1/Conversion.scala
@@ -145,6 +145,7 @@ object Conversion {
       .withFieldConst(_._type, "CaseTemplate")
       .withFieldComputed(_.customFields, _.customFields.map(_.toOutput).toSet)
       .withFieldComputed(_.tags, _.tags.map(_.toString).toSet)
+      .withFieldComputed(_.tasks, _.tasks.map(_.toOutput))
       .transform
   )