Forge Energy tunnel (color possibly TBD)

Author: robotgryphon

src/datagen/java/dev/compactmods/machines/datagen/RecipeGenerator.java

@@ -61,6 +61,14 @@ protected void buildCraftingRecipes(Consumer<FinishedRecipe> consumer) {
                 .unlockedBy("observer", RecipeProvider.has(Items.OBSERVER))
                 .save(consumer);
 
+        TunnelRecipeBuilder.tunnel(Tunnels.FORGE_ENERGY.get(), 2)
+                .requires(Items.GLOWSTONE_DUST)
+                .requires(Items.ENDER_PEARL)
+                .requires(Items.REDSTONE)
+                .requires(Items.OBSERVER)
+                .unlockedBy("observer", RecipeProvider.has(Items.OBSERVER))
+                .save(consumer);
+
         addMachineRecipes(consumer);
     }
 

src/main/java/dev/compactmods/machines/core/Tunnels.java

@@ -5,6 +5,7 @@
 import dev.compactmods.machines.tunnel.TunnelWallBlock;
 import dev.compactmods.machines.tunnel.TunnelWallEntity;
 import dev.compactmods.machines.tunnel.definitions.FluidTunnel;
+import dev.compactmods.machines.tunnel.definitions.ForgeEnergyTunnel;
 import dev.compactmods.machines.tunnel.definitions.UnknownTunnel;
 import dev.compactmods.machines.tunnel.definitions.ItemTunnel;
 import net.minecraft.resources.ResourceLocation;
@@ -66,6 +67,8 @@ public static TunnelDefinition getDefinition(ResourceLocation id) {
 
     public static final RegistryObject<TunnelDefinition> FLUID_TUNNEL_DEF = DEFINITIONS.register("fluid", FluidTunnel::new);
 
+    public static final RegistryObject<TunnelDefinition> FORGE_ENERGY = DEFINITIONS.register("energy", ForgeEnergyTunnel::new);
+
     // ================================================================================================================
     //   TUNNEL BLOCKS / TILES
     // ================================================================================================================

src/main/resources/assets/compactmachines/lang/en_us.json

@@ -53,6 +53,7 @@
   "item.compactmachines.personal_shrinking_device": "Personal Shrinking Device",
   "item.compactmachines.tunnels.item": "Item Tunnel",
   "item.compactmachines.tunnels.fluid": "Fluid Tunnel",
+  "item.compactmachines.tunnels.energy": "Energy Tunnel",
   "item.compactmachines.tunnels.redstone_in": "Redstone Tunnel (In)",
   "item.compactmachines.tunnels.redstone_out": "Redstone Tunnel (Out)",
   "itemGroup.compactmachines": "Compact Machines",

src/tunnels/java/dev/compactmods/machines/tunnel/definitions/ForgeEnergyTunnel.java

@@ -0,0 +1,85 @@
+package dev.compactmods.machines.tunnel.definitions;
+
+import com.google.common.collect.ImmutableSet;
+import dev.compactmods.machines.api.tunnels.TunnelDefinition;
+import dev.compactmods.machines.api.tunnels.capability.CapabilityTunnel;
+import dev.compactmods.machines.api.tunnels.lifecycle.TunnelInstance;
+import net.minecraft.core.BlockPos;
+import net.minecraft.core.Direction;
+import net.minecraft.nbt.CompoundTag;
+import net.minecraft.util.FastColor;
+import net.minecraftforge.common.capabilities.Capability;
+import net.minecraftforge.common.util.INBTSerializable;
+import net.minecraftforge.common.util.LazyOptional;
+import net.minecraftforge.energy.CapabilityEnergy;
+import net.minecraftforge.energy.EnergyStorage;
+import net.minecraftforge.energy.IEnergyStorage;
+import net.minecraftforge.registries.ForgeRegistryEntry;
+
+import javax.annotation.Nonnull;
+
+public class ForgeEnergyTunnel extends ForgeRegistryEntry<TunnelDefinition>
+        implements TunnelDefinition, CapabilityTunnel<ForgeEnergyTunnel.Instance> {
+    @Override
+    public int ringColor() {
+        return FastColor.ARGB32.color(255, 193, 100, 222);
+    }
+
+    @Override
+    public ImmutableSet<Capability<?>> getSupportedCapabilities() {
+        return ImmutableSet.of(CapabilityEnergy.ENERGY);
+    }
+
+    @Override
+    public <CapType> LazyOptional<CapType> getCapability(Capability<CapType> type, Instance instance) {
+        if (type == CapabilityEnergy.ENERGY) {
+            return instance.lazy().cast();
+        }
+
+        return LazyOptional.empty();
+    }
+
+    @Override
+    public Instance newInstance(BlockPos position, Direction side) {
+        return new Instance();
+    }
+
+    public static class Instance implements TunnelInstance, INBTSerializable<CompoundTag> {
+
+        private final int DEFAULT_STORAGE = 10000;
+        private EnergyStorage storage;
+        private final LazyOptional<IEnergyStorage> lazy;
+
+        public Instance() {
+            this.storage = new EnergyStorage(DEFAULT_STORAGE);
+            this.lazy = LazyOptional.of(this::getStorage);
+        }
+
+        @Nonnull
+        public EnergyStorage getStorage() {
+            return this.storage;
+        }
+
+        public LazyOptional<IEnergyStorage> lazy() {
+            return lazy;
+        }
+
+        @Override
+        public CompoundTag serializeNBT() {
+            CompoundTag data = new CompoundTag();
+            data.put("storage", storage.serializeNBT());
+            return data;
+        }
+
+        @Override
+        public void deserializeNBT(CompoundTag nbt) {
+            if(!nbt.contains("storage")) {
+                this.storage = new EnergyStorage(DEFAULT_STORAGE);
+                return;
+            }
+
+            final var stor = nbt.get("storage");
+            storage.deserializeNBT(stor);
+        }
+    }
+}