support parameterized type for generated tag keys

Author: Lulu13022002

paper-generator/src/main/java/io/papermc/generator/types/registry/GeneratedTagKeyType.java

@@ -1,12 +1,14 @@
 package io.papermc.generator.types.registry;
 
+import com.squareup.javapoet.ClassName;
 import com.squareup.javapoet.FieldSpec;
 import com.squareup.javapoet.JavaFile;
 import com.squareup.javapoet.MethodSpec;
 import com.squareup.javapoet.ParameterSpec;
 import com.squareup.javapoet.ParameterizedTypeName;
 import com.squareup.javapoet.TypeName;
 import com.squareup.javapoet.TypeSpec;
+import com.squareup.javapoet.WildcardTypeName;
 import io.papermc.generator.Main;
 import io.papermc.generator.registry.RegistryEntry;
 import io.papermc.generator.types.SimpleGenerator;
@@ -67,7 +69,17 @@ private TypeSpec.Builder keyHolderType() {
 
     @Override
     protected TypeSpec getTypeSpec() {
-        TypeName tagKeyType = ParameterizedTypeName.get(TagKey.class, this.entry.apiClass());
+        final TypeName apiType;
+        if (this.entry.genericArgCount() > 0) {
+            final TypeName[] args = new TypeName[this.entry.genericArgCount()];
+            for (int i = 0; i < args.length; i++) {
+                args[i] = WildcardTypeName.subtypeOf(Object.class);
+            }
+            apiType = ParameterizedTypeName.get(ClassName.get(this.entry.apiClass()), args);
+        } else {
+            apiType = ClassName.get(this.entry.apiClass());
+        }
+        TypeName tagKeyType = ParameterizedTypeName.get(ClassName.get(TagKey.class), apiType);
 
         TypeSpec.Builder typeBuilder = this.keyHolderType();
         MethodSpec.Builder createMethod = this.createMethod(tagKeyType);