summaryrefslogtreecommitdiff
path: root/src/cuchaz/enigma/convert/MatchesReader.java
blob: dac2f057a6026516aec4e911ec72b2956c706182 (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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
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.Entry;
import cuchaz.enigma.mapping.EntryFactory;
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 <T extends Entry> MemberMatches<T> readMembers(File file)
	throws IOException {
		try (BufferedReader in = new BufferedReader(new FileReader(file))) {
			MemberMatches<T> matches = new MemberMatches<T>();
			String line = null;
			while ((line = in.readLine()) != null) {
				readMemberMatch(matches, line);
			}
			return matches;
		}
	}

	private static <T extends Entry> void readMemberMatch(MemberMatches<T> matches, String line) {
		if (line.startsWith("!")) {
			T source = readEntry(line.substring(1));
			matches.addUnmatchableSourceEntry(source);
		} else {
			String[] parts = line.split(":", 2);
			T source = readEntry(parts[0]);
			T dest = readEntry(parts[1]);
			if (source != null && dest != null) {
				matches.addMatch(source, dest);
			} else if (source != null) {
				matches.addUnmatchedSourceEntry(source);
			} else if (dest != null) {
				matches.addUnmatchedDestEntry(dest);
			}
		}
	}

	@SuppressWarnings("unchecked")
	private static <T extends Entry> T readEntry(String in) {
		if (in.length() <= 0) {
			return null;
		}
		String[] parts = in.split(" ");
		if (parts.length == 3 && parts[2].indexOf('(') < 0) {
			return (T)new FieldEntry(
				new ClassEntry(parts[0]),
				parts[1],
				new Type(parts[2])
			);
		} else {
			assert(parts.length == 2 || parts.length == 3);
			if (parts.length == 2) {
				return (T)EntryFactory.getBehaviorEntry(parts[0], parts[1]);
			} else if (parts.length == 3) {
				return (T)EntryFactory.getBehaviorEntry(parts[0], parts[1], parts[2]);
			} else {
				throw new Error("Malformed behavior entry: " + in);
			}
		}
	}
}