path: root/src/main/scala/xyz/driver/pdsuidomain/formats/json/intervention.scala

package xyz.driver.pdsuidomain.formats.json

import spray.json._
import xyz.driver.pdsuicommon.domain.{LongId, StringId}
import xyz.driver.pdsuidomain.entities.InterventionType.DeliveryMethod
import xyz.driver.pdsuidomain.entities._

object intervention {
  import DefaultJsonProtocol._
  import common._

  implicit def interventionFormat: RootJsonFormat[InterventionWithArms] = new RootJsonFormat[InterventionWithArms] {
    override def write(obj: InterventionWithArms) =
      JsObject(
        "id"             -> obj.intervention.id.toJson,
        "name"           -> obj.intervention.name.toJson,
        "typeId"         -> obj.intervention.typeId.toJson,
        "dosage"         -> obj.intervention.dosage.toJson,
        "isActive"       -> obj.intervention.isActive.toJson,
        "arms"           -> obj.arms.map(_.armId).toJson,
        "trialId"        -> obj.intervention.trialId.toJson,
        "deliveryMethod" -> obj.intervention.deliveryMethod.toJson,
        "originalName"   -> obj.intervention.originalName.toJson,
        "originalDosage" -> obj.intervention.originalDosage.toJson,
        "originalType"   -> obj.intervention.originalType.toJson
      )

    override def read(json: JsValue): InterventionWithArms = json match {
      case JsObject(fields) =>
        val trialId = fields
          .get("trialId")
          .map(_.convertTo[StringId[Trial]])
          .getOrElse(deserializationError(s"Intervention json object does not contain `trialId` field: $json"))

        val typeId = fields
          .get("typeId")
          .flatMap(_.convertTo[Option[LongId[InterventionType]]])

        val name = fields
          .get("name")
          .map(_.convertTo[String])
          .getOrElse("")

        val dosage = fields
          .get("dosage")
          .map(_.convertTo[String])

        val isActive = fields
          .get("isActive")
          .exists(_.convertTo[Boolean])

        val deliveryMethod = fields
          .get("deliveryMethod")
          .flatMap(_.convertTo[Option[String]])

        val arms = fields
          .get("arms")
          .map(_.convertTo[List[LongId[SlotArm]]])
          .map(_ map (x => InterventionArm(armId = x, interventionId = LongId(0L))))
          .getOrElse(List.empty[InterventionArm])

        InterventionWithArms(
          intervention = Intervention(
            id = LongId(0L),
            trialId = trialId,
            name = name,
            originalName = name,
            typeId = typeId,
            originalType = None,
            dosage = dosage.getOrElse(""),
            originalDosage = dosage.getOrElse(""),
            isActive = isActive,
            deliveryMethod = deliveryMethod
          ),
          arms = arms
        )

      case _ => deserializationError(s"Expected Json Object as create Intervention json, but got $json")
    }
  }

  def applyUpdateToInterventionWithArms(json: JsValue, orig: InterventionWithArms): InterventionWithArms = json match {
    case JsObject(fields) =>
      val name = fields
        .get("name")
        .map(_.convertTo[String])

      val typeId = fields
        .get("typeId")
        .map(_.convertTo[Option[LongId[InterventionType]]])

      val dosage = fields
        .get("dosage")
        .map(_.convertTo[String])

      val isActive = fields
        .get("isActive")
        .map(_.convertTo[Boolean])

      val deliveryMethod = fields
        .get("deliveryMethod")
        .map(_.convertTo[Option[String]])
        .getOrElse(orig.intervention.deliveryMethod)

      val origIntervention = orig.intervention
      val arms = fields
        .get("arms")
        .map(_.convertTo[List[LongId[SlotArm]]].map(x => InterventionArm(x, orig.intervention.id)))

      orig.copy(
        intervention = origIntervention.copy(
          name = name.getOrElse(origIntervention.name),
          typeId = typeId.getOrElse(origIntervention.typeId),
          dosage = dosage.getOrElse(origIntervention.dosage),
          isActive = isActive.getOrElse(origIntervention.isActive),
          deliveryMethod = deliveryMethod
        ),
        arms = arms.getOrElse(orig.arms)
      )

    case _ => deserializationError(s"Expected Json Object as partial Intervention, but got $json")
  }

  implicit def interventionTypeFormat: JsonFormat[InterventionType] = new RootJsonFormat[InterventionType] {
    override def read(json: JsValue): InterventionType = json match {
      case JsObject(fields) =>
        val name = fields
          .get("name")
          .map(_.convertTo[String])
          .getOrElse(deserializationError(s"Intervention type json object does not contain `name` field: $json"))

        InterventionType
          .typeFromString(name)
          .getOrElse(deserializationError(s"Unknown intervention type: $name"))

      case _ => deserializationError(s"Expected Json Object as Intervention type, but got $json")
    }

    override def write(obj: InterventionType) =
      JsObject(
        "id"              -> obj.id.toJson,
        "name"            -> obj.name.toJson,
        "deliveryMethods" -> obj.deliveryMethods.map(DeliveryMethod.methodToString).toJson
      )
  }

}