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

package xyz.driver.pdsuidomain.entities

import java.nio.channels.ReadableByteChannel
import java.time.LocalDateTime

import com.fasterxml.jackson.annotation.JsonSubTypes.Type
import com.fasterxml.jackson.annotation.{JsonProperty, JsonSubTypes, JsonTypeInfo}
import xyz.driver.core.auth.User
import xyz.driver.entities.assays.PatientCase
import xyz.driver.entities.clinic.ClinicalRecord
import xyz.driver.pdsuicommon.domain._
import xyz.driver.pdsuicommon.logging._
import xyz.driver.pdsuicommon.utils.Utils
import xyz.driver.pdsuidomain.entities.MedicalRecord.Meta
import xyz.driver.pdsuidomain.entities.MedicalRecord.Meta.{Duplicate, Reorder, Rotation}

object MedicalRecord {

  @JsonTypeInfo(use = JsonTypeInfo.Id.NAME, property = "type")
  @JsonSubTypes(
    Array(
      new Type(value = classOf[Duplicate], name = "duplicate"),
      new Type(value = classOf[Reorder], name = "reorder"),
      new Type(value = classOf[Rotation], name = "rotation")
    ))
  trait Meta {
    @JsonProperty("type") def metaType: String
  }

  object Meta {

    final case class Duplicate(startPage: Double,
                               endPage: Double,
                               startOriginalPage: Double,
                               endOriginalPage: Option[Double])
        extends Meta {
      override val metaType = "duplicate"
    }

    object Duplicate {
      implicit def toPhiString(x: Duplicate): PhiString = {
        import x._
        phi"Duplicate(startPage=${Unsafe(startPage)}, endPage=${Unsafe(endPage)}, " +
          phi"startOriginalPage=${Unsafe(startOriginalPage)}, endOriginalPage=${Unsafe(endOriginalPage)}"
      }
    }

    final case class Reorder(items: Seq[Int]) extends Meta {
      override val metaType = "reorder"
    }

    object Reorder {
      implicit def toPhiString(x: Reorder): PhiString = {
        import x._
        phi"Reorder(items=${Unsafe(items.toString)})"
      }
    }

    final case class Rotation(items: Map[String, Int]) extends Meta {
      override val metaType = "rotation"
    }

    object Rotation {
      implicit def toPhiString(x: Rotation): PhiString = {
        import x._
        phi"Rotation(items=${Unsafe(items.toString)})"
      }
    }

    implicit def toPhiString(input: Meta): PhiString = input match {
      case x: Duplicate => Duplicate.toPhiString(x)
      case x: Reorder   => Reorder.toPhiString(x)
      case x: Rotation  => Rotation.toPhiString(x)
    }

  }

  // Product with Serializable fixes issue:
  // Set(New, Cleaned) has type Set[Status with Product with Serializable]
  sealed trait Status extends Product with Serializable {

    def oneOf(xs: Status*): Boolean = xs.contains(this)

    def oneOf(xs: Set[Status]): Boolean = xs.contains(this)

  }
  object Status {
    case object New           extends Status
    case object Cleaned       extends Status
    case object Reviewed      extends Status
    case object Organized     extends Status
    case object Done          extends Status
    case object Flagged       extends Status
    case object Archived      extends Status

    def fromString(status: String): Option[Status] = status match {
      case "New"           => Some(New)
      case "Cleaned"       => Some(Cleaned)
      case "Reviewed"      => Some(Reviewed)
      case "Organized"     => Some(Organized)
      case "Done"          => Some(Done)
      case "Flagged"       => Some(Flagged)
      case "Archived"      => Some(Archived)
      case _               => None
    }

    val All: Set[Status] = Set[Status](
      New,
      Cleaned,
      Reviewed,
      Organized,
      Done,
      Flagged,
      Archived
    )

    val AllPrevious: Set[Status] = Set[Status](New, Cleaned, Reviewed, Organized)

    implicit def toPhiString(x: Status): PhiString = Unsafe(Utils.getClassSimpleName(x.getClass))
  }

  sealed trait PdfSource

  object PdfSource {
    case object Empty extends PdfSource

    /** @param createResource Constructor of the resource which is represents the file */
    final case class Channel(createResource: () => ReadableByteChannel) extends PdfSource
  }

  implicit def toPhiString(x: MedicalRecord): PhiString = {
    import x._
    phi"MedicalRecord(id=$id, status=$status, assignee=${Unsafe(assignee)}, " +
      phi"previousAssignee=${Unsafe(previousAssignee)}, lastActiveUserId=${Unsafe(lastActiveUserId)})"
  }
}

final case class MedicalRecord(id: LongId[MedicalRecord],
                               status: MedicalRecord.Status,
                               previousStatus: Option[MedicalRecord.Status],
                               assignee: Option[xyz.driver.core.Id[User]],
                               previousAssignee: Option[xyz.driver.core.Id[User]],
                               lastActiveUserId: Option[xyz.driver.core.Id[User]],
                               patientId: UuidId[Patient],
                               requestId: xyz.driver.core.Id[ClinicalRecord],
                               disease: String,
                               caseId: Option[xyz.driver.core.Id[PatientCase]],
                               physician: Option[String],
                               meta: Option[TextJson[List[Meta]]],
                               lastUpdate: LocalDateTime,
                               totalPages: Int) {

  import MedicalRecord.Status._

  if (previousStatus.nonEmpty) {
    assert(AllPrevious.contains(previousStatus.get), s"Previous status has invalid value: ${previousStatus.get}")
  }

  //  TODO: with the current business logic code this constraint sometimes harmful
  //  require(status match {
  //    case MedicalRecord.Status.Done if assignee.isDefined => false
  //    case _ => true
  //  }, "Assignee can't be defined in Done status")
}