From 735a41b002d3548454cdf3c7ffb7e1fc378f6490 Mon Sep 17 00:00:00 2001
From: Angelika Serwa <angelika.serwa@swmansion.com>
Date: Tue, 18 Jun 2024 14:56:57 +0200
Subject: [PATCH] [WIP] Fix ios example

---
 Sources/FishjamClient/FishjamClientInternal.swift | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/Sources/FishjamClient/FishjamClientInternal.swift b/Sources/FishjamClient/FishjamClientInternal.swift
index 41d7cf9..68fffc8 100644
--- a/Sources/FishjamClient/FishjamClientInternal.swift
+++ b/Sources/FishjamClient/FishjamClientInternal.swift
@@ -7,6 +7,7 @@ internal class FishjamClientInternal: MembraneRTCDelegate, WebSocketDelegate {
     private var listener: FishjamClientListener
     private var websocketFactory: (String) -> FishjamWebsocket
     var webrtcClient: FishjamMembraneRTC?
+    private var isAuthenticated = false
 
     public init(listener: FishjamClientListener, websocketFactory: @escaping (String) -> FishjamWebsocket) {
         self.listener = listener
@@ -23,10 +24,12 @@ internal class FishjamClientInternal: MembraneRTCDelegate, WebSocketDelegate {
 
     func leave() {
         webrtcClient?.disconnect()
+        isAuthenticated = false
     }
 
     func cleanUp() {
         webrtcClient?.disconnect()
+        isAuthenticated = false
         webSocket?.disconnect()
         webSocket = nil
         onDisconnected()
@@ -77,6 +80,7 @@ internal class FishjamClientInternal: MembraneRTCDelegate, WebSocketDelegate {
         do {
             let peerMessage = try Fishjam_PeerMessage(serializedData: data)
             if case .authenticated(_) = peerMessage.content {
+                isAuthenticated = true
                 onAuthSuccess()
             } else if case .mediaEvent(_) = peerMessage.content {
                 receiveEvent(event: peerMessage.mediaEvent.data)
@@ -114,6 +118,10 @@ internal class FishjamClientInternal: MembraneRTCDelegate, WebSocketDelegate {
     }
 
     func onSendMediaEvent(event: SerializedMediaEvent) {
+        if (!isAuthenticated) {
+          print("Tried to send media event: \(event) before authentication")
+          return
+        }
         let mediaEvent =
             Fishjam_PeerMessage.with({
                 $0.mediaEvent = Fishjam_PeerMessage.MediaEvent.with({