1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
/*******************************************************************************
* Copyright (c) 2015 Jeff Martin.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the GNU Lesser General Public
* License v3.0 which accompanies this distribution, and is available at
* http://www.gnu.org/licenses/lgpl.html
* <p>
* Contributors:
* Jeff Martin - initial API and implementation
******************************************************************************/
package cuchaz.enigma;
import cuchaz.enigma.command.*;
import java.util.LinkedHashMap;
import java.util.Locale;
import java.util.Map;
public class CommandMain {
private static final Map<String, Command> COMMANDS = new LinkedHashMap<>();
public static void main(String... args) throws Exception {
try {
// process the command
if (args.length < 1)
throw new IllegalArgumentException("Requires a command");
String command = args[0].toLowerCase(Locale.ROOT);
Command cmd = COMMANDS.get(command);
if (cmd == null)
throw new IllegalArgumentException("Command not recognized: " + command);
if (!cmd.isValidArgument(args.length - 1)) {
throw new CommandHelpException(cmd);
}
String[] cmdArgs = new String[args.length - 1];
System.arraycopy(args, 1, cmdArgs, 0, args.length - 1);
try {
cmd.run(cmdArgs);
} catch (Exception ex) {
throw new CommandHelpException(cmd, ex);
}
} catch (CommandHelpException ex) {
System.err.println(ex.getMessage());
System.out.println(String.format("%s - %s", Constants.NAME, Constants.VERSION));
System.out.println("Command " + ex.command.name + " has encountered an error! Usage:");
printHelp(ex.command);
System.exit(1);
} catch (IllegalArgumentException ex) {
System.err.println(ex.getMessage());
printHelp();
System.exit(1);
}
}
private static void printHelp() {
System.out.println(String.format("%s - %s", Constants.NAME, Constants.VERSION));
System.out.println("Usage:");
System.out.println("\tjava -cp enigma.jar cuchaz.enigma.CommandMain <command>");
System.out.println("\twhere <command> is one of:");
for (Command command : COMMANDS.values()) {
printHelp(command);
}
}
private static void printHelp(Command command) {
System.out.println("\t\t" + command.name + " " + command.getUsage());
}
private static void register(Command command) {
Command old = COMMANDS.put(command.name, command);
if (old != null) {
System.err.println("Command " + old + " with name " + command.name + " has been substituted by " + command);
}
}
static {
register(new DeobfuscateCommand());
register(new DecompileCommand());
register(new ConvertMappingsCommand());
register(new ComposeMappingsCommand());
register(new InvertMappingsCommand());
register(new CheckMappingsCommand());
register(new MapSpecializedMethodsCommand());
}
private static final class CommandHelpException extends IllegalArgumentException {
final Command command;
CommandHelpException(Command command) {
this.command = command;
}
CommandHelpException(Command command, Throwable cause) {
super(cause);
this.command = command;
}
}
}
|
