finish loader

2 files changed, 172 insertions, 35 deletions

diff --git a/src/main/java/cc/polyfrost/oneconfig/loader/OneConfigLoader.java b/src/main/java/cc/polyfrost/oneconfig/loader/OneConfigLoader.java
deleted file mode 100644
index e7224a8..0000000
--- a/src/main/java/cc/polyfrost/oneconfig/loader/OneConfigLoader.java
+++ /dev/null
@@ -1,35 +0,0 @@
-package cc.polyfrost.oneconfig.loader;
-
-import net.minecraftforge.fml.relauncher.IFMLLoadingPlugin;
-
-import java.util.Map;
-
-
-public class OneConfigLoader implements IFMLLoadingPlugin {
-    private IFMLLoadingPlugin loader = null;
-
-    @Override
-    public String[] getASMTransformerClass() {
-        return loader.getASMTransformerClass();
-    }
-
-    @Override
-    public String getModContainerClass() {
-        return loader.getModContainerClass();
-    }
-
-    @Override
-    public String getSetupClass() {
-        return loader.getSetupClass();
-    }
-
-    @Override
-    public void injectData(Map<String, Object> data) {
-        loader.injectData(data);
-    }
-
-    @Override
-    public String getAccessTransformerClass() {
-        return loader.getAccessTransformerClass();
-    }
-}
diff --git a/src/main/java/cc/polyfrost/oneconfigloader/OneConfigLoader.java b/src/main/java/cc/polyfrost/oneconfigloader/OneConfigLoader.java
new file mode 100644
index 0000000..3ce3527
--- /dev/null
+++ b/src/main/java/cc/polyfrost/oneconfigloader/OneConfigLoader.java
@@ -0,0 +1,172 @@
+package cc.polyfrost.oneconfigloader;
+
+import com.google.gson.JsonElement;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonParser;
+import net.minecraft.launchwrapper.Launch;
+import net.minecraftforge.fml.relauncher.IFMLLoadingPlugin;
+
+import javax.net.ssl.HttpsURLConnection;
+import java.io.*;
+import java.lang.reflect.Method;
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.net.URLConnection;
+import java.nio.file.Files;
+import java.nio.file.StandardCopyOption;
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+import java.util.Arrays;
+import java.util.Map;
+
+
+public class OneConfigLoader implements IFMLLoadingPlugin {
+    private IFMLLoadingPlugin OneConfigTransformer = null;
+
+    public OneConfigLoader() {
+        File oneConfigDir = new File(Launch.minecraftHome, "OneConfig");
+        if (!oneConfigDir.exists() && !oneConfigDir.mkdir())
+            throw new IllegalStateException("Could not create OneConfig dir!");
+        File oneConfigFile = new File(oneConfigDir, "OneConfig (1.8.9).jar");
+        if (isInitialized(oneConfigFile)) return;
+        JsonElement json = getRequest("https://polyfrost.cc/static/oneconfig-versions.json");
+        if (json != null && json.isJsonObject()) {
+            JsonObject jsonObject = json.getAsJsonObject();
+            if (jsonObject.has("oneconfig") && jsonObject.getAsJsonObject("oneconfig").has("url")
+                    && jsonObject.getAsJsonObject("oneconfig").has("checksum")) {
+                String checksum = jsonObject.getAsJsonObject("oneconfig").get("checksum").getAsString();
+                String downloadUrl = jsonObject.getAsJsonObject("oneconfig").get("url").getAsString();
+                if (!oneConfigFile.exists() || !checksum.equals(getChecksum(oneConfigFile))) {
+                    System.out.println("Updating OneConfig...");
+                    File newOneConfigFile = new File(oneConfigDir, "OneConfig-NEW (1.8.9).jar");
+                    downloadFile(downloadUrl, newOneConfigFile);
+                    if (newOneConfigFile.exists() && checksum.equals(getChecksum(newOneConfigFile))) {
+                        try {
+                            Files.move(newOneConfigFile.toPath(), oneConfigFile.toPath(), StandardCopyOption.REPLACE_EXISTING);
+                            System.out.println("Updated OneConfig");
+                        } catch (IOException ignored) {
+                        }
+                    } else {
+                        if (newOneConfigFile.exists()) newOneConfigFile.delete();
+                        System.out.println("Failed to update OneConfig, trying to continue...");
+                    }
+                }
+            }
+        }
+        if (!oneConfigFile.exists()) throw new IllegalStateException("OneConfig jar doesn't exist");
+        addToClasspath(oneConfigFile);
+        try {
+            OneConfigTransformer = ((IFMLLoadingPlugin) Launch.classLoader.findClass("cc.polyfrost.oneconfig.lwjgl.plugin.LoadingPlugin").newInstance());
+        } catch (ClassNotFoundException | InstantiationException | IllegalAccessException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    private boolean isInitialized(File file) {
+        try {
+            URL url = file.toURI().toURL();
+            return Arrays.asList(((URLClassLoader) Launch.classLoader.getClass().getClassLoader()).getURLs()).contains(url);
+        } catch (MalformedURLException e) {
+            e.printStackTrace();
+        }
+        return false;
+    }
+
+    private void addToClasspath(File file) {
+        try {
+            URL url = file.toURI().toURL();
+            Launch.classLoader.addURL(url);
+            ClassLoader classLoader = Launch.classLoader.getClass().getClassLoader();
+            Method method = URLClassLoader.class.getDeclaredMethod("addURL", URL.class);
+            method.setAccessible(true);
+            method.invoke(classLoader, url);
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    private void downloadFile(String url, File location) {
+        try {
+            URLConnection con = new URL(url).openConnection();
+            con.setUseCaches(false);
+            con.setRequestProperty("User-Agent", "OneConfig-Loader");
+            InputStream in = con.getInputStream();
+            Files.copy(in, location.toPath(), StandardCopyOption.REPLACE_EXISTING);
+            in.close();
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
+
+    private static JsonElement getRequest(String site) {
+        try {
+            URL url = new URL(site);
+            HttpsURLConnection con = (HttpsURLConnection) url.openConnection();
+            con.setRequestMethod("GET");
+            int status = con.getResponseCode();
+            if (status != 200) {
+                System.out.println("API request failed, status code " + status);
+                return null;
+            }
+            BufferedReader in = new BufferedReader(new InputStreamReader(con.getInputStream()));
+            String inputLine;
+            StringBuilder content = new StringBuilder();
+            while ((inputLine = in.readLine()) != null) {
+                content.append(inputLine);
+            }
+            in.close();
+            JsonParser parser = new JsonParser();
+            return parser.parse(content.toString());
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+        return null;
+    }
+
+    private String getChecksum(File file) {
+        try (FileInputStream in = new FileInputStream(file)) {
+            MessageDigest digest = MessageDigest.getInstance("SHA-256");
+            byte[] buffer = new byte[1024];
+            int count;
+            while ((count = in.read(buffer)) != -1) {
+                digest.update(buffer, 0, count);
+            }
+            byte[] digested = digest.digest();
+            StringBuilder sb = new StringBuilder();
+            for (byte b : digested) {
+                sb.append(Integer.toString((b & 0xff) + 0x100, 16).substring(1));
+            }
+            System.out.println(sb);
+            return sb.toString();
+        } catch (IOException | NoSuchAlgorithmException e) {
+            e.printStackTrace();
+        }
+        return "";
+    }
+
+    @Override
+    public String[] getASMTransformerClass() {
+        return OneConfigTransformer.getASMTransformerClass();
+    }
+
+    @Override
+    public String getModContainerClass() {
+        return OneConfigTransformer.getModContainerClass();
+    }
+
+    @Override
+    public String getSetupClass() {
+        return OneConfigTransformer.getSetupClass();
+    }
+
+    @Override
+    public void injectData(Map<String, Object> data) {
+        OneConfigTransformer.injectData(data);
+    }
+
+    @Override
+    public String getAccessTransformerClass() {
+        return OneConfigTransformer.getAccessTransformerClass();
+    }
+}