summaryrefslogtreecommitdiff
path: root/src/cuchaz/enigma/convert/MatchesReader.java
blob: 1dd042daf78627466253fb5834427e89bb08d72b (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
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
package cuchaz.enigma.convert;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.Collection;
import java.util.List;

import com.beust.jcommander.internal.Lists;

import cuchaz.enigma.mapping.ClassEntry;
import cuchaz.enigma.mapping.FieldEntry;
import cuchaz.enigma.mapping.Type;


public class MatchesReader {
	
	public static ClassMatches readClasses(File file)
	throws IOException {
		try (BufferedReader in = new BufferedReader(new FileReader(file))) {
			ClassMatches matches = new ClassMatches();
			String line = null;
			while ((line = in.readLine()) != null) {
				matches.add(readClassMatch(line));
			}
			return matches;
		}
	}

	private static ClassMatch readClassMatch(String line)
	throws IOException {
		String[] sides = line.split(":", 2);
		return new ClassMatch(readClasses(sides[0]), readClasses(sides[1]));
	}

	private static Collection<ClassEntry> readClasses(String in) {
		List<ClassEntry> entries = Lists.newArrayList();
		for (String className : in.split(",")) {
			className = className.trim();
			if (className.length() > 0) {
				entries.add(new ClassEntry(className));
			}
		}
		return entries;
	}

	public static FieldMatches readFields(File file)
	throws IOException {
		try (BufferedReader in = new BufferedReader(new FileReader(file))) {
			FieldMatches matches = new FieldMatches();
			String line = null;
			while ((line = in.readLine()) != null) {
				readFieldMatch(matches, line);
			}
			return matches;
		}
	}

	private static void readFieldMatch(FieldMatches matches, String line) {
		String[] parts = line.split(":", 2);
		FieldEntry source = readField(parts[0]);
		FieldEntry dest = readField(parts[1]);
		if (source != null && dest != null) {
			matches.addMatch(source, dest);
		} else if (source != null) {
			matches.addUnmatchedSourceField(source);
		} else if (dest != null) {
			matches.addUnmatchedDestField(dest);
		}
	}

	private static FieldEntry readField(String in) {
		if (in.length() <= 0) {
			return null;
		}
		String[] parts = in.split(" ");
		assert(parts.length == 3);
		return new FieldEntry(
			new ClassEntry(parts[0]),
			parts[1],
			new Type(parts[2])
		);
	}
}