zaaarf/lillero-processor
1
0
Fork 0
mirror of https://github.com/zaaarf/lillero-processor.git synced 2024-11-22 04:14:49 +01:00
Code Issues Projects Releases Packages Wiki Activity

feat: generate dummies for all abstract methods, not just targets

Browse source
This commit is contained in:
zaaarf 2023-04-12 17:08:56 +02:00
parent 5973cf6d95
commit 527cef56e6
No known key found for this signature in database
GPG key ID: 82240E075E31FA4C
2 changed files with 19 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/main/java/ftbsc/lll/processor/LilleroProcessor.java
View file

@ -346,7 +346,7 @@ public class LilleroProcessor extends AbstractProcessor {
.addMethod(buildStringReturnMethod("targetClass", obfuscateInjectorMetadata ? targetClass.fqnObf : targetClass.fqn))
.addMethod(buildStringReturnMethod("methodName", obfuscateInjectorMetadata ? target.nameObf : target.name))
.addMethod(buildStringReturnMethod("methodDesc", obfuscateInjectorMetadata ? target.descriptorObf : target.descriptor))
.addMethods(generateDummies(targets))
.addMethods(generateDummies(cl))
.addMethod(generateInjector(toGenerate.get(injName), this.processingEnv))
.build();

33
src/main/java/ftbsc/lll/processor/tools/JavaPoetUtils.java
View file

@ -13,11 +13,7 @@ import ftbsc.lll.proxies.impl.FieldProxy;
import ftbsc.lll.proxies.impl.MethodProxy;
import javax.annotation.processing.ProcessingEnvironment;
import javax.lang.model.element.Element;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.Modifier;
import javax.lang.model.element.VariableElement;
import java.util.Collection;
import javax.lang.model.element.*;
import java.util.HashSet;
import static ftbsc.lll.processor.tools.ASTUtils.getProxyType;
@ -102,7 +98,7 @@ public class JavaPoetUtils {
con.addStatement(
"$L.setModifiers($L)",
builderName,
target == null ? 0 :mapModifiers(target.getModifiers())
target == null ? 0 : mapModifiers(target.getModifiers())
);
//set type(s)
@ -121,19 +117,26 @@ public class JavaPoetUtils {
}
/**
* Generates a {@link HashSet} of dummy overrides given a {@link Collection} stubs.
* @param dummies the stubs
* Generates a {@link HashSet} of dummy overrides for every abstract method in a given class,
* represented as a {@link TypeElement}.
* @param clazz the given class
* @return a {@link HashSet} containing the generated {@link MethodSpec}s
* @since 0.5.0
*/
public static HashSet<MethodSpec> generateDummies(Collection<ExecutableElement> dummies) {
public static HashSet<MethodSpec> generateDummies(TypeElement clazz) {
HashSet<MethodSpec> specs = new HashSet<>();
for(ExecutableElement d : dummies)
if(d.getModifiers().contains(Modifier.ABSTRACT))
specs.add(MethodSpec.overriding(d)
.addStatement("throw new $T($S)", RuntimeException.class, "This is a stub and should not have been called")
.build()
);
clazz
.getEnclosedElements()
.stream()
.filter(e -> e instanceof ExecutableElement)
.map(e -> (ExecutableElement) e)
.forEach(e -> {
if(e.getModifiers().contains(Modifier.ABSTRACT))
specs.add(MethodSpec.overriding(e)
.addStatement("throw new $T($S)", RuntimeException.class, "This is a stub and should not have been called")
.build()
);
});
return specs;
}