Compare commits

..

18 Commits

Author SHA1 Message Date
GitHub Action
203d7f414d Update version in pom.xml 2025-12-29 03:17:11 +00:00
kohlerpop1
95e352908d Minor fix to proxying requests. Finally preserves state and proxies all requests when enabled. 2025-12-28 22:12:30 -05:00
GitHub Action
3f00256634 Update version in pom.xml 2025-11-16 04:30:57 +00:00
kohlerpop1
089d8d6ed8 Minor changes to new social/gift events! 2025-11-15 23:17:39 -05:00
GosDev
4be74c45ff Implement Social Event instances with custom Users (#151)
* Add .of methods for custom User objects

Adds an extra .of() function that accepts a User object, instead of a username string.

* Use username if profile name not set

* Add combo argument

* Replace WebcastGiftMessage from the new constructor

* Fix argument naming error

* Fix wrong argument order
2025-11-15 22:57:43 -05:00
GitHub Action
db4d382e34 Update version in pom.xml 2025-11-09 05:09:03 +00:00
kohlerpop1
2590200205 Temp fix until Eulerstream passes currentViewers! 2025-11-09 00:06:43 -05:00
GitHub Action
4aefde8a0c Update version in pom.xml 2025-11-09 03:43:26 +00:00
kohlerpop1
6486519876 Fix Eulerstream websocket roomInfo incorrectly being mapped! 2025-11-08 22:41:09 -05:00
GitHub Action
96cf28e5d5 Update version in pom.xml 2025-10-07 03:11:13 +00:00
kohlerpop1
cfdced9645 Add direct method to provide sessionId and ttTargetIdc for sending chats from 1 client. 2025-10-06 23:08:10 -04:00
GitHub Action
7589a2ac4a Update version in pom.xml 2025-10-05 02:40:59 +00:00
kohlerpop1
a0c445656c Slight change of decoding for TikTokWebSocketEulerListener! 2025-10-04 22:38:47 -04:00
GitHub Action
1e78fdda89 Update version in pom.xml 2025-09-24 23:28:34 +00:00
David Kohler
57f33b2efa Change static schedulers to AsyncHandler to hold for heartbeat and reconnect logic. (#148) 2025-09-24 19:26:44 -04:00
GitHub Action
85cba9fff2 Update version in pom.xml 2025-09-19 12:31:49 +00:00
kohlerpop1
b7977469a0 Simply prevent copying connectionState. It is a control elsewhere so we should not copy default state! 2025-09-19 08:28:56 -04:00
GitHub Action
8910c6a491 Update version in pom.xml 2025-09-17 22:57:05 +00:00
29 changed files with 254 additions and 124 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.0-Release</version> <version>1.11.9-Release</version>
</parent> </parent>
<modelVersion>4.0.0</modelVersion> <modelVersion>4.0.0</modelVersion>
<artifactId>API</artifactId> <artifactId>API</artifactId>

View File

@@ -39,11 +39,16 @@ public class TikTokGiftComboEvent extends TikTokGiftEvent {
this.comboState = comboState; this.comboState = comboState;
} }
public TikTokGiftComboEvent(Gift gift, User host, User user, int combo, GiftComboStateType comboState) {
super(gift, user, host, combo);
this.comboState = comboState;
}
public static TikTokGiftComboEvent of(Gift gift, int combo, GiftComboStateType comboState) { public static TikTokGiftComboEvent of(Gift gift, int combo, GiftComboStateType comboState) {
return new TikTokGiftComboEvent( return new TikTokGiftComboEvent(gift, new User(0L, "Test", new Picture("")), WebcastGiftMessage.newBuilder().setComboCount(combo).build(), comboState);
gift, }
new User(0L, "Test", new Picture("")),
WebcastGiftMessage.newBuilder().setComboCount(combo).build(), public static TikTokGiftComboEvent of(Gift gift, User host, User user, int combo, GiftComboStateType comboState) {
comboState); return new TikTokGiftComboEvent(gift, host, user, combo, comboState);
} }
} }

View File

@@ -49,17 +49,25 @@ public class TikTokGiftEvent extends TikTokHeaderEvent {
} }
combo = msg.getComboCount(); combo = msg.getComboCount();
} }
public TikTokGiftEvent(Gift gift) { public TikTokGiftEvent(Gift gift, User user, User toUser, int combo) {
this.gift = gift; this.gift = gift;
user = new User(0L, "sender", new Picture("")); this.user = user;
toUser = new User(0L, "receiver", new Picture("")); this.toUser = toUser;
combo = 1; this.combo = combo;
} }
public static TikTokGiftEvent of(Gift gift) { public static TikTokGiftEvent of(Gift gift) {
return new TikTokGiftEvent(gift); return new TikTokGiftEvent(
gift,
new User(0L, "sender", new Picture("")),
new User(0L, "reviever", new Picture("")),
1
);
}
public static TikTokGiftEvent of(Gift gift, User user, User toUser) {
return new TikTokGiftEvent(gift, user, toUser, 1) ;
} }
public static TikTokGiftEvent of(String name, int id, int diamonds) { public static TikTokGiftEvent of(String name, int id, int diamonds) {

View File

@@ -44,10 +44,20 @@ public class TikTokFollowEvent extends TikTokHeaderEvent {
public static TikTokFollowEvent of(String userName) { public static TikTokFollowEvent of(String userName) {
return new TikTokFollowEvent(WebcastSocialMessage.newBuilder() return new TikTokFollowEvent(WebcastSocialMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder() .setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName) .setUsername(userName)
.setNickname(userName) .setNickname(userName)
.build()) .build())
.build()); .build());
}
public static TikTokFollowEvent of(User user) {
return new TikTokFollowEvent(WebcastSocialMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(user.getName())
.setNickname(user.getProfileName() != null ? user.getProfileName() : user.getName())
.setId(user.getId())
.build())
.build());
} }
} }

View File

@@ -48,13 +48,22 @@ public class TikTokJoinEvent extends TikTokHeaderEvent {
totalUsers = msg.getMemberCount(); totalUsers = msg.getMemberCount();
} }
public static TikTokJoinEvent of(String userName) public static TikTokJoinEvent of(String userName) {
{
return new TikTokJoinEvent(WebcastMemberMessage.newBuilder() return new TikTokJoinEvent(WebcastMemberMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder() .setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName) .setUsername(userName)
.setNickname(userName) .setNickname(userName)
.build()) .build())
.build()); .build());
}
public static TikTokJoinEvent of(User user) {
return new TikTokJoinEvent(WebcastMemberMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(user.getName())
.setNickname(user.getProfileName())
.setId(user.getId())
.build())
.build());
} }
} }

View File

@@ -56,15 +56,25 @@ public class TikTokLikeEvent extends TikTokHeaderEvent
totalLikes = msg.getTotal(); totalLikes = msg.getTotal();
} }
public static TikTokLikeEvent of(String userName, int likes) public static TikTokLikeEvent of(String userName, int likes) {
{
return new TikTokLikeEvent(WebcastLikeMessage.newBuilder() return new TikTokLikeEvent(WebcastLikeMessage.newBuilder()
.setCount(likes) .setCount(likes)
.setTotal(likes) .setTotal(likes)
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder() .setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName) .setUsername(userName)
.setNickname(userName) .setNickname(userName)
.build()) .build())
.build()); .build());
}
public static TikTokLikeEvent of(User user, int likes) {
return new TikTokLikeEvent(WebcastLikeMessage.newBuilder()
.setCount(likes)
.setTotal(likes)
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(user.getName())
.setNickname(user.getProfileName())
.build())
.build());
} }
} }

View File

@@ -49,10 +49,19 @@ public class TikTokShareEvent extends TikTokHeaderEvent {
public static TikTokShareEvent of(String userName, int shaders) { public static TikTokShareEvent of(String userName, int shaders) {
return new TikTokShareEvent(WebcastSocialMessage.newBuilder() return new TikTokShareEvent(WebcastSocialMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder() .setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(userName) .setUsername(userName)
.setNickname(userName) .setNickname(userName)
.build()) .build())
.build(), shaders); .build(), shaders);
}
public static TikTokShareEvent of(User user, int shaders) {
return new TikTokShareEvent(WebcastSocialMessage.newBuilder()
.setUser(io.github.jwdeveloper.tiktok.messages.data.User.newBuilder()
.setUsername(user.getName())
.setNickname(user.getProfileName())
.build())
.build(), shaders);
} }
} }

View File

@@ -95,6 +95,9 @@ public class ProxyClientSettings implements Iterator<ProxyData>, Iterable<ProxyD
public ProxyClientSettings clone() { public ProxyClientSettings clone() {
ProxyClientSettings settings = new ProxyClientSettings(); ProxyClientSettings settings = new ProxyClientSettings();
settings.setEnabled(enabled); settings.setEnabled(enabled);
settings.setAutoDiscard(autoDiscard);
settings.setFallback(fallback);
settings.setAllowWebsocket(allowWebsocket);
settings.setRotation(rotation); settings.setRotation(rotation);
settings.setIndex(index); settings.setIndex(index);
settings.setType(type); settings.setType(type);

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); boolean sendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc);
} }

View File

@@ -102,4 +102,12 @@ 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,12 +62,11 @@ message Text {
int32 type = 1; int32 type = 1;
TextFormat format = 2; TextFormat format = 2;
string stringValue = 11; string stringValue = 11;
oneof textPieceType TextPieceUser userValue = 21;
{ TextPieceGift giftValue = 22;
TextPieceUser userValue = 21; TextPieceHeart heartValue = 23;
TextPieceGift giftValue = 22;
}
TextPiecePatternRef patternRefValue = 24; TextPiecePatternRef patternRefValue = 24;
TextPieceImage imageValue = 25;
} }
message TextFormat { message TextFormat {
@@ -83,7 +82,7 @@ message Text {
message TextPieceGift { message TextPieceGift {
int32 giftId = 1; int32 giftId = 1;
PatternRef nameRef = 2; TextPiecePatternRef nameRef = 2;
ShowType showType = 3; // Enum ShowType showType = 3; // Enum
int64 colorId = 4; int64 colorId = 4;
} }
@@ -98,16 +97,19 @@ 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
@@ -151,7 +153,7 @@ message BadgeStruct {
bool is_customized = 24; bool is_customized = 24;
message CombineBadge { message CombineBadge {
int32 badge_display_type = 1; BadgeDisplayType 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.0-Release</version> <version>1.11.9-Release</version>
</parent> </parent>
<modelVersion>4.0.0</modelVersion> <modelVersion>4.0.0</modelVersion>

View File

@@ -23,6 +23,7 @@
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.*;
@@ -39,7 +40,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.CompletableFuture; import java.util.concurrent.*;
import java.util.function.Consumer; import java.util.function.Consumer;
import java.util.logging.Logger; import java.util.logging.Logger;
@@ -89,12 +90,11 @@ 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()) {
try { AsyncHandler.getReconnectScheduler().schedule(() -> {
Thread.sleep(clientSettings.getRetryConnectionTimeout().toMillis()); logger.info("Reconnecting");
} catch (Exception ignored) {} tikTokEventHandler.publish(this, new TikTokReconnectingEvent());
logger.info("Reconnecting"); this.connect();
tikTokEventHandler.publish(this, new TikTokReconnectingEvent()); }, clientSettings.getRetryConnectionTimeout().toMillis(), TimeUnit.MILLISECONDS);
this.connect();
} }
throw e; throw e;
} catch (Exception e) { } catch (Exception e) {
@@ -199,7 +199,12 @@ public class TikTokLiveClient implements LiveClient
@Override @Override
public boolean sendChat(String content) { public boolean sendChat(String content) {
return httpClient.sendChat(roomInfo, content); return sendChat(content, clientSettings.getSessionId(), clientSettings.getTtTargetIdc());
}
@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); // True global singleton - Static objects are located to serve as global dependance.registerSingleton(WebSocketHeartbeatTask.class);
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) { public boolean sendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc) {
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); return requestSendChat(roomInfo, content, sessionId, ttTargetIdc);
} catch (TikTokProxyRequestException ignored) {} } catch (TikTokProxyRequestException ignored) {}
} }
} }
return requestSendChat(roomInfo, content); return requestSendChat(roomInfo, content, sessionId, ttTargetIdc);
} }
public boolean requestSendChat(LiveRoomInfo roomInfo, String content) { public boolean requestSendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc) {
JsonObject body = new JsonObject(); JsonObject body = new JsonObject();
body.addProperty("content", content); body.addProperty("content", content);
body.addProperty("sessionId", clientSettings.getSessionId()); body.addProperty("sessionId", sessionId);
body.addProperty("ttTargetIdc", clientSettings.getTtTargetIdc()); body.addProperty("ttTargetIdc", ttTargetIdc);
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) { public boolean sendChat(LiveRoomInfo roomInfo, String content, String sessionId, String ttTargetIdc) {
// 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.connectionState = roomInfo.getConnectionState(); // This should not be copied - Controlled elsewhere!
} }
} }

View File

@@ -0,0 +1,22 @@
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

@@ -69,7 +69,7 @@ public class HttpClient {
} }
} }
public <T> ActionResult<T> toResponse(HttpResponse.BodyHandler<T> handler) { protected <T> ActionResult<T> toResponse(HttpResponse.BodyHandler<T> handler) {
return toHttpResponse(handler).map(HttpResponse::body); return toHttpResponse(handler).map(HttpResponse::body);
} }

View File

@@ -31,9 +31,11 @@ import java.io.IOException;
import java.net.*; import java.net.*;
import java.net.http.*; import java.net.http.*;
import java.net.http.HttpResponse.ResponseInfo; import java.net.http.HttpResponse.ResponseInfo;
import java.nio.ByteBuffer;
import java.security.*; import java.security.*;
import java.security.cert.X509Certificate; import java.security.cert.X509Certificate;
import java.util.*; import java.util.*;
import java.util.concurrent.Flow;
import java.util.stream.Collectors; import java.util.stream.Collectors;
public class HttpProxyClient extends HttpClient { public class HttpProxyClient extends HttpClient {
@@ -45,14 +47,14 @@ public class HttpProxyClient extends HttpClient {
this.proxySettings = httpClientSettings.getProxyClientSettings(); this.proxySettings = httpClientSettings.getProxyClientSettings();
} }
public ActionResult<HttpResponse<byte[]>> toResponse() { public <T> ActionResult<HttpResponse<T>> toHttpResponse(HttpResponse.BodyHandler<T> handler) {
return switch (proxySettings.getType()) { return switch (proxySettings.getType()) {
case HTTP, DIRECT -> handleHttpProxyRequest(); case HTTP, DIRECT -> handleHttpProxyRequest(handler);
default -> handleSocksProxyRequest(); default -> handleSocksProxyRequest(handler);
}; };
} }
public ActionResult<HttpResponse<byte[]>> handleHttpProxyRequest() { public <T> ActionResult<HttpResponse<T>> handleHttpProxyRequest(HttpResponse.BodyHandler<T> handler) {
var builder = java.net.http.HttpClient.newBuilder() var builder = java.net.http.HttpClient.newBuilder()
.followRedirects(java.net.http.HttpClient.Redirect.NORMAL) .followRedirects(java.net.http.HttpClient.Redirect.NORMAL)
.cookieHandler(new CookieManager()) .cookieHandler(new CookieManager())
@@ -67,7 +69,7 @@ public class HttpProxyClient extends HttpClient {
var client = builder.build(); var client = builder.build();
var request = prepareRequest(); var request = prepareRequest();
var response = client.send(request, HttpResponse.BodyHandlers.ofByteArray()); var response = client.send(request, handler);
if (response.statusCode() != 200) if (response.statusCode() != 200)
continue; continue;
return ActionResult.success(response); return ActionResult.success(response);
@@ -77,7 +79,7 @@ public class HttpProxyClient extends HttpClient {
throw new TikTokProxyRequestException(e); throw new TikTokProxyRequestException(e);
} catch (IOException e) { } catch (IOException e) {
if (e.getMessage().contains("503") && proxySettings.isFallback()) // Indicates proxy protocol is not supported if (e.getMessage().contains("503") && proxySettings.isFallback()) // Indicates proxy protocol is not supported
return super.toHttpResponse(HttpResponse.BodyHandlers.ofByteArray()); return super.toHttpResponse(handler);
throw new TikTokProxyRequestException(e); throw new TikTokProxyRequestException(e);
} catch (Exception e) { } catch (Exception e) {
throw new TikTokLiveRequestException(e); throw new TikTokLiveRequestException(e);
@@ -86,7 +88,7 @@ public class HttpProxyClient extends HttpClient {
throw new TikTokLiveRequestException("No more proxies available!"); throw new TikTokLiveRequestException("No more proxies available!");
} }
private ActionResult<HttpResponse<byte[]>> handleSocksProxyRequest() { private <T> ActionResult<HttpResponse<T>> handleSocksProxyRequest(HttpResponse.BodyHandler<T> handler) {
try { try {
SSLContext sc = SSLContext.getInstance("SSL"); SSLContext sc = SSLContext.getInstance("SSL");
sc.init(null, new TrustManager[]{ new X509TrustManager() { sc.init(null, new TrustManager[]{ new X509TrustManager() {
@@ -95,7 +97,8 @@ public class HttpProxyClient extends HttpClient {
public X509Certificate[] getAcceptedIssuers() { return null; } public X509Certificate[] getAcceptedIssuers() { return null; }
}}, null); }}, null);
URL url = toUri().toURL(); URI uri = toUri();
URL url = uri.toURL();
if (proxySettings.hasNext()) { if (proxySettings.hasNext()) {
try { try {
@@ -117,12 +120,22 @@ public class HttpProxyClient extends HttpClient {
var responseInfo = createResponseInfo(socksConnection.getResponseCode(), headers); var responseInfo = createResponseInfo(socksConnection.getResponseCode(), headers);
var response = createHttpResponse(body, toUri(), responseInfo); HttpResponse.BodySubscriber<T> subscriber = handler.apply(responseInfo);
subscriber.onSubscribe(new Flow.Subscription() {
@Override public void request(long n) {}
@Override public void cancel() {}
});
subscriber.onNext(List.of(ByteBuffer.wrap(body)));
subscriber.onComplete();
var response = createHttpResponse(subscriber.getBody().toCompletableFuture().join(), uri, responseInfo);
return ActionResult.success(response); return ActionResult.success(response);
} catch (IOException e) { } catch (IOException e) {
if (e.getMessage().contains("503") && proxySettings.isFallback()) // Indicates proxy protocol is not supported if (e.getMessage().contains("503") && proxySettings.isFallback()) // Indicates proxy protocol is not supported
return super.toHttpResponse(HttpResponse.BodyHandlers.ofByteArray()); return super.toHttpResponse(handler);
if (proxySettings.isAutoDiscard()) if (proxySettings.isAutoDiscard())
proxySettings.remove(); proxySettings.remove();
throw new TikTokProxyRequestException(e); throw new TikTokProxyRequestException(e);
@@ -160,7 +173,7 @@ public class HttpProxyClient extends HttpClient {
}; };
} }
private HttpResponse<byte[]> createHttpResponse(byte[] body, private <T> HttpResponse<T> createHttpResponse(T body,
URI uri, URI uri,
ResponseInfo info) { ResponseInfo info) {
return new HttpResponse<>() return new HttpResponse<>()
@@ -176,7 +189,7 @@ public class HttpProxyClient extends HttpClient {
} }
@Override @Override
public Optional<HttpResponse<byte[]>> previousResponse() { public Optional<HttpResponse<T>> previousResponse() {
return Optional.empty(); return Optional.empty();
} }
@@ -186,7 +199,7 @@ public class HttpProxyClient extends HttpClient {
} }
@Override @Override
public byte[] body() { public T body() {
return body; return body;
} }

View File

@@ -29,7 +29,7 @@ import io.github.jwdeveloper.tiktok.data.models.users.User;
import io.github.jwdeveloper.tiktok.data.requests.LiveUserData; import io.github.jwdeveloper.tiktok.data.requests.LiveUserData;
import io.github.jwdeveloper.tiktok.exceptions.TikTokLiveRequestException; import io.github.jwdeveloper.tiktok.exceptions.TikTokLiveRequestException;
import java.util.List; import java.util.*;
import java.util.logging.Logger; import java.util.logging.Logger;
public class LiveUserDataMapper public class LiveUserDataMapper
@@ -66,7 +66,6 @@ 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);
@@ -92,9 +91,51 @@ 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 e) { } catch (JsonSyntaxException | IllegalStateException | NullPointerException 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(Optional.ofNullable(roomInfoJson.get("currentViewers")).filter(JsonElement::isJsonPrimitive).map(JsonElement::getAsInt).orElse(0));
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(webSocketClient); heartbeatTask.stop();
} }
webSocketClient = null; webSocketClient = null;
} }

View File

@@ -22,6 +22,7 @@
*/ */
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.*;
@@ -29,47 +30,30 @@ import java.util.concurrent.*;
public class WebSocketHeartbeatTask public class WebSocketHeartbeatTask
{ {
// Single shared pool for all heartbeat tasks private ScheduledFuture<?> task;
private static final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1, r -> { private Long commTime;
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 byte[] heartbeatBytes = Base64.getDecoder().decode("MgJwYjoCaGI="); // Used to be '3A026862' aka ':\x02hb', now is '2\x02pb:\x02hb'. private final static 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(webSocket); // remove existing task if any stop(); // remove existing task if any
tasks.put(webSocket, scheduler.scheduleAtFixedRate(() -> { task = AsyncHandler.getHeartBeatScheduler().scheduleAtFixedRate(() -> {
try { try {
if (webSocket.isOpen()) { if (webSocket.isOpen()) {
webSocket.send(heartbeatBytes); webSocket.send(heartbeatBytes);
commTime.put(webSocket, System.currentTimeMillis()); commTime = System.currentTimeMillis();
} else { } else if (commTime != null && System.currentTimeMillis() - commTime >= 60_000) // Stop if disconnected longer than 60s
Long time = commTime.get(webSocket); stop();
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(webSocket); stop();
} }
}, 0, pingTaskTime, TimeUnit.MILLISECONDS)); }, 0, pingTaskTime, TimeUnit.MILLISECONDS);
} }
public void stop(WebSocket webSocket) { public void stop() {
ScheduledFuture<?> future = tasks.remove(webSocket); if (task != null)
if (future != null) task.cancel(true);
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.map(oMsg.getAsJsonObject("data").getAsJsonObject("data").getAsJsonObject("raw").toString(), liveClient.getLogger()); LiveUserData.Response data = LiveUserDataMapper.mapEulerstream(oMsg.getAsJsonObject("data"), 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,6 +67,7 @@ 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.0-Release</version> <version>1.11.9-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.0-Release</version> <version>1.11.9-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.0-Release</version> <version>1.11.9-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.0-Release</version> <version>1.11.9-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.0-Release</version> <version>1.11.9-Release</version>
</parent> </parent>
<modelVersion>4.0.0</modelVersion> <modelVersion>4.0.0</modelVersion>