2525import javax .annotation .Nonnull ;
2626import javax .annotation .Nullable ;
2727
28+ import org .jetbrains .annotations .NotNull ;
29+
2830import mod .gottsch .forge .gottschcore .enums .IRarity ;
2931import mod .gottsch .forge .treasure2 .Treasure ;
3032import mod .gottsch .forge .treasure2 .api .TreasureApi ;
@@ -208,19 +210,33 @@ protected void onContentsChanged(int slot) {
208210
209211 @ Override
210212 public boolean isItemValid (int slot , @ Nonnull ItemStack stack ) {
211- // TODO add check for locked property and return false is enabled
212- // return ForgeHooks.getBurnTime(stack, RecipeType.SMELTING) > 0;
213- return true ;
213+ return !isLocked ();
214214 }
215215
216216 @ Nonnull
217217 @ Override
218218 public ItemStack insertItem (int slot , @ Nonnull ItemStack stack , boolean simulate ) {
219- // if (ForgeHooks.getBurnTime(stack, RecipeType.SMELTING) <= 0 ) {
220- // return stack ;
221- // }
219+ if (isLocked () ) {
220+ return ItemStack . EMPTY ;
221+ }
222222 return super .insertItem (slot , stack , simulate );
223223 }
224+
225+ @ Override
226+ public @ NotNull ItemStack getStackInSlot (int slot ) {
227+ if (isLocked ()) {
228+ return ItemStack .EMPTY ;
229+ }
230+ return super .getStackInSlot (slot );
231+ }
232+
233+ @ Override
234+ public void setStackInSlot (int slot , @ NotNull ItemStack stack ) {
235+ if (isLocked ()) {
236+ return ;
237+ }
238+ super .setStackInSlot (slot , stack );
239+ }
224240 };
225241 }
226242
@@ -229,7 +245,7 @@ public <T> LazyOptional<T> getCapability(Capability<T> cap, Direction side) {
229245 if (cap == CapabilityItemHandler .ITEM_HANDLER_CAPABILITY ) {
230246 return instanceHandler .cast ();
231247 }
232- return super .getCapability (cap );
248+ return super .getCapability (cap , side );
233249 }
234250
235251 /**
0 commit comments