Somewhat working.

Author: Lassebq

src/main/java/org/mcphackers/mcp/MCP.java

@@ -8,13 +8,14 @@
 import org.mcphackers.mcp.tools.VersionsParser;
 
 import java.io.IOException;
+import java.net.URISyntaxException;
 import java.nio.file.Files;
 import java.nio.file.Paths;
 import java.util.*;
 
 public class MCP {
 
-	public static final String VERSION = "v0.1";
+	public static final String VERSION = "Test";
     public static EnumMode mode = null;
     public static EnumMode helpCommand = null;
     public static MCPLogger logger;
@@ -30,24 +31,31 @@ public class MCP {
             .fgCyan().a(" |_|  \\_\\___|\\__|_|  \\___/").fgYellow().a("|_|  |_|\\_____|_|     ").a('\n')
             .fgDefault();
 
-    private static boolean checkIfUpdating(String[] args) throws IOException {
-    	if(args.length == 2) {
+    private static boolean checkIfUpdating(String[] args) throws URISyntaxException, IOException {
+    	System.out.println(Paths.get(MCP.class
+			          .getProtectionDomain()
+			          .getCodeSource()
+			          .getLocation()
+			          .toURI()));
+    	if(args.length >= 2) {
 			if(args[0].equals("update")) {
+				Files.deleteIfExists(Paths.get(args[1]));
+				Files.copy(Paths.get(MCPConfig.UPDATE_JAR), Paths.get(args[1]));
 				Util.runCommand(new String[] {
 					Util.getJava(),
 					"-jar",
 					args[1]
 				});
-				return true;
 			}
+			return true;
     	}
 		return false;
 	}
 
     public static void main(String[] args) throws Exception {
-    	if(checkIfUpdating(args)) return;
-		Files.deleteIfExists(Paths.get(MCPConfig.UPDATE_JAR));
     	SelfCommandPrompt.runWithCMD(SelfCommandPrompt.suggestAppId(), "RetroMCP " + VERSION, args);
+    	if(checkIfUpdating(args)) System.exit(0);
+		//Files.deleteIfExists(Paths.get(MCPConfig.UPDATE_JAR));
     	AnsiConsole.systemInstall();
         logger = new MCPLogger();
         config = new MCPConfig();

src/main/java/org/mcphackers/mcp/tasks/TaskDownloadUpdate.java

@@ -14,6 +14,8 @@
 import org.mcphackers.mcp.tools.FileUtil;
 import org.mcphackers.mcp.tools.Util;
 
+import jredfox.selfcmd.SelfCommandPrompt;
+
 public class TaskDownloadUpdate extends Task {
 
 	private static final String API = "https://api.github.com/repos/MCPHackers/RetroMCP-Java/releases/latest";
@@ -39,6 +41,7 @@ public void doTask() throws Exception {
 						continue;
 					}
 					FileUtil.downloadFile(new URL(assetObj.getString("browser_download_url")), Paths.get(MCPConfig.UPDATE_JAR));
+					//if(!Files.exists(Paths.get(MCPConfig.UPDATE_JAR)))Files.copy(Paths.get("test.jar"), Paths.get(MCPConfig.UPDATE_JAR));
 				}
 			}
 			MCP.logger.info("Press return key to continue");
@@ -49,14 +52,15 @@ public void doTask() throws Exception {
 			          .getLocation()
 			          .toURI());
 			if(!Files.isDirectory(jarPath)) {
-				Util.runCommand(new String[] {
+				String[] cmd = new String[] {
 					Util.getJava(),
 					"-jar",
 					MCPConfig.UPDATE_JAR,
 					"update",
 					jarPath.toString()
-				});
-				System.exit(0);
+				};
+				Util.runCommand(cmd);
+				SelfCommandPrompt.shutdown();
 			}
 			else {
 				throw new IOException("Running from a folder! Aborting");

src/main/java/org/mcphackers/mcp/tools/Util.java

@@ -51,7 +51,9 @@ public void run() {
 
     public static void runCommand(String[] cmd) throws IOException {
         ProcessBuilder procBuilder = new ProcessBuilder(cmd);
-        procBuilder.start();
+        Process proc = procBuilder.start();
+        while(proc.isAlive());
+        proc.exitValue();
     }
 
     public static OS getOperatingSystem() {