Ability to map roles back to Driver's roles

2 files changed, 31 insertions, 14 deletions

diff --git a/src/main/scala/xyz/driver/pdsuicommon/domain/User.scala b/src/main/scala/xyz/driver/pdsuicommon/domain/User.scala
index cfc2270..63eef85 100644
--- a/src/main/scala/xyz/driver/pdsuicommon/domain/User.scala
+++ b/src/main/scala/xyz/driver/pdsuicommon/domain/User.scala
@@ -131,4 +131,35 @@ object User {
         Set.empty[xyz.driver.pdsuicommon.domain.User.Role]
     }
   }
+
+  def mapRolesToDriver(pdsuiRole: xyz.driver.pdsuicommon.domain.User.Role): Set[xyz.driver.core.auth.Role] = {
+    pdsuiRole match {
+      case xyz.driver.pdsuicommon.domain.User.Role.SystemUser =>
+        Set(xyz.driver.entities.auth.AdministratorRole)
+      case xyz.driver.pdsuicommon.domain.User.Role.RecordAdmin =>
+        Set(xyz.driver.entities.auth.RecordAdmin)
+      case xyz.driver.pdsuicommon.domain.User.Role.RecordCleaner =>
+        Set(xyz.driver.entities.auth.RecordCleaner)
+      case xyz.driver.pdsuicommon.domain.User.Role.RecordOrganizer =>
+        Set(xyz.driver.entities.auth.RecordOrganizer)
+      case xyz.driver.pdsuicommon.domain.User.Role.DocumentExtractor =>
+        Set(xyz.driver.entities.auth.DocumentExtractor)
+      case xyz.driver.pdsuicommon.domain.User.Role.TrialSummarizer =>
+        Set(xyz.driver.entities.auth.TrialSummarizer)
+      case xyz.driver.pdsuicommon.domain.User.Role.CriteriaCurator =>
+        Set(xyz.driver.entities.auth.CriteriaCurator)
+      case xyz.driver.pdsuicommon.domain.User.Role.TrialAdmin =>
+        Set(xyz.driver.entities.auth.TrialAdmin)
+      case xyz.driver.pdsuicommon.domain.User.Role.EligibilityVerifier =>
+        Set(xyz.driver.entities.auth.EligibilityVerifier)
+      case xyz.driver.pdsuicommon.domain.User.Role.TreatmentMatchingAdmin =>
+        Set(xyz.driver.entities.auth.TreatmentMatchingAdmin)
+      case xyz.driver.pdsuicommon.domain.User.Role.RoutesCurator =>
+        Set(xyz.driver.entities.auth.RoutesCurator)
+      case xyz.driver.pdsuicommon.domain.User.Role.ResearchOncologist =>
+        Set(xyz.driver.entities.auth.ResearchOncologist)
+      case _ =>
+        Set.empty[xyz.driver.core.auth.Role]
+    }
+  }
 }
diff --git a/src/main/scala/xyz/driver/pdsuidomain/entities/LinkedPatient.scala b/src/main/scala/xyz/driver/pdsuidomain/entities/LinkedPatient.scala
deleted file mode 100644
index 28f279f..0000000
--- a/src/main/scala/xyz/driver/pdsuidomain/entities/LinkedPatient.scala
+++ /dev/null
@@ -1,14 +0,0 @@
-package xyz.driver.pdsuidomain.entities
-
-import xyz.driver.pdsuicommon.domain._
-import xyz.driver.pdsuicommon.logging._
-
-object LinkedPatient {
-
-  implicit def toPhiString(x: LinkedPatient): PhiString = {
-    import x._
-    phi"LinkedPatient(userId=$userId, patientId=$patientId, trialId=$trialId)"
-  }
-}
-
-case class LinkedPatient(userId: StringId[User], patientId: UuidId[Patient], trialId: StringId[Trial])