Use AbstractRecipeViewingScreen and remove static from ScreenHelper

Signed-off-by: shedaniel <daniel@shedaniel.me>

1 files changed, 235 insertions, 0 deletions

diff --git a/runtime/src/main/java/me/shedaniel/rei/impl/REIHelperImpl.java b/runtime/src/main/java/me/shedaniel/rei/impl/REIHelperImpl.java
new file mode 100644
index 000000000..5f36fd2cd
--- /dev/null
+++ b/runtime/src/main/java/me/shedaniel/rei/impl/REIHelperImpl.java
@@ -0,0 +1,235 @@
+/*
+ * This file is licensed under the MIT License, part of Roughly Enough Items.
+ * Copyright (c) 2018, 2019, 2020 shedaniel
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+package me.shedaniel.rei.impl;
+
+import com.google.common.collect.Iterables;
+import com.google.common.collect.Sets;
+import com.mojang.blaze3d.platform.Window;
+import me.shedaniel.architectury.event.events.GuiEvent;
+import me.shedaniel.architectury.event.events.client.ClientTickEvent;
+import me.shedaniel.math.Rectangle;
+import me.shedaniel.rei.api.ConfigManager;
+import me.shedaniel.rei.api.ConfigObject;
+import me.shedaniel.rei.api.REIHelper;
+import me.shedaniel.rei.api.REIOverlay;
+import me.shedaniel.rei.api.gui.config.SearchFieldLocation;
+import me.shedaniel.rei.api.gui.widgets.TextField;
+import me.shedaniel.rei.api.gui.widgets.Tooltip;
+import me.shedaniel.rei.api.registry.screen.DisplayBoundsProvider;
+import me.shedaniel.rei.api.registry.screen.OverlayDecider;
+import me.shedaniel.rei.api.registry.screen.ScreenRegistry;
+import me.shedaniel.rei.gui.ContainerScreenOverlay;
+import me.shedaniel.rei.gui.OverlaySearchField;
+import me.shedaniel.rei.gui.RecipeScreen;
+import net.fabricmc.api.EnvType;
+import net.fabricmc.api.Environment;
+import net.minecraft.client.Minecraft;
+import net.minecraft.client.gui.screens.Screen;
+import net.minecraft.client.gui.screens.inventory.AbstractContainerScreen;
+import net.minecraft.resources.ResourceLocation;
+import net.minecraft.world.InteractionResult;
+import org.jetbrains.annotations.ApiStatus;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.LinkedHashSet;
+import java.util.Optional;
+
+@ApiStatus.Internal
+@Environment(EnvType.CLIENT)
+public class REIHelperImpl implements REIHelper {
+    private static final ResourceLocation DISPLAY_TEXTURE = new ResourceLocation("roughlyenoughitems", "textures/gui/display.png");
+    private static final ResourceLocation DISPLAY_TEXTURE_DARK = new ResourceLocation("roughlyenoughitems", "textures/gui/display_dark.png");
+    @ApiStatus.Internal
+    public static boolean isWithinRecipeViewingScreen = false;
+    private ContainerScreenOverlay overlay;
+    private OverlaySearchField searchField;
+    private AbstractContainerScreen<?> previousContainerScreen = null;
+    private Screen previousScreen = null;
+    private LinkedHashSet<RecipeScreen> lastRecipeScreen = Sets.newLinkedHashSetWithExpectedSize(10);
+    
+    /**
+     * @return the instance of screen helper
+     * @see REIHelper#getInstance()
+     */
+    @ApiStatus.Internal
+    public static REIHelperImpl getInstance() {
+        return (REIHelperImpl) REIHelper.getInstance();
+    }
+    
+    @Override
+    public void queueTooltip(@Nullable Tooltip tooltip) {
+        if (overlay != null && tooltip != null) {
+            overlay.addTooltip(tooltip);
+        }
+    }
+    
+    @Override
+    @Nullable
+    public TextField getSearchTextField() {
+        if (searchField == null) {
+            searchField = new OverlaySearchField(0, 0, 0, 0);
+        }
+        
+        return searchField;
+    }
+    
+    @Nullable
+    public static OverlaySearchField getSearchField() {
+        return (OverlaySearchField) getInstance().getSearchTextField();
+    }
+    
+    public void storeRecipeScreen(RecipeScreen screen) {
+        while (lastRecipeScreen.size() >= 10)
+            lastRecipeScreen.remove(Iterables.get(lastRecipeScreen, 0));
+        lastRecipeScreen.add(screen);
+    }
+    
+    public boolean hasLastRecipeScreen() {
+        return !lastRecipeScreen.isEmpty();
+    }
+    
+    public Screen getLastRecipeScreen() {
+        RecipeScreen screen = Iterables.getLast(lastRecipeScreen);
+        lastRecipeScreen.remove(screen);
+        screen.recalculateCategoryPage();
+        return (Screen) screen;
+    }
+    
+    @Override
+    public boolean isOverlayVisible() {
+        return ConfigObject.getInstance().isOverlayVisible();
+    }
+    
+    @Override
+    public void toggleOverlayVisible() {
+        ConfigObject.getInstance().setOverlayVisible(!ConfigObject.getInstance().isOverlayVisible());
+        ConfigManager.getInstance().saveConfig();
+    }
+    
+    @Override
+    public Optional<REIOverlay> getOverlay(boolean reset) {
+        if (overlay == null || reset) {
+            overlay = new ContainerScreenOverlay();
+            overlay.init();
+            getSearchField().setFocused(false);
+        }
+        
+        return Optional.ofNullable(overlay);
+    }
+    
+    @Override
+    @Nullable
+    public AbstractContainerScreen<?> getPreviousContainerScreen() {
+        return previousContainerScreen;
+    }
+    
+    @Override
+    @Nullable
+    public Screen getPreviousScreen() {
+        return previousScreen;
+    }
+    
+    public void setPreviousScreen(Screen previousScreen) {
+        if (previousScreen == null || previousScreen.getClass().getName().contains(".rei.")) {
+            return;
+        }
+        
+        this.previousScreen = previousScreen;
+        
+        if (previousScreen instanceof AbstractContainerScreen) {
+            this.previousContainerScreen = (AbstractContainerScreen<?>) previousScreen;
+        }
+    }
+    
+    @Override
+    public boolean isDarkThemeEnabled() {
+        return ConfigObject.getInstance().isUsingDarkTheme();
+    }
+    
+    @Override
+    public ResourceLocation getDefaultDisplayTexture() {
+        return isDarkThemeEnabled() ? DISPLAY_TEXTURE_DARK : DISPLAY_TEXTURE;
+    }
+    
+    @Override
+    public SearchFieldLocation getContextualSearchFieldLocation() {
+        Window window = Minecraft.getInstance().getWindow();
+        for (OverlayDecider decider : ScreenRegistry.getInstance().getDeciders(Minecraft.getInstance().screen.getClass())) {
+            if (decider instanceof DisplayBoundsProvider) {
+                Rectangle containerBounds = ((DisplayBoundsProvider<Screen>) decider).getScreenBounds(Minecraft.getInstance().screen);
+                if (window.getGuiScaledHeight() - 20 <= containerBounds.getMaxY()) {
+                    return SearchFieldLocation.BOTTOM_SIDE;
+                } else break;
+            }
+        }
+        
+        return ConfigObject.getInstance().getSearchFieldLocation();
+    }
+    
+    @Override
+    public Rectangle calculateEntryListArea() {
+        Rectangle bounds = ScreenRegistry.getInstance().getOverlayBounds(ConfigObject.getInstance().getDisplayPanelLocation(), Minecraft.getInstance().screen);
+        SearchFieldLocation searchFieldLocation = getContextualSearchFieldLocation();
+        
+        int yOffset = 2;
+        if (searchFieldLocation == SearchFieldLocation.TOP_SIDE) yOffset += 24;
+        if (!ConfigObject.getInstance().isEntryListWidgetScrolled()) yOffset += 22;
+        int heightOffset = 0;
+        if (searchFieldLocation == SearchFieldLocation.BOTTOM_SIDE) heightOffset += 24;
+        return new Rectangle(bounds.x, bounds.y + yOffset, bounds.width, bounds.height - 1 - yOffset - heightOffset);
+    }
+    
+    @Override
+    public Rectangle calculateFavoritesListArea() {
+        Rectangle bounds = ScreenRegistry.getInstance().getOverlayBounds(ConfigObject.getInstance().getDisplayPanelLocation().mirror(), Minecraft.getInstance().screen);
+        
+        int yOffset = 8;
+        if (ConfigObject.getInstance().doesShowUtilsButtons()) yOffset += 50;
+        else if (!ConfigObject.getInstance().isLowerConfigButton()) yOffset += 25;
+        return new Rectangle(bounds.x, bounds.y + yOffset, bounds.width, bounds.height - 3 - yOffset);
+    }
+    
+    @Override
+    public void startReload() {
+        getOverlay().ifPresent(REIOverlay::queueReloadOverlay);
+        lastRecipeScreen.clear();
+    }
+    
+    @Override
+    public void endReload() {
+        getOverlay().ifPresent(REIOverlay::queueReloadOverlay);
+    }
+    
+    public void onInitializeClient() {
+        GuiEvent.INIT_PRE.register((screen, widgets, children) -> {
+            if (previousContainerScreen != screen && screen instanceof AbstractContainerScreen)
+                previousContainerScreen = (AbstractContainerScreen<?>) screen;
+            return InteractionResult.PASS;
+        });
+        ClientTickEvent.CLIENT_POST.register(minecraft -> {
+            if (isOverlayVisible() && getSearchField() != null)
+                getSearchField().tick();
+        });
+    }
+}