feat: Add support for Augment Items (#22)

Author: cutekibry

src/main/java/dev/latvian/mods/kubejs/thermal/KubeJSThermalPlugin.java

@@ -5,8 +5,14 @@
 import cofh.thermal.core.init.registries.TCoreRecipeTypes;
 import dev.latvian.mods.kubejs.KubeJSPlugin;
 import dev.latvian.mods.kubejs.recipe.schema.RegisterRecipeSchemasEvent;
+import dev.latvian.mods.kubejs.registry.RegistryInfo;
 
 public class KubeJSThermalPlugin extends KubeJSPlugin {
+	@Override
+	public void init() {
+		RegistryInfo.ITEM.addType("thermal_augment", ThermalAugmentItemBuilder.class, ThermalAugmentItemBuilder::new);
+	}
+
 	@Override
 	public void registerRecipeSchemas(RegisterRecipeSchemasEvent event) {
 		event.namespace(ModIds.ID_THERMAL)

src/main/java/dev/latvian/mods/kubejs/thermal/ThermalAugmentItemBuilder.java

@@ -0,0 +1,54 @@
+package dev.latvian.mods.kubejs.thermal;
+
+import cofh.core.util.helpers.AugmentDataHelper;
+import cofh.thermal.lib.common.item.AugmentItem;
+import dev.latvian.mods.kubejs.item.ItemBuilder;
+import dev.latvian.mods.kubejs.typings.Info;
+import net.minecraft.resources.ResourceLocation;
+import net.minecraft.world.item.Item;
+
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+import static cofh.lib.util.constants.NBTTags.TAG_AUGMENT_BASE_MOD;
+import static cofh.lib.util.constants.NBTTags.TAG_AUGMENT_TYPE_UPGRADE;
+
+public class ThermalAugmentItemBuilder extends ItemBuilder {
+	public transient String augmentType;
+	public transient LinkedHashMap<String, Float> thermalMods;
+
+	public ThermalAugmentItemBuilder(ResourceLocation i) {
+		super(i);
+		augmentType = TAG_AUGMENT_TYPE_UPGRADE;
+		thermalMods = new LinkedHashMap<>();
+	}
+
+	@Info("Sets Thermal augment type, default is Thermal upgrade augment type.")
+	public ThermalAugmentItemBuilder augmentType(String type) {
+		augmentType = type;
+		return this;
+	}
+
+	@Info("Sets a Thermal augment numeric modifier by key. Can be called multiple times for different keys.")
+	public ThermalAugmentItemBuilder thermalMod(String key, float value) {
+		thermalMods.put(key, value);
+		return this;
+	}
+
+	@Info("Convenience alias for setting Thermal BaseMod.")
+	public ThermalAugmentItemBuilder baseMod(float value) {
+		thermalMods.put(TAG_AUGMENT_BASE_MOD, value);
+		return this;
+	}
+
+	@Override
+	public Item createObject() {
+		var builder = AugmentDataHelper.builder().type(augmentType);
+		for (Map.Entry<String, Float> entry : thermalMods.entrySet()) {
+			builder.mod(entry.getKey(), entry.getValue());
+		}
+
+		return new AugmentItem(createItemProperties(),
+				builder.build());
+	}
+}