Add HasPolyCborCodec

commons-core/src/main/scala/com/avsystem/commons/serialization/cbor/CborAdtMetadata.scala

@@ -6,6 +6,7 @@ import com.avsystem.commons.meta._
 import com.avsystem.commons.misc.ValueOf
 import com.avsystem.commons.serialization.GenCodec.OOOFieldsObjectCodec
 import com.avsystem.commons.serialization._
+import monix.execution.atomic.AtomicBoolean
 
 /**
   * Like [[HasGenCodec]] but generates a codec optimized for writing and reading CBOR via [[CborOutput]] and
@@ -253,3 +254,25 @@ trait CborAdtInstances[T] {
   def cborCodec: GenObjectCodec[T] =
     metadata.setup(_.validate()).adjustCodec(stdCodec)
 }
+
+trait CborAdtPolyInstances[C[_]] {
+  def stdCodec[T: GenCodec]: GenObjectCodec[C[T]]
+  def metadata[T: GenCodec]: CborAdtMetadata[C[T]]
+  def cborCodec[T: GenCodec](validate: Boolean): GenObjectCodec[C[T]] =
+    metadata.setup { metadata =>
+      if (validate) metadata.validate()
+    }.adjustCodec(stdCodec)
+}
+
+/**
+  * Like [[HasCborCodec]] but for parameterized (generic) data types.
+  */
+abstract class HasPolyCborCodec[C[_]](implicit instances: MacroInstances[CborOptimizedCodecs, CborAdtPolyInstances[C]]) {
+  private val alreadyValidated = AtomicBoolean(false)
+
+  implicit def codec[T: GenCodec]: GenObjectCodec[C[T]] = {
+    val codec = instances(CborOptimizedCodecs, this).cborCodec(!alreadyValidated.get())
+    if (!alreadyValidated.get()) alreadyValidated.set(true)
+    codec
+  }
+}

commons-core/src/test/scala/com/avsystem/commons/serialization/cbor/CborInputOutputTest.scala

@@ -28,6 +28,16 @@ case class CustomKeysRecord(
 )
 object CustomKeysRecord extends HasCborCodec[CustomKeysRecord]
 
+@cborDiscriminator(0)
+sealed trait GenericSealedTrait[+T]
+object GenericSealedTrait extends HasPolyCborCodec[GenericSealedTrait] {
+  @cborKey(0)
+  case class Success[+T](value: T) extends GenericSealedTrait[T]
+  @cborKey(1)
+  case class Failure(message: String) extends GenericSealedTrait[Nothing]
+
+}
+
 @cborDiscriminator(0)
 sealed trait CustomKeysFlatUnion extends Product with Serializable
 object CustomKeysFlatUnion extends HasCborCodec[CustomKeysFlatUnion] {
@@ -211,6 +221,19 @@ class CborInputOutputTest extends AnyFunSuite {
   test("chunked byte string") {
     assert(CborInput.readRawCbor[Bytes](RawCbor.fromHex("5F426162426162426162FF")) == Bytes("ababab"))
   }
+
+  test("generic sealed trait") {
+    val success = GenericSealedTrait.Success[Int](234)
+    val successCbor = CborOutput.writeRawCbor[GenericSealedTrait[Int]](success)
+    val decodedSuccess = CborInput.readRawCbor[GenericSealedTrait[Int]](successCbor)
+    assert(success == decodedSuccess)
+
+    val failure = GenericSealedTrait.Failure("error")
+    val failureCbor = CborOutput.writeRawCbor[GenericSealedTrait[String]](failure)
+    val decodedFailure = CborInput.readRawCbor[GenericSealedTrait[String]](failureCbor)
+    assert(failure == decodedFailure)
+  }
+
 }
 
 class CborGenCodecRoundtripTest extends GenCodecRoundtripTest {