summaryrefslogtreecommitdiff
path: root/enigma-cli/src/main/java/cuchaz/enigma/command/ComposeMappingsCommand.java
blob: 7e9002d8089afb7ac1dcc8a5aa2c7c1ffb405761 (plain) (blame)
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
package cuchaz.enigma.command;

import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;

import cuchaz.enigma.translation.mapping.EntryMapping;
import cuchaz.enigma.translation.mapping.MappingOperations;
import cuchaz.enigma.translation.mapping.serde.MappingFileNameFormat;
import cuchaz.enigma.translation.mapping.serde.MappingParseException;
import cuchaz.enigma.translation.mapping.serde.MappingSaveParameters;
import cuchaz.enigma.translation.mapping.tree.EntryTree;
import cuchaz.enigma.utils.Utils;

public class ComposeMappingsCommand extends Command {
	public ComposeMappingsCommand() {
		super("compose-mappings");
	}

	@Override
	public String getUsage() {
		return "<left-format> <left> <right-format> <right> <result-format> <result> <keep-mode>";
	}

	@Override
	public boolean isValidArgument(int length) {
		return length == 7;
	}

	@Override
	public void run(String... args) throws IOException, MappingParseException {
		MappingSaveParameters saveParameters = new MappingSaveParameters(MappingFileNameFormat.BY_DEOBF);

		EntryTree<EntryMapping> left = MappingCommandsUtil.read(args[0], Paths.get(args[1]), saveParameters);
		EntryTree<EntryMapping> right = MappingCommandsUtil.read(args[2], Paths.get(args[3]), saveParameters);
		EntryTree<EntryMapping> result = MappingOperations.compose(left, right, args[6].equals("left") || args[6].equals("both"), args[6].equals("right") || args[6].equals("both"));

		Path output = Paths.get(args[5]);
		Utils.delete(output);
		MappingCommandsUtil.write(result, args[4], output, saveParameters);
	}
}