Modpack file importing

Author: North-West-Wind

src/ml/northwestwind/Config.java

@@ -11,7 +11,7 @@
 import java.util.stream.Collectors;
 
 public class Config {
-    public static File directory, modpackDir, profileDir, exportDir;
+    public static File directory, modpackDir, profileDir, exportDir, tempDir;
 
     public static void run(String[] args) {
         if (args.length < 3) {
@@ -58,6 +58,7 @@ public static void load() {
             if (!profileDir.exists() || !profileDir.isDirectory()) profileDir.mkdir();
             exportDir = new File(directory.getPath() + File.separator + "exported");
             if (!exportDir.exists() || !exportDir.isDirectory()) exportDir.mkdir();
+            tempDir = new File(directory.getPath() + File.separator + "tmp");
         } catch(Exception e) {
             e.printStackTrace();
         }

src/ml/northwestwind/Main.java

@@ -4,8 +4,9 @@
 
 import java.util.Arrays;
 
-import static org.fusesource.jansi.Ansi.*;
-import static org.fusesource.jansi.Ansi.Color.*;
+import static org.fusesource.jansi.Ansi.Color.GREEN;
+import static org.fusesource.jansi.Ansi.Color.YELLOW;
+import static org.fusesource.jansi.Ansi.ansi;
 
 public class Main {
     public static void main(String[] args) {

src/ml/northwestwind/Mod.java

@@ -5,10 +5,8 @@
 import org.json.simple.JSONArray;
 import org.json.simple.JSONObject;
 
-import java.io.File;
 import java.io.PrintWriter;
 import java.net.URLEncoder;
-import java.nio.charset.Charset;
 import java.util.Arrays;
 import java.util.HashMap;
 import java.util.Map;

src/ml/northwestwind/Modpack.java

@@ -6,7 +6,10 @@
 import org.json.simple.JSONObject;
 import org.json.simple.parser.JSONParser;
 
-import java.io.*;
+import java.io.File;
+import java.io.FileReader;
+import java.io.IOException;
+import java.io.PrintWriter;
 import java.util.*;
 import java.util.concurrent.atomic.AtomicInteger;
 
@@ -58,14 +61,14 @@ private static void install(String[] ids) {
         }
     }
 
-    private static void copyFromOverride(String folder, String overridesName) throws IOException {
+    public static void copyFromOverride(String folder, String overridesName) throws IOException {
         File overrides = new File(folder + File.separator + overridesName);
         if (!overrides.exists()) return;
         FileUtils.copyDirectory(overrides, new File(folder));
         FileUtils.deleteDirectory(overrides);
     }
 
-    private static void downloadMods(String folder) {
+    public static void downloadMods(String folder) {
         System.out.println(Ansi.ansi().fg(Ansi.Color.CYAN).a("Starting mod download").reset());
         try {
             File modsFolder = new File(folder + File.separator + "mods");
@@ -95,7 +98,7 @@ private static void downloadMods(String folder) {
                     StringBuilder formatted = new StringBuilder(String.format("[%d/%d] [S: %d | F: %d] Downloaded %s", ++i, array.size(), suc, fai, name));
                     int ll = formatted.length() - lastlength;
                     if (ll > 0) for (int ii = 0; ii < ll; ii++) formatted.append(" ");
-                    System.out.print(Ansi.ansi().fg(Ansi.Color.GREEN).a(formatted.toString()).reset()+"\r");
+                    System.out.print(Ansi.ansi().fg(Ansi.Color.GREEN).a(formatted.toString()).reset().a("\r"));
                     System.out.flush();
                     lastlength = formatted.length();
                 }

src/ml/northwestwind/Profile.java

@@ -30,6 +30,7 @@ public static void run(String[] args) {
         else if (cmd.equalsIgnoreCase("export")) export(args);
         else if (cmd.equalsIgnoreCase("edit")) edit(args[0]);
         else if (cmd.equalsIgnoreCase("delete")) delete(args);
+        else if (cmd.equals("import")) importProf(args);
         else Utils.invalid();
     }
 
@@ -491,6 +492,53 @@ private static void update(String[] args) {
         }
     }
 
+    private static void importProf(String[] paths) {
+        if (!Config.tempDir.exists() || !Config.tempDir.isDirectory()) Config.tempDir.mkdir();
+        else {
+            System.out.println(Ansi.ansi().fg(Ansi.Color.RED).a("Another instance is importing a modpack. Please wait until it is finished."));
+            return;
+        }
+        for (String path : paths) {
+            File zipped = new File(path);
+            try {
+                if (!zipped.exists()) throw new Exception("The file " + path + " does not exist");
+                if (zipped.isFile()) {
+                    FileUtils.copyFileToDirectory(zipped, Config.tempDir);
+                    if (!Utils.unzip(Config.tempDir.getPath() + File.separator + zipped.getName())) throw new Exception("Failed to extract modpack content of "+path);
+                } else FileUtils.copyDirectoryToDirectory(zipped, Config.tempDir);
+                File manifest = new File(Config.tempDir.getPath() + File.separator + "manifest.json");
+                if (!manifest.exists() || !manifest.isFile()) throw new Exception("Modpack is missing manifest. Cannot convert to profile.");
+                Modpack.copyFromOverride(Config.tempDir.getPath(), (String) ((JSONObject) parser.parse(new FileReader(manifest))).get("overrides"));
+                Modpack.downloadMods(Config.tempDir.getPath());
+                JSONObject json = (JSONObject) parser.parse(new FileReader(manifest));
+                String slug = ((String) json.get("name")).toLowerCase().replaceAll("[^a-z0-9]", "-");
+                FileUtils.moveDirectoryToDirectory(Config.tempDir, Config.profileDir, true);
+                File packFolder = new File(Config.profileDir.getPath() + File.separator + "tmp");
+                File newFolder = new File(Config.profileDir.getPath() + File.separator + slug);
+                if (!packFolder.renameTo(newFolder)) {
+                    System.out.println(Ansi.ansi().fg(Ansi.Color.RED).a("Failed to rename modpack, but we are continuing anyway."));
+                    newFolder = packFolder;
+                }
+                JSONObject profile = new JSONObject();
+                JSONObject minecraft = (JSONObject) json.get("minecraft");
+                String[] launcher = ((String) ((JSONObject) ((JSONArray) minecraft.get("modLoaders")).get(0)).get("id")).split("-", 2);
+                profile.put("name", json.get("name"));
+                profile.put("mcVer", minecraft.get("version"));
+                profile.put("launcher", launcher[0]);
+                profile.put("modVer", launcher[1]);
+                PrintWriter pw = new PrintWriter(newFolder.getPath() + File.separator + "profile.json");
+                pw.write(profile.toJSONString());
+
+                pw.flush();
+                pw.close();
+                manifest.delete();
+                System.out.println(Ansi.ansi().fg(Ansi.Color.GREEN).a("Converted modpack " + slug + " into profile."));
+            } catch (Exception e) {
+                e.printStackTrace();
+            }
+        }
+    }
+
     public static void printHelp(String prefix) {
         System.out.println(prefix + "profile: Commands for custom profiles.");
         System.out.println(prefix + "\tcreate: Create a profile.");
@@ -508,5 +556,7 @@ public static void printHelp(String prefix) {
         System.out.println(prefix + "\tupdate: Update mods in the profile.");
         System.out.println(prefix + "\t\targ <profile>: Name of the profile to target.");
         System.out.println(prefix + "\t\targ [ID]: The ID of mods. Omit to check for updates.");
+        System.out.println(prefix + "\timport: Import a downloaded modpack.");
+        System.out.println(prefix + "\t\targ <path>: Path to the zip or directory.");
     }
 }

src/ml/northwestwind/Utils.java

@@ -1,6 +1,5 @@
 package ml.northwestwind;
 
-import org.apache.commons.io.Charsets;
 import org.apache.commons.io.FileUtils;
 import org.fusesource.jansi.Ansi;
 import org.json.simple.JSONArray;
@@ -10,7 +9,6 @@
 import java.io.*;
 import java.net.HttpURLConnection;
 import java.net.URL;
-import java.net.URLEncoder;
 import java.nio.charset.StandardCharsets;
 import java.util.Arrays;
 import java.util.List;