Plex/src/main/java/dev/plex/util/UpdateChecker.java

package dev.plex.util;

import com.google.common.base.Charsets;
import com.google.gson.Gson;
import com.google.gson.JsonObject;
import com.google.gson.JsonSyntaxException;
import dev.plex.Plex;
import dev.plex.PlexBase;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import javax.annotation.Nonnull;
import net.kyori.adventure.text.Component;
import net.kyori.adventure.text.format.NamedTextColor;
import org.bukkit.ChatColor;

public class UpdateChecker extends PlexBase
{
    private static final String DOWNLOAD_PAGE = "https://ci.plex.us.org/job/Plex/";

    // Adapted from Paper
    private int fetchDistanceFromGitHub(@Nonnull String repo, @Nonnull String branch, @Nonnull String hash)
    {
        try
        {
            HttpURLConnection connection = (HttpURLConnection)new URL("https://api.github.com/repos/" + repo + "/compare/" + branch + "..." + hash).openConnection();
            connection.connect();
            if (connection.getResponseCode() == HttpURLConnection.HTTP_NOT_FOUND)
            {
                return -2; // Unknown commit
            }
            try (BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream(), Charsets.UTF_8)))
            {
                JsonObject obj = new Gson().fromJson(reader, JsonObject.class);
                String status = obj.get("status").getAsString();
                return switch (status)
                        {
                            case "identical" -> 0;
                            case "behind" -> obj.get("behind_by").getAsInt();
                            default -> -1;
                        };
            }
            catch (JsonSyntaxException | NumberFormatException e)
            {
                e.printStackTrace();
                return -1;
            }
        }
        catch (IOException e)
        {
            e.printStackTrace();
            return -1;
        }
    }

    public boolean getUpdateStatusMessage(@Nonnull String repo, @Nonnull String branch)
    {
        int distance;
        distance = fetchDistanceFromGitHub(repo, branch, Plex.build.head);

        switch (distance)
        {
            case -1 -> {
                PlexLog.log(ChatColor.RED + "There was an error checking for updates.");
                return false;
            }
            case 0 -> {
                PlexLog.log(ChatColor.GREEN + "Your version of Plex is up to date!");
                return true;
            }
            case -2 -> {
                PlexLog.log(ChatColor.RED + "Unknown version, unable to check for updates.");
                return false;
            }
            default -> {
                PlexLog.log(Component.text("Your version of Plex is not up to date!", NamedTextColor.RED)
                        .append(Component.newline())
                        .append(Component.text("Download the new version at: ")
                                .append(Component.text(DOWNLOAD_PAGE, NamedTextColor.RED))));
                return true;
            }
        }
    }
}