Move machine upgrades to a data map

Author: Technici4n

docs/ADDING_MACHINES.md

@@ -18,15 +18,16 @@ MIMachineEvents.addEbfTiers(event => {
 ```
 
 ## Adding electric machine upgrades
-You can add electric machine upgrades using KubeJS.
-This will allow them to be used in all the electric machines to increase the maximum EU that the machine can handle.
-
-Example:
-```js
-MIMachineEvents.registerUpgrades(event => {
-  // id of the upgrade item, number of added EU per upgrade
-  event.register("minecraft:diamond", 42);
-});
+You can add electric machine upgrades using the `modern_industrialization:machine_upgrades` item data map.
+For example, in `data/modern_industrialization/data_maps/item/machine_upgrades.json`:
+```json
+{
+  "values": {
+    "minecraft:diamond": {
+      "extraMaxEu": 42
+    }
+  }
+}
 ```
 
 ## Registering a recipe type

src/generated/resources/data/modern_industrialization/data_maps/item/machine_upgrades.json

@@ -0,0 +1,19 @@
+{
+  "values": {
+    "modern_industrialization:advanced_upgrade": {
+      "extraMaxEu": 8
+    },
+    "modern_industrialization:basic_upgrade": {
+      "extraMaxEu": 2
+    },
+    "modern_industrialization:highly_advanced_upgrade": {
+      "extraMaxEu": 128
+    },
+    "modern_industrialization:quantum_upgrade": {
+      "extraMaxEu": 999999999
+    },
+    "modern_industrialization:turbo_upgrade": {
+      "extraMaxEu": 32
+    }
+  }
+}

src/main/java/aztech/modern_industrialization/MI.java

@@ -199,6 +199,7 @@ public MI(IEventBus modBus, Dist dist) {
         modBus.addListener(RegisterDataMapTypesEvent.class, event -> {
             event.register(MIDataMaps.FLUID_FUELS);
             event.register(MIDataMaps.ITEM_PIPE_UPGRADES);
+            event.register(MIDataMaps.MACHINE_UPGRADES);
         });
 
         if (MIConfig.loadAe2Compat()) {

src/main/java/aztech/modern_industrialization/api/datamaps/MIDataMaps.java

@@ -45,6 +45,15 @@ public final class MIDataMaps {
             .synced(ItemPipeUpgrade.CODEC, true)
             .build();
 
+    /**
+     * Items that can be added to electric machines, to increase the maximum EU/t they can support.
+     */
+    public static final DataMapType<Item, MachineUpgrade> MACHINE_UPGRADES = DataMapType
+            .builder(
+                    MI.id("machine_upgrades"), Registries.ITEM, MachineUpgrade.CODEC)
+            .synced(MachineUpgrade.CODEC, true)
+            .build();
+
     private MIDataMaps() {
     }
 }

src/main/java/aztech/modern_industrialization/compat/kubejs/machine/RegisterUpgradesEventJS.java renamed to src/main/java/aztech/modern_industrialization/api/datamaps/MachineUpgrade.java

@@ -21,14 +21,15 @@
  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  * SOFTWARE.
  */
-package aztech.modern_industrialization.compat.kubejs.machine;
+package aztech.modern_industrialization.api.datamaps;
 
-import aztech.modern_industrialization.machines.components.UpgradeComponent;
-import dev.latvian.mods.kubejs.event.KubeEvent;
-import net.minecraft.resources.ResourceLocation;
+import com.mojang.serialization.Codec;
+import com.mojang.serialization.codecs.RecordCodecBuilder;
+import net.minecraft.util.ExtraCodecs;
 
-public class RegisterUpgradesEventJS implements KubeEvent {
-    public static void register(ResourceLocation itemId, long extraEu) {
-        UpgradeComponent.registerUpgrade(itemId, extraEu);
-    }
+public record MachineUpgrade(int extraMaxEu) {
+    public static final Codec<MachineUpgrade> CODEC = RecordCodecBuilder.create(
+            instance -> instance
+                    .group(ExtraCodecs.POSITIVE_INT.fieldOf("extraMaxEu").forGetter(MachineUpgrade::extraMaxEu))
+                    .apply(instance, MachineUpgrade::new));
 }

src/main/java/aztech/modern_industrialization/compat/kubejs/LoadedKubeJSProxy.java

@@ -69,11 +69,6 @@ public void fireRegisterMachinesEvent() {
         MIMachineKubeJSEvents.REGISTER_MACHINES.post(new RegisterMachinesEventJS());
     }
 
-    @Override
-    public void fireRegisterUpgradesEvent() {
-        MIMachineKubeJSEvents.REGISTER_UPGRADES.post(new RegisterUpgradesEventJS());
-    }
-
     @Override
     public void fireAddMultiblockSlotsEvent(String category, SlotPositions.Builder itemInputs, SlotPositions.Builder itemOutputs,
             SlotPositions.Builder fluidInputs, SlotPositions.Builder fluidOutputs) {

src/main/java/aztech/modern_industrialization/compat/kubejs/machine/MIMachineKubeJSEvents.java

@@ -34,7 +34,6 @@ public interface MIMachineKubeJSEvents {
     EventHandler REGISTER_RECIPE_TYPES = EVENT_GROUP.startup("registerRecipeTypes", () -> RegisterRecipeTypesEventJS.class);
     EventHandler REGISTER_CASINGS = EVENT_GROUP.startup("registerCasings", () -> RegisterCasingsEventJS.class);
     EventHandler REGISTER_MACHINES = EVENT_GROUP.startup("registerMachines", () -> RegisterMachinesEventJS.class);
-    EventHandler REGISTER_UPGRADES = EVENT_GROUP.startup("registerUpgrades", () -> RegisterUpgradesEventJS.class);
     TargetedEventHandler<String> ADD_MULTIBLOCK_SLOTS = EVENT_GROUP.startup("addMultiblockSlots", () -> AddMultiblockSlotsEventJS.class)
             .requiredTarget(EventTargetType.STRING);
     EventHandler ADD_EBF_TIERS = EVENT_GROUP.startup("addEbfTiers", () -> AddEbfTiersEventJS.class);

src/main/java/aztech/modern_industrialization/datagen/datamap/MIDataMapProvider.java

@@ -29,6 +29,7 @@
 import aztech.modern_industrialization.api.datamaps.FluidFuel;
 import aztech.modern_industrialization.api.datamaps.ItemPipeUpgrade;
 import aztech.modern_industrialization.api.datamaps.MIDataMaps;
+import aztech.modern_industrialization.api.datamaps.MachineUpgrade;
 import aztech.modern_industrialization.definition.FluidDefinition;
 import aztech.modern_industrialization.definition.ItemDefinition;
 import java.util.concurrent.CompletableFuture;
@@ -51,6 +52,7 @@ protected void gather() {
 
         gatherFluidFuels();
         gatherItemPipeUpgrades();
+        gatherMachineUpgrades();
     }
 
     private void gatherFurnaceFuels() {
@@ -105,4 +107,16 @@ private void gatherItemPipeUpgrades() {
     private void addItemPipeUpgrade(ItemDefinition<?> itemDefinition, int maxExtractedItems) {
         builder(MIDataMaps.ITEM_PIPE_UPGRADES).add(itemDefinition.getId(), new ItemPipeUpgrade(maxExtractedItems), false);
     }
+
+    private void gatherMachineUpgrades() {
+        addMachineUpgrade(MIItem.BASIC_UPGRADE, 2);
+        addMachineUpgrade(MIItem.ADVANCED_UPGRADE, 8);
+        addMachineUpgrade(MIItem.TURBO_UPGRADE, 32);
+        addMachineUpgrade(MIItem.HIGHLY_ADVANCED_UPGRADE, 128);
+        addMachineUpgrade(MIItem.QUANTUM_UPGRADE, 999999999);
+    }
+
+    private void addMachineUpgrade(ItemDefinition<?> itemDefinition, int extraMaxEu) {
+        builder(MIDataMaps.MACHINE_UPGRADES).add(itemDefinition.getId(), new MachineUpgrade(extraMaxEu), false);
+    }
 }

src/main/java/aztech/modern_industrialization/machines/components/UpgradeComponent.java

@@ -23,17 +23,12 @@
  */
 package aztech.modern_industrialization.machines.components;
 
-import aztech.modern_industrialization.MIItem;
+import aztech.modern_industrialization.api.datamaps.MIDataMaps;
 import aztech.modern_industrialization.compat.kubejs.KubeJSProxy;
 import aztech.modern_industrialization.machines.IComponent;
 import aztech.modern_industrialization.machines.MachineBlockEntity;
-import java.util.HashMap;
-import java.util.Map;
-import java.util.Objects;
 import net.minecraft.core.HolderLookup;
-import net.minecraft.core.registries.BuiltInRegistries;
 import net.minecraft.nbt.CompoundTag;
-import net.minecraft.resources.ResourceLocation;
 import net.minecraft.world.InteractionHand;
 import net.minecraft.world.ItemInteractionResult;
 import net.minecraft.world.entity.player.Player;
@@ -43,37 +38,13 @@
 public class UpgradeComponent implements IComponent.ServerOnly, DropableComponent {
 
     private ItemStack itemStack = ItemStack.EMPTY;
-    private final static Map<ResourceLocation, Long> UPGRADES = new HashMap<>();
 
     public static long getExtraEu(ItemLike item) {
-        return UPGRADES.getOrDefault(BuiltInRegistries.ITEM.getKey(item.asItem()), 0L);
-    }
-
-    public static void registerUpgrade(ItemLike item, long extraEu) {
-        registerUpgrade(BuiltInRegistries.ITEM.getKey(item.asItem()), extraEu);
-    }
-
-    public static void registerUpgrade(ResourceLocation itemId, long extraEu) {
-        Objects.requireNonNull(itemId);
-
-        if (extraEu <= 0) {
-            throw new IllegalArgumentException("extraEu must be positive");
-        }
-
-        if (UPGRADES.containsKey(itemId)) {
-            throw new IllegalArgumentException("Upgrade already registered:" + itemId);
-        }
-
-        UPGRADES.put(itemId, extraEu);
+        var data = item.asItem().builtInRegistryHolder().getData(MIDataMaps.MACHINE_UPGRADES);
+        return data == null ? 0 : data.extraMaxEu();
     }
 
     static {
-        registerUpgrade(MIItem.BASIC_UPGRADE.asItem(), 2L);
-        registerUpgrade(MIItem.ADVANCED_UPGRADE.asItem(), 8L);
-        registerUpgrade(MIItem.TURBO_UPGRADE.asItem(), 32L);
-        registerUpgrade(MIItem.HIGHLY_ADVANCED_UPGRADE.asItem(), 128L);
-        registerUpgrade(MIItem.QUANTUM_UPGRADE.asItem(), 999999999L);
-
         KubeJSProxy.instance.fireRegisterUpgradesEvent();
     }