From a552e75456af2d790d5f709ad3bbe70e69173f01 Mon Sep 17 00:00:00 2001
From: Marvin Bertin <marvin.bertin@gmail.com>
Date: Mon, 2 Oct 2017 13:21:08 -0700
Subject: update ApiEligibilityArm with diseases

---
 .../json/eligibilityarm/ApiEligibilityArm.scala    | 47 ++++++++++++++--------
 1 file changed, 30 insertions(+), 17 deletions(-)

(limited to 'src/main/scala/xyz/driver/pdsuidomain/formats')

diff --git a/src/main/scala/xyz/driver/pdsuidomain/formats/json/eligibilityarm/ApiEligibilityArm.scala b/src/main/scala/xyz/driver/pdsuidomain/formats/json/eligibilityarm/ApiEligibilityArm.scala
index e7eb022..4d6d0a2 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/formats/json/eligibilityarm/ApiEligibilityArm.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/formats/json/eligibilityarm/ApiEligibilityArm.scala
@@ -1,20 +1,27 @@
 package xyz.driver.pdsuidomain.formats.json.eligibilityarm
 
-import xyz.driver.pdsuicommon.domain.{LongId, StringId}
-import xyz.driver.pdsuidomain.entities.EligibilityArm
 import play.api.libs.functional.syntax._
 import play.api.libs.json._
+import xyz.driver.pdsuicommon.domain.{LongId, StringId}
+import xyz.driver.pdsuidomain.entities.{EligibilityArm, EligibilityArmDisease, EligibilityArmWithDiseases, Trial}
 
-final case class ApiEligibilityArm(id: Long, name: String, originalName: String, trialId: String) {
+final case class ApiEligibilityArm(id: Long, name: String, originalName: String, trialId: String, diseases: Seq[String]) {
 
-  def toDomain: EligibilityArm = EligibilityArm(
-    id = LongId(this.id),
-    name = this.name,
-    originalName = this.originalName,
-    trialId = StringId(this.trialId),
-    deleted = None // if we have an ApiEligibilityArm object, the EligibilityArm itself has not been deleted
-  )
+  def toDomain: EligibilityArmWithDiseases = {
+    val eligibilityArm = EligibilityArm(
+      id = LongId(this.id),
+      name = this.name,
+      originalName = this.originalName,
+      trialId = StringId(this.trialId),
+      deleted = None // if we have an ApiEligibilityArm object, the EligibilityArm itself has not been deleted
+    )
 
+    EligibilityArmWithDiseases(eligibilityArm, this.diseases.map { disease =>
+      val condition = Trial.Condition.fromString(disease)
+        .getOrElse(throw new NoSuchElementException(s"unknown condition $disease"))
+      EligibilityArmDisease(eligibilityArm.id, condition)
+    })
+  }
 }
 
 object ApiEligibilityArm {
@@ -23,13 +30,19 @@ object ApiEligibilityArm {
     (JsPath \ "id").format[Long] and
       (JsPath \ "name").format[String] and
       (JsPath \ "originalName").format[String] and
-      (JsPath \ "trialId").format[String]
+      (JsPath \ "trialId").format[String] and
+      (JsPath \ "diseases").format[Seq[String]]
   )(ApiEligibilityArm.apply, unlift(ApiEligibilityArm.unapply))
 
-  def fromDomain(arm: EligibilityArm): ApiEligibilityArm = ApiEligibilityArm(
-    id = arm.id.id,
-    name = arm.name,
-    originalName = arm.originalName,
-    trialId = arm.trialId.id
-  )
+  def fromDomain(eligibilityArmWithDiseases: EligibilityArmWithDiseases): ApiEligibilityArm = {
+    import eligibilityArmWithDiseases.{eligibilityArm, eligibilityArmDiseases}
+
+    ApiEligibilityArm(
+      id = eligibilityArm.id.id,
+      name = eligibilityArm.name,
+      originalName = eligibilityArm.originalName,
+      trialId = eligibilityArm.trialId.id,
+      diseases = eligibilityArmDiseases.map(_.disease.toString)
+    )
+  }
 }
-- 
cgit v1.2.3