summaryrefslogtreecommitdiff
path: root/src/main/java/cuchaz/enigma/convert/MatchesReader.java
blob: ee5e48289d3eb5e67bc1750a7132e9e905299b78 (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
104
105
106
107
108
109
/*******************************************************************************
 * 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.convert;

import com.google.common.collect.Lists;

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 cuchaz.enigma.mapping.*;


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;
            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;
            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);
            }
        }
    }
}