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

package xyz.driver.pdsuidomain.fakes.entities.rep

import xyz.driver.core.generators._
import xyz.driver.pdsuicommon.domain.{LongId, UuidId}
import xyz.driver.pdsuidomain.entities._
import xyz.driver.pdsuidomain.entities.export.patient.{
  ExportPatientLabel,
  ExportPatientLabelEvidence,
  ExportPatientLabelEvidenceDocument,
  ExportPatientWithLabels
}
import xyz.driver.pdsuidomain.fakes.entities.common.{nextLocalDate, nextLongId}

object ExportPatientGen {
  private val maxItemsInCollectionNumber = 3

  def nextDocumentType(documentTypeId: LongId[DocumentType] = nextLongId): DocumentType = {
    DocumentType(
      documentTypeId,
      nextString()
    )
  }

  def nextProviderType(providerTypeId: LongId[ProviderType] = nextLongId): ProviderType = {
    ProviderType(
      providerTypeId,
      nextString()
    )
  }

  def nextExportPatientLabelEvidenceDocument(documentId: LongId[Document]): ExportPatientLabelEvidenceDocument = {
    ExportPatientLabelEvidenceDocument(
      documentId = documentId,
      requestId = RecordRequestId(nextUuid()),
      documentType = nextDocumentType(),
      providerType = nextProviderType(),
      date = nextLocalDate
    )
  }

  def nextExportPatientLabelEvidence(documentId: LongId[Document]): ExportPatientLabelEvidence = {
    ExportPatientLabelEvidence(
      id = nextLongId[ExtractedData],
      value = Common.nextFuzzyValue(),
      evidenceText = nextString(),
      document = nextExportPatientLabelEvidenceDocument(documentId)
    )
  }

  def nextExportPatientLabel(documentId: LongId[Document]): ExportPatientLabel = {
    ExportPatientLabel(
      id = nextLongId[Label],
      evidences = List.fill(
        nextInt(maxItemsInCollectionNumber, minValue = 0)
      )(nextExportPatientLabelEvidence(documentId))
    )
  }

  def nextExportPatientWithLabels(documentId: LongId[Document]): ExportPatientWithLabels = {
    ExportPatientWithLabels(
      patientId = UuidId[xyz.driver.pdsuidomain.entities.Patient](nextUuid()),
      labelVersion = scala.util.Random.nextLong(),
      labels = List.fill(
        nextInt(maxItemsInCollectionNumber, minValue = 0)
      )(nextExportPatientLabel(documentId))
    )
  }

}