update ApiCreateEligibilityArm

1 files changed, 16 insertions, 8 deletions

diff --git a/src/main/scala/xyz/driver/pdsuidomain/formats/json/eligibilityarm/ApiEligibilityCreateArm.scala b/src/main/scala/xyz/driver/pdsuidomain/formats/json/eligibilityarm/ApiEligibilityCreateArm.scala
index 6f55c10..6cf71a2 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/formats/json/eligibilityarm/ApiEligibilityCreateArm.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/formats/json/eligibilityarm/ApiEligibilityCreateArm.scala
@@ -1,17 +1,25 @@
 package xyz.driver.pdsuidomain.formats.json.eligibilityarm
 
 import xyz.driver.pdsuicommon.domain.{LongId, StringId}
-import xyz.driver.pdsuidomain.entities.EligibilityArm
+import xyz.driver.pdsuidomain.entities.{EligibilityArm, EligibilityArmDisease, EligibilityArmWithDiseases, Trial}
 import play.api.libs.json.{Format, Json}
 
-final case class ApiCreateEligibilityArm(name: String, trialId: String) {
+final case class ApiCreateEligibilityArm(name: String, trialId: String, diseases: Seq[String]) {
 
-  def toDomain = EligibilityArm(
-    id = LongId(0),
-    name = name,
-    trialId = StringId(trialId),
-    originalName = name
-  )
+  def toDomain: EligibilityArmWithDiseases = {
+    val eligibilityArm = EligibilityArm(
+      id = LongId(0),
+      name = name,
+      trialId = StringId(trialId),
+      originalName = name
+    )
+
+    EligibilityArmWithDiseases(eligibilityArm, diseases.map { disease =>
+      val condition = Trial.Condition.fromString(disease)
+        .getOrElse(throw new NoSuchElementException(s"unknown condition $disease"))
+      EligibilityArmDisease(eligibilityArm.id, condition)
+    })
+  }
 }
 
 object ApiCreateEligibilityArm {