EmptyBucketDispenseBehavior.java

package net.glowstone.dispenser;

import net.glowstone.block.GlowBlock;
import net.glowstone.block.ItemTable;
import net.glowstone.block.blocktype.BlockDispenser;
import net.glowstone.block.blocktype.BlockLiquid;
import net.glowstone.block.blocktype.BlockType;
import net.glowstone.block.state.GlowDispenser;
import org.bukkit.Material;
import org.bukkit.inventory.ItemStack;

public class EmptyBucketDispenseBehavior extends DefaultDispenseBehavior {
    private DefaultDispenseBehavior defaultBehavior = new DefaultDispenseBehavior();

    @Override
    protected ItemStack dispenseStack(GlowBlock block, ItemStack stack) {
        GlowDispenser dispenser = (GlowDispenser) block.getState();
        GlowBlock target = block.getRelative(BlockDispenser.getFacing(block));
        BlockLiquid liquid = collectableLiquidAtBlock(target);
        if (liquid == null) {
            return super.dispenseStack(block, stack);
        }
        Material bucket = liquid.getBucketType();
        target.setType(Material.AIR);
        stack.setAmount(stack.getAmount() - 1);
        if (stack.getAmount() == 0) {
            stack.setAmount(1);
            stack.setType(bucket);
        } else {
            ItemStack toPlace = new ItemStack(bucket);
            ItemStack remaining = dispenser.placeInDispenser(toPlace);
            if (remaining != null) {
                defaultBehavior.dispense(block, remaining);
            }
        }

        return stack;
    }

    private BlockLiquid collectableLiquidAtBlock(GlowBlock target) {
        Material material = target.getType();
        if (material == null || material == Material.AIR) {
            return null;
        }

        BlockType type = ItemTable.instance().getBlock(material);
        if (!(type instanceof BlockLiquid)) {
            return null;
        }

        BlockLiquid liquid = (BlockLiquid) type;
        if (!liquid.isCollectible(target.getState())) {
            return null;
        }

        return liquid;
    }
}