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
|
package moe.nea.zwirn;
import net.fabricmc.stitch.commands.tinyv2.*;
import org.jetbrains.annotations.NotNull;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.stream.Collectors;
public class RenameTask {
private final TinyFile tinyFile;
private final List<Zwirn.RenameCommand> newNamespaceOrder;
private final int[] namespaceRemapOrder;
private final SimpleRemapper remapper;
public RenameTask(@NotNull TinyFile tinyFile, @NotNull List<Zwirn.RenameCommand> newNamespaceOrder) {
this.tinyFile = tinyFile;
this.newNamespaceOrder = newNamespaceOrder;
namespaceRemapOrder = newNamespaceOrder.stream().mapToInt(
it -> tinyFile.getHeader().getNamespaces().indexOf(it.oldNamespaceName())
).toArray();
this.remapper = new SimpleRemapper(
classOnlyRename(), "__old", "__new"
);
}
public TinyFile classOnlyRename() {
List<String> namespaces = new ArrayList<>();
namespaces.add("__old");
namespaces.add("__new");
return new TinyFile(
new TinyHeader(namespaces, 2, 0, new HashMap<>()),
tinyFile.getClassEntries().stream().map(
it -> new TinyClass(
Arrays.asList(
it.getClassNames().get(0),
it.getClassNames().get(namespaceRemapOrder[0]))
)
).collect(Collectors.toList())
);
}
private List<String> rename(List<String> strings) {
List<String> newNames = new ArrayList<>(namespaceRemapOrder.length);
for (int j : namespaceRemapOrder) {
newNames.add(strings.get(j));
}
return newNames;
}
public TinyFile rename() {
return new TinyFile(
new TinyHeader(newNamespaceOrder.stream().map(it -> it.newNamespaceName()).collect(Collectors.toList()), 2, 0, tinyFile.getHeader().getProperties()),
tinyFile.getClassEntries().stream().map(this::renameClass).collect(Collectors.toList())
);
}
private TinyClass renameClass(TinyClass tinyClass) {
return new TinyClass(
rename(tinyClass.getClassNames()),
tinyClass.getMethods().stream().map(this::renameMethod).collect(Collectors.toList()),
tinyClass.getFields().stream().map(this::renameField).collect(Collectors.toList()),
tinyClass.getComments()
);
}
private TinyField renameField(TinyField tinyField) {
var names = rename(tinyField.getFieldNames());
return new TinyField(
remapper.remapFieldDescriptor(tinyField.getFieldDescriptorInFirstNamespace()),
names,
tinyField.getComments()
);
}
private TinyMethod renameMethod(TinyMethod tinyMethod) {
var names = rename(tinyMethod.getMethodNames());
return new TinyMethod(
remapper.remapMethodDescriptor(tinyMethod.getMethodDescriptorInFirstNamespace()),
names,
tinyMethod.getParameters().stream().map(this::renameMethodParameter).collect(Collectors.toList()),
tinyMethod.getLocalVariables().stream().map(this::renameVariable).collect(Collectors.toList()),
tinyMethod.getComments()
);
}
private TinyMethodParameter renameMethodParameter(TinyMethodParameter tinyMethodParameter) {
return new TinyMethodParameter(
tinyMethodParameter.getLvIndex(),
rename(tinyMethodParameter.getParameterNames()),
tinyMethodParameter.getComments()
);
}
private TinyLocalVariable renameVariable(TinyLocalVariable tinyLocalVariable) {
return new TinyLocalVariable(
tinyLocalVariable.getLvIndex(), tinyLocalVariable.getLvStartOffset(),
tinyLocalVariable.getLvTableIndex(),
rename(tinyLocalVariable.getLocalVariableNames()),
tinyLocalVariable.getComments()
);
}
}
|
