-
Notifications
You must be signed in to change notification settings - Fork 18
/
DispenseCauldronRecipe.java
83 lines (73 loc) · 3.09 KB
/
DispenseCauldronRecipe.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package knightminer.inspirations.recipes.dispenser;
import knightminer.inspirations.common.Config;
import knightminer.inspirations.library.InspirationsTags;
import knightminer.inspirations.recipes.recipe.inventory.VanillaCauldronInventory;
import knightminer.inspirations.recipes.tileentity.CauldronTileEntity;
import net.minecraft.block.BlockState;
import net.minecraft.block.CauldronBlock;
import net.minecraft.block.DispenserBlock;
import net.minecraft.dispenser.DefaultDispenseItemBehavior;
import net.minecraft.dispenser.IBlockSource;
import net.minecraft.dispenser.IDispenseItemBehavior;
import net.minecraft.item.ItemStack;
import net.minecraft.tileentity.DispenserTileEntity;
import net.minecraft.util.math.BlockPos;
import net.minecraft.world.World;
import slimeknights.mantle.util.TileEntityHelper;
import java.util.Optional;
import java.util.function.Consumer;
/**
* Dispenser behavior logic for cauldron recipes
*/
public class DispenseCauldronRecipe extends DefaultDispenseItemBehavior {
private static final DefaultDispenseItemBehavior DEFAULT = new DefaultDispenseItemBehavior();
private final IDispenseItemBehavior fallback;
/**
* Creates a new instance of the dispenser logic
* @param fallback Fallback if no cauldron is in front of the dispenser
*/
public DispenseCauldronRecipe(IDispenseItemBehavior fallback) {
this.fallback = fallback;
}
@Override
protected ItemStack dispenseStack(IBlockSource source, ItemStack stack) {
if (!stack.getItem().isIn(InspirationsTags.Items.DISP_CAULDRON_RECIPES)) {
return fallback.dispense(source, stack);
}
// find cauldron, quit if missing
World world = source.getWorld();
BlockPos pos = source.getBlockPos().offset(source.getBlockState().get(DispenserBlock.FACING));
BlockState state = world.getBlockState(pos);
if (!(state.getBlock() instanceof CauldronBlock)) {
return fallback.dispense(source, stack);
}
// create consumer to add items
DispenserTileEntity dispenser = source.getBlockTileEntity();
Consumer<ItemStack> addItems = item -> {
if (dispenser.addItemStack(stack) < 0) {
DEFAULT.dispense(source, stack);
}
};
// use tile entity if extended, it handles everything
if (Config.extendedCauldron.getAsBoolean()) {
// if we have the tile entity, run. If missing, fallback to vanilla logic
Optional<CauldronTileEntity> cauldron = TileEntityHelper.getTile(CauldronTileEntity.class, world, pos);
if (cauldron.isPresent()) {
ItemStack newStack = cauldron.get().handleDispenser(stack, addItems);
// nonnull means we did something
if (newStack != null) {
return newStack;
}
// null means we did nothing, dispense the stack
return DEFAULT.dispense(source, stack);
}
}
// use the vanilla logic to run the recipe if no TE
VanillaCauldronInventory inventory = new VanillaCauldronInventory(world, pos, state, stack, addItems);
if (inventory.handleRecipe()) {
return inventory.getStack();
}
// no recipe? dispense directly
return DEFAULT.dispense(source, stack);
}
}