Merge remote-tracking branch 'origin/master' into proposal-tweak

1 files changed, 140 insertions, 0 deletions

diff --git a/src/main/java/cuchaz/enigma/command/Command.java b/src/main/java/cuchaz/enigma/command/Command.java
new file mode 100644
index 0000000..b107fb6
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/command/Command.java
@@ -0,0 +1,140 @@
+package cuchaz.enigma.command;
+
+import cuchaz.enigma.Deobfuscator;
+import cuchaz.enigma.ProgressListener;
+import cuchaz.enigma.translation.mapping.EntryMapping;
+import cuchaz.enigma.translation.mapping.serde.MappingFormat;
+import cuchaz.enigma.translation.mapping.tree.EntryTree;
+
+import java.io.File;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.jar.JarFile;
+
+public abstract class Command {
+        public final String name;
+
+        protected Command(String name) {
+                this.name = name;
+        }
+
+        public abstract String getUsage();
+
+        public abstract boolean isValidArgument(int length);
+
+        public abstract void run(String... args) throws Exception;
+
+        protected static Deobfuscator getDeobfuscator(Path fileMappings, JarFile jar) throws Exception {
+                System.out.println("Reading jar...");
+                Deobfuscator deobfuscator = new Deobfuscator(jar);
+                if (fileMappings != null) {
+                        System.out.println("Reading mappings...");
+                        EntryTree<EntryMapping> mappings = chooseEnigmaFormat(fileMappings).read(fileMappings, new ConsoleProgressListener());
+                        deobfuscator.setMappings(mappings);
+                }
+                return deobfuscator;
+        }
+
+        protected static MappingFormat chooseEnigmaFormat(Path path) {
+                if (Files.isDirectory(path)) {
+                        return MappingFormat.ENIGMA_DIRECTORY;
+                } else {
+                        return MappingFormat.ENIGMA_FILE;
+                }
+        }
+
+        protected static File getWritableFile(String path) {
+                if (path == null) {
+                        return null;
+                }
+                File file = new File(path).getAbsoluteFile();
+                File dir = file.getParentFile();
+                if (dir == null) {
+                        throw new IllegalArgumentException("Cannot write file: " + path);
+                }
+                // quick fix to avoid stupid stuff in Gradle code
+                if (!dir.isDirectory()) {
+                        dir.mkdirs();
+                }
+                return file;
+        }
+
+        protected static File getWritableFolder(String path) {
+                if (path == null) {
+                        return null;
+                }
+                File dir = new File(path).getAbsoluteFile();
+                if (!dir.exists()) {
+                        throw new IllegalArgumentException("Cannot write to folder: " + dir);
+                }
+                return dir;
+        }
+
+        protected static File getReadableFile(String path) {
+                if (path == null) {
+                        return null;
+                }
+                File file = new File(path).getAbsoluteFile();
+                if (!file.exists()) {
+                        throw new IllegalArgumentException("Cannot find file: " + file.getAbsolutePath());
+                }
+                return file;
+        }
+
+        protected static Path getReadablePath(String path) {
+                if (path == null) {
+                        return null;
+                }
+                Path file = Paths.get(path).toAbsolutePath();
+                if (!Files.exists(file)) {
+                        throw new IllegalArgumentException("Cannot find file: " + file.toString());
+                }
+                return file;
+        }
+
+        protected static String getArg(String[] args, int i, String name, boolean required) {
+                if (i >= args.length) {
+                        if (required) {
+                                throw new IllegalArgumentException(name + " is required");
+                        } else {
+                                return null;
+                        }
+                }
+                return args[i];
+        }
+
+        public static class ConsoleProgressListener implements ProgressListener {
+
+                private static final int ReportTime = 5000; // 5s
+
+                private int totalWork;
+                private long startTime;
+                private long lastReportTime;
+
+                @Override
+                public void init(int totalWork, String title) {
+                        this.totalWork = totalWork;
+                        this.startTime = System.currentTimeMillis();
+                        this.lastReportTime = this.startTime;
+                        System.out.println(title);
+                }
+
+                @Override
+                public void step(int numDone, String message) {
+                        long now = System.currentTimeMillis();
+                        boolean isLastUpdate = numDone == this.totalWork;
+                        boolean shouldReport = isLastUpdate || now - this.lastReportTime > ReportTime;
+
+                        if (shouldReport) {
+                                int percent = numDone * 100 / this.totalWork;
+                                System.out.println(String.format("\tProgress: %3d%%", percent));
+                                this.lastReportTime = now;
+                        }
+                        if (isLastUpdate) {
+                                double elapsedSeconds = (now - this.startTime) / 1000.0;
+                                System.out.println(String.format("Finished in %.1f seconds", elapsedSeconds));
+                        }
+                }
+        }
+}