1 files changed, 28 insertions, 1 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/entities/PatientCriterion.scala b/src/main/scala/xyz/driver/pdsuidomain/entities/PatientCriterion.scala
index 193e8cb..e6d466a 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/entities/PatientCriterion.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/entities/PatientCriterion.scala
@@ -2,7 +2,7 @@ package xyz.driver.pdsuidomain.entities
 
 import java.time.LocalDateTime
 
-import xyz.driver.entities.labels.LabelValue
+import xyz.driver.entities.labels.{Label, LabelValue}
 import xyz.driver.pdsuicommon.domain.{LongId, StringId}
 import xyz.driver.pdsuicommon.logging._
 
@@ -57,6 +57,33 @@ final case class PatientCriterion(id: LongId[PatientCriterion],
   def isIneligibleForEv: Boolean = eligibilityStatus === LabelValue.No && isVerified
 }
 
+final case class DraftPatientCriterion(id: LongId[PatientCriterion],
+                                       eligibilityStatus: Option[LabelValue],
+                                       isVerified: Option[Boolean]) {
+  def applyTo(orig: PatientCriterion) = {
+    orig.copy(
+      eligibilityStatus = eligibilityStatus.getOrElse(orig.eligibilityStatus),
+      isVerified = isVerified.getOrElse(orig.isVerified)
+    )
+  }
+}
+
+object DraftPatientCriterion {
+  implicit def toPhiString(x: DraftPatientCriterion): PhiString = {
+    phi"DraftPatientCriterion(id=${x.id}, eligibilityStatus=${Unsafe(x.eligibilityStatus)}, isVerified=${x.isVerified})"
+  }
+}
+
+final case class RichPatientCriterion(patientCriterion: PatientCriterion,
+                                      labelId: LongId[Label],
+                                      armList: List[PatientCriterionArm])
+
+object RichPatientCriterion {
+  implicit def toPhiString(x: RichPatientCriterion): PhiString = {
+    phi"RichPatientCriterion(patientCriterion=${x.patientCriterion}, labelId=${x.labelId}, arms=${x.armList})"
+  }
+}
+
 object PatientCriterionArm {
 
   implicit def toPhiString(x: PatientCriterionArm): PhiString = {