42 lines
2.0 KiB
Java
42 lines
2.0 KiB
Java
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.serverLevel()).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.serverLevel()).getClaimAt(itemEntity.blockPosition());
|
|
return claim != null && !claim.canInteract(p, PermissionRegistry.PICKUP, itemEntity.blockPosition(), false);
|
|
}).orElse(false))
|
|
info.cancel();
|
|
}
|
|
}
|