PDSUI-2188 Created and fixed spray json formats for tric and rep

1 files changed, 60 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/intervention.scala b/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/intervention.scala
new file mode 100644
index 0000000..a8ce950
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/intervention.scala
@@ -0,0 +1,60 @@
+package xyz.driver.pdsuidomain.formats.json.sprayformats
+
+import spray.json._
+import xyz.driver.pdsuicommon.domain.LongId
+import xyz.driver.pdsuidomain.entities._
+
+object intervention {
+  import DefaultJsonProtocol._
+  import common._
+
+  implicit val interventionWriter: JsonWriter[InterventionWithArms] = new JsonWriter[InterventionWithArms] {
+    override def write(obj: InterventionWithArms) =
+      JsObject(
+        "id"                  -> obj.intervention.id.toJson,
+        "name"                -> obj.intervention.name.toJson,
+        "typeId"              -> obj.intervention.typeId.toJson,
+        "description"         -> obj.intervention.description.toJson,
+        "isActive"            -> obj.intervention.isActive.toJson,
+        "arms"                -> obj.arms.map(_.armId).toJson,
+        "trialId"             -> obj.intervention.trialId.toJson,
+        "originalName"        -> obj.intervention.originalName.toJson,
+        "originalDescription" -> obj.intervention.originalDescription.toJson,
+        "originalType"        -> obj.intervention.originalType.toJson
+      )
+  }
+
+  def applyUpdateToInterventionWithArms(json: JsValue, orig: InterventionWithArms): InterventionWithArms = json match {
+    case JsObject(fields) =>
+      val typeId = fields
+        .get("typeId")
+        .map(_.convertTo[LongId[InterventionType]])
+
+      val description = fields
+        .get("description")
+        .map(_.convertTo[String])
+
+      val isActive = fields
+        .get("isActive")
+        .map(_.convertTo[Boolean])
+
+      val origIntervention = orig.intervention
+      val arms = fields
+        .get("arms")
+        .map(_.convertTo[List[LongId[Arm]]].map(x => InterventionArm(x, orig.intervention.id)))
+
+      orig.copy(
+        intervention = origIntervention.copy(
+          typeId = typeId.orElse(origIntervention.typeId),
+          description = description.getOrElse(origIntervention.description),
+          isActive = isActive.getOrElse(origIntervention.isActive)
+        ),
+        arms = arms.getOrElse(orig.arms)
+      )
+
+    case _ => deserializationError(s"Expected Json Object as partial Intervention, but got $json")
+  }
+
+  implicit val interventionTypeFormat: RootJsonFormat[InterventionType] = jsonFormat2(InterventionType.apply)
+
+}