codec (fix): Support Scala 3 Enum in AnyCodec to make RPC work (#3276)

Fixes #3275

airframe-codec/src/main/scala-2/wvlet/airrame/codec/AnyCodecCompat.scala

@@ -0,0 +1,18 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package wvlet.airframe.codec
+
+trait AnyCodecCompat {
+  protected def isEnum(o: Any): Boolean = false
+}

airframe-codec/src/main/scala-3/wvlet/airframe/codec/AnyCodecCompat.scala

@@ -0,0 +1,23 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package wvlet.airframe.codec
+
+trait AnyCodecCompat {
+  protected def isEnum(o: Any): Boolean = {
+    o match {
+      case _: scala.runtime.EnumValue => true
+      case _                          => false
+    }
+  }
+}

airframe-codec/src/main/scala/wvlet/airframe/codec/PrimitiveCodec.scala

@@ -923,7 +923,8 @@ object PrimitiveCodec {
   class AnyCodec(
       codecFactory: MessageCodecFactory = MessageCodecFactory.defaultFactoryForJSON,
       knownSurfaces: Seq[Surface] = Seq.empty
-  ) extends MessageCodec[Any] {
+  ) extends MessageCodec[Any]
+      with AnyCodecCompat {
 
     private val knownSurfaceTable = knownSurfaces.map(s => s.rawType -> s).toMap[Class[_], Surface]
 
@@ -991,6 +992,9 @@ object PrimitiveCodec {
           }
         case v: Throwable =>
           ThrowableCodec.pack(p, v)
+        case e if isEnum(e) =>
+          // Scala 3 EnumValue
+          StringCodec.pack(p, e.toString)
         case _ =>
           val cl = v.getClass
           knownSurfaceTable.get(cl) match {

airframe-codec/src/test/scala-3/wvlet/airframe/codec/Scala3EnumCodecTest.scala

@@ -13,6 +13,7 @@
  */
 package wvlet.airframe.codec
 
+import wvlet.airframe.codec.PrimitiveCodec.AnyCodec
 import wvlet.airframe.json.JSON
 import wvlet.airframe.msgpack.spi.MessagePack
 import wvlet.airspec.AirSpec
@@ -70,3 +71,9 @@ object Scala3EnumCodecTest extends AirSpec:
     val json  = codec.toJson(p)
     json shouldBe """{}"""
   }
+
+  test("Pack with AnyCodec") {
+    val codec   = AnyCodec.default
+    val msgpack = codec.toMsgPack(Color.Green)
+    MessageCodec.of[Color].unpack(msgpack) shouldBe Color.Green
+  }

airframe-integration-test-api/src/main/scala/wvlet/airframe/test/api/HelloRPC.scala

@@ -35,24 +35,8 @@ object HelloRPC extends RxRouterProvider {
   )
 }
 
-sealed trait Status {
-  def isDone: Boolean
-
-  def name: String = toString
-}
-
-object Status {
-  case object OK extends Status {
-    override def isDone: Boolean = true
-  }
-
-  case object NG extends Status {
-    override def isDone: Boolean = true
-  }
-
-  def all: Seq[Status] = Seq(OK, NG)
-
-  def unapply(s: String): Option[Status] = {
-    all.find(_.toString == s)
-  }
+enum Status(isDone: Boolean) {
+  def name: String = this.toString
+  case OK extends Status(isDone = true)
+  case NG extends Status(isDone = true)
 }

airframe-integration-test/src/test/scala/wvlet/airframe/test/api/HelloRPCTest.scala

@@ -17,7 +17,6 @@ import wvlet.airframe.Design
 import wvlet.airframe.http.{Http, RPCEncoding, RxRouter}
 import wvlet.airframe.http.netty.{Netty, NettyServer}
 import wvlet.airframe.test.api.HelloRPC.VariousParams
-import wvlet.airframe.test.api.Status
 import wvlet.airspec.AirSpec
 
 class HelloRPCTest extends AirSpec {

build.sbt

@@ -140,6 +140,11 @@ val scala2Only = Seq[Setting[_]](
   crossScalaVersions := uptoScala2
 )
 
+val scala3Only = Seq[Setting[_]](
+  scalaVersion       := SCALA_3,
+  crossScalaVersions := List(SCALA_3)
+)
+
 // Do not run tests concurrently to avoid JMX registration failures
 val runTestSequentially = Seq[Setting[_]](Test / parallelExecution := false)
 
@@ -225,9 +230,7 @@ lazy val jvmProjects: Seq[ProjectReference] = communityBuildProjects ++ Seq[Proj
   finagle,
   benchmark,
   sql,
-  examples,
-  integrationTestApi,
-  integrationTest
+  examples
 )
 
 // Scala.js build (Scala 2.12, 2.13, and 3.x)
@@ -316,7 +319,9 @@ lazy val projectDotty =
       rx.jvm,
       rxHtml.jvm,
       sql,
-      ulid.jvm
+      ulid.jvm,
+      integrationTestApi,
+      integrationTest
     )
 
 lazy val docs =
@@ -1007,28 +1012,30 @@ lazy val dottyTest =
     )
     .dependsOn(log.jvm, surface.jvm, di.jvm, codec.jvm)
 
+// Integration test for Scala 3
 lazy val integrationTestApi =
   project
     .in(file("airframe-integration-test-api"))
     .settings(buildSettings)
     .settings(noPublish)
     .settings(
-      name               := "airframe-integration-test-api",
-      description        := "APIs for integration test",
-      crossScalaVersions := targetScalaVersions
+      scala3Only,
+      name        := "airframe-integration-test-api",
+      description := "APIs for integration test"
     )
     .dependsOn(http.jvm)
 
+// Integration test for Scala 3
 lazy val integrationTest =
   project
     .in(file("airframe-integration-test"))
     .enablePlugins(AirframeHttpPlugin)
     .settings(buildSettings)
     .settings(noPublish)
     .settings(
+      scala3Only,
       name                := "airframe-integration-test",
       description         := "integration test project",
-      crossScalaVersions  := targetScalaVersions,
       airframeHttpClients := Seq("wvlet.airframe.test.api:rpc")
     )
     .dependsOn(integrationTestApi, netty)