summaryrefslogtreecommitdiff
path: root/main/api/src/mill/api/JsonFormatters.scala
blob: 23d7fad09e93d937513fd6d37ccf7b95047c09b9 (plain) (blame)
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 mill.api

import upickle.default.{ReadWriter => RW}
import scala.util.matching.Regex

object JsonFormatters extends JsonFormatters

/**
 * Defines various default JSON formatters used in mill.
 */
trait JsonFormatters {
  implicit val pathReadWrite: RW[os.Path] = upickle.default.readwriter[String]
    .bimap[os.Path](
      _.toString,
      os.Path(_)
    )

  implicit val regexReadWrite: RW[Regex] = upickle.default.readwriter[String]
    .bimap[Regex](
      _.pattern.toString,
      _.r
    )

  implicit val bytesReadWrite: RW[os.Bytes] = upickle.default.readwriter[String]
    .bimap(
      o => java.util.Base64.getEncoder.encodeToString(o.array),
      str => new os.Bytes(java.util.Base64.getDecoder.decode(str))
    )

  implicit lazy val crFormat: RW[os.CommandResult] = upickle.default.macroRW

  implicit val stackTraceRW = upickle.default.readwriter[ujson.Obj].bimap[StackTraceElement](
    ste => ujson.Obj(
      "declaringClass" -> ujson.Str(ste.getClassName),
      "methodName" -> ujson.Str(ste.getMethodName),
      "fileName" -> ujson.Str(ste.getFileName),
      "lineNumber" -> ujson.Num(ste.getLineNumber)
    ),
    {
      case json: ujson.Obj =>
        new StackTraceElement(
          json("declaringClass").str.toString,
          json("methodName").str.toString,
          json("fileName").str.toString,
          json("lineNumber").num.toInt
        )
    }
  )
}