1 files changed, 56 insertions, 0 deletions

diff --git a/src/main/java/lv/enes/mc/eris_alchemy/mixin/ContainerOpenersCounterMixin.java b/src/main/java/lv/enes/mc/eris_alchemy/mixin/ContainerOpenersCounterMixin.java
new file mode 100644
index 0000000..13b8669
--- /dev/null
+++ b/src/main/java/lv/enes/mc/eris_alchemy/mixin/ContainerOpenersCounterMixin.java
@@ -0,0 +1,56 @@
+package lv.enes.mc.eris_alchemy.mixin;
+
+import jakarta.annotation.Nonnull;
+import lv.enes.mc.eris_alchemy.utils.ContainerOpenersCounterUtil;
+import net.minecraft.core.BlockPos;
+import net.minecraft.server.level.ServerPlayer;
+import net.minecraft.world.entity.player.Player;
+import net.minecraft.world.level.Level;
+import net.minecraft.world.level.block.entity.ContainerOpenersCounter;
+import net.minecraft.world.level.block.state.BlockState;
+import net.minecraft.world.level.entity.EntityTypeTest;
+import net.minecraft.world.phys.AABB;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.Unique;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+import java.util.List;
+import java.util.Map;
+import java.util.WeakHashMap;
+
+@Mixin(ContainerOpenersCounter.class)
+public abstract class ContainerOpenersCounterMixin
+                implements ContainerOpenersCounterUtil.ContainerOpenersCounterSuper {
+        @Shadow protected abstract boolean isOwnContainer(Player player);
+
+        @Unique
+        private final static Map<ContainerOpenersCounterMixin, List<ServerPlayer>> openers = new WeakHashMap<>();
+
+        @Inject(method = "recheckOpeners", at = @At("RETURN"))
+        public void onRecheckOpeners(Level world, BlockPos pos, BlockState state, CallbackInfo ci) {
+                int x = pos.getX();
+                int y = pos.getY();
+                int z = pos.getZ();
+                var boundingBox = new AABB(
+                                x - 5.0, y - 5.0, z - 5.0,
+                                x + 5.0, y + 5.0, z + 5.0
+                );
+                var players = world.getEntities(EntityTypeTest.forClass(ServerPlayer.class), boundingBox, this::isOwnContainer);
+                if (openers.containsKey(this)) {
+                        var list = openers.get(this);
+                        list.clear();
+                        list.addAll(players);
+                } else {
+                        openers.put(this, players);
+                }
+        }
+
+        @Nonnull
+        @Override
+        public List<ServerPlayer> lv_enes_mc$getOpeners() {
+                return openers.getOrDefault(this, List.of());
+        }
+}