mirror of
https://github.com/jwdeveloper/TikTokLiveJava.git
synced 2026-02-27 08:49:40 -05:00
Optimized a few methods when calling and using the Api!
This commit is contained in:
@@ -136,12 +136,10 @@ public class TikTokLiveClientBuilder implements LiveClientBuilder {
|
|||||||
handler.setFormatter(new Formatter() {
|
handler.setFormatter(new Formatter() {
|
||||||
@Override
|
@Override
|
||||||
public String format(LogRecord record) {
|
public String format(LogRecord record) {
|
||||||
var sb = new StringBuilder();
|
return ConsoleColors.GREEN + "[" + record.getLoggerName() + "] " +
|
||||||
sb.append(ConsoleColors.GREEN).append("[").append(record.getLoggerName()).append("] ");
|
ConsoleColors.GREEN + "[" + record.getLevel() + "]: " +
|
||||||
sb.append(ConsoleColors.GREEN).append("[").append(record.getLevel()).append("]: ");
|
ConsoleColors.WHITE_BRIGHT + record.getMessage() +
|
||||||
sb.append(ConsoleColors.WHITE_BRIGHT).append(record.getMessage());
|
ConsoleColors.RESET + "\n";
|
||||||
sb.append(ConsoleColors.RESET).append("\n");
|
|
||||||
return sb.toString();
|
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
logger.setUseParentHandlers(false);
|
logger.setUseParentHandlers(false);
|
||||||
|
|||||||
@@ -22,58 +22,30 @@
|
|||||||
*/
|
*/
|
||||||
package io.github.jwdeveloper.tiktok.http;
|
package io.github.jwdeveloper.tiktok.http;
|
||||||
|
|
||||||
import lombok.SneakyThrows;
|
|
||||||
|
|
||||||
import java.net.URLEncoder;
|
import java.net.URLEncoder;
|
||||||
import java.nio.charset.StandardCharsets;
|
import java.nio.charset.StandardCharsets;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
public class HttpUtils
|
public class HttpUtils
|
||||||
{
|
{
|
||||||
public static String parseParameters(String url, Map<String,Object> parameters)
|
public static String parseParameters(String url, Map<String,Object> parameters)
|
||||||
{
|
{
|
||||||
var parameterString = "";
|
if (parameters.isEmpty())
|
||||||
if (!parameters.isEmpty()) {
|
return url;
|
||||||
var builder = new StringBuilder();
|
|
||||||
builder.append("?");
|
|
||||||
var first = false;
|
|
||||||
for (var param : parameters.entrySet()) {
|
|
||||||
|
|
||||||
if (first) {
|
return url+ "?" + parameters.entrySet().stream().map(entry -> entry.getKey()+"="+entry.getValue()).collect(Collectors.joining("&"));
|
||||||
builder.append("&");
|
|
||||||
}
|
|
||||||
builder.append(param.getKey()).append("=").append(param.getValue());
|
|
||||||
first = true;
|
|
||||||
}
|
|
||||||
parameterString = builder.toString();
|
|
||||||
}
|
|
||||||
|
|
||||||
return url+parameterString;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@SneakyThrows
|
|
||||||
public static String parseParametersEncode(String url, Map<String,Object> parameters)
|
public static String parseParametersEncode(String url, Map<String,Object> parameters)
|
||||||
{
|
{
|
||||||
|
if (parameters.isEmpty())
|
||||||
|
return url;
|
||||||
|
|
||||||
var parameterString = "";
|
return url+ "?" + parameters.entrySet().stream().map(entry -> {
|
||||||
if (!parameters.isEmpty()) {
|
String encodedKey = URLEncoder.encode(entry.getKey(), StandardCharsets.UTF_8);
|
||||||
var builder = new StringBuilder();
|
String encodedValue = URLEncoder.encode(entry.getValue().toString(), StandardCharsets.UTF_8);
|
||||||
builder.append("?");
|
return encodedKey+"="+encodedValue;
|
||||||
var first = false;
|
}).collect(Collectors.joining("&"));
|
||||||
for (var param : parameters.entrySet()) {
|
|
||||||
|
|
||||||
if (first) {
|
|
||||||
builder.append("&");
|
|
||||||
}
|
|
||||||
|
|
||||||
final String encodedKey = URLEncoder.encode(param.getKey(), StandardCharsets.UTF_8);
|
|
||||||
final String encodedValue = URLEncoder.encode(param.getValue().toString(), StandardCharsets.UTF_8);
|
|
||||||
builder.append(encodedKey).append("=").append(encodedValue);
|
|
||||||
first = true;
|
|
||||||
}
|
|
||||||
parameterString = builder.toString();
|
|
||||||
}
|
|
||||||
|
|
||||||
return url+parameterString;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -73,7 +73,7 @@ public class TikTokApiService {
|
|||||||
var message = roomData.get("message").getAsString();
|
var message = roomData.get("message").getAsString();
|
||||||
|
|
||||||
if (message.equals("params_error")) {
|
if (message.equals("params_error")) {
|
||||||
throw new TikTokLiveRequestException("fetchRoomIdFromTiktokApi -> Unable to fetch roomID, contact with developer");
|
throw new TikTokLiveRequestException("fetchRoomIdFromTiktokApi -> Unable to fetch roomID, contact the developer");
|
||||||
}
|
}
|
||||||
if (message.equals("user_not_found")) {
|
if (message.equals("user_not_found")) {
|
||||||
return new TikTokUserInfo(TikTokUserInfo.UserStatus.NotFound, "", -1);
|
return new TikTokUserInfo(TikTokUserInfo.UserStatus.NotFound, "", -1);
|
||||||
@@ -81,7 +81,11 @@ public class TikTokApiService {
|
|||||||
//live -> status 2
|
//live -> status 2
|
||||||
//live paused -> 3
|
//live paused -> 3
|
||||||
//not live -> status 4
|
//not live -> status 4
|
||||||
var data = roomData.getAsJsonObject("data");
|
var element = roomData.get("data");
|
||||||
|
if (element.isJsonNull()) {
|
||||||
|
return new TikTokUserInfo(TikTokUserInfo.UserStatus.NotFound, "", -1);
|
||||||
|
}
|
||||||
|
var data = element.getAsJsonObject();
|
||||||
var user = data.getAsJsonObject("user");
|
var user = data.getAsJsonObject("user");
|
||||||
var roomId = user.get("roomId").getAsString();
|
var roomId = user.get("roomId").getAsString();
|
||||||
var status = user.get("status").getAsInt();
|
var status = user.get("status").getAsInt();
|
||||||
|
|||||||
@@ -22,9 +22,8 @@
|
|||||||
*/
|
*/
|
||||||
package io.github.jwdeveloper.tiktok.http;
|
package io.github.jwdeveloper.tiktok.http;
|
||||||
|
|
||||||
import java.util.HashMap;
|
import java.util.*;
|
||||||
import java.util.Map;
|
import java.util.stream.Collectors;
|
||||||
import java.util.Set;
|
|
||||||
|
|
||||||
public class TikTokCookieJar {
|
public class TikTokCookieJar {
|
||||||
private final Map<String, String> cookies;
|
private final Map<String, String> cookies;
|
||||||
@@ -40,13 +39,7 @@ public class TikTokCookieJar {
|
|||||||
cookies.put(key, value);
|
cookies.put(key, value);
|
||||||
}
|
}
|
||||||
|
|
||||||
public String parseCookies()
|
public String parseCookies() {
|
||||||
{
|
return cookies.entrySet().stream().map(entry -> entry.getKey()+"="+entry.getValue()+";").collect(Collectors.joining());
|
||||||
var sb = new StringBuilder();
|
|
||||||
for(var entry : cookies.entrySet())
|
|
||||||
{
|
|
||||||
sb.append(entry.getKey()).append("=").append(entry.getValue()).append(";");
|
|
||||||
}
|
|
||||||
return sb.toString();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -45,8 +45,7 @@ public class TikTokHttpClient {
|
|||||||
this.tikTokCookieJar = tikTokCookieJar;
|
this.tikTokCookieJar = tikTokCookieJar;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void setSessionId(String sessionId)
|
public void setSessionId(String sessionId) {
|
||||||
{
|
|
||||||
tikTokCookieJar.set("sessionid", sessionId);
|
tikTokCookieJar.set("sessionid", sessionId);
|
||||||
tikTokCookieJar.set("sessionid_ss", sessionId);
|
tikTokCookieJar.set("sessionid_ss", sessionId);
|
||||||
tikTokCookieJar.set("sid_tt", sessionId);
|
tikTokCookieJar.set("sid_tt", sessionId);
|
||||||
@@ -54,14 +53,12 @@ public class TikTokHttpClient {
|
|||||||
|
|
||||||
|
|
||||||
public String getLivestreamPage(String userName) {
|
public String getLivestreamPage(String userName) {
|
||||||
|
|
||||||
var url = Constants.TIKTOK_URL_WEB + "@" + userName + "/live/";
|
var url = Constants.TIKTOK_URL_WEB + "@" + userName + "/live/";
|
||||||
var get = getRequest(url, null);
|
var get = getRequest(url, null);
|
||||||
return get;
|
return get;
|
||||||
}
|
}
|
||||||
|
|
||||||
public JsonObject getJsonFromTikTokApi(String path, Map<String,Object> params)
|
public JsonObject getJsonFromTikTokApi(String path, Map<String,Object> params) {
|
||||||
{
|
|
||||||
var get = getRequest(Constants.TIKTOK_URL_WEB + path, params);
|
var get = getRequest(Constants.TIKTOK_URL_WEB + path, params);
|
||||||
var json = JsonParser.parseString(get);
|
var json = JsonParser.parseString(get);
|
||||||
var jsonObject = json.getAsJsonObject();
|
var jsonObject = json.getAsJsonObject();
|
||||||
@@ -117,7 +114,6 @@ public class TikTokHttpClient {
|
|||||||
{
|
{
|
||||||
var split = cookie.split(";")[0].split("=");
|
var split = cookie.split(";")[0].split("=");
|
||||||
|
|
||||||
|
|
||||||
var key = split[0];
|
var key = split[0];
|
||||||
var value = split[1];
|
var value = split[1];
|
||||||
tikTokCookieJar.set(key, value);
|
tikTokCookieJar.set(key, value);
|
||||||
@@ -133,7 +129,6 @@ public class TikTokHttpClient {
|
|||||||
|
|
||||||
|
|
||||||
private String getSignedUrl(String url, Map<String, Object> parameters) {
|
private String getSignedUrl(String url, Map<String, Object> parameters) {
|
||||||
|
|
||||||
var fullUrl = HttpUtils.parseParameters(url,parameters);
|
var fullUrl = HttpUtils.parseParameters(url,parameters);
|
||||||
var signParams = new TreeMap<String,Object>();
|
var signParams = new TreeMap<String,Object>();
|
||||||
signParams.put("client", "ttlive-java");
|
signParams.put("client", "ttlive-java");
|
||||||
@@ -143,7 +138,6 @@ public class TikTokHttpClient {
|
|||||||
var request = requestFactory.setQueries(signParams);
|
var request = requestFactory.setQueries(signParams);
|
||||||
var content = request.get(Constants.TIKTOK_SIGN_API);
|
var content = request.get(Constants.TIKTOK_SIGN_API);
|
||||||
|
|
||||||
|
|
||||||
try {
|
try {
|
||||||
var json = JsonParser.parseString(content);
|
var json = JsonParser.parseString(content);
|
||||||
var jsonObject = json.getAsJsonObject();
|
var jsonObject = json.getAsJsonObject();
|
||||||
|
|||||||
@@ -42,6 +42,7 @@ import java.util.HashMap;
|
|||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.TreeMap;
|
import java.util.TreeMap;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
public class TikTokHttpRequestFactory implements TikTokHttpRequest {
|
public class TikTokHttpRequestFactory implements TikTokHttpRequest {
|
||||||
private final CookieManager cookieManager;
|
private final CookieManager cookieManager;
|
||||||
@@ -122,18 +123,16 @@ public class TikTokHttpRequestFactory implements TikTokHttpRequest {
|
|||||||
public TikTokHttpRequest setQueries(Map<String, Object> queries) {
|
public TikTokHttpRequest setQueries(Map<String, Object> queries) {
|
||||||
if (queries == null)
|
if (queries == null)
|
||||||
return this;
|
return this;
|
||||||
var testMap = new TreeMap<String, Object>(queries);
|
var testMap = new TreeMap<>(queries);
|
||||||
query = String.join("&", testMap.entrySet().stream().map(x ->
|
query = testMap.entrySet().stream().map(x -> {
|
||||||
{
|
|
||||||
var key = x.getKey();
|
var key = x.getKey();
|
||||||
var value = "";
|
|
||||||
try {
|
try {
|
||||||
value = URLEncoder.encode(x.getValue().toString(), StandardCharsets.UTF_8);
|
return key+"="+URLEncoder.encode(x.getValue().toString(), StandardCharsets.UTF_8);
|
||||||
} catch (Exception e) {
|
} catch (Exception e) {
|
||||||
e.printStackTrace();
|
e.printStackTrace();
|
||||||
|
return key + "=";
|
||||||
}
|
}
|
||||||
return key + "=" + value;
|
}).collect(Collectors.joining("&"));
|
||||||
}).toList());
|
|
||||||
return this;
|
return this;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -162,10 +161,7 @@ public class TikTokHttpRequestFactory implements TikTokHttpRequest {
|
|||||||
var map = new HashMap<String, List<String>>();
|
var map = new HashMap<String, List<String>>();
|
||||||
map.put(key, List.of(value));
|
map.put(key, List.of(value));
|
||||||
cookieManager.put(uri, map);
|
cookieManager.put(uri, map);
|
||||||
|
|
||||||
}
|
}
|
||||||
return response.body();
|
return response.body();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
@@ -115,7 +115,7 @@ public class TikTokListenersManager implements ListenersManager {
|
|||||||
throw new TikTokEventListenerMethodException(e);
|
throw new TikTokEventListenerMethodException(e);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
eventsMap.computeIfAbsent(eventClazz, (a) -> new ArrayList<EventConsumer<?>>()).add(eventMethodRef);
|
eventsMap.computeIfAbsent(eventClazz, (a) -> new ArrayList<>()).add(eventMethodRef);
|
||||||
}
|
}
|
||||||
return new ListenerBindingModel(listener, eventsMap);
|
return new ListenerBindingModel(listener, eventsMap);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,47 @@
|
|||||||
|
package io.github.jwdeveloper.tiktok.http;
|
||||||
|
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.junit.jupiter.api.Assertions;
|
||||||
|
|
||||||
|
import java.util.*;
|
||||||
|
|
||||||
|
public class HttpUtilsTest
|
||||||
|
{
|
||||||
|
@Test
|
||||||
|
public void parseParameters_EmptyParameters_ShouldHaveNoParameters()
|
||||||
|
{
|
||||||
|
String parsed = HttpUtils.parseParameters("https://webcast.tiktok.com/webcast/im/fetch/", new HashMap<>());
|
||||||
|
|
||||||
|
Assertions.assertEquals("https://webcast.tiktok.com/webcast/im/fetch/", parsed);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void parseParameters_ValidParameters_ShouldConstructValidURL()
|
||||||
|
{
|
||||||
|
LinkedHashMap<String, Object> testMap = new LinkedHashMap<>();
|
||||||
|
testMap.put("room_id", 1);
|
||||||
|
testMap.put("uniqueId", "randomName");
|
||||||
|
String parsed = HttpUtils.parseParameters("https://webcast.tiktok.com/webcast/im/fetch/", testMap);
|
||||||
|
|
||||||
|
Assertions.assertEquals("https://webcast.tiktok.com/webcast/im/fetch/?room_id=1&uniqueId=randomName", parsed);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void parseParametersEncode_EmptyParameters_ShouldHaveNoParameters()
|
||||||
|
{
|
||||||
|
String parsed = HttpUtils.parseParametersEncode("https://webcast.tiktok.com/webcast/im/fetch/", new HashMap<>());
|
||||||
|
|
||||||
|
Assertions.assertEquals("https://webcast.tiktok.com/webcast/im/fetch/", parsed);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void parseParametersEncode_ValidParameters_ShouldConstructValidURL()
|
||||||
|
{
|
||||||
|
LinkedHashMap<String, Object> testMap = new LinkedHashMap<>();
|
||||||
|
testMap.put("room_id", 1);
|
||||||
|
testMap.put("root_referer", "https://www.tiktok.com/");
|
||||||
|
String parsed = HttpUtils.parseParametersEncode("https://webcast.tiktok.com/webcast/im/fetch/", testMap);
|
||||||
|
|
||||||
|
Assertions.assertEquals("https://webcast.tiktok.com/webcast/im/fetch/?room_id=1&root_referer=https%3A%2F%2Fwww.tiktok.com%2F", parsed);
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user