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

1 files changed, 88 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/trial.scala b/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/trial.scala
new file mode 100644
index 0000000..e7b6d54
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/trial.scala
@@ -0,0 +1,88 @@
+package xyz.driver.pdsuidomain.formats.json.sprayformats
+
+import spray.json._
+import xyz.driver.core.json.EnumJsonFormat
+import xyz.driver.pdsuicommon.domain.{LongId, UuidId}
+import xyz.driver.pdsuidomain.entities._
+
+object trial {
+  import DefaultJsonProtocol._
+  import common._
+  import Trial._
+
+  implicit val trialStatusFormat = new EnumJsonFormat[Status](
+    "New"            -> Status.New,
+    "ReviewSummary"  -> Status.ReviewSummary,
+    "Summarized"     -> Status.Summarized,
+    "PendingUpdate"  -> Status.PendingUpdate,
+    "Update"         -> Status.Update,
+    "ReviewCriteria" -> Status.ReviewCriteria,
+    "Done"           -> Status.Done,
+    "Flagged"        -> Status.Flagged,
+    "Archived"       -> Status.Archived
+  )
+
+  implicit val conditionFormat = new EnumJsonFormat[Condition](
+    "Breast"   -> Condition.Breast,
+    "Lung"     -> Condition.Lung,
+    "Prostate" -> Condition.Prostate
+  )
+
+  implicit val trialWriter: JsonWriter[Trial] = new JsonWriter[Trial] {
+    override def write(obj: Trial) =
+      JsObject(
+        "id"                    -> obj.id.toJson,
+        "externalid"            -> obj.externalId.toJson,
+        "lastUpdate"            -> obj.lastUpdate.toJson,
+        "status"                -> obj.status.toJson,
+        "assignee"              -> obj.assignee.toJson,
+        "previousStatus"        -> obj.previousStatus.toJson,
+        "previousAssignee"      -> obj.previousAssignee.toJson,
+        "lastActiveUser"        -> obj.lastActiveUserId.toJson,
+        "condition"             -> obj.condition.toJson,
+        "phase"                 -> obj.phase.toJson,
+        "hypothesisId"          -> obj.hypothesisId.toJson,
+        "studyDesignId"         -> obj.studyDesignId.toJson,
+        "originalStudyDesignId" -> obj.originalStudyDesign.toJson,
+        "isPartner"             -> obj.isPartner.toJson,
+        "overview"              -> obj.overview.toJson,
+        "overviewTemplate"      -> obj.overviewTemplate.toJson,
+        "isUpdated"             -> obj.isUpdated.toJson,
+        "title"                 -> obj.title.toJson,
+        "originalTitle"         -> obj.originalTitle.toJson
+      )
+  }
+
+  def applyUpdateToTrial(json: JsValue, orig: Trial): Trial = json match {
+    case JsObject(fields) =>
+      val hypothesisId = fields
+        .get("hypothesisId")
+        .map(_.convertTo[UuidId[Hypothesis]])
+        .orElse(orig.hypothesisId)
+
+      val studyDesignId = fields
+        .get("studyDesignId")
+        .map(_.convertTo[LongId[StudyDesign]])
+        .orElse(orig.studyDesignId)
+
+      val overview = fields
+        .get("overview")
+        .map(_.convertTo[String])
+        .orElse(orig.overview)
+
+      val title = fields
+        .get("title")
+        .map(_.convertTo[String])
+        .getOrElse(orig.title)
+
+      orig.copy(
+        hypothesisId = hypothesisId,
+        studyDesignId = studyDesignId,
+        overview = overview,
+        title = title
+      )
+
+    case _ => deserializationError(s"Expected Json Object as Trial, but got $json")
+  }
+
+}