Optimized a few methods when calling and using the Api!

This commit is contained in:
kohlerpop1
2023-12-21 21:40:11 -05:00
parent 7a4c7fecbd
commit 0b9f1570d0
8 changed files with 86 additions and 82 deletions

View File

@@ -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);

View File

@@ -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;
} }
} }

View File

@@ -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();

View File

@@ -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();
} }
} }

View File

@@ -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();
@@ -157,4 +151,4 @@ public class TikTokHttpClient {
} }
} }
} }

View File

@@ -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();
} }
}
}

View File

@@ -115,8 +115,8 @@ 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);
} }
} }

View File

@@ -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);
}
}