1 files changed, 15 insertions, 19 deletions
diff --git a/src/main/java/me/shedaniel/rei/plugin/autocrafting/AutoBlastingBookHandler.java b/src/main/java/me/shedaniel/rei/plugin/autocrafting/AutoBlastingBookHandler.java
index dbdea81bf..855ea7ee1 100644
--- a/src/main/java/me/shedaniel/rei/plugin/autocrafting/AutoBlastingBookHandler.java
+++ b/src/main/java/me/shedaniel/rei/plugin/autocrafting/AutoBlastingBookHandler.java
@@ -5,35 +5,31 @@
 
 package me.shedaniel.rei.plugin.autocrafting;
 
-import me.shedaniel.rei.api.AutoCraftingHandler;
-import me.shedaniel.rei.api.RecipeDisplay;
+import me.shedaniel.rei.api.AutoTransferHandler;
 import me.shedaniel.rei.client.ScreenHelper;
-import me.shedaniel.rei.gui.ContainerScreenOverlay;
 import me.shedaniel.rei.listeners.RecipeBookGuiHooks;
 import me.shedaniel.rei.plugin.blasting.DefaultBlastingDisplay;
-import net.minecraft.client.MinecraftClient;
 import net.minecraft.client.gui.screen.Screen;
-import net.minecraft.client.gui.screen.ingame.AbstractContainerScreen;
 import net.minecraft.client.gui.screen.ingame.BlastFurnaceScreen;
 import net.minecraft.container.BlastFurnaceContainer;
 
-import java.util.function.Supplier;
-
-public class AutoBlastingBookHandler implements AutoCraftingHandler {
+public class AutoBlastingBookHandler implements AutoTransferHandler {
     @Override
-    public boolean handle(Supplier<RecipeDisplay> displaySupplier, MinecraftClient minecraft, Screen recipeViewingScreen, AbstractContainerScreen<?> parentScreen, ContainerScreenOverlay overlay) {
-        DefaultBlastingDisplay display = (DefaultBlastingDisplay) displaySupplier.get();
-        BlastFurnaceScreen furnaceScreen = (BlastFurnaceScreen) parentScreen;
-        minecraft.openScreen(furnaceScreen);
+    public Result handle(Context context) {
+        if (!(context.getContainerScreen() instanceof BlastFurnaceScreen) || !(context.getRecipe() instanceof DefaultBlastingDisplay))
+            return Result.createNotApplicable();
+        if (!((DefaultBlastingDisplay) context.getRecipe()).getOptionalRecipe().isPresent() || !context.getMinecraft().player.getRecipeBook().contains(((DefaultBlastingDisplay) context.getRecipe()).getOptionalRecipe().get()))
+            return Result.createNotApplicable();
+        if (!context.isActuallyCrafting())
+            return Result.createSuccessful();
+        
+        DefaultBlastingDisplay display = (DefaultBlastingDisplay) context.getRecipe();
+        BlastFurnaceScreen furnaceScreen = (BlastFurnaceScreen) context.getContainerScreen();
+        context.getMinecraft().openScreen(furnaceScreen);
         ((RecipeBookGuiHooks) furnaceScreen.getRecipeBookGui()).rei_getGhostSlots().reset();
         BlastFurnaceContainer container = furnaceScreen.getContainer();
-        minecraft.interactionManager.clickRecipe(container.syncId, display.getOptionalRecipe().get(), Screen.hasShiftDown());
+        context.getMinecraft().interactionManager.clickRecipe(container.syncId, display.getOptionalRecipe().get(), Screen.hasShiftDown());
         ScreenHelper.getLastOverlay().init();
-        return true;
-    }
-    
-    @Override
-    public boolean canHandle(Supplier<RecipeDisplay> displaySupplier, MinecraftClient minecraft, Screen recipeViewingScreen, AbstractContainerScreen<?> parentScreen, ContainerScreenOverlay overlay) {
-        return parentScreen instanceof BlastFurnaceScreen && displaySupplier.get() instanceof DefaultBlastingDisplay && ((DefaultBlastingDisplay) displaySupplier.get()).getOptionalRecipe().isPresent() && minecraft.player.getRecipeBook().contains(((DefaultBlastingDisplay) displaySupplier.get()).getOptionalRecipe().get());
+        return Result.createSuccessful();
     }
 }