Make LiveChannel thread safe

core/src/scalive/WebSocketMessage.scala

@@ -0,0 +1,102 @@
+package scalive
+
+import scalive.WebSocketMessage.LiveResponse
+import scalive.WebSocketMessage.Payload
+import scalive.WebSocketMessage.Payload.EventType
+import zio.Chunk
+import zio.json.*
+import zio.json.ast.Json
+
+final case class WebSocketMessage(
+  // Live session ID, auto increment defined by the client on join
+  joinRef: Option[Int],
+  // Message ID, global auto increment defined by the client on every message
+  messageRef: Int,
+  // LiveView instance id
+  topic: String,
+  eventType: String,
+  payload: WebSocketMessage.Payload)
+object WebSocketMessage:
+  given JsonCodec[WebSocketMessage] = JsonCodec[Json].transformOrFail(
+    {
+      case Json.Arr(
+            Chunk(joinRef, Json.Str(messageRef), Json.Str(topic), Json.Str(eventType), payload)
+          ) =>
+        val payloadParsed = eventType match
+          case "heartbeat" => Right(Payload.Heartbeat)
+          case "phx_join"  => payload.as[Payload.Join]
+          case "event"     => payload.as[Payload.Event]
+          case s           => Left(s"Unknown event type : $s")
+
+        payloadParsed.map(
+          WebSocketMessage(
+            joinRef.asString.map(_.toInt),
+            messageRef.toInt,
+            topic,
+            eventType,
+            _
+          )
+        )
+      case v => Left(s"Could not parse socket message ${v.toJson}")
+    },
+    m =>
+      Json.Arr(
+        m.joinRef.map(ref => Json.Str(ref.toString)).getOrElse(Json.Null),
+        Json.Str(m.messageRef.toString),
+        Json.Str(m.topic),
+        Json.Str(m.eventType),
+        m.payload.match
+          case Payload.Heartbeat => Json.Obj.empty
+          case p: Payload.Join   => p.toJsonAST.getOrElse(throw new IllegalArgumentException())
+          case p: Payload.Reply  => p.toJsonAST.getOrElse(throw new IllegalArgumentException())
+          case p: Payload.Event  => p.toJsonAST.getOrElse(throw new IllegalArgumentException())
+      )
+  )
+
+  enum Payload:
+    case Heartbeat
+    case Join(
+      url: String,
+      // params: Map[String, String],
+      session: String,
+      static: Option[String],
+      sticky: Boolean)
+    case Reply(status: String, response: LiveResponse)
+    case Event(`type`: Payload.EventType, event: String, value: Map[String, String])
+  object Payload:
+    given JsonCodec[Payload.Join]    = JsonCodec.derived
+    given JsonEncoder[Payload.Reply] = JsonEncoder.derived
+    given JsonCodec[Payload.Event]   = JsonCodec.derived
+
+    enum EventType:
+      case Click
+    object EventType:
+      given JsonCodec[EventType] = JsonCodec[String].transformOrFail(
+        {
+          case "click" => Right(Click)
+          case s       => Left(s"Unsupported event type: $s")
+        },
+        { case Click =>
+          "click"
+        }
+      )
+
+  enum LiveResponse:
+    case Empty
+    case InitDiff(rendered: scalive.Diff)
+    case Diff(diff: scalive.Diff)
+  object LiveResponse:
+    given JsonEncoder[LiveResponse] =
+      JsonEncoder[Json].contramap {
+        case Empty              => Json.Obj.empty
+        case InitDiff(rendered) =>
+          Json.Obj(
+            "liveview_version" -> Json.Str("1.1.8"),
+            "rendered"         -> rendered.toJsonAST.getOrElse(throw new IllegalArgumentException())
+          )
+        case Diff(diff) =>
+          Json.Obj(
+            "diff" -> diff.toJsonAST.getOrElse(throw new IllegalArgumentException())
+          )
+      }
+end WebSocketMessage