fix: auto refresh websocket connection

(cherry picked from commit 38cea5a)

Author: nanang-ab

src/main/java/net/accelbyte/sdk/api/lobby/ws_models/RefreshTokenRequest.java

@@ -0,0 +1,68 @@
+/*
+ * Copyright (c) 2022 AccelByte Inc. All Rights Reserved
+ * This is licensed software from AccelByte Inc, for limitations
+ * and restrictions contact your company contract manager.
+ *
+ * Code generated. DO NOT EDIT.
+ */
+
+package net.accelbyte.sdk.api.lobby.ws_models;
+
+import lombok.Builder;
+import lombok.Getter;
+import lombok.Setter;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import static net.accelbyte.sdk.core.util.Helper.generateUUID;
+import static net.accelbyte.sdk.core.util.Helper.parseWSM;
+
+@Getter
+@Setter
+public class RefreshTokenRequest {
+  private String id;
+
+  private String token;
+
+  private RefreshTokenRequest() {}
+
+  @Builder
+  // @deprecated 2022-08-29 - All args constructor may cause problems. Use builder instead.
+  @Deprecated
+  public RefreshTokenRequest(String id, String token) {
+    this.id = id;
+    this.token = token;
+  }
+
+  public static String getType() {
+    return "refreshTokenRequest";
+  }
+
+  public static RefreshTokenRequest createFromWSM(String message) {
+    RefreshTokenRequest result = new RefreshTokenRequest();
+    Map<String, String> response = parseWSM(message);
+    result.id = response.get("id") != null ? response.get("id") : null;
+    result.token = response.get("token") != null ? response.get("token") : null;
+    return result;
+  }
+
+  public String toWSM() {
+    StringBuilder stringBuilder = new StringBuilder();
+    stringBuilder.append("type: ").append(RefreshTokenRequest.getType());
+    if (id != null) {
+      stringBuilder.append("\n").append("id: ").append(id);
+    } else {
+      stringBuilder.append("\n").append("id: ").append(generateUUID());
+    }
+    stringBuilder.append("\n").append("token: ").append(token);
+    return stringBuilder.toString();
+  }
+
+  public static Map<String, String> getFieldInfo() {
+    Map<String, String> result = new HashMap<>();
+    result.put("id", "id");
+    result.put("token", "token");
+    return result;
+  }
+}

src/main/java/net/accelbyte/sdk/api/lobby/ws_models/RefreshTokenResponse.java

@@ -0,0 +1,68 @@
+/*
+ * Copyright (c) 2022 AccelByte Inc. All Rights Reserved
+ * This is licensed software from AccelByte Inc, for limitations
+ * and restrictions contact your company contract manager.
+ *
+ * Code generated. DO NOT EDIT.
+ */
+
+package net.accelbyte.sdk.api.lobby.ws_models;
+
+import lombok.Builder;
+import lombok.Getter;
+import lombok.Setter;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import static net.accelbyte.sdk.core.util.Helper.generateUUID;
+import static net.accelbyte.sdk.core.util.Helper.parseWSM;
+
+@Getter
+@Setter
+public class RefreshTokenResponse {
+  private String id;
+
+  private String code;
+
+  private RefreshTokenResponse() {}
+
+  @Builder
+  // @deprecated 2022-08-29 - All args constructor may cause problems. Use builder instead.
+  @Deprecated
+  public RefreshTokenResponse(String id, String code) {
+    this.id = id;
+    this.code = code;
+  }
+
+  public static String getType() {
+    return "refreshTokenResponse";
+  }
+
+  public static RefreshTokenResponse createFromWSM(String message) {
+    RefreshTokenResponse result = new RefreshTokenResponse();
+    Map<String, String> response = parseWSM(message);
+    result.id = response.get("id") != null ? response.get("id") : null;
+    result.code = response.get("code") != null ? response.get("code") : null;
+    return result;
+  }
+
+  public String toWSM() {
+    StringBuilder stringBuilder = new StringBuilder();
+    stringBuilder.append("type: ").append(RefreshTokenResponse.getType());
+    if (id != null) {
+      stringBuilder.append("\n").append("id: ").append(id);
+    } else {
+      stringBuilder.append("\n").append("id: ").append(generateUUID());
+    }
+    stringBuilder.append("\n").append("code: ").append(code);
+    return stringBuilder.toString();
+  }
+
+  public static Map<String, String> getFieldInfo() {
+    Map<String, String> result = new HashMap<>();
+    result.put("id", "id");
+    result.put("code", "code");
+    return result;
+  }
+}

src/main/java/net/accelbyte/sdk/core/client/OkhttpWebSocketClient.java

@@ -6,23 +6,27 @@
 
 package net.accelbyte.sdk.core.client;
 
-import java.util.concurrent.TimeUnit;
+import lombok.extern.java.Log;
+import net.accelbyte.sdk.api.lobby.ws_models.RefreshTokenRequest;
 import net.accelbyte.sdk.core.repository.ConfigRepository;
 import net.accelbyte.sdk.core.repository.TokenRepository;
-import okhttp3.*;
+import net.accelbyte.sdk.core.repository.TokenRepositoryCallback;
+import net.accelbyte.sdk.core.util.Helper;
+import okhttp3.OkHttpClient;
+import okhttp3.Request;
+import okhttp3.WebSocket;
+import okhttp3.WebSocketListener;
 
-public class OkhttpWebSocketClient {
-  private WebSocket websocket;
+import java.util.concurrent.TimeUnit;
 
-  private OkhttpWebSocketClient(WebSocket websocket) {
-    this.websocket = websocket;
-  }
+@Log
+public class OkhttpWebSocketClient extends TokenRepositoryCallback {
 
   public static OkhttpWebSocketClient create(
-      ConfigRepository configRepository,
-      TokenRepository tokenRepository,
-      WebSocketListener listener)
-      throws Exception {
+          ConfigRepository configRepository,
+          TokenRepository tokenRepository,
+          WebSocketListener listener)
+          throws Exception {
     final OkHttpClient client = new OkHttpClient.Builder().readTimeout(0, TimeUnit.SECONDS).build();
     String baseURL = configRepository.getBaseURL();
     if (baseURL == null || baseURL.isEmpty()) {
@@ -31,14 +35,20 @@ public static OkhttpWebSocketClient create(
     String url = configRepository.getBaseURL() + "/lobby/";
     String accessToken = tokenRepository.getToken();
     Request request =
-        new Request.Builder()
-            .url(url)
-            .addHeader("Authorization", String.format("Bearer %s", accessToken))
-            .build();
+            new Request.Builder()
+                    .url(url)
+                    .addHeader("Authorization", String.format("Bearer %s", accessToken))
+                    .build();
     WebSocket websocket = client.newWebSocket(request, listener);
-    OkhttpWebSocketClient websocketClient = new OkhttpWebSocketClient(websocket);
+    OkhttpWebSocketClient webSocketClient = new OkhttpWebSocketClient(websocket);
+    tokenRepository.registerTokenRepositoryCallback(webSocketClient);
+    return webSocketClient;
+  }
 
-    return websocketClient;
+  private final WebSocket websocket;
+
+  private OkhttpWebSocketClient(WebSocket websocket) {
+    this.websocket = websocket;
   }
 
   public void sendMessage(String message) {
@@ -52,4 +62,14 @@ public void close(int code, String reason) {
       this.websocket.close(code, reason);
     }
   }
+
+  @Override
+  public void onAccessTokenRefreshed(String newToken) {
+    log.info("send websocket refresh token request because token refreshed");
+    RefreshTokenRequest request = RefreshTokenRequest.builder()
+            .id(Helper.generateUUID())
+            .token(newToken)
+            .build();
+    sendMessage(request.toWSM());
+  }
 }

src/main/java/net/accelbyte/sdk/core/repository/DefaultTokenRepository.java

@@ -6,9 +6,20 @@
 
 package net.accelbyte.sdk.core.repository;
 
+import com.google.common.base.Strings;
+
+import java.util.Objects;
+import java.util.concurrent.CopyOnWriteArrayList;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+
 public class DefaultTokenRepository implements TokenRepository {
   private String accessToken = null;
 
+  private CopyOnWriteArrayList<TokenRepositoryCallback> callbacks = new CopyOnWriteArrayList<>();
+
+  private final ExecutorService executor = Executors.newCachedThreadPool();
+
   public DefaultTokenRepository() {}
 
   private static final class InstanceHolder {
@@ -31,6 +42,32 @@ public void removeToken() {
 
   @Override
   public void storeToken(String accessToken) {
+    if (Objects.isNull(accessToken) || accessToken.equals(this.accessToken)) {
+      return;
+    }
+    String oldToken = this.accessToken;
     this.accessToken = accessToken;
+
+    // only notify if previous token not empty (i.e. previously already logged in)
+    if (!Strings.isNullOrEmpty(oldToken)) {
+      notifyOnAccessTokenRefreshed(accessToken);
+    }
+  }
+
+  protected void notifyOnAccessTokenRefreshed(String newToken) {
+    for (TokenRepositoryCallback callback: callbacks) {
+      executor.execute(() -> callback.onAccessTokenRefreshed(newToken));
+    }
   }
+
+  @Override
+  public boolean registerTokenRepositoryCallback(TokenRepositoryCallback callback) {
+    return callbacks.add(callback);
+  }
+
+  @Override
+  public boolean unregisterTokenRepositoryCallback(TokenRepositoryCallback callback) {
+    return callbacks.remove(callback);
+  }
+
 }

src/main/java/net/accelbyte/sdk/core/repository/TokenRepository.java

@@ -12,4 +12,12 @@ public interface TokenRepository {
   String getToken() throws Exception;
 
   void removeToken() throws Exception;
+
+  default boolean registerTokenRepositoryCallback(TokenRepositoryCallback callback) {
+    return true;
+  }
+
+  default boolean unregisterTokenRepositoryCallback(TokenRepositoryCallback callback) {
+    return true;
+  }
 }

src/main/java/net/accelbyte/sdk/core/repository/TokenRepositoryCallback.java

@@ -0,0 +1,7 @@
+package net.accelbyte.sdk.core.repository;
+
+abstract public class TokenRepositoryCallback {
+
+    public void onAccessTokenRefreshed(String newToken) {}
+
+}