small cleanup

Prunoideae · Apr 7, 2022 · bdeacc3 · bdeacc3
1 parent afcdcb2
commit bdeacc3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/main/java/com/prunoideae/probejs/compiler/RegistryCompiler.java b/src/main/java/com/prunoideae/probejs/compiler/RegistryCompiler.java
@@ -66,6 +66,9 @@ public List<String> format(Integer indent, Integer stepIndent) {
             for (RegistryObjectBuilderTypes.BuilderType<?> builder : types.types.values()) {
                 formatted.add(" ".repeat(stepped) + "create(id: string, type: %s): %s;".formatted(stringG.toJson(builder.type()), FormatterClass.formatTypeParameterized(new TypeInfoClass(builder.builderClass()))));
             }
+            if (types.getDefaultType() != null) {
+                formatted.add(" ".repeat(stepped) + "create(id: string): %s;".formatted(FormatterClass.formatTypeParameterized(new TypeInfoClass(types.getDefaultType().builderClass()))));
+            }
             formatted.add(" ".repeat(indent) + "}");
             return formatted;
         }

diff --git a/src/main/java/com/prunoideae/probejs/formatter/SpecialTypes.java b/src/main/java/com/prunoideae/probejs/formatter/SpecialTypes.java
@@ -9,9 +9,6 @@
 import com.prunoideae.probejs.info.type.TypeInfoVariable;
 import dev.latvian.mods.kubejs.item.ingredient.IngredientJS;
 
-import java.lang.reflect.Method;
-import java.lang.reflect.Modifier;
-import java.lang.reflect.TypeVariable;
 import java.util.*;
 
 public class SpecialTypes {