From a259d844a3cba99867be6415999b8c697b31adfb Mon Sep 17 00:00:00 2001
From: zaaarf <zaaarf@proton.me>
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(-)

(limited to 'src/main/java')

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