6 files changed, 228 insertions, 29 deletions

diff --git a/src/main/java/lv/enes/mc/eris_alchemy/block/AlchemicalChestBlock.java b/src/main/java/lv/enes/mc/eris_alchemy/block/AlchemicalChestBlock.java
index d50ee77..eb7cb6e 100644
--- a/src/main/java/lv/enes/mc/eris_alchemy/block/AlchemicalChestBlock.java
+++ b/src/main/java/lv/enes/mc/eris_alchemy/block/AlchemicalChestBlock.java
@@ -3,15 +3,15 @@ package lv.enes.mc.eris_alchemy.block;
 import jakarta.annotation.Nonnull;
 import jakarta.annotation.Nullable;
 import lv.enes.mc.eris_alchemy.ErisAlchemyRegistry;
-import lv.enes.mc.eris_alchemy.block.entity.AlchemicalChestBlockEntity;
+import lv.enes.mc.eris_alchemy.block.entity.AlchemicalChestEntity;
 import net.minecraft.core.BlockPos;
 import net.minecraft.network.chat.Component;
 import net.minecraft.world.level.block.entity.BlockEntity;
 import net.minecraft.world.level.block.state.BlockState;
 
-public class AlchemicalChestBlock extends ChestLikeBlock<AlchemicalChestBlockEntity>
-{
-        public static final Component CONTAINER_TITLE = Component.translatable("container.eris_alchemy.alchemical_chest");
+public class AlchemicalChestBlock extends ChestLikeBlock<AlchemicalChestEntity> {
+        public static final Component CONTAINER_TITLE
+                        = Component.translatable("container.eris_alchemy.alchemical_chest");
 
         public AlchemicalChestBlock(Properties properties) {
                 super(properties, () -> ErisAlchemyRegistry.BlockEntities.ALCHEMICAL_CHEST);
@@ -26,6 +26,6 @@ public class AlchemicalChestBlock extends ChestLikeBlock<AlchemicalChestBlockEnt
         @Nullable
         @Override
         public BlockEntity newBlockEntity(BlockPos pos, BlockState state) {
-                return new AlchemicalChestBlockEntity(pos, state);
+                return new AlchemicalChestEntity(pos, state);
         }
 }
diff --git a/src/main/java/lv/enes/mc/eris_alchemy/block/ChestLikeBlock.java b/src/main/java/lv/enes/mc/eris_alchemy/block/ChestLikeBlock.java
index e50563a..ade24fb 100644
--- a/src/main/java/lv/enes/mc/eris_alchemy/block/ChestLikeBlock.java
+++ b/src/main/java/lv/enes/mc/eris_alchemy/block/ChestLikeBlock.java
@@ -1,7 +1,7 @@
 package lv.enes.mc.eris_alchemy.block;
 
 import jakarta.annotation.Nonnull;
-import lv.enes.mc.eris_alchemy.block.entity.ChestLikeBlockEntity;
+import lv.enes.mc.eris_alchemy.block.entity.ChestLikeEntity;
 import net.minecraft.core.BlockPos;
 import net.minecraft.core.Direction;
 import net.minecraft.network.chat.Component;
@@ -33,7 +33,7 @@ import net.minecraft.world.phys.shapes.VoxelShape;
 
 import java.util.function.Supplier;
 
-public abstract class ChestLikeBlock<E extends ChestLikeBlockEntity>
+public abstract class ChestLikeBlock<E extends ChestLikeEntity>
                 extends AbstractChestBlock<E>
                 implements SimpleWaterloggedBlock
 {
@@ -85,7 +85,7 @@ public abstract class ChestLikeBlock<E extends ChestLikeBlockEntity>
 
         @Override
         public MenuProvider getMenuProvider(BlockState state, Level world, BlockPos pos) {
-                if (world.getBlockEntity(pos) instanceof ChestLikeBlockEntity entity) {
+                if (world.getBlockEntity(pos) instanceof ChestLikeEntity entity) {
                         return new SimpleMenuProvider(entity, getContainerTitle());
                 }
                 return null;
diff --git a/src/main/java/lv/enes/mc/eris_alchemy/block/EnergyCondenserBlock.java b/src/main/java/lv/enes/mc/eris_alchemy/block/EnergyCondenserBlock.java
new file mode 100644
index 0000000..86ac061
--- /dev/null
+++ b/src/main/java/lv/enes/mc/eris_alchemy/block/EnergyCondenserBlock.java
@@ -0,0 +1,56 @@
+package lv.enes.mc.eris_alchemy.block;
+
+import jakarta.annotation.Nonnull;
+import jakarta.annotation.Nullable;
+import lv.enes.mc.eris_alchemy.EMC;
+import lv.enes.mc.eris_alchemy.EmcStorage;
+import lv.enes.mc.eris_alchemy.ErisAlchemyRegistry;
+import lv.enes.mc.eris_alchemy.block.entity.EnergyCondenserEntity;
+import net.minecraft.core.BlockPos;
+import net.minecraft.nbt.CompoundTag;
+import net.minecraft.network.chat.Component;
+import net.minecraft.world.item.ItemStack;
+import net.minecraft.world.item.TooltipFlag;
+import net.minecraft.world.level.BlockGetter;
+import net.minecraft.world.level.block.entity.BlockEntity;
+import net.minecraft.world.level.block.state.BlockState;
+
+import java.util.List;
+
+public class EnergyCondenserBlock extends ChestLikeBlock<EnergyCondenserEntity> implements EmcStorage {
+        public static final Component CONTAINER_TITLE
+                        = Component.translatable("container.eris_alchemy.energy_condenser");
+
+        public EnergyCondenserBlock(Properties properties) {
+                super(properties, () -> ErisAlchemyRegistry.BlockEntities.ENERGY_CONDENSER);
+        }
+
+        @Override
+        public void appendHoverText(
+                        ItemStack stack,
+                        @Nullable BlockGetter world,
+                        List<Component> tooltip,
+                        TooltipFlag options
+        ) {
+                if (getStoredEmc(stack) >= 0.1) {
+                        tooltip.add(Component.literal("Stored EMC: %s".formatted(EMC.formatEmc(getStoredEmc(stack)))));
+                }
+        }
+
+        @Override
+        @Nonnull
+        protected Component getContainerTitle() {
+                return CONTAINER_TITLE;
+        }
+
+        @Override
+        public double getStoredEmc(ItemStack stack, CompoundTag blockEntityData) {
+                return blockEntityData.getDouble("stored_emc");
+        }
+
+        @Nullable
+        @Override
+        public BlockEntity newBlockEntity(BlockPos pos, BlockState state) {
+                return new EnergyCondenserEntity(pos, state);
+        }
+}
diff --git a/src/main/java/lv/enes/mc/eris_alchemy/block/entity/AlchemicalChestBlockEntity.java b/src/main/java/lv/enes/mc/eris_alchemy/block/entity/AlchemicalChestEntity.java
index bac49b6..38ed902 100644
--- a/src/main/java/lv/enes/mc/eris_alchemy/block/entity/AlchemicalChestBlockEntity.java
+++ b/src/main/java/lv/enes/mc/eris_alchemy/block/entity/AlchemicalChestEntity.java
@@ -3,37 +3,30 @@ package lv.enes.mc.eris_alchemy.block.entity;
 import jakarta.annotation.Nonnull;
 import lv.enes.mc.eris_alchemy.ErisAlchemyRegistry;
 import lv.enes.mc.eris_alchemy.menu.AlchemicalChestMenu;
-import lv.enes.mc.eris_alchemy.menu.ChestLikeBlockMenu;
+import lv.enes.mc.eris_alchemy.menu.ChestLikeMenu;
 import net.minecraft.core.BlockPos;
 import net.minecraft.core.NonNullList;
-import net.minecraft.network.chat.Component;
 import net.minecraft.world.entity.player.Inventory;
 import net.minecraft.world.item.ItemStack;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.block.state.BlockState;
 
-public class AlchemicalChestBlockEntity extends ChestLikeBlockEntity {
+public class AlchemicalChestEntity extends ChestLikeEntity {
         private final static int WIDTH = 13;
         private final static int HEIGHT = 8;
 
         private final NonNullList<ItemStack> items = NonNullList.withSize(WIDTH * HEIGHT, ItemStack.EMPTY);
 
-        public AlchemicalChestBlockEntity(BlockPos pos, BlockState state) {
+        public AlchemicalChestEntity(BlockPos pos, BlockState state) {
                 super(ErisAlchemyRegistry.BlockEntities.ALCHEMICAL_CHEST, pos, state);
         }
 
         @Nonnull
         @Override
-        protected ChestLikeBlockMenu createMenu(int syncId, Inventory playerInventory) {
+        protected ChestLikeMenu createMenu(int syncId, Inventory playerInventory) {
                 return new AlchemicalChestMenu(syncId, playerInventory, this);
         }
 
-        @Nonnull
-        @Override
-        protected Component getDefaultName() {
-                return Component.translatable("container.eris_alchemy.alchemical_chest");
-        }
-
         @Override
         protected NonNullList<ItemStack> getItems() {
                 return items;
diff --git a/src/main/java/lv/enes/mc/eris_alchemy/block/entity/ChestLikeBlockEntity.java b/src/main/java/lv/enes/mc/eris_alchemy/block/entity/ChestLikeEntity.java
index 48b3ad7..362b054 100644
--- a/src/main/java/lv/enes/mc/eris_alchemy/block/entity/ChestLikeBlockEntity.java
+++ b/src/main/java/lv/enes/mc/eris_alchemy/block/entity/ChestLikeEntity.java
@@ -2,10 +2,11 @@ package lv.enes.mc.eris_alchemy.block.entity;
 
 import jakarta.annotation.Nonnull;
 import jakarta.annotation.Nullable;
-import lv.enes.mc.eris_alchemy.menu.ChestLikeBlockMenu;
+import lv.enes.mc.eris_alchemy.menu.ChestLikeMenu;
 import net.minecraft.core.BlockPos;
 import net.minecraft.core.NonNullList;
 import net.minecraft.nbt.CompoundTag;
+import net.minecraft.network.chat.Component;
 import net.minecraft.sounds.SoundEvents;
 import net.minecraft.sounds.SoundSource;
 import net.minecraft.world.Container;
@@ -17,19 +18,19 @@ import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.block.entity.*;
 import net.minecraft.world.level.block.state.BlockState;
 
-public abstract class ChestLikeBlockEntity extends BaseContainerBlockEntity implements LidBlockEntity {
+public abstract class ChestLikeEntity extends BaseContainerBlockEntity implements LidBlockEntity {
         protected static final int EVENT_INTERACTED = 1;
 
         protected final ChestLidController lidController = new ChestLidController();
         protected final ContainerOpenersCounter openersCounter = new ContainerOpenersCounter() {
                 @Override
                 protected void onOpen(Level world, BlockPos pos, BlockState state) {
-                        ChestLikeBlockEntity.this.onOpen(world, pos);
+                        ChestLikeEntity.this.onOpen(world, pos);
                 }
 
                 @Override
                 protected void onClose(Level world, BlockPos pos, BlockState state) {
-                        ChestLikeBlockEntity.this.onClose(world, pos);
+                        ChestLikeEntity.this.onClose(world, pos);
                 }
 
                 @Override
@@ -39,15 +40,15 @@ public abstract class ChestLikeBlockEntity extends BaseContainerBlockEntity impl
 
                 @Override
                 protected boolean isOwnContainer(Player player) {
-                        if (player.containerMenu instanceof ChestLikeBlockMenu menu) {
-                                return menu.getContainer() == ChestLikeBlockEntity.this;
+                        if (player.containerMenu instanceof ChestLikeMenu menu) {
+                                return menu.getContainer() == ChestLikeEntity.this;
                         }
                         return false;
                 }
         };
 
-        public ChestLikeBlockEntity(
-                        BlockEntityType<? extends ChestLikeBlockEntity> type,
+        public ChestLikeEntity(
+                        BlockEntityType<? extends ChestLikeEntity> type,
                         BlockPos pos,
                         BlockState state
         ) {
@@ -69,6 +70,12 @@ public abstract class ChestLikeBlockEntity extends BaseContainerBlockEntity impl
 
         @Nonnull
         @Override
+        protected Component getDefaultName() {
+                return getBlockState().getBlock().getName();
+        }
+
+        @Nonnull
+        @Override
         public ItemStack getItem(int slot) {
                 return getItems().get(slot);
         }
@@ -134,7 +141,7 @@ public abstract class ChestLikeBlockEntity extends BaseContainerBlockEntity impl
         public ItemStack removeItem(int slot, int amount) {
                 var stack = ContainerHelper.removeItem(getItems(), slot, amount);
                 if (!stack.isEmpty()) {
-                        this.setChanged();
+                        setChanged();
                 }
                 return stack;
         }
@@ -157,7 +164,7 @@ public abstract class ChestLikeBlockEntity extends BaseContainerBlockEntity impl
                 if (stack.getCount() > getMaxStackSize()) {
                         stack.setCount(getMaxStackSize());
                 }
-                this.setChanged();
+                setChanged();
         }
 
         @Override
diff --git a/src/main/java/lv/enes/mc/eris_alchemy/block/entity/EnergyCondenserEntity.java b/src/main/java/lv/enes/mc/eris_alchemy/block/entity/EnergyCondenserEntity.java
new file mode 100644
index 0000000..dfe07a4
--- /dev/null
+++ b/src/main/java/lv/enes/mc/eris_alchemy/block/entity/EnergyCondenserEntity.java
@@ -0,0 +1,143 @@
+package lv.enes.mc.eris_alchemy.block.entity;
+
+import jakarta.annotation.Nonnull;
+import lv.enes.mc.eris_alchemy.EMC;
+import lv.enes.mc.eris_alchemy.ErisAlchemyRegistry;
+import lv.enes.mc.eris_alchemy.menu.ChestLikeMenu;
+import lv.enes.mc.eris_alchemy.menu.EnergyCondenserMenu;
+import lv.enes.mc.eris_alchemy.utils.ContainerOpenersCounterUtil;
+import lv.enes.mc.eris_alchemy.utils.SyncedValue.SyncedDouble;
+import net.fabricmc.fabric.api.screenhandler.v1.ExtendedScreenHandlerFactory;
+import net.minecraft.core.BlockPos;
+import net.minecraft.core.NonNullList;
+import net.minecraft.nbt.CompoundTag;
+import net.minecraft.network.FriendlyByteBuf;
+import net.minecraft.server.level.ServerPlayer;
+import net.minecraft.world.entity.player.Inventory;
+import net.minecraft.world.item.ItemStack;
+import net.minecraft.world.level.Level;
+import net.minecraft.world.level.block.Block;
+import net.minecraft.world.level.block.state.BlockState;
+
+import java.util.stream.IntStream;
+
+public class EnergyCondenserEntity extends ChestLikeEntity implements ExtendedScreenHandlerFactory {
+        private final static int WIDTH = 13;
+        private final static int HEIGHT = 7;
+
+        private final NonNullList<ItemStack> items = NonNullList.withSize(WIDTH * HEIGHT + 1, ItemStack.EMPTY);
+
+        public final SyncedDouble storedEmc = new SyncedDouble(0);
+
+        public EnergyCondenserEntity(BlockPos pos, BlockState state) {
+                super(ErisAlchemyRegistry.BlockEntities.ENERGY_CONDENSER, pos, state);
+        }
+
+        @Nonnull
+        @Override
+        protected ChestLikeMenu createMenu(int syncId, Inventory playerInventory) {
+                return new EnergyCondenserMenu(syncId, playerInventory, this);
+        }
+
+        @Override
+        protected NonNullList<ItemStack> getItems() {
+                return items;
+        }
+
+        @Override
+        protected Block getParent() {
+                return ErisAlchemyRegistry.Blocks.ENERGY_CONDENSER;
+        }
+
+        public double getStoredEmc() {
+                return storedEmc.getValue();
+        }
+
+        public SyncedDouble getStoredEmcSyncer() {
+                return storedEmc;
+        }
+
+        @Override
+        public void load(CompoundTag nbt) {
+                super.load(nbt);
+                storedEmc.setValue(nbt.getDouble("stored_emc"));
+        }
+
+        @Override
+        protected void saveAdditional(CompoundTag nbt) {
+                super.saveAdditional(nbt);
+                nbt.putDouble("stored_emc", storedEmc.getValue());
+        }
+
+        public void setStoredEmc(double storedEmc) {
+                this.storedEmc.setValue(storedEmc);
+                setChanged();
+        }
+
+        @Override
+        public void tick(Level world, BlockPos pos, BlockState state) {
+                super.tick(world, pos, state);
+                EMC.getInstance(world).get(items.get(0)).ifPresent(cost -> {
+                        tryConsumeEmc(world, cost);
+                        tryCloneTemplate(cost);
+                });
+
+                this.storedEmc.syncIfChanged(ContainerOpenersCounterUtil.getOpeners(openersCounter));
+        }
+
+        @Override
+        public void writeScreenOpeningData(ServerPlayer player, FriendlyByteBuf buf) {
+                storedEmc.serialize(buf);
+        }
+
+        private void tryCloneTemplate(double cost) {
+                if (cost > getStoredEmc()) {
+                        return;
+                }
+
+                var template = items.get(0);
+
+                items.stream()
+                                .skip(1) // skip template
+                                .filter(stack -> ItemStack.isSameItemSameTags(template, stack))
+                                .filter(stack -> stack.getCount() < getMaxStackSize() && stack.getCount() < stack.getMaxStackSize())
+                                .findFirst()
+                                .ifPresentOrElse(
+                                                stack -> {
+                                                        stack.setCount(stack.getCount() + 1);
+                                                        setStoredEmc(getStoredEmc() - cost);
+                                                        setChanged();
+                                                },
+                                                () -> IntStream.range(1, items.size())
+                                                                .filter(i -> items.get(i).isEmpty())
+                                                                .findFirst()
+                                                                .ifPresent(emptySlot -> {
+                                                                        items.set(emptySlot, template.copyWithCount(1));
+                                                                        setStoredEmc(getStoredEmc() - cost);
+                                                                        setChanged();
+                                                                })
+                                );
+        }
+
+        private void tryConsumeEmc(Level world, double cost) {
+                if (cost <= getStoredEmc()) {
+                        return;
+                }
+
+                var emc = EMC.getInstance(world);
+                var template = items.get(0);
+                var sacrifice = items.stream()
+                                .skip(1) // skip the template
+                                .filter(stack -> !ItemStack.isSameItemSameTags(template, stack))
+                                .flatMap(stack -> emc.get(stack)
+                                                .stream()
+                                                .mapToObj(emcValue -> new StackEmcPair(stack, emcValue)))
+                                .findFirst();
+                sacrifice.ifPresent(pair -> {
+                        pair.stack.setCount(pair.stack.getCount() - 1);
+                        setStoredEmc(getStoredEmc() + pair.emc());
+                });
+        }
+
+        private record StackEmcPair(ItemStack stack, double emc) {}
+}