From cb8823eb0b446d5c1b9b580e5578866e691771d8 Mon Sep 17 00:00:00 2001
From: liach
Date: Wed, 15 May 2019 22:03:13 -0700
Subject: Feature/weave (#138)

* Add weave/stitch style command system to enigma

Also fixed divide by zero stupidity

Signed-off-by: liach <liach@users.noreply.github.com>

* Add tests for package access index and command

Signed-off-by: liach <liach@users.noreply.github.com>

* Minor tweaks

Signed-off-by: liach <liach@users.noreply.github.com>
---
 src/main/java/cuchaz/enigma/command/Command.java | 140 +++++++++++++++++++++++
 1 file changed, 140 insertions(+)
 create mode 100644 src/main/java/cuchaz/enigma/command/Command.java

(limited to 'src/main/java/cuchaz/enigma/command/Command.java')

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));
+			}
+		}
+	}
+}
-- 
cgit v1.2.3