Please wait. This can take some minutes ...
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
net.guizhanss.guizhanlib.updater.UpdaterTask Maven / Gradle / Ivy
package net.guizhanss.guizhanlib.updater;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import net.guizhanss.guizhanlib.utils.JsonUtil;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;
import java.nio.charset.StandardCharsets;
import java.text.MessageFormat;
import java.util.logging.Level;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
/**
* Auto update task.
*
* @author ybw0014
*/
class UpdaterTask implements Runnable {
private final AbstractGuizhanBuildsUpdater updater;
private String workingDirectory;
private JsonObject repoInfo = null;
private JsonObject updateInfo = null;
UpdaterTask(AbstractGuizhanBuildsUpdater updater) {
this.updater = updater;
}
@Override
public void run() {
getRepoInfo();
String format = getVersionFormat();
if (format == null) {
updater.log(Level.SEVERE, updater.getLocalizedString("invalid_version_format", "Cannot get version format. Maybe the developer did not configure the updater correctly?"));
return;
}
if (!checkVersion(format)) {
updater.log(Level.WARNING, updater.getLocalizedString("invalid_version", "You are using an unofficial version, which is not downloaded from Guizhan Builds. The updater has stopped."));
return;
}
if (hasUpdate()) {
if (updater.checkOnly()) {
sendUpdateNotification();
} else {
update();
}
}
}
/**
* Get repository information.
*/
private void getRepoInfo() {
try {
URL repos = new URL(updater.getReposFileURL());
JsonObject reposJson = (JsonObject) JsonUtil.parse(fetch(repos));
String key = updater.getRepoKey();
JsonElement repoInfo = null;
while (key != null) {
repoInfo = JsonUtil.getFromPath(reposJson, key);
if (repoInfo == null) {
break;
}
if (JsonUtil.getFromPath((JsonObject) repoInfo, "type").getAsString().equals("redirect")) {
key = JsonUtil.getFromPath((JsonObject) repoInfo, "options.repo").getAsString();
} else {
key = null;
}
}
if (repoInfo == null) {
throw new IllegalStateException("Repository information is not found");
}
this.repoInfo = (JsonObject) repoInfo;
// Get working directory
JsonElement customDir = JsonUtil.getFromPath(this.repoInfo, "options.customDir");
if (customDir != null) {
this.workingDirectory = customDir.getAsString();
} else {
this.workingDirectory = MessageFormat.format(
"{0}/{1}/{2}",
updater.getUser(),
updater.getRepo(),
updater.getBranch()
);
}
} catch (MalformedURLException | IllegalStateException ex) {
updater.log(Level.SEVERE, updater.getLocalizedString("cannot_find_repo", "Cannot find repo in Guizhan Builds."), ex);
}
}
/**
* Get the version format.
*
* @return the version format, {@code null} if failed
*/
@Nullable
private String getVersionFormat() {
try {
return JsonUtil.getFromPath(repoInfo, "options.target.version").getAsString();
} catch (IllegalStateException | NullPointerException ex) {
return null;
}
}
/**
* Check if version format matches.
*
* @param format The version format.
*
* @return Whether the version format matches.
*/
private boolean checkVersion(String format) {
String regex = format.replace("(", "\\(")
.replace(")", "\\)")
.replace("{version}", "\\d{1,6}")
.replace("{git_commit}", "([a-z0-9]{7})")
.replace("{year}", "\\d{1,4}")
.replace("{month}", "\\d{1,2}")
.replace("{day}", "\\d{1,2}");
Pattern pattern = Pattern.compile(regex);
Matcher m = pattern.matcher(updater.getPlugin().getDescription().getVersion());
return m.matches();
}
/**
* Check if there is new build.
*
* @return Whether new build exists.
*/
private boolean hasUpdate() {
try {
// Retrieve builds info
URL buildsUrl = new URL(updater.getBuildsInfo(workingDirectory));
JsonObject buildsJson = (JsonObject) JsonUtil.parse(fetch(buildsUrl));
JsonArray builds = (JsonArray) JsonUtil.getFromPath(buildsJson, "builds");
JsonObject build = null;
for (int i = builds.size() - 1; i >= 0; i--) {
build = (JsonObject) builds.get(i);
if (build.get("success").getAsBoolean()) break;
build = null;
}
if (build == null) {
updater.log(Level.SEVERE, updater.getLocalizedString("no_successful_builds", "There is no successful build in Guizhan Builds. How did you do this?"));
return false;
}
String pluginName = JsonUtil.getFromPath(this.repoInfo, "options.target.name").getAsString();
boolean needUpdate = !MessageFormat.format("{0}-{1}.jar", pluginName, updater.getPlugin().getDescription().getVersion()).equals(build.get("target").getAsString());
if (!needUpdate) {
updater.log(Level.INFO, updater.getLocalizedString("latest", "{0} is up-to-date."), updater.getPlugin().getName());
return false;
}
updateInfo = build;
return true;
} catch (MalformedURLException | NullPointerException ex) {
updater.log(Level.SEVERE, updater.getLocalizedString("an_error_has_occurred", "An error has occurred while checking for update"));
return false;
}
}
/**
* Send update notification.
*/
private void sendUpdateNotification() {
updater.log(Level.INFO, updater.getLocalizedString("need_update", "{0} needs to be updated."), updater.getPlugin().getName());
updater.log(Level.INFO, updater.getLocalizedString("download_notification", "Since this updater is set to check update only, you need to download the new version of {0} manually from Guizhan Builds, or disable checkOnly option."), updater.getPlugin().getName());
}
/**
* Download and install the latest version
*/
private void update() {
String targetFilename = updateInfo.get("target").getAsString();
updater.log(Level.INFO, updater.getLocalizedString("need_update", "{0} needs to be updated."), updater.getPlugin().getName());
updater.log(Level.INFO, updater.getLocalizedString("downloading", "Downloading {0} - Build {1}"), updater.getPlugin().getName(), updateInfo.get("id").getAsString());
try {
BufferedInputStream input = new BufferedInputStream(new URL(updater.getTargetUrl(workingDirectory, targetFilename)).openStream());
FileOutputStream output = new FileOutputStream(new File("plugins/" + updater.getPlugin().getServer().getUpdateFolder(), updater.getFile().getName()));
byte[] data = new byte[1024];
int read;
while ((read = input.read(data, 0, 1024)) != -1) {
output.write(data, 0, read);
}
} catch (Exception ex) {
updater.log(Level.SEVERE, ex, MessageFormat.format(updater.getLocalizedString("update_fail", "Failed to update {0}"), updater.getPlugin().getName()));
return;
}
updater.log(Level.INFO, " ");
updater.log(Level.INFO, updater.getLocalizedString("info_auto_update", "============== Auto update =============="));
updater.log(Level.INFO, updater.getLocalizedString("info_auto_update_1", "Downloaded {0} Build {1}"), updater.getPlugin().getName(), updateInfo.get("id").getAsString());
updater.log(Level.INFO, updater.getLocalizedString("info_auto_update_2", "Restart server to update"));
updater.log(Level.INFO, " ");
}
/**
* Fetch information from {@link URL}.
*
* @param url The {@link URL} of resource.
*
* @return The content {@link String}.
*/
@Nullable
private String fetch(@Nonnull URL url) {
try {
StringBuilder content = new StringBuilder();
URLConnection connection = url.openConnection();
connection.setConnectTimeout(10_000);
connection.addRequestProperty("User-Agent", "Guizhan Updater");
connection.setDoOutput(true);
try (BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream(), StandardCharsets.UTF_8))) {
String line = reader.readLine();
while (line != null) {
content.append(line);
line = reader.readLine();
}
}
return content.toString();
} catch (IOException | NullPointerException ex) {
updater.log(Level.WARNING, updater.getLocalizedString("cannot_fetch_info", "Cannot fetch info from Guizhan Builds"), ex);
return null;
}
}
}