diff options
Diffstat (limited to 'shared/src')
-rw-r--r-- | shared/src/main/scala/DerivedFormats.scala | 12 | ||||
-rw-r--r-- | shared/src/test/scala/FieldNameTests.scala | 72 |
2 files changed, 81 insertions, 3 deletions
diff --git a/shared/src/main/scala/DerivedFormats.scala b/shared/src/main/scala/DerivedFormats.scala index eabfa82..4a0b8b4 100644 --- a/shared/src/main/scala/DerivedFormats.scala +++ b/shared/src/main/scala/DerivedFormats.scala @@ -9,11 +9,16 @@ import scala.language.experimental.macros trait DerivedFormats { self: BasicFormats => type Typeclass[T] = JsonFormat[T] + /** Convert the name of a parameter to that of a field in a JSON object. This + * method can be overriden to use alternative naming conventions. */ + def extractFieldName(paramName: String): String = paramName + def combine[T](ctx: CaseClass[JsonFormat, T]): JsonFormat[T] = new JsonFormat[T] { override def write(value: T): JsValue = { val fields: Seq[(String, JsValue)] = ctx.parameters.map { param => - param.label -> param.typeclass.write(param.dereference(value)) + extractFieldName(param.label) -> param.typeclass.write( + param.dereference(value)) } JsObject(fields: _*) } @@ -24,10 +29,11 @@ trait DerivedFormats { self: BasicFormats => ctx.rawConstruct(Seq.empty) } else { ctx.construct { param => + val fieldName = extractFieldName(param.label) val fieldValue = if (param.option) { - obj.fields.getOrElse(param.label, JsNull) + obj.fields.getOrElse(fieldName, JsNull) } else { - obj.fields(param.label) + obj.fields(fieldName) } param.typeclass.read(fieldValue) } diff --git a/shared/src/test/scala/FieldNameTests.scala b/shared/src/test/scala/FieldNameTests.scala new file mode 100644 index 0000000..f1b76a9 --- /dev/null +++ b/shared/src/test/scala/FieldNameTests.scala @@ -0,0 +1,72 @@ +package spray.json + +import org.scalatest._ + +trait SnakeCaseFormats { self: DerivedFormats => + override def extractFieldName(paramName: String) = + FieldNaming.substituteCamel(paramName, '_') +} +trait KebabCaseFormats { self: DerivedFormats => + override def extractFieldName(paramName: String) = + FieldNaming.substituteCamel(paramName, '-') +} + +object FieldNaming { + + @inline final private def isLower(ch: Char): Boolean = + ((ch & 0x20) != 0) || (ch == '_') + + @inline def substituteCamel(paramName: String, substitute: Char) = { + val length = paramName.length + val builder = new StringBuilder(length) + var i = 0 + while (i < length) { + val cur = paramName(i) + if (isLower(cur) && i + 1 < length) { + builder.append(cur) + val next = paramName(i + 1) + if (!isLower(next)) { + builder.append(substitute) + builder.append((next ^ 0x20).toChar) + } else { + builder.append(next) + } + i += 1 + } else { + builder.append((cur ^ 0x20).toChar) + } + i += 1 + } + builder.result() + } + +} + +class FieldNameTests extends FlatSpec with FormatTests { + + case class A(camelCASE: String, `__a_aB__`: Int, `a-a_B`: Int) + case class B(camelCaseA: A) + + trait All extends DefaultJsonProtocol with DerivedFormats { + implicit val bFormat = jsonFormat[B] + } + + { + object Protocol extends All with SnakeCaseFormats + import Protocol._ + "snake_case" should behave like checkRoundtrip( + B(A("helloWorld", 0, 0)), + """{"camel_case_a":{"camel_case":"helloWorld","__a_a_b__":0,"a-a_b":0}}""" + ) + } + + { + object Protocol extends All with KebabCaseFormats + import Protocol._ + "kebab-case" should behave like checkRoundtrip( + B(A("helloWorld", 0, 0)), + """{"camel-case-a":{"camel-case":"helloWorld","__a_a-b__":0,"a-a_b":0}}""" + ) + } + +} |