make owned allays not able to pickup locked items or if claim disables it close #223

common/src/main/java/io/github/flemmli97/flan/mixin/AllayMixin.java

@@ -0,0 +1,41 @@
+package io.github.flemmli97.flan.mixin;
+
+import io.github.flemmli97.flan.api.permission.PermissionRegistry;
+import io.github.flemmli97.flan.claim.Claim;
+import io.github.flemmli97.flan.claim.ClaimStorage;
+import io.github.flemmli97.flan.player.IOwnedItem;
+import net.minecraft.world.entity.animal.allay.Allay;
+import net.minecraft.world.entity.animal.allay.AllayAi;
+import net.minecraft.world.entity.item.ItemEntity;
+import net.minecraft.world.item.ItemStack;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
+
+@Mixin(Allay.class)
+public abstract class AllayMixin {
+
+    @Inject(method = "wantsToPickUp", at = @At("HEAD"), cancellable = true)
+    private void onWantingPickup(ItemStack stack, CallbackInfoReturnable<Boolean> info) {
+        Allay allay = (Allay) (Object) this;
+        if (AllayAi.getLikedPlayer(allay).map(p -> {
+            Claim claim = ClaimStorage.get(p.getLevel()).getClaimAt(allay.blockPosition());
+            return claim != null && (!claim.canInteract(p, PermissionRegistry.PICKUP, allay.blockPosition(), false));
+        }).orElse(false))
+            info.setReturnValue(false);
+    }
+
+    @Inject(method = "pickUpItem", at = @At("HEAD"), cancellable = true)
+    private void onPickupItem(ItemEntity itemEntity, CallbackInfo info) {
+        if (AllayAi.getLikedPlayer((Allay) (Object) this).map(p -> {
+            IOwnedItem ownedItem = (IOwnedItem) itemEntity;
+            if (p.getUUID().equals(ownedItem.getPlayerOrigin()))
+                return true;
+            Claim claim = ClaimStorage.get(p.getLevel()).getClaimAt(itemEntity.blockPosition());
+            return claim != null && !claim.canInteract(p, PermissionRegistry.PICKUP, itemEntity.blockPosition(), false);
+        }).orElse(false))
+            info.cancel();
+    }
+}

common/src/main/java/io/github/flemmli97/flan/mixin/DragonEggBlockMixin.java

@@ -21,7 +21,7 @@ import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
 
 @Mixin(DragonEggBlock.class)
-public class DragonEggBlockMixin {
+public abstract class DragonEggBlockMixin {
 
     @Unique
     private Player flanTempPlayer;

common/src/main/java/io/github/flemmli97/flan/mixin/MossBlockCheckMixin.java

@@ -10,7 +10,7 @@ import org.spongepowered.asm.mixin.injection.Inject;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
 
 @Mixin(BoneMealItem.class)
-public class MossBlockCheckMixin {
+public abstract class MossBlockCheckMixin {
 
     @Inject(method = "useOn", at = @At("HEAD"), cancellable = true)
     private void check(UseOnContext context, CallbackInfoReturnable<InteractionResult> info) {

common/src/main/resources/flan.mixins.json

@@ -29,7 +29,8 @@
     "EnderMitePearlCheck",
     "SculkSensorMixin",
     "SculkShriekerMixin",
-    "DragonEggBlockMixin"
+    "DragonEggBlockMixin",
+    "AllayMixin"
   ],
   "server": [
   ],