Fix invalid generated code from NBT Compound argument in annotation e…

…xample project
willkroboth · Dec 12, 2023 · 1782863 · 1782863
1 parent 64c6ec3
commit 1782863
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/commandapi-annotations/src/main/java/dev/jorel/commandapi/annotations/Annotations.java b/commandapi-annotations/src/main/java/dev/jorel/commandapi/annotations/Annotations.java
@@ -305,8 +305,12 @@ private <T extends Annotation> Map<Integer, String> emitArgumentsAndGenerateArgu
 
  // Handle return types
  Primitive primitive = getPrimitive(argumentAnnotation);
- if(primitive.value().length == 1) {
- argumentMapping.put(i - 1, primitive.value()[0]);
+ if (argumentAnnotation instanceof ANBTCompoundArgument) {
+ argumentMapping.put(i - 1, parameter.asType().toString());
+ } else {
+ if(primitive.value().length == 1) {
+ argumentMapping.put(i - 1, primitive.value()[0]);
+ }
  }
  }