From 0aa1b1be9bd113bb520241a3aed821cad09e28ec Mon Sep 17 00:00:00 2001
From: ilotterytea <iltsu@alright.party>
Date: Tue, 21 Jan 2025 01:06:26 +0500
Subject: feat: set authorized on identity success

---
 .../java/kz/ilotterytea/frogartha/sessions/IdentityClient.java    | 4 ++++
 .../java/kz/ilotterytea/frogartha/sessions/SessionClient.java     | 1 +
 .../java/kz/ilotterytea/frogartha/sessions/SessionHandlers.java   | 8 +++++++-
 3 files changed, 12 insertions(+), 1 deletion(-)

(limited to 'core')

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);
+        }
     }
 }
-- 
cgit v1.2.3