summaryrefslogtreecommitdiff
path: root/src/main/java/cuchaz/enigma/mapping/FieldMapping.java
blob: 22ba307ef1df19ad9f73237c9195043bd67b6e00 (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
110
/*******************************************************************************
 * 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.mapping;

import cuchaz.enigma.throwables.IllegalNameException;

public class FieldMapping implements Comparable<FieldMapping>, MemberMapping<FieldEntry> {

    private String obfName;
    private String deobfName;
    private Type obfType;
    private Mappings.EntryModifier modifier;

    public FieldMapping(String obfName, Type obfType, String deobfName, Mappings.EntryModifier modifier) {
        this.obfName = obfName;
        this.deobfName = NameValidator.validateFieldName(deobfName);
        this.obfType = obfType;
        this.modifier = modifier;
    }

    public FieldMapping(FieldMapping other, ClassNameReplacer obfClassNameReplacer) {
        this.obfName = other.obfName;
        this.deobfName = other.deobfName;
        this.modifier = other.modifier;
        this.obfType = new Type(other.obfType, obfClassNameReplacer);
    }

    @Override
    public FieldEntry getObfEntry(ClassEntry classEntry) {
        return new FieldEntry(classEntry, this.obfName, this.obfType);
    }

    @Override
    public String getObfName() {
        return this.obfName;
    }

    public String getDeobfName() {
        return this.deobfName;
    }

    public void setDeobfName(String val) {
        this.deobfName = NameValidator.validateFieldName(val);
    }

    public void setObfName(String name) {
        try
        {
            NameValidator.validateMethodName(name);
        } catch (IllegalNameException ex)
        {
            // Invalid name, damn obfuscation! Map to a deob name with another name to avoid issues
            if (this.deobfName == null)
            {
                System.err.println("WARNING: " + name + " is conflicting, auto deobfuscate to " + (name + "_auto_deob"));
                setDeobfName(name + "_auto_deob");
            }
        }
        this.obfName = name;
    }

    public Type getObfType() {
        return this.obfType;
    }

    public void setObfType(Type val) {
        this.obfType = val;
    }

    public void setModifier(Mappings.EntryModifier modifier)
    {
        this.modifier = modifier;
    }

    public Mappings.EntryModifier getModifier()
    {
        return modifier;
    }

    @Override
    public int compareTo(FieldMapping other) {
        return (this.obfName + this.obfType).compareTo(other.obfName + other.obfType);
    }

    public boolean renameObfClass(final String oldObfClassName, final String newObfClassName) {
        // rename obf classes in the type
        Type newType = new Type(this.obfType, className ->
        {
            if (className.equals(oldObfClassName)) {
                return newObfClassName;
            }
            return null;
        });

        if (!newType.equals(this.obfType)) {
            this.obfType = newType;
            return true;
        }
        return false;
    }

}