From a259d844a3cba99867be6415999b8c697b31adfb Mon Sep 17 00:00:00 2001 From: zaaarf Date: Thu, 13 Apr 2023 11:12:41 +0200 Subject: fix: injector generated correctly --- .../ftbsc/lll/processor/tools/JavaPoetUtils.java | 23 ++++++++-------------- 1 file changed, 8 insertions(+), 15 deletions(-) diff --git a/src/main/java/ftbsc/lll/processor/tools/JavaPoetUtils.java b/src/main/java/ftbsc/lll/processor/tools/JavaPoetUtils.java index b09ffba..875ea67 100644 --- a/src/main/java/ftbsc/lll/processor/tools/JavaPoetUtils.java +++ b/src/main/java/ftbsc/lll/processor/tools/JavaPoetUtils.java @@ -151,27 +151,20 @@ public class JavaPoetUtils { MethodSpec.Builder injectBuilder = MethodSpec.methodBuilder("inject") .addModifiers(Modifier.PUBLIC) .returns(void.class) - .addAnnotation(Override.class); - - int argumentCount = inj.injector.getParameters().size(); - - if(argumentCount == 2) { - injectBuilder - .addParameter(ParameterSpec.builder( - TypeName.get(env - .getElementUtils() - .getTypeElement("org.objectweb.asm.tree.ClassNode").asType()), "clazz") - .build()); - } - - injectBuilder + .addAnnotation(Override.class) + .addParameter(ParameterSpec.builder( + TypeName.get(env + .getElementUtils() + .getTypeElement("org.objectweb.asm.tree.ClassNode").asType()), "clazz") + .build()) .addParameter(ParameterSpec.builder( TypeName.get(env .getElementUtils() .getTypeElement("org.objectweb.asm.tree.MethodNode").asType()), "main") .build()); - if(argumentCount == 2) injectBuilder.addStatement("super.$L(clazz, main)", inj.injector.getSimpleName()); + if(inj.injector.getParameters().size() == 2) + injectBuilder.addStatement("super.$L(clazz, main)", inj.injector.getSimpleName()); else injectBuilder.addStatement("super.$L(main)", inj.injector.getSimpleName()); return injectBuilder.build(); -- cgit v1.2.3-56-ga3b1