scalapb.json4s.StructFormat.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scalapb-json4s_2.12 Show documentation
scalapb-json4s
The newest version!
package scalapb.json4s

import com.google.protobuf.struct.Value.Kind
import org.json4s.JsonAST._
import com.google.protobuf.struct
import scala.collection.compat._

object StructFormat {
  def structValueWriter(v: struct.Value): JValue =
    v.kind match {
      case Kind.Empty              => JNull
      case Kind.NullValue(value)   => JNull
      case Kind.NumberValue(value) => JDouble(value)
      case Kind.StringValue(value) => JString(value)
      case Kind.BoolValue(value)   => JBool(value)
      case Kind.StructValue(value) => structWriter(value)
      case Kind.ListValue(value)   => listValueWriter(value)
    }

  def structValueParser(v: JValue): struct.Value = {
    val kind: struct.Value.Kind = v match {
      case JNothing      => Kind.NullValue(struct.NullValue.NULL_VALUE)
      case JNull         => Kind.NullValue(struct.NullValue.NULL_VALUE)
      case JString(s)    => Kind.StringValue(value = s)
      case JDouble(num)  => Kind.NumberValue(value = num)
      case JDecimal(num) => Kind.NumberValue(value = num.toDouble)
      case JLong(num)    => Kind.NumberValue(value = num.toDouble)
      case JInt(num)     => Kind.NumberValue(value = num.toDouble)
      case JBool(value)  => Kind.BoolValue(value = value)
      case obj: JObject  => Kind.StructValue(value = structParser(obj))
      case arr: JArray   => Kind.ListValue(listValueParser(arr))
      case JSet(set)     => throw new RuntimeException("Unsupported")
    }
    struct.Value(kind = kind)
  }

  def structParser(v: JValue): struct.Struct =
    v match {
      case JObject(fields) =>
        struct.Struct(
          fields = fields.map(kv => (kv._1 -> structValueParser(kv._2))).toMap
        )
      case _ => throw new JsonFormatException("Expected an object")
    }

  def structWriter(v: struct.Struct): JValue =
    JObject(v.fields.view.mapValues(structValueWriter(_)).toList)

  def listValueParser(v: JValue): struct.ListValue =
    v match {
      case JArray(elems) =>
        com.google.protobuf.struct.ListValue(elems.map(structValueParser))
      case _ => throw new JsonFormatException("Expected a list")
    }

  def listValueWriter(v: struct.ListValue): JArray =
    JArray(v.values.map(structValueWriter(_)).toList)

  def nullValueParser(v: JValue): struct.NullValue =
    v match {
      case JNull =>
        com.google.protobuf.struct.NullValue.NULL_VALUE
      case _ => throw new JsonFormatException("Expected a null")
    }

  def nullValueWriter(v: struct.NullValue): JValue = JNull
}