7241036e61
Instead use NativeImage on 1.13+ via newly introduced Image class. See Image docs for details. Also fixes an issue with thumbnail taking on 1.13+: The original ScreenshotHelper method is now async, luckily there's a sync variant and as a bonus it directly returns a NativeImage so we can skip having to read the screenshot from disk.
278 lines
10 KiB
Java
Executable File
278 lines
10 KiB
Java
Executable File
package com.replaymod.online.api;
|
|
|
|
import com.google.gson.Gson;
|
|
import com.google.gson.JsonElement;
|
|
import com.google.gson.JsonParseException;
|
|
import com.google.gson.JsonParser;
|
|
import com.mojang.authlib.exceptions.AuthenticationException;
|
|
import com.replaymod.core.ReplayMod;
|
|
import com.replaymod.core.versions.MCVer;
|
|
import com.replaymod.online.AuthenticationHash;
|
|
import com.replaymod.online.api.replay.ReplayModApiMethods;
|
|
import com.replaymod.online.api.replay.SearchQuery;
|
|
import com.replaymod.online.api.replay.holders.*;
|
|
import de.johni0702.minecraft.gui.versions.Image;
|
|
import net.minecraft.client.MinecraftClient;
|
|
import org.apache.commons.io.FileUtils;
|
|
import org.apache.commons.io.IOUtils;
|
|
|
|
import javax.net.ssl.HttpsURLConnection;
|
|
import java.io.BufferedInputStream;
|
|
import java.io.File;
|
|
import java.io.FileOutputStream;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.net.URL;
|
|
import java.util.List;
|
|
import java.util.function.Consumer;
|
|
|
|
import static com.replaymod.core.utils.Utils.SSL_SOCKET_FACTORY;
|
|
|
|
public class ApiClient {
|
|
|
|
private static final MinecraftClient mc = MCVer.getMinecraft();
|
|
private static final Gson gson = new Gson();
|
|
private static final JsonParser jsonParser = new JsonParser();
|
|
|
|
private final AuthData authData;
|
|
|
|
public ApiClient(AuthData authData) {
|
|
this.authData = authData;
|
|
}
|
|
|
|
public boolean isLoggedIn() {
|
|
return authData.getUserName() != null && authData.getAuthKey() != null;
|
|
}
|
|
|
|
public String getAuthKey() {
|
|
return authData.getAuthKey();
|
|
}
|
|
|
|
public void register(String userName, String eMail, String password) throws AuthenticationException, IOException, ApiException {
|
|
AuthenticationHash authenticationHash = sessionserverJoin();
|
|
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.register);
|
|
builder.put("username", userName);
|
|
builder.put("email", eMail);
|
|
builder.put("password", password);
|
|
builder.put("uuid", mc.getSession().getProfile().getId().toString());
|
|
builder.put("mcusername", authenticationHash.username);
|
|
builder.put("timelong", authenticationHash.currentTime);
|
|
builder.put("randomlong", authenticationHash.randomLong);
|
|
AuthKey auth = invokeAndReturn(builder, AuthKey.class);
|
|
|
|
authData.setData(userName, auth.getAuth());
|
|
}
|
|
|
|
public AuthData.AuthResult login(String userName, String password) {
|
|
try {
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.login);
|
|
builder.put("user", userName);
|
|
builder.put("pw", password);
|
|
builder.put("mod", true);
|
|
AuthKey result = invokeAndReturn(builder, AuthKey.class);
|
|
authData.setData(userName, result.getAuth());
|
|
return AuthData.AuthResult.SUCCESS;
|
|
} catch(ApiException e) {
|
|
return AuthData.AuthResult.INVALID_DATA;
|
|
} catch(Exception e) {
|
|
return AuthData.AuthResult.IO_ERROR;
|
|
}
|
|
}
|
|
|
|
public AuthData.AuthResult logout() {
|
|
try {
|
|
authData.setData(null, null);
|
|
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.logout);
|
|
builder.put("auth", authData.getAuthKey());
|
|
builder.put("mod", true);
|
|
invokeAndReturn(builder, Success.class);
|
|
return AuthData.AuthResult.SUCCESS;
|
|
} catch(ApiException e) {
|
|
return AuthData.AuthResult.INVALID_DATA;
|
|
} catch(Exception e) {
|
|
return AuthData.AuthResult.IO_ERROR;
|
|
}
|
|
}
|
|
|
|
private AuthenticationHash sessionserverJoin() throws AuthenticationException {
|
|
AuthenticationHash hash = new AuthenticationHash();
|
|
|
|
mc.getSessionService().joinServer(
|
|
mc.getSession().getProfile(), mc.getSession().getAccessToken(), hash.hash);
|
|
|
|
return hash;
|
|
}
|
|
|
|
public AuthConfirmation checkAuthkey(String auth) {
|
|
try {
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.check_authkey);
|
|
builder.put("auth", auth);
|
|
return invokeAndReturn(builder, AuthConfirmation.class);
|
|
} catch(Exception e) {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public FileInfo[] getFileInfo(List<Integer> ids) throws IOException, ApiException {
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.file_details);
|
|
builder.put("id", buildListString(ids));
|
|
return invokeAndReturn(builder, FileInfo[].class);
|
|
}
|
|
|
|
public FileInfo[] searchFiles(SearchQuery query) throws IOException, ApiException {
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.search);
|
|
return invokeAndReturn(builder.toString() + query.buildQuery(), SearchResult.class).getResults();
|
|
}
|
|
|
|
public String getTranslation(String languageCode) throws IOException, ApiException {
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.get_language);
|
|
builder.put("language", languageCode);
|
|
return SimpleApiClient.invokeUrl(builder.toString());
|
|
}
|
|
|
|
public Image downloadThumbnail(int file) throws IOException {
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.get_thumbnail);
|
|
builder.put("id", file);
|
|
URL url = new URL(builder.toString());
|
|
HttpsURLConnection connection = (HttpsURLConnection) url.openConnection();
|
|
connection.setSSLSocketFactory(SSL_SOCKET_FACTORY);
|
|
try (InputStream in = connection.getInputStream()) {
|
|
return Image.read(in);
|
|
}
|
|
}
|
|
|
|
private boolean cancelDownload = false;
|
|
|
|
public void downloadFile(int file, File target, Consumer<Float> listener) throws IOException, ApiException {
|
|
cancelDownload = false;
|
|
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.download_file);
|
|
builder.put("auth", authData.getAuthKey());
|
|
builder.put("id", file);
|
|
String url = builder.toString();
|
|
URL website = new URL(url);
|
|
HttpsURLConnection con = (HttpsURLConnection) website.openConnection();
|
|
con.setSSLSocketFactory(SSL_SOCKET_FACTORY);
|
|
|
|
int fileSize = con.getContentLength();
|
|
|
|
InputStream is = con.getInputStream();
|
|
|
|
if(con.getResponseCode() == 200) {
|
|
BufferedInputStream bin = new BufferedInputStream(is);
|
|
FileOutputStream fout = new FileOutputStream(target);
|
|
try {
|
|
final byte data[] = new byte[1024];
|
|
int count;
|
|
int read = 0;
|
|
while ((count = bin.read(data, 0, 1024)) != -1) {
|
|
if(cancelDownload) {
|
|
bin.close();
|
|
fout.close();
|
|
|
|
FileUtils.deleteQuietly(target);
|
|
|
|
return;
|
|
}
|
|
|
|
fout.write(data, 0, count);
|
|
read += count;
|
|
listener.accept((float) read / fileSize);
|
|
}
|
|
} finally {
|
|
bin.close();
|
|
fout.close();
|
|
}
|
|
} else {
|
|
JsonElement element = jsonParser.parse(IOUtils.toString(is));
|
|
try {
|
|
ApiError err = gson.fromJson(element, ApiError.class);
|
|
if(err.getDesc() != null) {
|
|
throw new ApiException(err);
|
|
}
|
|
} catch(JsonParseException e) {
|
|
throw new ApiException(IOUtils.toString(is));
|
|
}
|
|
}
|
|
}
|
|
|
|
public void cancelDownload() {
|
|
this.cancelDownload = true;
|
|
}
|
|
|
|
public void rateFile(int file, Rating.RatingType rating) throws IOException, ApiException {
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.rate_file);
|
|
builder.put("auth", authData.getAuthKey());
|
|
builder.put("id", file);
|
|
builder.put("rating", rating.getKey());
|
|
invokeAndReturn(builder, Success.class);
|
|
}
|
|
|
|
public FileRating[] getRatedFiles() throws IOException, ApiException {
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.get_ratings);
|
|
builder.put("auth", authData.getAuthKey());
|
|
|
|
return invokeAndReturn(builder, RatedFiles.class).getRated();
|
|
}
|
|
|
|
public void favFile(int file, boolean fav) throws IOException, ApiException {
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.fav_file);
|
|
builder.put("auth", authData.getAuthKey());
|
|
builder.put("id", file);
|
|
builder.put("fav", fav);
|
|
invokeAndReturn(builder, Success.class);
|
|
}
|
|
|
|
public int[] getFavorites() throws IOException, ApiException {
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.get_favorites);
|
|
builder.put("auth", authData.getAuthKey());
|
|
|
|
return invokeAndReturn(builder, Favorites.class).getFavorited();
|
|
}
|
|
|
|
public void removeFile(int file) throws IOException, ApiException {
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.remove_file);
|
|
builder.put("auth", authData.getAuthKey());
|
|
builder.put("id", file);
|
|
invokeAndReturn(builder, Success.class);
|
|
}
|
|
|
|
public boolean isVersionUpToDate(String versionIdentifier) throws IOException, ApiException {
|
|
//in a development environment, getContainer().getVersion() will return ${version}
|
|
if(versionIdentifier.equals("${version}")) return true;
|
|
|
|
QueryBuilder builder = new QueryBuilder(ReplayModApiMethods.up_to_date);
|
|
builder.put("version", versionIdentifier);
|
|
builder.put("minecraft", ReplayMod.getMinecraftVersion());
|
|
return invokeAndReturn(builder, Success.class).isSuccess();
|
|
}
|
|
|
|
private <T> T invokeAndReturn(QueryBuilder builder, Class<T> classOfT) throws IOException, ApiException {
|
|
return invokeAndReturn(builder.toString(), classOfT);
|
|
}
|
|
|
|
private <T> T invokeAndReturn(String url, Class<T> classOfT) throws IOException, ApiException {
|
|
JsonElement ele = GsonApiClient.invokeJson(url);
|
|
return gson.fromJson(ele, classOfT);
|
|
}
|
|
|
|
@SuppressWarnings("rawtypes")
|
|
private String buildListString(List idList) {
|
|
if(idList == null) return null;
|
|
|
|
String ids = "";
|
|
Integer x = 0;
|
|
for(Object id : idList) {
|
|
x++;
|
|
ids += id.toString();
|
|
if(x != idList.size()) {
|
|
ids += ",";
|
|
}
|
|
}
|
|
|
|
return ids;
|
|
}
|
|
}
|