RightClickHarvesting.java

package com.pam.harvestcraft.addons;

import com.pam.harvestcraft.HarvestCraft;
import com.pam.harvestcraft.blocks.growables.BlockPamFruit;
import com.pam.harvestcraft.blocks.growables.BlockPamFruitLog;
import com.pam.harvestcraft.blocks.growables.PamCropGrowable;
import net.minecraft.block.BlockCarrot;
import net.minecraft.block.BlockCrops;
import net.minecraft.block.BlockPotato;
import net.minecraft.block.state.IBlockState;
import net.minecraft.enchantment.EnchantmentHelper;
import net.minecraft.entity.item.EntityItem;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.init.Enchantments;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.util.EnumHand;
import net.minecraft.util.math.BlockPos;
import net.minecraft.world.World;
import net.minecraftforge.common.MinecraftForge;
import net.minecraftforge.event.entity.player.PlayerInteractEvent;
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;

import java.util.Iterator;
import java.util.List;

public class RightClickHarvesting {

    public static final RightClickHarvesting instance = new RightClickHarvesting();

    private RightClickHarvesting() {}

    public void register() {
        MinecraftForge.EVENT_BUS.register(this);
    }

    @SuppressWarnings("unused")
    @SubscribeEvent
    public void onPlayerInteract(PlayerInteractEvent.RightClickBlock event) {
        if (!HarvestCraft.config.enableEasyHarvest) return;

        if (event.getWorld().isRemote) {
            return;
        }

        if (event.getEntityPlayer() == null) return;
        if (event.getHand() != EnumHand.MAIN_HAND) return;

        final IBlockState blockState = event.getWorld().getBlockState(event.getPos());

        if (blockState.getBlock() instanceof BlockCrops) {
            harvestCrops(blockState, event.getEntityPlayer(), event.getWorld(), event.getPos());
        }

        if (blockState.getBlock() instanceof BlockPamFruit || blockState.getBlock() instanceof BlockPamFruitLog) {
            harvestFruit(blockState, event.getEntityPlayer(), event.getWorld(), event.getPos());
        }
    }

    private static void harvestCrops(IBlockState blockState, EntityPlayer player, World world, BlockPos blockPos) {
        final BlockCrops crops = (BlockCrops) blockState.getBlock();
        if (crops.isMaxAge(blockState)) {
            final ItemStack stack = player.getHeldItemMainhand();
            final int fortune = EnchantmentHelper.getEnchantmentLevel(Enchantments.FORTUNE, stack);

            final List<ItemStack> drops = crops.getDrops(world, blockPos, blockState, fortune);

            // This removes exactly one seed from drops in order to make this more fair compared to vanilla
            // as one seed stays planted.
            final Item seedItem = crops.getItemDropped(blockState, world.rand, fortune);
            if (seedItem != null) for (Iterator<ItemStack> iterator = drops.iterator(); iterator.hasNext();) {
                final ItemStack drop = iterator.next();

                // Remove a seed, then break.
                if (! (drop.getItem() == seedItem) || crops instanceof BlockCarrot || crops instanceof BlockPotato) {
                    iterator.remove();
                    break;
                }
            }

            // Reset growth level
            world.setBlockState(blockPos, crops.withAge(0));

            for (ItemStack drop : drops) {
                dropItem(drop, world, blockPos);
            }
        }
    }

    private static void harvestFruit(IBlockState blockState, EntityPlayer player, World world, BlockPos blockPos) {
        final PamCropGrowable blockPamFruit =  (PamCropGrowable) blockState.getBlock();

        if (blockPamFruit.isMature(blockState)) {
            final ItemStack stack = player.getHeldItemMainhand();
            final int fortune = EnchantmentHelper.getEnchantmentLevel(Enchantments.FORTUNE, stack);
            final List<ItemStack> drops = blockPamFruit.getDrops(world, blockPos, blockState, fortune);

            // This removes exactly one fruit from drops in order to make this more fair compared to "vanilla"
            // as one fruit stays planted.
            if (drops.size() > 0) drops.remove(drops.size() - 1);

            world.setBlockState(blockPos, blockState.withProperty(blockPamFruit.getAgeProperty(), 0), 3);

            for (ItemStack drop : drops) {
                dropItem(drop, world, blockPos);
            }
        }
    }

    private static void dropItem(ItemStack itemStack, World world, BlockPos pos) {
        if (world.restoringBlockSnapshots || world.isRemote) return;

        float f = 0.5F;
        double d0 = (world.rand.nextFloat() * f) + 0.25D;
        double d1 = (world.rand.nextFloat() * f) + 0.25D;
        double d2 = (world.rand.nextFloat() * f) + 0.25D;

        final EntityItem entityItem = new EntityItem(world, pos.getX() + d0, pos.getY() + d1, pos.getZ() + d2, itemStack);
        entityItem.setDefaultPickupDelay();
        world.spawnEntity(entityItem);
    }

}