src/main/scala/xyz/driver/pdsuidomain/formats/json/extracteddata/ApiExtractedData.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

package xyz.driver.pdsuidomain.formats.json.extracteddata

import xyz.driver.pdsuidomain.formats.json.label.ApiExtractedDataLabel
import play.api.libs.json._
import play.api.data.validation._
import play.api.libs.functional.syntax._
import xyz.driver.pdsuicommon.json.JsonSerializer
import xyz.driver.pdsuidomain.services.ExtractedDataService.RichExtractedData

// The specification: https://driverinc.atlassian.net/wiki/pages/viewpage.action?pageId=33423387
// Note, that there is "Extracted data object or Temporary extracted data object" in specification
// ApiExtractedData represents both types
final case class ApiExtractedData(id: Long,
                                  documentId: Long,
                                  keywordId: Option[Long],
                                  evidence: Option[String],
                                  meta: Option[String],
                                  // An empty list and no-existent list are different cases
                                  labels: Option[List[ApiExtractedDataLabel]])

object ApiExtractedData {

  implicit val format: Format[ApiExtractedData] = (
    (JsPath \ "id").format[Long] and
      (JsPath \ "documentId").format[Long] and
      (JsPath \ "keywordId").formatNullable[Long] and
      (JsPath \ "evidence").formatNullable[String] and
      (JsPath \ "meta").formatNullable[String] and
      (JsPath \ "labels").formatNullable[List[ApiExtractedDataLabel]](
        Format(
          Reads
            .of[List[ApiExtractedDataLabel]]
            .filter(ValidationError("empty labels"))({
              case x if x.nonEmpty => true
              case _               => false
            }),
          Writes.of[List[ApiExtractedDataLabel]]
        ))
  )(ApiExtractedData.apply, unlift(ApiExtractedData.unapply))

  def fromDomain(extractedDataWithLabels: RichExtractedData) = ApiExtractedData(
    id = extractedDataWithLabels.extractedData.id.id,
    documentId = extractedDataWithLabels.extractedData.documentId.id,
    keywordId = extractedDataWithLabels.extractedData.keywordId.map(_.id),
    evidence = extractedDataWithLabels.extractedData.evidenceText,
    meta = extractedDataWithLabels.extractedData.meta.map(x => JsonSerializer.serialize(x.content)),
    labels = Option(extractedDataWithLabels.labels.map(ApiExtractedDataLabel.fromDomain))
  )
}