Add session id to websocket connection to get authenticated WS as well as optional customizable type for disconnecting websocket client in various ways.

kohlerpop1 · kohlerpop1 · commit 7e5909979320 · 2025-05-21T18:43:14.000-04:00
diff --git a/API/src/main/java/io/github/jwdeveloper/tiktok/live/LiveClient.java b/API/src/main/java/io/github/jwdeveloper/tiktok/live/LiveClient.java
@@ -51,9 +51,17 @@ public interface LiveClient {
 
     /**
      * Disconnects the connection.
+     * @param type
+     * <p>0 - Normal - Initiates disconnection and returns
+     * <p>1 - Disconnects blocking and returns after closure
+     * <p>2 - Disconnects and kills connection to websocket
+     * <p>Default {@link #disconnect()} is 0
      */
-    void disconnect();
+    void disconnect(int type);
 
+    default void disconnect() {
+        disconnect(0);
+    }
 
     /**
      * Use to manually invoke event
diff --git a/API/src/main/java/io/github/jwdeveloper/tiktok/websocket/LiveSocketClient.java b/API/src/main/java/io/github/jwdeveloper/tiktok/websocket/LiveSocketClient.java
@@ -27,6 +27,6 @@
 
 public interface LiveSocketClient {
     void start(LiveConnectionData.Response webcastResponse, LiveClient tikTokLiveClient);
-    void stop();
+    void stop(int type);
     boolean isConnected();
 }
diff --git a/Client/src/main/java/io/github/jwdeveloper/tiktok/TikTokLiveClient.java b/Client/src/main/java/io/github/jwdeveloper/tiktok/TikTokLiveClient.java
@@ -153,9 +153,9 @@ public void tryConnect() {
         tikTokEventHandler.publish(this, new TikTokRoomInfoEvent(roomInfo));
     }
 
-    public void disconnect() {
+    public void disconnect(int type) {
         if (webSocketClient.isConnected())
-            webSocketClient.stop();
+            webSocketClient.stop(type);
 		if (!roomInfo.hasConnectionState(ConnectionState.DISCONNECTED))
 			setState(ConnectionState.DISCONNECTED);
 	}
diff --git a/Client/src/main/java/io/github/jwdeveloper/tiktok/TikTokLiveHttpClient.java b/Client/src/main/java/io/github/jwdeveloper/tiktok/TikTokLiveHttpClient.java
@@ -199,6 +199,8 @@ protected ActionResult<HttpResponse<byte[]>> getByteResponse(String room_id) {
             .withParam("client", "ttlive-java")
             .withParam("room_id", room_id);
 
+        if (clientSettings.getSessionId() != null) // Allows receiving of all comments and Subscribe Events
+            builder.withParam("session_id", clientSettings.getSessionId());
         if (clientSettings.getApiKey() != null)
             builder.withParam("apiKey", clientSettings.getApiKey());
 
diff --git a/Client/src/main/java/io/github/jwdeveloper/tiktok/websocket/TikTokWebSocketClient.java b/Client/src/main/java/io/github/jwdeveloper/tiktok/websocket/TikTokWebSocketClient.java
@@ -28,6 +28,7 @@
 import io.github.jwdeveloper.tiktok.exceptions.*;
 import io.github.jwdeveloper.tiktok.live.*;
 import org.java_websocket.client.WebSocketClient;
+import org.java_websocket.framing.CloseFrame;
 
 import javax.net.ssl.*;
 import java.net.Proxy;
@@ -56,18 +57,18 @@ public TikTokWebSocketClient(
     @Override
     public void start(LiveConnectionData.Response connectionData, LiveClient liveClient) {
         if (isConnected())
-			stop();
+			stop(0);
 
         messageHandler.handle(liveClient, connectionData.getWebcastResponse());
 
         var headers = new HashMap<>(clientSettings.getHttpSettings().getHeaders());
         headers.put("Cookie", connectionData.getWebsocketCookies());
         webSocketClient = new TikTokWebSocketListener(connectionData.getWebsocketUrl(),
-                headers,
-                clientSettings.getHttpSettings().getTimeout().toMillisPart(),
-                messageHandler,
-                tikTokEventHandler,
-                liveClient);
+            headers,
+            clientSettings.getHttpSettings().getTimeout().toMillisPart(),
+            messageHandler,
+            tikTokEventHandler,
+            liveClient);
 
         ProxyClientSettings proxyClientSettings = clientSettings.getHttpSettings().getProxyClientSettings();
         if (proxyClientSettings.isEnabled() && proxyClientSettings.isAllowWebsocket())
@@ -128,9 +129,19 @@ public boolean tryProxyConnection(ProxyClientSettings proxySettings, ProxyData p
         }
     }
 
-    public void stop() {
+    public void stop(int type) {
         if (isConnected()) {
-            webSocketClient.close();
+            switch (type) {
+				case 1 -> {
+					try {
+						webSocketClient.closeBlocking();
+					} catch (InterruptedException e) {
+                        throw new TikTokLiveException("Failed to stop the websocket");
+                    }
+				}
+                case 2 -> webSocketClient.closeConnection(CloseFrame.NORMAL, "");
+                default -> webSocketClient.close();
+            }
             heartbeatTask.stop();
         }
         webSocketClient = null;
diff --git a/Client/src/main/java/io/github/jwdeveloper/tiktok/websocket/TikTokWebSocketOfflineClient.java b/Client/src/main/java/io/github/jwdeveloper/tiktok/websocket/TikTokWebSocketOfflineClient.java
@@ -44,7 +44,7 @@ public void start(LiveConnectionData.Response webcastResponse, LiveClient tikTok
     }
 
     @Override
-    public void stop() {
+    public void stop(int type) {
 		if (liveClient != null)
 			handler.publish(liveClient, new TikTokDisconnectedEvent("Stopping"));
 	}

Original file line number	Diff line number	Diff line change
`@@ -27,6 +27,6 @@`
`27`	`27`
`28`	`28`	`public interface LiveSocketClient {`
`29`	`29`	`void start(LiveConnectionData.Response webcastResponse, LiveClient tikTokLiveClient);`
`30`		`- void stop();`
	`30`	`+ void stop(int type);`
`31`	`31`	`boolean isConnected();`
`32`	`32`	`}`
Original file line number	Diff line number	Diff line change
`@@ -153,9 +153,9 @@ public void tryConnect() {`
`153`	`153`	`tikTokEventHandler.publish(this, new TikTokRoomInfoEvent(roomInfo));`
`154`	`154`	`}`
`155`	`155`
`156`		`- public void disconnect() {`
	`156`	`+ public void disconnect(int type) {`
`157`	`157`	`if (webSocketClient.isConnected())`
`158`		`- webSocketClient.stop();`
	`158`	`+ webSocketClient.stop(type);`
`159`	`159`	`if (!roomInfo.hasConnectionState(ConnectionState.DISCONNECTED))`
`160`	`160`	`setState(ConnectionState.DISCONNECTED);`
`161`	`161`	`}`
Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,7 @@ public void start(LiveConnectionData.Response webcastResponse, LiveClient tikTok`
`44`	`44`	`}`
`45`	`45`
`46`	`46`	`@Override`
`47`		`- public void stop() {`
	`47`	`+ public void stop(int type) {`
`48`	`48`	`if (liveClient != null)`
`49`	`49`	`handler.publish(liveClient, new TikTokDisconnectedEvent("Stopping"));`
`50`	`50`	`}`