4.0.14: Better widgets system

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

1 files changed, 242 insertions, 0 deletions

diff --git a/src/main/java/me/shedaniel/rei/impl/InternalWidgets.java b/src/main/java/me/shedaniel/rei/impl/InternalWidgets.java
new file mode 100644
index 000000000..24668e2ae
--- /dev/null
+++ b/src/main/java/me/shedaniel/rei/impl/InternalWidgets.java
@@ -0,0 +1,242 @@
+package me.shedaniel.rei.impl;
+
+import com.google.common.collect.Lists;
+import it.unimi.dsi.fastutil.ints.IntList;
+import me.shedaniel.math.api.Rectangle;
+import me.shedaniel.math.impl.PointHelper;
+import me.shedaniel.rei.api.*;
+import me.shedaniel.rei.api.widgets.Button;
+import me.shedaniel.rei.api.widgets.Widgets;
+import me.shedaniel.rei.gui.toast.CopyRecipeIdentifierToast;
+import me.shedaniel.rei.gui.widget.LateRenderable;
+import me.shedaniel.rei.gui.widget.Widget;
+import me.shedaniel.rei.gui.widget.WidgetWithBounds;
+import me.shedaniel.rei.utils.CollectionUtils;
+import net.minecraft.client.MinecraftClient;
+import net.minecraft.client.gui.Element;
+import net.minecraft.client.gui.screen.ingame.HandledScreen;
+import net.minecraft.client.resource.language.I18n;
+import net.minecraft.util.Formatting;
+import org.jetbrains.annotations.ApiStatus;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.Collections;
+import java.util.List;
+import java.util.Objects;
+import java.util.function.Supplier;
+
+@ApiStatus.Internal
+public final class InternalWidgets {
+    private InternalWidgets() {}
+    
+    public static Widget createAutoCraftingButtonWidget(Rectangle displayBounds, me.shedaniel.math.Rectangle rectangle, String text, Supplier<RecipeDisplay> displaySupplier, List<Widget> setupDisplay, RecipeCategory<?> category) {
+        HandledScreen<?> handledScreen = ScreenHelper.getLastHandledScreen();
+        boolean[] visible = {false};
+        List<String>[] errorTooltip = new List[]{null};
+        Button autoCraftingButton = Widgets.createButton(rectangle, text)
+                .focusable(false)
+                .onClick(button -> {
+                    AutoTransferHandler.Context context = AutoTransferHandler.Context.create(true, handledScreen, displaySupplier.get());
+                    for (AutoTransferHandler autoTransferHandler : RecipeHelper.getInstance().getSortedAutoCraftingHandler())
+                        try {
+                            AutoTransferHandler.Result result = autoTransferHandler.handle(context);
+                            if (result.isSuccessful())
+                                return;
+                        } catch (Exception e) {
+                            e.printStackTrace();
+                        }
+                    MinecraftClient.getInstance().openScreen(handledScreen);
+                    ScreenHelper.getLastOverlay().init();
+                })
+                .onRender(button -> {
+                    button.setEnabled(false);
+                    List<String> error = null;
+                    int color = 0;
+                    visible[0] = false;
+                    IntList redSlots = null;
+                    AutoTransferHandler.Context context = AutoTransferHandler.Context.create(false, handledScreen, displaySupplier.get());
+                    for (AutoTransferHandler autoTransferHandler : RecipeHelper.getInstance().getSortedAutoCraftingHandler()) {
+                        try {
+                            AutoTransferHandler.Result result = autoTransferHandler.handle(context);
+                            if (result.isApplicable())
+                                visible[0] = true;
+                            if (result.isSuccessful()) {
+                                button.setEnabled(true);
+                                error = null;
+                                color = 0;
+                                redSlots = null;
+                                break;
+                            } else if (result.isApplicable()) {
+                                if (error == null) {
+                                    error = Lists.newArrayList();
+                                }
+                                error.add(result.getErrorKey());
+                                color = result.getColor();
+                                if (result.getIntegers() != null && !result.getIntegers().isEmpty())
+                                    redSlots = result.getIntegers();
+                            }
+                        } catch (Exception e) {
+                            e.printStackTrace();
+                        }
+                    }
+                    if (!visible[0]) {
+                        button.setEnabled(false);
+                        if (error == null) {
+                            error = Lists.newArrayList();
+                        } else {
+                            error.clear();
+                        }
+                        error.add("error.rei.no.handlers.applicable");
+                    }
+                    if ((button.containsMouse(PointHelper.ofMouse()) || button.isFocused()) && category instanceof TransferRecipeCategory && redSlots != null) {
+                        ((TransferRecipeCategory<RecipeDisplay>) category).renderRedSlots(setupDisplay, displayBounds, displaySupplier.get(), redSlots);
+                    }
+                    errorTooltip[0] = error == null || error.isEmpty() ? null : Lists.newArrayList();
+                    if (errorTooltip[0] != null) {
+                        for (String s : error) {
+                            if (errorTooltip[0].stream().noneMatch(ss -> ss.equalsIgnoreCase(s)))
+                                errorTooltip[0].add(s);
+                        }
+                    }
+                    button.setTint(color);
+                })
+                .textColor((button, mouse) -> {
+                    if (!visible[0]) {
+                        return 10526880;
+                    } else if (button.isEnabled() && (button.containsMouse(mouse) || button.isFocused())) {
+                        return 16777120;
+                    }
+                    return 14737632;
+                })
+                .textureId((button, mouse) -> !visible[0] ? 0 : (button.containsMouse(mouse) || button.isFocused()) && button.isEnabled() ? 4 : 1)
+                .tooltipSupplier(button -> {
+                    String str = "";
+                    if (errorTooltip[0] == null) {
+                        if (((ClientHelperImpl) ClientHelper.getInstance()).isYog.get())
+                            str += I18n.translate("text.auto_craft.move_items.yog");
+                        else
+                            str += I18n.translate("text.auto_craft.move_items");
+                    } else {
+                        if (errorTooltip[0].size() > 1)
+                            str += Formatting.RED.toString() + I18n.translate("error.rei.multi.errors") + "\n";
+                        str += CollectionUtils.mapAndJoinToString(errorTooltip[0], s -> Formatting.RED.toString() + (errorTooltip[0].size() > 1 ? "- " : "") + I18n.translate(s), "\n");
+                    }
+                    if (MinecraftClient.getInstance().options.advancedItemTooltips) {
+                        str += displaySupplier.get().getRecipeLocation().isPresent() ? I18n.translate("text.rei.recipe_id", Formatting.GRAY.toString(), displaySupplier.get().getRecipeLocation().get().toString()) : "";
+                    }
+                    return str;
+                });
+        return new WidgetWithBounds() {
+            @Override
+            public @NotNull Rectangle getBounds() {
+                return autoCraftingButton.getBounds();
+            }
+            
+            @Override
+            public List<? extends Element> children() {
+                return Collections.singletonList(autoCraftingButton);
+            }
+            
+            @Override
+            public void render(int mouseX, int mouseY, float delta) {
+                autoCraftingButton.render(mouseX, mouseY, delta);
+            }
+            
+            @Override
+            public boolean keyPressed(int int_1, int int_2, int int_3) {
+                if (displaySupplier.get().getRecipeLocation().isPresent() && ConfigObject.getInstance().getCopyRecipeIdentifierKeybind().matchesKey(int_1, int_2) && containsMouse(PointHelper.ofMouse())) {
+                    minecraft.keyboard.setClipboard(displaySupplier.get().getRecipeLocation().get().toString());
+                    if (ConfigObject.getInstance().isToastDisplayedOnCopyIdentifier()) {
+                        CopyRecipeIdentifierToast.addToast(I18n.translate("msg.rei.copied_recipe_id"), I18n.translate("msg.rei.recipe_id_details", displaySupplier.get().getRecipeLocation().get().toString()));
+                    }
+                    return true;
+                }
+                return super.keyPressed(int_1, int_2, int_3);
+            }
+        };
+    }
+    
+    public static LateRenderable wrapLateRenderable(WidgetWithBounds widget) {
+        return new LateRenderableWidgetWithBounds(widget);
+    }
+    
+    public static LateRenderable wrapLateRenderable(Widget widget) {
+        return new LateRenderableWidget(widget);
+    }
+    
+    public static Widget mergeWidgets(Widget widget1, Widget widget2) {
+        return new MergedWidget(widget2, widget1);
+    }
+    
+    private static class MergedWidget extends Widget {
+        private final List<Widget> widgets;
+        
+        public MergedWidget(Widget widget1, Widget widget2) {
+            this.widgets = Lists.newArrayList(Objects.requireNonNull(widget1), Objects.requireNonNull(widget2));
+        }
+        
+        @Override
+        public void render(int mouseX, int mouseY, float delta) {
+            for (Widget widget : widgets) {
+                widget.setZ(getZ());
+                widget.render(mouseX, mouseY, delta);
+            }
+        }
+        
+        @Override
+        public List<? extends Element> children() {
+            return widgets;
+        }
+    }
+    
+    private static class LateRenderableWidget extends Widget implements LateRenderable {
+        private final Widget widget;
+        
+        private LateRenderableWidget(Widget widget) {
+            this.widget = widget;
+        }
+        
+        
+        @Override
+        public void lateRender(int mouseX, int mouseY, float delta) {
+            this.widget.setZ(getZ());
+            this.widget.render(mouseX, mouseY, delta);
+        }
+        
+        @Override
+        public void render(int mouseX, int mouseY, float delta) {}
+        
+        @Override
+        public List<? extends Element> children() {
+            return Collections.singletonList(this.widget);
+        }
+    }
+    
+    private static class LateRenderableWidgetWithBounds extends WidgetWithBounds implements LateRenderable {
+        private final WidgetWithBounds widget;
+        
+        private LateRenderableWidgetWithBounds(WidgetWithBounds widget) {
+            this.widget = widget;
+        }
+        
+        
+        @Override
+        public void lateRender(int mouseX, int mouseY, float delta) {
+            this.widget.setZ(getZ());
+            this.widget.render(mouseX, mouseY, delta);
+        }
+        
+        @Override
+        public @NotNull Rectangle getBounds() {
+            return this.widget.getBounds();
+        }
+        
+        @Override
+        public void render(int mouseX, int mouseY, float delta) {}
+        
+        @Override
+        public List<? extends Element> children() {
+            return Collections.singletonList(this.widget);
+        }
+    }
+}