1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2018 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.core.jsoniter
21 
22 import com.github.plokhotnyuk.jsoniter_scala.core._
23 import enumeratum.values.{StringEnum, StringEnumEntry}
24 
25 import scala.collection.mutable.HashMap
26 
27 @SuppressWarnings(Array("org.wartremover.warts.LeakingSealed"))
28 trait JsoniterEnum[T <: StringEnumEntry] extends StringEnum[T] {
29 
30   @SuppressWarnings(Array("org.wartremover.warts.MutableDataStructures"))
31   private val kvs: HashMap[String, T] = this.values.map(v => (v.value, v)).to(HashMap)
32 
33   @SuppressWarnings(Array("org.wartremover.warts.Null", "org.wartremover.warts.AsInstanceOf"))
34   implicit val codec: JsonValueCodec[T] =
35     new JsonValueCodec[T] {
36 
37       def decodeValue(in: JsonReader, default: T): T =
38         kvs(in.readString(null))
39 
40       def encodeValue(wrapper: T, out: JsonWriter): Unit =
41         out.writeVal(wrapper.value)
42 
43       def nullValue: T = null.asInstanceOf[T]
44     }
45 }