[FEATURE] Implement custom java download

Author: gravit0

components/launcher-api/src/main/java/pro/gravit/launcher/base/LauncherConfig.java

@@ -56,6 +56,10 @@ public final class LauncherConfig extends StreamObject {
     public LauncherEnvironment environment;
     @LauncherInject("runtimeconfig.buildNumber")
     public long buildNumber;
+    @LauncherInject("runtimeconfig.customJavaDownload")
+    public Map<String, String> customJavaDownload;
+    @LauncherInject("runtimeconfig.forceUseCustomJava")
+    public boolean forceUseCustomJava;
 
     private static class ModernModulesClass {
         @LauncherInject("launcher.modules")

components/launcher-runtime/src/main/java/pro/gravit/launcher/runtime/backend/ClientDownloadImpl.java

@@ -43,7 +43,12 @@ CompletableFuture<LauncherBackendAPI.ReadyProfile> downloadProfile(ClientProfile
             return downloadAsset(profile.getAssetDir(), profile.getAssetUpdateMatcher(), profile.getAssetIndex(), callback);
         })).thenCompose(assetDir -> {
             assetRef.set(assetDir);
-            return CompletableFuture.completedFuture((DownloadedDir)null); // TODO Custom Java
+            Path javaPath = settings.getSelectedJava().getPath();
+            if(javaPath.startsWith(DirBridge.dirUpdates)) {
+                String javaDirName = DirBridge.dirUpdates.relativize(javaPath).getFileName().toString();
+                return downloadDir(javaDirName, null, callback);
+            }
+            return CompletableFuture.completedFuture((DownloadedDir)null);
         }).thenCompose(javaDir -> {
             javaRef.set(javaDir);
             return CompletableFuture.completedFuture(null);

components/launcher-runtime/src/main/java/pro/gravit/launcher/runtime/backend/LauncherBackendImpl.java

@@ -27,10 +27,7 @@
 import pro.gravit.launcher.runtime.managers.SettingsManager;
 import pro.gravit.launcher.runtime.utils.HWIDProvider;
 import pro.gravit.launcher.runtime.utils.LauncherUpdater;
-import pro.gravit.utils.helper.IOHelper;
-import pro.gravit.utils.helper.JavaHelper;
-import pro.gravit.utils.helper.LogHelper;
-import pro.gravit.utils.helper.SecurityHelper;
+import pro.gravit.utils.helper.*;
 
 import java.io.IOException;
 import java.net.URI;
@@ -40,6 +37,7 @@
 import java.time.LocalDateTime;
 import java.time.ZoneOffset;
 import java.time.temporal.ChronoUnit;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import java.util.UUID;
@@ -48,6 +46,8 @@
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 import java.util.function.Function;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 import java.util.stream.Collectors;
 
 public class LauncherBackendImpl implements LauncherBackendAPI, TextureUploadExtension {
@@ -244,13 +244,16 @@ public CompletableFuture<byte[]> fetchTexture(Texture texture) {
         return CompletableFuture.failedFuture(new UnsupportedOperationException());
     }
 
-    @SuppressWarnings({"rawtypes", "unchecked"})
     @Override
     public CompletableFuture<List<Java>> getAvailableJava() {
         if(availableJavas == null) {
             if(availableJavasFuture == null) {
                 availableJavasFuture = CompletableFuture.supplyAsync(() -> {
-                    return (List) JavaHelper.findJava(); // TODO: Custom Java
+                    List<Java> javas = getCustomJava();
+                    if(!Launcher.getConfig().forceUseCustomJava && !javas.isEmpty()) {
+                        javas.addAll(JavaHelper.findJava());
+                    }
+                    return javas;
                 }, executorService).thenApply(e -> {
                     availableJavas = e;
                     return e;
@@ -261,6 +264,54 @@ public CompletableFuture<List<Java>> getAvailableJava() {
         return CompletableFuture.completedFuture(availableJavas);
     }
 
+    private static final Pattern JAVA_VERSION_PATTERN = Pattern.compile(
+            "Java (?<version>.+) b(?<build>.+) (?<os>.+) (?<arch>.+) javafx (?<javafx>.+)");
+
+    public List<Java> getCustomJava() {
+        List<Java> versions = new ArrayList<>();
+        for (Map.Entry<String, String> entry : Launcher.getConfig().customJavaDownload.entrySet()) {
+            String javaDir = entry.getKey();
+            String javaVersionString = entry.getValue();
+            Matcher matcher = JAVA_VERSION_PATTERN.matcher(javaVersionString);
+            if (matcher.matches()) {
+                String os = matcher.group("os");
+                int version = Integer.parseInt(matcher.group("version"));
+                int build = Integer.parseInt(matcher.group("build"));
+                JVMHelper.ARCH arch = JVMHelper.ARCH.valueOf(matcher.group("arch"));
+                boolean javafx = Boolean.parseBoolean(matcher.group("javafx"));
+                if (!isArchAvailable(arch)) {
+                    continue;
+                }
+                if (!JVMHelper.OS_TYPE.name.equals(os)) {
+                    continue;
+                }
+                Path javaDirectory = DirBridge.dirUpdates.resolve(javaDir);
+                LogHelper.debug("In-Launcher Java Version found: Java %d b%d %s javafx %s", version, build,
+                        arch.name, Boolean.toString(javafx));
+                JavaHelper.JavaVersion javaVersion = new JavaHelper.JavaVersion(javaDirectory, version, build,
+                        arch, javafx);
+                versions.add(javaVersion);
+            } else {
+                LogHelper.warning("Java Version: %s does not match", javaVersionString);
+            }
+        }
+        return versions;
+    }
+
+    public boolean isArchAvailable(JVMHelper.ARCH arch) {
+        if (JVMHelper.ARCH_TYPE == arch) {
+            return true;
+        }
+        if (arch == JVMHelper.ARCH.X86_64 && JVMHelper.OS_TYPE == JVMHelper.OS.MUSTDIE
+                && ((JVMHelper.ARCH_TYPE == JVMHelper.ARCH.X86 && !JVMHelper.isJVMMatchesSystemArch())
+                || JVMHelper.ARCH_TYPE == JVMHelper.ARCH.ARM64)) {
+            return true;
+        }
+        return arch == JVMHelper.ARCH.X86_64
+                && JVMHelper.OS_TYPE == JVMHelper.OS.MACOSX
+                && JVMHelper.ARCH_TYPE == JVMHelper.ARCH.ARM64;
+    }
+
     @Override
     public CompletableFuture<ServerPingInfo> pingServer(ProfileFeatureAPI.ClientProfile profile) {
         return pingFutures.computeIfAbsent(profile.getUUID(), (k) -> {

components/launcher-runtime/src/main/java/pro/gravit/launcher/runtime/backend/ProfileSettingsImpl.java

@@ -124,6 +124,9 @@ public void disableOptional(ProfileFeatureAPI.OptionalMod mod, ChangedOptionalSt
 
     @Override
     public JavaHelper.JavaVersion getSelectedJava() {
+        if(selectedJava == null) {
+            selectedJava = getRecommendedJava();
+        }
         return selectedJava;
     }
 

components/launchserver/src/main/java/pro/gravit/launchserver/binary/tasks/MainBuildTask.java

@@ -113,6 +113,8 @@ protected void initProps(BuildContext context) {
         context.properties.put("launchercore.env", server.config.env);
         context.properties.put("launcher.memory", server.config.launcher.memoryLimit);
         context.properties.put("launcher.customJvmOptions", server.config.launcher.customJvmOptions);
+        context.properties.put("runtimeconfig.customJavaDownload", server.config.launcher.customJavaDownload);
+        context.properties.put("runtimeconfig.forceUseCustomJava", server.config.launcher.forceUseCustomJava);
         if (server.config.launcher.encryptRuntime) {
             String runtimeEncryptKey = SecurityHelper.randomStringToken();
             context.pipelineContext.putProperty("runtimeEncryptKey", runtimeEncryptKey);

components/launchserver/src/main/java/pro/gravit/launchserver/config/LaunchServerConfig.java

@@ -251,6 +251,8 @@ public static class LauncherConf {
         public boolean certificatePinning;
         public boolean encryptRuntime;
         public List<String> customJvmOptions = new ArrayList<>();
+        public Map<String, String> customJavaDownload = new HashMap<>();
+        public boolean forceUseCustomJava;
         public int memoryLimit = 256;
     }