Compare commits

..

1 Commits

Author SHA1 Message Date
kohlerpop1
ffde21afac Only copy roomInfo if its valid and null check safety! 2025-09-17 18:54:14 -04:00
20 changed files with 71 additions and 134 deletions

View File

@@ -5,7 +5,7 @@
<parent> <parent>
<artifactId>TikTokLiveJava</artifactId> <artifactId>TikTokLiveJava</artifactId>
<groupId>io.github.jwdeveloper.tiktok</groupId> <groupId>io.github.jwdeveloper.tiktok</groupId>
<version>1.11.5-Release</version> <version>1.11.0-Release</version>
</parent> </parent>
<modelVersion>4.0.0</modelVersion> <modelVersion>4.0.0</modelVersion>
<artifactId>API</artifactId> <artifactId>API</artifactId>

View File

@@ -66,5 +66,5 @@ public interface LiveHttpClient
LiveConnectionData.Response fetchLiveConnectionData(LiveConnectionData.Request request); LiveConnectionData.Response fetchLiveConnectionData(LiveConnectionData.Request request);
boolean sendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc); boolean sendChat(LiveRoomInfo roomInfo, String content);
} }

View File

@@ -102,12 +102,4 @@ public interface LiveClient {
* <p>We cannot fix this as it is a TikTok issue, not a library issue. * <p>We cannot fix this as it is a TikTok issue, not a library issue.
*/ */
boolean sendChat(String content); boolean sendChat(String content);
/**
* Send a chat message to the connected room
* @return true if successful, otherwise false
* @apiNote This is known to return true on some sessionIds despite failing!
* <p>We cannot fix this as it is a TikTok issue, not a library issue.
*/
boolean sendChat(String content, String sessionId, String ttTargetIdc);
} }

View File

@@ -62,11 +62,12 @@ message Text {
int32 type = 1; int32 type = 1;
TextFormat format = 2; TextFormat format = 2;
string stringValue = 11; string stringValue = 11;
TextPieceUser userValue = 21; oneof textPieceType
TextPieceGift giftValue = 22; {
TextPieceHeart heartValue = 23; TextPieceUser userValue = 21;
TextPieceGift giftValue = 22;
}
TextPiecePatternRef patternRefValue = 24; TextPiecePatternRef patternRefValue = 24;
TextPieceImage imageValue = 25;
} }
message TextFormat { message TextFormat {
@@ -82,7 +83,7 @@ message Text {
message TextPieceGift { message TextPieceGift {
int32 giftId = 1; int32 giftId = 1;
TextPiecePatternRef nameRef = 2; PatternRef nameRef = 2;
ShowType showType = 3; // Enum ShowType showType = 3; // Enum
int64 colorId = 4; int64 colorId = 4;
} }
@@ -97,19 +98,16 @@ message Text {
bool withColon = 2; bool withColon = 2;
} }
message PatternRef {
string key = 1;
string default_pattern = 2;
}
enum ShowType { enum ShowType {
SHOW_TYPE_NORMAL = 0; SHOW_TYPE_NORMAL = 0;
SHOW_TYPE_FADE_IN_OUT = 1; SHOW_TYPE_FADE_IN_OUT = 1;
} }
message TextPieceHeart {
string color = 1;
}
message TextPieceImage {
Image image_model = 1;
}
} }
// @Image // @Image
@@ -153,7 +151,7 @@ message BadgeStruct {
bool is_customized = 24; bool is_customized = 24;
message CombineBadge { message CombineBadge {
BadgeDisplayType badge_display_type = 1; int32 badge_display_type = 1;
Image icon = 2; Image icon = 2;
TextBadge text = 3; TextBadge text = 3;
string str = 4; string str = 4;

View File

@@ -5,7 +5,7 @@
<parent> <parent>
<artifactId>TikTokLiveJava</artifactId> <artifactId>TikTokLiveJava</artifactId>
<groupId>io.github.jwdeveloper.tiktok</groupId> <groupId>io.github.jwdeveloper.tiktok</groupId>
<version>1.11.5-Release</version> <version>1.11.0-Release</version>
</parent> </parent>
<modelVersion>4.0.0</modelVersion> <modelVersion>4.0.0</modelVersion>

View File

@@ -23,7 +23,6 @@
package io.github.jwdeveloper.tiktok; package io.github.jwdeveloper.tiktok;
import com.google.protobuf.ByteString; import com.google.protobuf.ByteString;
import io.github.jwdeveloper.tiktok.common.AsyncHandler;
import io.github.jwdeveloper.tiktok.data.events.*; import io.github.jwdeveloper.tiktok.data.events.*;
import io.github.jwdeveloper.tiktok.data.events.common.TikTokEvent; import io.github.jwdeveloper.tiktok.data.events.common.TikTokEvent;
import io.github.jwdeveloper.tiktok.data.events.control.*; import io.github.jwdeveloper.tiktok.data.events.control.*;
@@ -40,7 +39,7 @@ import io.github.jwdeveloper.tiktok.websocket.*;
import lombok.Getter; import lombok.Getter;
import java.util.Base64; import java.util.Base64;
import java.util.concurrent.*; import java.util.concurrent.CompletableFuture;
import java.util.function.Consumer; import java.util.function.Consumer;
import java.util.logging.Logger; import java.util.logging.Logger;
@@ -90,11 +89,12 @@ public class TikTokLiveClient implements LiveClient
tikTokEventHandler.publish(this, new TikTokDisconnectedEvent("Exception: " + e.getMessage())); tikTokEventHandler.publish(this, new TikTokDisconnectedEvent("Exception: " + e.getMessage()));
if (e instanceof TikTokLiveOfflineHostException && clientSettings.isRetryOnConnectionFailure()) { if (e instanceof TikTokLiveOfflineHostException && clientSettings.isRetryOnConnectionFailure()) {
AsyncHandler.getReconnectScheduler().schedule(() -> { try {
logger.info("Reconnecting"); Thread.sleep(clientSettings.getRetryConnectionTimeout().toMillis());
tikTokEventHandler.publish(this, new TikTokReconnectingEvent()); } catch (Exception ignored) {}
this.connect(); logger.info("Reconnecting");
}, clientSettings.getRetryConnectionTimeout().toMillis(), TimeUnit.MILLISECONDS); tikTokEventHandler.publish(this, new TikTokReconnectingEvent());
this.connect();
} }
throw e; throw e;
} catch (Exception e) { } catch (Exception e) {
@@ -199,12 +199,7 @@ public class TikTokLiveClient implements LiveClient
@Override @Override
public boolean sendChat(String content) { public boolean sendChat(String content) {
return sendChat(content, clientSettings.getSessionId(), clientSettings.getTtTargetIdc()); return httpClient.sendChat(roomInfo, content);
}
@Override
public boolean sendChat(String content, String sessionId, String ttTargetIdc) {
return httpClient.sendChat(roomInfo, content, sessionId, ttTargetIdc);
} }
public void connectAsync(Consumer<LiveClient> onConnection) { public void connectAsync(Consumer<LiveClient> onConnection) {

View File

@@ -133,7 +133,7 @@ public class TikTokLiveClientBuilder implements LiveClientBuilder {
//networking //networking
dependance.registerSingleton(HttpClientFactory.class); dependance.registerSingleton(HttpClientFactory.class);
dependance.registerSingleton(WebSocketHeartbeatTask.class); dependance.registerSingleton(WebSocketHeartbeatTask.class); // True global singleton - Static objects are located to serve as global
if (clientSettings.isOffline()) { if (clientSettings.isOffline()) {
dependance.registerSingleton(LiveSocketClient.class, TikTokWebSocketOfflineClient.class); dependance.registerSingleton(LiveSocketClient.class, TikTokWebSocketOfflineClient.class);
dependance.registerSingleton(LiveHttpClient.class, TikTokLiveHttpOfflineClient.class); dependance.registerSingleton(LiveHttpClient.class, TikTokLiveHttpOfflineClient.class);

View File

@@ -182,23 +182,23 @@ public class TikTokLiveHttpClient implements LiveHttpClient
} }
@Override @Override
public boolean sendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc) { public boolean sendChat(LiveRoomInfo roomInfo, String content) {
var proxyClientSettings = clientSettings.getHttpSettings().getProxyClientSettings(); var proxyClientSettings = clientSettings.getHttpSettings().getProxyClientSettings();
if (proxyClientSettings.isEnabled()) { if (proxyClientSettings.isEnabled()) {
while (proxyClientSettings.hasNext()) { while (proxyClientSettings.hasNext()) {
try { try {
return requestSendChat(roomInfo, content, sessionId, ttTargetIdc); return requestSendChat(roomInfo, content);
} catch (TikTokProxyRequestException ignored) {} } catch (TikTokProxyRequestException ignored) {}
} }
} }
return requestSendChat(roomInfo, content, sessionId, ttTargetIdc); return requestSendChat(roomInfo, content);
} }
public boolean requestSendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc) { public boolean requestSendChat(LiveRoomInfo roomInfo, String content) {
JsonObject body = new JsonObject(); JsonObject body = new JsonObject();
body.addProperty("content", content); body.addProperty("content", content);
body.addProperty("sessionId", sessionId); body.addProperty("sessionId", clientSettings.getSessionId());
body.addProperty("ttTargetIdc", ttTargetIdc); body.addProperty("ttTargetIdc", clientSettings.getTtTargetIdc());
body.addProperty("roomId", roomInfo.getRoomId()); body.addProperty("roomId", roomInfo.getRoomId());
HttpClientBuilder builder = httpFactory.client(clientSettings.isUseEulerstreamEnterprise() ? TIKTOK_CHAT_ENTERPRISE_URL : TIKTOK_CHAT_URL) HttpClientBuilder builder = httpFactory.client(clientSettings.isUseEulerstreamEnterprise() ? TIKTOK_CHAT_ENTERPRISE_URL : TIKTOK_CHAT_URL)
.withHeader("Content-Type", "application/json"); .withHeader("Content-Type", "application/json");

View File

@@ -64,7 +64,7 @@ public class TikTokLiveHttpOfflineClient implements LiveHttpClient {
} }
@Override @Override
public boolean sendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc) { public boolean sendChat(LiveRoomInfo roomInfo, String content) {
// DO NOTHING // DO NOTHING
return false; return false;
} }

View File

@@ -70,6 +70,6 @@ public class TikTokRoomInfo implements LiveRoomInfo
this.hostName = roomInfo.getHostName(); this.hostName = roomInfo.getHostName();
this.title = roomInfo.getTitle(); this.title = roomInfo.getTitle();
this.language = roomInfo.getLanguage(); this.language = roomInfo.getLanguage();
// this.connectionState = roomInfo.getConnectionState(); // This should not be copied - Controlled elsewhere! this.connectionState = roomInfo.getConnectionState();
} }
} }

View File

@@ -1,22 +0,0 @@
package io.github.jwdeveloper.tiktok.common;
import lombok.Getter;
import java.util.concurrent.*;
public class AsyncHandler
{
@Getter
private static final ScheduledExecutorService heartBeatScheduler = Executors.newScheduledThreadPool(1, r -> {
Thread t = new Thread(r, "heartbeat-pool");
t.setDaemon(true);
return t;
});
@Getter
private static final ScheduledExecutorService reconnectScheduler = Executors.newScheduledThreadPool(0, r -> {
Thread t = new Thread(r, "reconnect-pool");
t.setDaemon(true);
return t;
});
}

View File

@@ -66,6 +66,7 @@ public class LiveUserDataMapper
roomInfo.setTitle(liveRoom.get("title").getAsString()); roomInfo.setTitle(liveRoom.get("title").getAsString());
roomInfo.setStartTime(liveRoom.get("startTime").getAsLong()); roomInfo.setStartTime(liveRoom.get("startTime").getAsLong());
roomInfo.setTitle(liveRoom.get("title").getAsString());
roomInfo.setViewersCount(liveRoom.getAsJsonObject("liveRoomStats").get("userCount").getAsInt()); roomInfo.setViewersCount(liveRoom.getAsJsonObject("liveRoomStats").get("userCount").getAsInt());
roomInfo.setTotalViewersCount(liveRoom.getAsJsonObject("liveRoomStats").get("enterCount").getAsInt()); roomInfo.setTotalViewersCount(liveRoom.getAsJsonObject("liveRoomStats").get("enterCount").getAsInt());
roomInfo.setAgeRestricted(jsonObject.get("statusCode").getAsInt() == TikTokLiveHttpClient.TIKTOK_AGE_RESTRICTED_CODE); roomInfo.setAgeRestricted(jsonObject.get("statusCode").getAsInt() == TikTokLiveHttpClient.TIKTOK_AGE_RESTRICTED_CODE);
@@ -91,51 +92,9 @@ public class LiveUserDataMapper
roomInfo.setHostName(foundUser.getName()); roomInfo.setHostName(foundUser.getName());
return new LiveUserData.Response(json, statusEnum, roomInfo); return new LiveUserData.Response(json, statusEnum, roomInfo);
} catch (JsonSyntaxException | IllegalStateException | NullPointerException e) { } catch (JsonSyntaxException | IllegalStateException e) {
logger.warning("Malformed Json: '"+json+"' - Error Message: "+e.getMessage()); logger.warning("Malformed Json: '"+json+"' - Error Message: "+e.getMessage());
return new LiveUserData.Response(json, LiveUserData.UserStatus.NotFound, null); return new LiveUserData.Response(json, LiveUserData.UserStatus.NotFound, null);
} }
} }
public static LiveUserData.Response mapEulerstream(JsonObject jsonObject, Logger logger) {
try {
JsonObject roomInfoJson = jsonObject.getAsJsonObject("roomInfo");
JsonObject userJson = jsonObject.getAsJsonObject("user");
var roomId = roomInfoJson.get("id").getAsString();
var status = roomInfoJson.get("status").getAsInt();
TikTokRoomInfo roomInfo = new TikTokRoomInfo();
roomInfo.setRoomId(roomId);
roomInfo.setTitle(roomInfoJson.get("title").getAsString());
roomInfo.setStartTime(roomInfoJson.get("startTime").getAsLong());
roomInfo.setViewersCount(roomInfoJson.get("currentViewers").getAsInt());
roomInfo.setTotalViewersCount(roomInfoJson.get("totalViewers").getAsInt());
var statusEnum = switch (status) {
case 2 -> LiveUserData.UserStatus.Live;
case 3 -> LiveUserData.UserStatus.LivePaused;
case 4 -> LiveUserData.UserStatus.Offline;
default -> LiveUserData.UserStatus.NotFound;
};
User foundUser = new User(
Long.parseLong(userJson.get("numericUid").getAsString()),
userJson.get("uniqueId").getAsString(),
userJson.get("nickname").getAsString(),
userJson.get("signature").getAsString(),
new Picture(userJson.get("avatarUrl").getAsString()),
userJson.get("following").getAsLong(),
userJson.get("followers").getAsLong(),
List.of());
roomInfo.setHost(foundUser);
roomInfo.setHostName(foundUser.getName());
return new LiveUserData.Response(jsonObject.toString(), statusEnum, roomInfo);
} catch (JsonSyntaxException | IllegalStateException | NullPointerException e) {
logger.warning("Malformed Json: '"+jsonObject.toString()+"' - Error Message: "+e.getMessage());
return new LiveUserData.Response(jsonObject.toString(), LiveUserData.UserStatus.NotFound, null);
}
}
} }

View File

@@ -142,7 +142,7 @@ public class TikTokWebSocketClient implements LiveSocketClient {
case DISCONNECT -> webSocketClient.closeConnection(CloseFrame.NORMAL, ""); case DISCONNECT -> webSocketClient.closeConnection(CloseFrame.NORMAL, "");
default -> webSocketClient.close(); default -> webSocketClient.close();
} }
heartbeatTask.stop(); heartbeatTask.stop(webSocketClient);
} }
webSocketClient = null; webSocketClient = null;
} }

View File

@@ -22,7 +22,6 @@
*/ */
package io.github.jwdeveloper.tiktok.websocket; package io.github.jwdeveloper.tiktok.websocket;
import io.github.jwdeveloper.tiktok.common.AsyncHandler;
import org.java_websocket.WebSocket; import org.java_websocket.WebSocket;
import java.util.*; import java.util.*;
@@ -30,30 +29,47 @@ import java.util.concurrent.*;
public class WebSocketHeartbeatTask public class WebSocketHeartbeatTask
{ {
private ScheduledFuture<?> task; // Single shared pool for all heartbeat tasks
private Long commTime; private static final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1, r -> {
Thread t = new Thread(r, "heartbeat-pool");
t.setDaemon(true);
return t;
});
private static final Map<WebSocket, ScheduledFuture<?>> tasks = new ConcurrentHashMap<>();
private static final Map<WebSocket, Long> commTime = new ConcurrentHashMap<>();
private final static byte[] heartbeatBytes = Base64.getDecoder().decode("MgJwYjoCaGI="); // Used to be '3A026862' aka ':\x02hb', now is '2\x02pb:\x02hb'. private final byte[] heartbeatBytes = Base64.getDecoder().decode("MgJwYjoCaGI="); // Used to be '3A026862' aka ':\x02hb', now is '2\x02pb:\x02hb'.
public void run(WebSocket webSocket, long pingTaskTime) { public void run(WebSocket webSocket, long pingTaskTime) {
stop(); // remove existing task if any stop(webSocket); // remove existing task if any
task = AsyncHandler.getHeartBeatScheduler().scheduleAtFixedRate(() -> { tasks.put(webSocket, scheduler.scheduleAtFixedRate(() -> {
try { try {
if (webSocket.isOpen()) { if (webSocket.isOpen()) {
webSocket.send(heartbeatBytes); webSocket.send(heartbeatBytes);
commTime = System.currentTimeMillis(); commTime.put(webSocket, System.currentTimeMillis());
} else if (commTime != null && System.currentTimeMillis() - commTime >= 60_000) // Stop if disconnected longer than 60s } else {
stop(); Long time = commTime.get(webSocket);
if (time != null && System.currentTimeMillis() - time >= 60_000) // Stop if disconnected longer than 60s
stop(webSocket);
}
} catch (Exception e) { } catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
stop(); stop(webSocket);
} }
}, 0, pingTaskTime, TimeUnit.MILLISECONDS); }, 0, pingTaskTime, TimeUnit.MILLISECONDS));
} }
public void stop() { public void stop(WebSocket webSocket) {
if (task != null) ScheduledFuture<?> future = tasks.remove(webSocket);
task.cancel(true); if (future != null)
future.cancel(true);
commTime.remove(webSocket);
}
public void shutdown() {
tasks.values().forEach(f -> f.cancel(true));
commTime.clear();
scheduler.shutdownNow();
} }
} }

View File

@@ -56,7 +56,7 @@ public class TikTokWebSocketEulerListener extends TikTokWebSocketListener
switch (oMsg.get("type").getAsString()) { // Should only receive these 2 types ever switch (oMsg.get("type").getAsString()) { // Should only receive these 2 types ever
case "workerInfo" -> liveClient.getLogger().info(oMsg.toString()); // Always 1st message case "workerInfo" -> liveClient.getLogger().info(oMsg.toString()); // Always 1st message
case "roomInfo" -> { // Always 2nd message case "roomInfo" -> { // Always 2nd message
LiveUserData.Response data = LiveUserDataMapper.mapEulerstream(oMsg.getAsJsonObject("data"), liveClient.getLogger()); LiveUserData.Response data = LiveUserDataMapper.map(oMsg.getAsJsonObject("data").getAsJsonObject("data").getAsJsonObject("raw").toString(), liveClient.getLogger());
liveClient.getRoomInfo().copy(data.getRoomInfo()); liveClient.getRoomInfo().copy(data.getRoomInfo());
eventHandler.publish(liveClient, new TikTokRoomInfoEvent(liveClient.getRoomInfo())); eventHandler.publish(liveClient, new TikTokRoomInfoEvent(liveClient.getRoomInfo()));
} }
@@ -67,7 +67,6 @@ public class TikTokWebSocketEulerListener extends TikTokWebSocketListener
} else } else
throw new IllegalArgumentException("Invalid JsonObject: "+element); throw new IllegalArgumentException("Invalid JsonObject: "+element);
} catch (Exception e) { } catch (Exception e) {
e.printStackTrace();
eventHandler.publish(liveClient, new TikTokErrorEvent(e)); eventHandler.publish(liveClient, new TikTokErrorEvent(e));
} }
if (isOpen()) { if (isOpen()) {

View File

@@ -41,7 +41,7 @@
<parent> <parent>
<artifactId>TikTokLiveJava</artifactId> <artifactId>TikTokLiveJava</artifactId>
<groupId>io.github.jwdeveloper.tiktok</groupId> <groupId>io.github.jwdeveloper.tiktok</groupId>
<version>1.11.5-Release</version> <version>1.11.0-Release</version>
</parent> </parent>
<modelVersion>4.0.0</modelVersion> <modelVersion>4.0.0</modelVersion>

View File

@@ -6,7 +6,7 @@
<parent> <parent>
<groupId>io.github.jwdeveloper.tiktok</groupId> <groupId>io.github.jwdeveloper.tiktok</groupId>
<artifactId>TikTokLiveJava</artifactId> <artifactId>TikTokLiveJava</artifactId>
<version>1.11.5-Release</version> <version>1.11.0-Release</version>
</parent> </parent>

View File

@@ -5,7 +5,7 @@
<parent> <parent>
<artifactId>TikTokLiveJava</artifactId> <artifactId>TikTokLiveJava</artifactId>
<groupId>io.github.jwdeveloper.tiktok</groupId> <groupId>io.github.jwdeveloper.tiktok</groupId>
<version>1.11.5-Release</version> <version>1.11.0-Release</version>
</parent> </parent>
<modelVersion>4.0.0</modelVersion> <modelVersion>4.0.0</modelVersion>
<artifactId>extension-recorder</artifactId> <artifactId>extension-recorder</artifactId>

View File

@@ -7,7 +7,7 @@
<groupId>io.github.jwdeveloper.tiktok</groupId> <groupId>io.github.jwdeveloper.tiktok</groupId>
<artifactId>TikTokLiveJava</artifactId> <artifactId>TikTokLiveJava</artifactId>
<packaging>pom</packaging> <packaging>pom</packaging>
<version>1.11.5-Release</version> <version>1.11.0-Release</version>
<modules> <modules>
<module>API</module> <module>API</module>
<module>Client</module> <module>Client</module>

View File

@@ -5,7 +5,7 @@
<parent> <parent>
<artifactId>TikTokLiveJava</artifactId> <artifactId>TikTokLiveJava</artifactId>
<groupId>io.github.jwdeveloper.tiktok</groupId> <groupId>io.github.jwdeveloper.tiktok</groupId>
<version>1.11.5-Release</version> <version>1.11.0-Release</version>
</parent> </parent>
<modelVersion>4.0.0</modelVersion> <modelVersion>4.0.0</modelVersion>