feat: set authorized on identity success

3 files changed, 12 insertions, 1 deletions

diff --git a/core/src/main/java/kz/ilotterytea/frogartha/sessions/IdentityClient.java b/core/src/main/java/kz/ilotterytea/frogartha/sessions/IdentityClient.java
index 886ff7a..aa8389b 100644
--- a/core/src/main/java/kz/ilotterytea/frogartha/sessions/IdentityClient.java
+++ b/core/src/main/java/kz/ilotterytea/frogartha/sessions/IdentityClient.java
@@ -35,6 +35,10 @@ public class IdentityClient {
         return isAuthorized;
     }
 
+    public void setAuthorized(boolean authorized) {
+        isAuthorized = authorized;
+    }
+
     public boolean isProcessing() {
         return isProcessing;
     }
diff --git a/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionClient.java b/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionClient.java
index 204bea7..6cc5470 100644
--- a/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionClient.java
+++ b/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionClient.java
@@ -57,6 +57,7 @@ public class SessionClient extends WebSocketClient {
     @Override
     public void onClose(int code, String reason, boolean remote) {
         log.info("Connection closed! Reason: {} {}", code, reason);
+        game.getIdentityClient().setAuthorized(false);
     }
 
     @Override
diff --git a/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionHandlers.java b/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionHandlers.java
index 8c7487b..9b43fc8 100644
--- a/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionHandlers.java
+++ b/core/src/main/java/kz/ilotterytea/frogartha/sessions/SessionHandlers.java
@@ -2,14 +2,20 @@ package kz.ilotterytea.frogartha.sessions;
 
 import kz.ilotterytea.frogartha.FrogarthaGame;
 import kz.ilotterytea.frogartha.domain.Acknowledge;
+import kz.ilotterytea.frogartha.domain.Identity;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 public class SessionHandlers {
     private static final Logger log = LoggerFactory.getLogger(SessionHandlers.class);
-    private static final SessionClient client = FrogarthaGame.getInstance().getSessionClient();
+    private static final FrogarthaGame game = FrogarthaGame.getInstance();
+    private static final SessionClient client = game.getSessionClient();
 
     public static void handleAcknowledge(Acknowledge acknowledge) {
         log.info("The server was acknowledged: {}", acknowledge);
+
+        if (acknowledge.getPayload() instanceof Identity) {
+            game.getIdentityClient().setAuthorized(true);
+        }
     }
 }