Close #17

1 files changed, 286 insertions, 0 deletions

diff --git a/src/main/java/me/shedaniel/rei/gui/ContainerScreenOverlay.java b/src/main/java/me/shedaniel/rei/gui/ContainerScreenOverlay.java
new file mode 100644
index 000000000..3da3443cc
--- /dev/null
+++ b/src/main/java/me/shedaniel/rei/gui/ContainerScreenOverlay.java
@@ -0,0 +1,286 @@
+package me.shedaniel.rei.gui;
+
+import com.google.common.collect.Lists;
+import com.mojang.blaze3d.platform.GlStateManager;
+import me.shedaniel.rei.RoughlyEnoughItemsCore;
+import me.shedaniel.rei.client.ClientHelper;
+import me.shedaniel.rei.client.GuiHelper;
+import me.shedaniel.rei.gui.widget.*;
+import net.minecraft.client.MinecraftClient;
+import net.minecraft.client.gui.ContainerScreen;
+import net.minecraft.client.gui.GuiEventListener;
+import net.minecraft.client.gui.Screen;
+import net.minecraft.client.render.GuiLighting;
+import net.minecraft.client.resource.language.I18n;
+import net.minecraft.client.util.Window;
+import net.minecraft.item.ItemStack;
+import net.minecraft.util.math.MathHelper;
+
+import java.awt.*;
+import java.util.Collections;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.stream.Collectors;
+
+public class ContainerScreenOverlay extends Screen {
+    
+    private static final List<QueuedTooltip> QUEUED_TOOLTIPS = Lists.newArrayList();
+    public static String searchTerm = "";
+    private static int page = 0;
+    private static ItemListOverlay itemListOverlay;
+    private final List<IWidget> widgets = Lists.newArrayList();
+    private Rectangle rectangle;
+    private Window window;
+    private ButtonWidget buttonLeft, buttonRight;
+    private int lastLeft;
+    
+    public void onInitialized() {
+        //Update Variables
+        this.widgets.clear();
+        this.window = MinecraftClient.getInstance().window;
+        this.rectangle = calculateBoundary();
+        this.lastLeft = getLeft();
+        widgets.add(this.itemListOverlay = new ItemListOverlay(page));
+        this.itemListOverlay.updateList(getItemListArea(), page, searchTerm);
+        
+        widgets.add(buttonLeft = new ButtonWidget(rectangle.x, rectangle.y + 5, 16, 16, "<") {
+            @Override
+            public void onPressed(int button, double mouseX, double mouseY) {
+                page--;
+                if (page < 0)
+                    page = getTotalPage();
+                itemListOverlay.updateList(getItemListArea(), page, searchTerm);
+            }
+        });
+        widgets.add(buttonRight = new ButtonWidget(rectangle.x + rectangle.width - 18, rectangle.y + 5, 16, 16, ">") {
+            @Override
+            public void onPressed(int button, double mouseX, double mouseY) {
+                page++;
+                if (page > getTotalPage())
+                    page = 0;
+                itemListOverlay.updateList(getItemListArea(), page, searchTerm);
+            }
+        });
+        page = MathHelper.clamp(page, 0, getTotalPage());
+        widgets.add(new ButtonWidget(10, 10, 40, 20, "") {
+            @Override
+            public void draw(int int_1, int int_2, float float_1) {
+                this.text = getCheatModeText();
+                super.draw(int_1, int_2, float_1);
+            }
+            
+            @Override
+            public void onPressed(int button, double mouseX, double mouseY) {
+                ClientHelper.setCheating(!ClientHelper.isCheating());
+            }
+        });
+        widgets.add(new ButtonWidget(10, 35, 40, 20, I18n.translate("text.rei.config")) {
+            @Override
+            public void onPressed(int button, double mouseX, double mouseY) {
+                ClientHelper.openConfigWindow(GuiHelper.getLastContainerScreen());
+            }
+        });
+        this.widgets.add(new LabelWidget(rectangle.x + (rectangle.width / 2), rectangle.y + 10, "") {
+            @Override
+            public void draw(int mouseX, int mouseY, float partialTicks) {
+                page = MathHelper.clamp(page, 0, getTotalPage());
+                this.text = String.format("%s/%s", page + 1, getTotalPage() + 1);
+                super.draw(mouseX, mouseY, partialTicks);
+            }
+        });
+        if (GuiHelper.searchField == null)
+            GuiHelper.searchField = new TextFieldWidget(0, 0, 0, 0) {
+                @Override
+                public boolean mouseClicked(double double_1, double double_2, int int_1) {
+                    if (isVisible() && getBounds().contains(double_1, double_2) && int_1 == 1) {
+                        setText("");
+                        return true;
+                    }
+                    return super.mouseClicked(double_1, double_2, int_1);
+                }
+            };
+        GuiHelper.searchField.setChangedListener(s -> {
+            searchTerm = s;
+            itemListOverlay.updateList(page, searchTerm);
+        });
+        GuiHelper.searchField.setBounds(getTextFieldArea());
+        this.widgets.add(GuiHelper.searchField);
+        GuiHelper.searchField.setText(searchTerm);
+        if (RoughlyEnoughItemsCore.getConfigHelper().showCraftableOnlyButton())
+            this.widgets.add(new CraftableToggleButtonWidget(getCraftableToggleArea()) {
+                @Override
+                public void onPressed(int button, double mouseX, double mouseY) {
+                    RoughlyEnoughItemsCore.getConfigHelper().toggleCraftableOnly();
+                    itemListOverlay.updateList(page, searchTerm);
+                }
+            });
+        
+        this.itemListOverlay.updateList(getItemListArea(), page, searchTerm);
+        this.listeners.addAll(widgets);
+    }
+    
+    private Rectangle getTextFieldArea() {
+        int widthRemoved = RoughlyEnoughItemsCore.getConfigHelper().showCraftableOnlyButton() ? 22 : 0;
+        if (RoughlyEnoughItemsCore.getConfigHelper().sideSearchField())
+            return new Rectangle(rectangle.x + 2, window.getScaledHeight() - 22, rectangle.width - 6 - widthRemoved, 18);
+        if (MinecraftClient.getInstance().currentScreen instanceof RecipeViewingWidgetScreen) {
+            RecipeViewingWidgetScreen widget = (RecipeViewingWidgetScreen) MinecraftClient.getInstance().currentScreen;
+            return new Rectangle(widget.getBounds().x, window.getScaledHeight() - 22, widget.getBounds().width - widthRemoved, 18);
+        }
+        return new Rectangle(GuiHelper.getLastMixinContainerScreen().getContainerLeft(), window.getScaledHeight() - 22, GuiHelper.getLastMixinContainerScreen().getContainerWidth() - widthRemoved, 18);
+    }
+    
+    private Rectangle getCraftableToggleArea() {
+        Rectangle searchBoxArea = getTextFieldArea();
+        searchBoxArea.setLocation(searchBoxArea.x + searchBoxArea.width + 4, searchBoxArea.y - 1);
+        searchBoxArea.setSize(20, 20);
+        return searchBoxArea;
+    }
+    
+    private String getCheatModeText() {
+        return I18n.translate(String.format("%s%s", "text.rei.", ClientHelper.isCheating() ? "cheat" : "nocheat"));
+    }
+    
+    private Rectangle getItemListArea() {
+        return new Rectangle(rectangle.x + 2, rectangle.y + 24, rectangle.width - 4, rectangle.height - (RoughlyEnoughItemsCore.getConfigHelper().sideSearchField() ? 27 + 22 : 27));
+    }
+    
+    public Rectangle getRectangle() {
+        return rectangle;
+    }
+    
+    public void render(int mouseX, int mouseY, float partialTicks) {
+        List<ItemStack> currentStacks = ClientHelper.getInventoryItemsTypes();
+        if (getLeft() != lastLeft)
+            onInitialized();
+        else if (RoughlyEnoughItemsCore.getConfigHelper().craftableOnly() && (!hasSameListContent(new LinkedList<>(GuiHelper.inventoryStacks), currentStacks) || (currentStacks.size() != GuiHelper.inventoryStacks.size()))) {
+            GuiHelper.inventoryStacks = ClientHelper.getInventoryItemsTypes();
+            itemListOverlay.updateList(page, searchTerm);
+        }
+        GlStateManager.color4f(1.0F, 1.0F, 1.0F, 1.0F);
+        GuiLighting.disable();
+        this.draw(mouseX, mouseY, partialTicks);
+        GuiLighting.disable();
+        QUEUED_TOOLTIPS.stream().filter(queuedTooltip -> queuedTooltip != null).forEach(queuedTooltip -> MinecraftClient.getInstance().currentScreen.drawTooltip(queuedTooltip.text, queuedTooltip.mouse.x, queuedTooltip.mouse.y));
+        QUEUED_TOOLTIPS.clear();
+        GuiLighting.disable();
+    }
+    
+    private boolean hasSameListContent(List<ItemStack> list1, List<ItemStack> list2) {
+        Collections.sort(list1, (itemStack, t1) -> {
+            return itemStack.getDisplayName().getFormattedText().compareToIgnoreCase(t1.getDisplayName().getFormattedText());
+        });
+        Collections.sort(list2, (itemStack, t1) -> {
+            return itemStack.getDisplayName().getFormattedText().compareToIgnoreCase(t1.getDisplayName().getFormattedText());
+        });
+        String lastString = String.join("", list1.stream().map(itemStack -> {
+            return itemStack.getDisplayName().getFormattedText();
+        }).collect(Collectors.toList())), currentString = String.join("", list2.stream().map(itemStack -> {
+            return itemStack.getDisplayName().getFormattedText();
+        }).collect(Collectors.toList()));
+        return lastString.equals(currentString);
+    }
+    
+    public void addTooltip(QueuedTooltip queuedTooltip) {
+        QUEUED_TOOLTIPS.add(queuedTooltip);
+    }
+    
+    @Override
+    public void draw(int int_1, int int_2, float float_1) {
+        if (!GuiHelper.isOverlayVisible())
+            return;
+        widgets.forEach(widget -> {
+            GuiLighting.disable();
+            widget.draw(int_1, int_2, float_1);
+        });
+        GuiLighting.disable();
+        itemListOverlay.draw(int_1, int_2, float_1);
+        GuiLighting.disable();
+        super.draw(int_1, int_2, float_1);
+    }
+    
+    private Rectangle calculateBoundary() {
+        int startX = GuiHelper.getLastMixinContainerScreen().getContainerLeft() + GuiHelper.getLastMixinContainerScreen().getContainerWidth() + 10;
+        int width = window.getScaledWidth() - startX;
+        if (MinecraftClient.getInstance().currentScreen instanceof RecipeViewingWidgetScreen) {
+            RecipeViewingWidgetScreen widget = (RecipeViewingWidgetScreen) MinecraftClient.getInstance().currentScreen;
+            startX = widget.getBounds().x + widget.getBounds().width + 10;
+            width = window.getScaledWidth() - startX;
+        }
+        return new Rectangle(startX, 0, width, window.getScaledHeight());
+    }
+    
+    private int getLeft() {
+        if (MinecraftClient.getInstance().currentScreen instanceof RecipeViewingWidgetScreen) {
+            RecipeViewingWidgetScreen widget = (RecipeViewingWidgetScreen) MinecraftClient.getInstance().currentScreen;
+            return widget.getBounds().x;
+        }
+        return GuiHelper.getLastMixinContainerScreen().getContainerLeft();
+    }
+    
+    private int getTotalPage() {
+        return MathHelper.ceil(itemListOverlay.getCurrentDisplayed().size() / itemListOverlay.getTotalSlotsPerPage());
+    }
+    
+    @Override
+    public boolean mouseScrolled(double amount) {
+        if (rectangle.contains(ClientHelper.getMouseLocation())) {
+            if (amount > 0 && buttonLeft.enabled)
+                buttonLeft.onPressed(0, 0, 0);
+            else if (amount < 0 && buttonRight.enabled)
+                buttonRight.onPressed(0, 0, 0);
+            else
+                return false;
+            return true;
+        }
+        for(IWidget widget : widgets)
+            if (widget.mouseScrolled(amount))
+                return true;
+        return false;
+    }
+    
+    @Override
+    public boolean keyPressed(int int_1, int int_2, int int_3) {
+        for(GuiEventListener listener : listeners)
+            if (listener.keyPressed(int_1, int_2, int_3))
+                return true;
+        Point point = ClientHelper.getMouseLocation();
+        ItemStack itemStack = null;
+        for(IWidget widget : itemListOverlay.getListeners())
+            if (widget instanceof ItemSlotWidget && ((ItemSlotWidget) widget).isHighlighted(point.x, point.y)) {
+                itemStack = ((ItemSlotWidget) widget).getCurrentStack();
+                break;
+            }
+        if (itemStack == null && MinecraftClient.getInstance().currentScreen instanceof RecipeViewingWidgetScreen) {
+            RecipeViewingWidgetScreen recipeViewingWidget = (RecipeViewingWidgetScreen) MinecraftClient.getInstance().currentScreen;
+            for(GuiEventListener entry : recipeViewingWidget.getEntries())
+                if (entry instanceof ItemSlotWidget && ((HighlightableWidget) entry).isHighlighted(point.x, point.y)) {
+                    itemStack = ((ItemSlotWidget) entry).getCurrentStack();
+                    break;
+                }
+        }
+        if (itemStack == null && MinecraftClient.getInstance().currentScreen instanceof ContainerScreen)
+            if (GuiHelper.getLastMixinContainerScreen().getHoveredSlot() != null)
+                itemStack = GuiHelper.getLastMixinContainerScreen().getHoveredSlot().getStack();
+        if (itemStack != null && !itemStack.isEmpty()) {
+            if (ClientHelper.RECIPE.matchesKey(int_1, int_2))
+                return ClientHelper.executeRecipeKeyBind(this, itemStack);
+            else if (ClientHelper.USAGE.matchesKey(int_1, int_2))
+                return ClientHelper.executeUsageKeyBind(this, itemStack);
+        }
+        if (ClientHelper.HIDE.matchesKey(int_1, int_2)) {
+            GuiHelper.toggleOverlayVisible();
+            return true;
+        }
+        return false;
+    }
+    
+    @Override
+    public boolean charTyped(char char_1, int int_1) {
+        for(GuiEventListener listener : listeners)
+            if (listener.charTyped(char_1, int_1))
+                return true;
+        return super.charTyped(char_1, int_1);
+    }
+    
+}