add-scala3-enum-put-get-derivation

Author: drewfeelsblue

modules/core/src/main/scala-3/doobie/util/GetPlatform.scala

@@ -4,4 +4,23 @@
 
 package doobie.util
 
-trait GetPlatform {}
+import scala.deriving.Mirror
+import scala.compiletime.constValue
+import scala.reflect.Enum
+
+trait GetPlatform {
+  private def of[A](name: String, cases: List[A], labels: List[String]): Get[A] =
+    Get[String].temap { caseName =>
+      labels.indexOf(caseName) match {
+        case -1 => Left(s"enum $name does not contain case: $caseName")
+        case i  => Right(cases(i))
+      }
+    }
+
+  inline final def deriveEnumString[A <: Enum](using mirror: Mirror.SumOf[A]): Get[A] =
+    of(
+      constValue[mirror.MirroredLabel],
+      summonSingletonCases[mirror.MirroredElemTypes, A](constValue[mirror.MirroredLabel]),
+      summonLabels[mirror.MirroredElemLabels]
+    )
+}

modules/core/src/main/scala-3/doobie/util/PutPlatform.scala

@@ -4,4 +4,14 @@
 
 package doobie.util
 
-trait PutPlatform
+import scala.compiletime.constValue
+import scala.deriving.Mirror
+import scala.reflect.Enum
+
+trait PutPlatform {
+  inline final def deriveEnumString[A <: Enum](using mirror: Mirror.SumOf[A]): Put[A] =
+    val _ = summonSingletonCases[mirror.MirroredElemTypes, A](constValue[mirror.MirroredLabel])
+    val labels = summonLabels[mirror.MirroredElemLabels]
+
+    Put[String].contramap(a => labels(mirror.ordinal(a)))
+}

modules/core/src/main/scala-3/doobie/util/package.scala

@@ -0,0 +1,22 @@
+// Copyright (c) 2013-2020 Rob Norris and Contributors
+// This software is licensed under the MIT License (MIT).
+// For more information see LICENSE or https://opensource.org/licenses/MIT
+
+package doobie.util
+
+import scala.compiletime.{constValue, erasedValue, error, summonInline}
+import scala.deriving.Mirror
+
+private[util] inline final def summonLabels[T <: Tuple]: List[String] =
+  inline erasedValue[T] match
+    case _: EmptyTuple => Nil
+    case _: (t *: ts)  => constValue[t].asInstanceOf[String] :: summonLabels[ts]
+
+private[util] inline final def summonSingletonCases[T <: Tuple, A](inline typeName: String): List[A] =
+  inline erasedValue[T] match
+    case _: EmptyTuple => Nil
+    case _: (h *: t) =>
+      inline summonInline[Mirror.Of[h]] match
+        case m: Mirror.Singleton => m.fromProduct(EmptyTuple).asInstanceOf[A] :: summonSingletonCases[t, A](typeName)
+        case m: Mirror =>
+          error("Enum " + typeName + " contains non singleton case " + constValue[m.MirroredLabel])

modules/core/src/test/scala-2/doobie/util/GetSuitePlatform.scala

@@ -0,0 +1,7 @@
+// Copyright (c) 2013-2020 Rob Norris and Contributors
+// This software is licensed under the MIT License (MIT).
+// For more information see LICENSE or https://opensource.org/licenses/MIT
+
+package doobie.util
+
+trait GetSuitePlatform {}

modules/core/src/test/scala-2/doobie/util/PutSuitePlatform.scala

@@ -0,0 +1,7 @@
+// Copyright (c) 2013-2020 Rob Norris and Contributors
+// This software is licensed under the MIT License (MIT).
+// For more information see LICENSE or https://opensource.org/licenses/MIT
+
+package doobie.util
+
+trait PutSuitePlatform {}

modules/core/src/test/scala-3/doobie/util/GetSuitePlatform.scala

@@ -0,0 +1,40 @@
+// Copyright (c) 2013-2020 Rob Norris and Contributors
+// This software is licensed under the MIT License (MIT).
+// For more information see LICENSE or https://opensource.org/licenses/MIT
+
+package doobie.util
+
+import Predef.augmentString
+
+trait GetSuitePlatform { self: munit.FunSuite =>
+
+  enum EnumContainsNonSinletonCase {
+    case One
+    case Two(i: Int)
+  }
+
+  enum EnumWithOnlySingletonCases {
+    case One, Two, Three
+  }
+
+  sealed trait NotEnum
+
+  case object NotEnum1 extends NotEnum
+
+  test("Get should not be derived for enum with non singleton case") {
+    val compileError = compileErrors("Get.deriveEnumString[EnumContainsNonSinletonCase]")
+
+    assert(compileError.contains("Enum EnumContainsNonSinletonCase contains non singleton case Two"))
+  }
+
+  test("Get should be derived for enum with only singleton cases") {
+    Get.deriveEnumString[EnumWithOnlySingletonCases]
+  }
+
+  test("Get should not be derived for sealed trait") {
+    val compileError = compileErrors("Get.deriveEnumString[NotEnum]")
+
+    assert(compileError.contains(
+      "Type argument GetSuitePlatform.this.NotEnum does not conform to upper bound scala.reflect.Enum"))
+  }
+}

modules/core/src/test/scala-3/doobie/util/PutSuitePlatform.scala

@@ -0,0 +1,40 @@
+// Copyright (c) 2013-2020 Rob Norris and Contributors
+// This software is licensed under the MIT License (MIT).
+// For more information see LICENSE or https://opensource.org/licenses/MIT
+
+package doobie.util
+
+import Predef.augmentString
+
+trait PutSuitePlatform { self: munit.FunSuite =>
+
+  enum EnumContainsNonSinletonCase {
+    case One
+    case Two(i: Int)
+  }
+
+  enum EnumWithOnlySingletonCases {
+    case One, Two, Three
+  }
+
+  sealed trait NotEnum
+
+  case object NotEnum1 extends NotEnum
+
+  test("Put should not be derived for enum with non singleton case") {
+    val compileError = compileErrors("Put.deriveEnumString[EnumContainsNonSinletonCase]")
+
+    assert(compileError.contains("Enum EnumContainsNonSinletonCase contains non singleton case Two"))
+  }
+
+  test("Put should be derived for enum with only singleton cases") {
+    Put.deriveEnumString[EnumWithOnlySingletonCases]
+  }
+
+  test("Put should not be derived for sealed trait") {
+    val compileError = compileErrors("Put.deriveEnumString[NotEnum]")
+
+    assert(compileError.contains(
+      "Type argument PutSuitePlatform.this.NotEnum does not conform to upper bound scala.reflect.Enum"))
+  }
+}

modules/core/src/test/scala/doobie/util/GetSuite.scala

@@ -9,7 +9,7 @@ import doobie.enumerated.JdbcType
 import doobie.testutils.VoidExtensions
 import doobie.util.transactor.Transactor
 
-class GetSuite extends munit.CatsEffectSuite {
+class GetSuite extends munit.CatsEffectSuite with GetSuitePlatform {
 
   case class X(x: Int)
   case class Q(x: String)

modules/core/src/test/scala/doobie/util/PutSuite.scala

@@ -8,7 +8,7 @@ import cats.effect.IO
 import doobie.testutils.VoidExtensions
 import doobie.util.transactor.Transactor
 
-class PutSuite extends munit.FunSuite {
+class PutSuite extends munit.FunSuite with PutSuitePlatform {
   case class X(x: Int)
 
   case class Q(x: String)