Adding domain entitiesv0.1.0

1 files changed, 51 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/pdsuidomain/entities/export/trial/ExportTrialWithLabels.scala b/src/main/scala/xyz/driver/pdsuidomain/entities/export/trial/ExportTrialWithLabels.scala
new file mode 100644
index 0000000..251f6fb
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuidomain/entities/export/trial/ExportTrialWithLabels.scala
@@ -0,0 +1,51 @@
+package xyz.driver.pdsuidomain.entities.export.trial
+
+import java.time.LocalDateTime
+
+import xyz.driver.pdsuicommon.domain.{StringId, UuidId}
+import xyz.driver.pdsuicommon.logging._
+import xyz.driver.pdsuidomain.entities.{RawTrialLabel, Trial}
+
+import scala.collection.breakOut
+
+case class ExportTrialWithLabels(nctId: StringId[Trial],
+                                 trialId: UuidId[Trial],
+                                 condition: String,
+                                 lastReviewed: LocalDateTime,
+                                 labelVersion: Long,
+                                 arms: List[ExportTrialArm],
+                                 criteria: List[ExportTrialLabelCriterion])
+
+object ExportTrialWithLabels {
+
+  implicit def toPhiString(x: ExportTrialWithLabels): PhiString = {
+    import x._
+    phi"TrialWithLabels(nctId=$nctId, trialId=$trialId, condition=${Unsafe(condition)}, " +
+      phi"lastReviewed=$lastReviewed, labelVersion=${Unsafe(labelVersion)}, arms=$arms, criteria=$criteria)"
+  }
+
+  def fromRaw(rawData: List[RawTrialLabel]): ExportTrialWithLabels = {
+    val trials: Set[StringId[Trial]] = rawData.map(_.nctId)(breakOut)
+
+    assert(trials.size == 1, "There are more than one trials in the rawData")
+    val trial = rawData.head
+
+    ExportTrialWithLabels(
+      nctId = trial.nctId,
+      trialId = trial.trialId,
+      condition = trial.condition,
+      lastReviewed = trial.lastReviewed,
+      labelVersion = 1, // TODO It is needed to replace this mock label version.
+      arms = rawData.groupBy(_.armId).map { case (armId, rawTrials) =>
+        ExportTrialArm(armId, rawTrials.head.armName)
+      }(breakOut),
+      criteria = rawData.groupBy { x =>
+        (x.criterionId, x.labelId)
+      }.map {
+        case (_, rawTrialLabels) =>
+          val armIds = rawTrialLabels.map(_.criterionArmId).toSet
+          ExportTrialLabelCriterion.fromRaw(rawTrialLabels.head, armIds)
+      }(breakOut)
+    )
+  }
+}