path: root/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala

package xyz.driver.pdsuidomain.fakes.entities

import xyz.driver.core.auth.User
import xyz.driver.core.generators._
import xyz.driver.entities.labels.{Label, LabelCategory}
import xyz.driver.pdsuicommon.domain.LongId
import xyz.driver.pdsuidomain.ListResponse
import xyz.driver.pdsuidomain.entities._
import xyz.driver.pdsuidomain.services.CriterionService.RichCriterion

object trialcuration {

  import common._
  import xyz.driver.core.generators
  import xyz.driver.pdsuidomain.entities.InterventionType._

  private val maxItemsInCollectionNumber: Int = 5

  def nextTrial(): Trial = Trial(
    id = nextStringId[Trial],
    externalId = nextUuidId[Trial],
    status = nextTrialStatus,
    assignee = Option(generators.nextId[User]),
    previousStatus = Option(nextPreviousTrialStatus),
    previousAssignee = Option(generators.nextId[User]),
    lastActiveUserId = Option(generators.nextId[User]),
    lastUpdate = nextLocalDateTime,
    phase = generators.nextString(),
    hypothesisId = Option(nextUuidId[Hypothesis]),
    studyDesignId = Option(nextLongId[StudyDesign]),
    originalStudyDesign = Option(generators.nextString()),
    isPartner = generators.nextBoolean(),
    overview = Option(generators.nextString()),
    overviewTemplate = generators.nextString(),
    isUpdated = generators.nextBoolean(),
    title = generators.nextString(),
    originalTitle = generators.nextString()
  )

  def nextArm(): Arm = Arm(
    id = nextLongId[Arm],
    name = generators.nextString(),
    originalName = generators.nextString(),
    trialId = nextStringId[Trial],
    deleted = Option(nextLocalDateTime)
  )

  def nextCriterion(): Criterion = Criterion(
    id = nextLongId[Criterion],
    trialId = nextStringId[Trial],
    text = Option(generators.nextString()),
    isCompound = generators.nextBoolean(),
    meta = generators.nextString(),
    inclusion = Option(generators.nextBoolean())
  )

  def nextCriterionLabel(criterionId: LongId[Criterion]): CriterionLabel = CriterionLabel(
    id = nextLongId[CriterionLabel],
    labelId = Option(nextLongId[Label]),
    criterionId = criterionId,
    categoryId = Option(nextLongId[LabelCategory]),
    value = Option(generators.nextBoolean()),
    isDefining = generators.nextBoolean()
  )

  def nextRichCriterion(): RichCriterion = {
    val criterion = nextCriterion()
    RichCriterion(
      criterion = criterion,
      armIds = Seq(nextLongId[EligibilityArm], nextLongId[EligibilityArm]),
      labels = Seq(
        nextCriterionLabel(criterion.id),
        nextCriterionLabel(criterion.id)
      )
    )
  }

  def nextIntervention(): Intervention = Intervention(
    id = nextLongId[Intervention],
    trialId = nextStringId[Trial],
    name = generators.nextString(),
    originalName = generators.nextString(),
    typeId = Option(nextLongId[InterventionType]),
    originalType = Option(generators.nextString()),
    dosage = generators.nextString(),
    originalDosage = generators.nextString(),
    isActive = generators.nextBoolean(),
    deliveryMethod = Option(generators.nextString())
  )

  def nextInterventionArm(interventionId: LongId[Intervention]): InterventionArm = InterventionArm(
    interventionId = interventionId,
    armId = nextLongId[SlotArm]
  )

  def nextInterventionWithArms(): InterventionWithArms = {
    val intervention = nextIntervention()
    InterventionWithArms(
      intervention = intervention,
      arms = List(
        nextInterventionArm(intervention.id),
        nextInterventionArm(intervention.id),
        nextInterventionArm(intervention.id)
      )
    )
  }

  def nextTrialIssue(): TrialIssue = TrialIssue(
    id = nextLongId[TrialIssue],
    userId = generators.nextId[User],
    trialId = nextStringId[Trial],
    lastUpdate = nextLocalDateTime,
    isDraft = generators.nextBoolean(),
    text = generators.nextString(),
    evidence = generators.nextString(),
    archiveRequired = generators.nextBoolean(),
    meta = generators.nextString()
  )

  def nextTrialHistory(): TrialHistory = TrialHistory(
    id = nextLongId[TrialHistory],
    executor = generators.nextId[User],
    trialId = nextStringId[Trial],
    state = nextTrialState,
    action = nextTrialAction,
    created = nextLocalDateTime,
    comment = generators.nextOption(generators.nextString())
  )

  def nextHypothesis(): Hypothesis = Hypothesis(
    id = nextUuidId[Hypothesis],
    name = generators.nextString(),
    treatmentType = generators.nextString(),
    description = generators.nextString()
  )

  def nextStudyDesign(): StudyDesign = generators.oneOf[StudyDesign](StudyDesign.All: _*)

  def nextInterventionType(): InterventionType = generators.oneOf[InterventionType](
    Chemotherapy,
    TargetedTherapy,
    Immunotherapy,
    HormoneTherapy,
    Other,
    Radiation,
    SurgeryProcedure
  )

  def nextEligibilityArm(): EligibilityArm = EligibilityArm(
    id = nextLongId,
    name = nextString(),
    originalName = nextString(),
    trialId = nextStringId
  )

  def nextEligibilityArmDisease(): EligibilityArmDisease = EligibilityArmDisease(
    eligibilityArmId = nextLongId,
    disease = nextCancerType
  )

  private def nextEligibilityArmDiseaseCollection(count: Int): Seq[EligibilityArmDisease] =
    Seq.fill(count)(nextEligibilityArmDisease())

  def nextEligibilityArmWithDiseases(): EligibilityArmWithDiseases = {
    val entity = nextEligibilityArm()
    val id     = entity.id
    val collection = nextEligibilityArmDiseaseCollection(
      nextInt(maxItemsInCollectionNumber, minValue = 0)
    ).map(_.copy(eligibilityArmId = id))

    EligibilityArmWithDiseases(
      entity,
      collection
    )
  }

  def nextSlotArm(): SlotArm = SlotArm(
    id = nextLongId,
    name = nextString(),
    originalName = nextString(),
    trialId = nextStringId
  )

  def nextTrialListResponse(): ListResponse[Trial] = {
    val xs: Seq[Trial] = Seq.fill(3)(nextTrial())
    nextListResponse(xs)
  }

  def nextTrialIssueListResponse(): ListResponse[TrialIssue] = {
    val xs: Seq[TrialIssue] = Seq.fill(3)(nextTrialIssue())
    nextListResponse(xs)
  }

  def nextTrialHistoryListResponse(): ListResponse[TrialHistory] = {
    val xs: Seq[TrialHistory] = Seq.fill(3)(nextTrialHistory())
    nextListResponse(xs)
  }

  def nextArmListResponse(): ListResponse[Arm] = {
    val xs: Seq[Arm] = Seq.fill(3)(nextArm())
    nextListResponse(xs)
  }

  def nextInterventionWithArmsListResponse(): ListResponse[InterventionWithArms] = {
    val xs: Seq[InterventionWithArms] = Seq.fill(3)(nextInterventionWithArms())
    nextListResponse(xs)
  }

  def nextEligibilityArmWithDiseasesListResponse(): ListResponse[EligibilityArmWithDiseases] = {
    val xs: Seq[EligibilityArmWithDiseases] = Seq.fill(3)(nextEligibilityArmWithDiseases())
    nextListResponse(xs)
  }

  def nextSlotArmListResponse(): ListResponse[SlotArm] = {
    val xs: Seq[SlotArm] = Seq.fill(3)(nextSlotArm())
    nextListResponse(xs)
  }

  def nextRichCriterionListResponse(): ListResponse[RichCriterion] = {
    val xs: Seq[RichCriterion] = Seq.fill(3)(nextRichCriterion())
    nextListResponse(xs)
  }

  def nextInterventionTypeListResponse(): ListResponse[InterventionType] = {
    val xs: Seq[InterventionType] = Seq.fill(3)(nextInterventionType())
    nextListResponse(xs)
  }

  def nextStudyDesignListResponse(): ListResponse[StudyDesign] = {
    val xs: Seq[StudyDesign] = Seq.fill(3)(nextStudyDesign())
    nextListResponse(xs)
  }

  def nextHypothesesListResponse(): ListResponse[Hypothesis] = {
    val xs: Seq[Hypothesis] = Seq.fill(3)(nextHypothesis())
    nextListResponse(xs)
  }

}