61 lines
2.4 KiB
Java
61 lines
2.4 KiB
Java
package wtf.hak.survivalfabric.mixin;
|
|
|
|
import net.minecraft.block.Block;
|
|
import net.minecraft.block.BlockState;
|
|
import net.minecraft.block.CropBlock;
|
|
import net.minecraft.entity.player.PlayerEntity;
|
|
import net.minecraft.item.HoeItem;
|
|
import net.minecraft.item.Item;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.server.world.ServerWorld;
|
|
import net.minecraft.stat.Stats;
|
|
import net.minecraft.util.Hand;
|
|
import net.minecraft.util.math.BlockPos;
|
|
import net.minecraft.world.GameMode;
|
|
import net.minecraft.world.World;
|
|
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 wtf.hak.survivalfabric.config.ConfigManager;
|
|
|
|
import java.util.List;
|
|
|
|
@Mixin(Block.class)
|
|
public abstract class BlockMixin {
|
|
|
|
@Inject(method = "onBreak", at = @At("HEAD"), cancellable = true)
|
|
public void onBreak(World world, BlockPos pos, BlockState state, PlayerEntity player, CallbackInfoReturnable<BlockState> cir) {
|
|
if (world.isClient()) return;
|
|
|
|
if (state.getBlock() instanceof CropBlock && ConfigManager.getConfig().replenishEnabled) {
|
|
ItemStack mainHand = player.getStackInHand(Hand.MAIN_HAND);
|
|
if (mainHand.getItem() instanceof HoeItem) {
|
|
Item seedItem = state.getBlock().asItem();
|
|
Block seedBlock = state.getBlock();
|
|
List<ItemStack> drops = Block.getDroppedStacks(state, (ServerWorld) world, pos, null, player, mainHand);
|
|
if (removeIfAvailable(drops, seedItem)) {
|
|
if (player.getGameMode() != GameMode.CREATIVE) {
|
|
for (ItemStack drop : drops) {
|
|
Block.dropStack(world, pos, drop);
|
|
}
|
|
}
|
|
world.getServer().executeSync(() -> world.setBlockState(pos, seedBlock.getDefaultState()));
|
|
player.incrementStat(Stats.USED.getOrCreateStat(seedItem));
|
|
cir.cancel();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private boolean removeIfAvailable(List<ItemStack> drops, Item item) {
|
|
for (ItemStack drop : drops) {
|
|
if (drop.getItem() == item) {
|
|
drop.decrement(1);
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
}
|