#1414 Add tasks in case creation API

TheHive-Project · Jun 28, 2020 · 5d335ae · 5d335ae
1 parent b98ef5f
commit 5d335ae
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/dto/src/main/scala/org/thp/thehive/dto/v1/Task.scala b/dto/src/main/scala/org/thp/thehive/dto/v1/Task.scala
@@ -5,12 +5,11 @@ import java.util.Date
 import play.api.libs.json.{Json, OFormat, OWrites}
 
 case class InputTask(
-    caseId: String,
     title: String,
     group: Option[String] = None,
     description: Option[String] = None,
     status: Option[String] = None,
-    flag: Boolean = false,
+    flag: Option[Boolean] = None,
     startDate: Option[Date] = None,
     endDate: Option[Date] = None,
     order: Option[Int] = None,

diff --git a/thehive/app/org/thp/thehive/controllers/v1/Conversion.scala b/thehive/app/org/thp/thehive/controllers/v1/Conversion.scala
@@ -168,6 +168,7 @@ object Conversion {
         .withFieldComputed(_.status, _.status.fold(TaskStatus.Waiting)(TaskStatus.withName))
         .withFieldComputed(_.order, _.order.getOrElse(0))
         .withFieldComputed(_.group, _.group.getOrElse("default"))
+        .withFieldComputed(_.flag, _.flag.getOrElse(false))
         .transform
   }
 

diff --git a/thehive/app/org/thp/thehive/controllers/v1/TaskCtrl.scala b/thehive/app/org/thp/thehive/controllers/v1/TaskCtrl.scala
@@ -50,10 +50,12 @@ class TaskCtrl @Inject() (
   def create: Action[AnyContent] =
     entrypoint("create task")
       .extract("task", FieldsParser[InputTask])
+      .extract("caseId", FieldsParser[String])
       .authTransaction(db) { implicit request => implicit graph =>
         val inputTask: InputTask = request.body("task")
+        val caseId: String       = request.body("caseId")
         for {
-          case0        <- caseSrv.getOrFail(inputTask.caseId)
+          case0        <- caseSrv.getOrFail(caseId)
           createdTask  <- taskSrv.create(inputTask.toTask, None)
           organisation <- organisationSrv.getOrFail(request.organisation)
           _            <- shareSrv.shareTask(createdTask, case0, organisation)