62 lines
2.2 KiB
Java
62 lines
2.2 KiB
Java
package pm.c7.scout.mixin;
|
|
|
|
import net.minecraft.entity.player.PlayerEntity;
|
|
import net.minecraft.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.CallbackInfoReturnable;
|
|
import org.spongepowered.asm.mixin.injection.callback.LocalCapture;
|
|
import pm.c7.scout.ScoutUtil;
|
|
import pm.c7.scout.config.ScoutConfigHandler;
|
|
import pm.c7.scout.item.BaseBagItem;
|
|
|
|
import java.util.function.Predicate;
|
|
|
|
@Mixin(PlayerEntity.class)
|
|
public class PlayerEntityMixin {
|
|
@Inject(method = "getArrowType", at = @At(value = "INVOKE_ASSIGN", target = "Lnet/minecraft/item/RangedWeaponItem;getProjectiles()Ljava/util/function/Predicate;"), locals = LocalCapture.CAPTURE_FAILHARD, cancellable = true)
|
|
public void scout$arrowsFromBags(ItemStack stack, CallbackInfoReturnable<ItemStack> cir, Predicate<ItemStack> predicate, ItemStack itemStack) {
|
|
if ((boolean) ScoutConfigHandler.getConfigValue("useArrows").value()) {
|
|
var self = (PlayerEntity) (Object) this;
|
|
var leftPouch = ScoutUtil.findBagItem(self, BaseBagItem.BagType.POUCH, false);
|
|
var rightPouch = ScoutUtil.findBagItem(self, BaseBagItem.BagType.POUCH, true);
|
|
var satchel = ScoutUtil.findBagItem(self, BaseBagItem.BagType.SATCHEL, false);
|
|
|
|
if (!leftPouch.isEmpty()) {
|
|
BaseBagItem item = (BaseBagItem) leftPouch.getItem();
|
|
var inv = item.getInventory(leftPouch);
|
|
|
|
for(int i = 0; i < inv.size(); ++i) {
|
|
ItemStack invStack = inv.getStack(i);
|
|
if (predicate.test(invStack)) {
|
|
cir.setReturnValue(invStack);
|
|
}
|
|
}
|
|
}
|
|
if (!rightPouch.isEmpty()) {
|
|
BaseBagItem item = (BaseBagItem) rightPouch.getItem();
|
|
var inv = item.getInventory(rightPouch);
|
|
|
|
for(int i = 0; i < inv.size(); ++i) {
|
|
ItemStack invStack = inv.getStack(i);
|
|
if (predicate.test(invStack)) {
|
|
cir.setReturnValue(invStack);
|
|
}
|
|
}
|
|
}
|
|
if (!satchel.isEmpty()) {
|
|
BaseBagItem item = (BaseBagItem) satchel.getItem();
|
|
var inv = item.getInventory(satchel);
|
|
|
|
for(int i = 0; i < inv.size(); ++i) {
|
|
ItemStack invStack = inv.getStack(i);
|
|
if (predicate.test(invStack)) {
|
|
cir.setReturnValue(invStack);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|