wip more

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

1 files changed, 348 insertions, 0 deletions

diff --git a/runtime/src/main/java/me/shedaniel/rei/impl/InternalWidgets.java b/runtime/src/main/java/me/shedaniel/rei/impl/InternalWidgets.java
new file mode 100644
index 000000000..9256367e0
--- /dev/null
+++ b/runtime/src/main/java/me/shedaniel/rei/impl/InternalWidgets.java
@@ -0,0 +1,348 @@
+/*
+ * 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.Lists;
+import com.mojang.blaze3d.vertex.PoseStack;
+import it.unimi.dsi.fastutil.ints.IntList;
+import me.shedaniel.math.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.fabricmc.api.EnvType;
+import net.fabricmc.api.Environment;
+import net.minecraft.ChatFormatting;
+import net.minecraft.client.Minecraft;
+import net.minecraft.client.gui.components.events.GuiEventListener;
+import net.minecraft.client.gui.screens.inventory.AbstractContainerScreen;
+import net.minecraft.client.resources.language.I18n;
+import net.minecraft.network.chat.Component;
+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
+@Environment(EnvType.CLIENT)
+public final class InternalWidgets {
+    private InternalWidgets() {}
+    
+    public static Widget createAutoCraftingButtonWidget(Rectangle displayBounds, Rectangle rectangle, Component text, Supplier<RecipeDisplay> displaySupplier, List<Widget> setupDisplay, RecipeCategory<?> category) {
+        AbstractContainerScreen<?> containerScreen = REIHelper.getInstance().getPreviousContainerScreen();
+        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, containerScreen, displaySupplier.get());
+                    for (AutoTransferHandler autoTransferHandler : RecipeHelper.getInstance().getSortedAutoCraftingHandler())
+                        try {
+                            AutoTransferHandler.Result result = autoTransferHandler.handle(context);
+                            if (result.isBlocking()) {
+                                if (result.isReturningToScreen()) {
+                                    break;
+                                }
+                                return;
+                            }
+                        } catch (Exception e) {
+                            e.printStackTrace();
+                        }
+                    Minecraft.getInstance().setScreen(containerScreen);
+                    ScreenHelper.getLastOverlay().init();
+                })
+                .onRender((matrices, button) -> {
+                    button.setEnabled(false);
+                    List<String> error = null;
+                    int color = 0;
+                    visible[0] = false;
+                    IntList redSlots = null;
+                    AutoTransferHandler.Context context = AutoTransferHandler.Context.create(false, containerScreen, 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;
+                            } 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();
+                            }
+                            
+                            if (result.isBlocking()) break;
+                        } 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(matrices, 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.getInstance().isYog.get())
+                            str += I18n.get("text.auto_craft.move_items.yog");
+                        else
+                            str += I18n.get("text.auto_craft.move_items");
+                    } else {
+                        if (errorTooltip[0].size() > 1)
+                            str += ChatFormatting.RED.toString() + I18n.get("error.rei.multi.errors") + "\n";
+                        str += CollectionUtils.mapAndJoinToString(errorTooltip[0], s -> ChatFormatting.RED.toString() + (errorTooltip[0].size() > 1 ? "- " : "") + I18n.get(s), "\n");
+                    }
+                    if (Minecraft.getInstance().options.advancedItemTooltips) {
+                        str += displaySupplier.get().getRecipeLocation().isPresent() ? I18n.get("text.rei.recipe_id", ChatFormatting.GRAY.toString(), displaySupplier.get().getRecipeLocation().get().toString()) : "";
+                    }
+                    return str;
+                });
+        return new WidgetWithBounds() {
+            @Override
+            public @NotNull Rectangle getBounds() {
+                return autoCraftingButton.getBounds();
+            }
+            
+            @Override
+            public List<? extends GuiEventListener> children() {
+                return Collections.singletonList(autoCraftingButton);
+            }
+            
+            @Override
+            public void render(PoseStack matrices, int mouseX, int mouseY, float delta) {
+                autoCraftingButton.render(matrices, mouseX, mouseY, delta);
+            }
+            
+            @Override
+            public boolean keyPressed(int keyCode, int scanCode, int modifiers) {
+                if (displaySupplier.get().getRecipeLocation().isPresent() && ConfigObject.getInstance().getCopyRecipeIdentifierKeybind().matchesKey(keyCode, scanCode) && containsMouse(PointHelper.ofMouse())) {
+                    minecraft.keyboardHandler.setClipboard(displaySupplier.get().getRecipeLocation().get().toString());
+                    if (ConfigObject.getInstance().isToastDisplayedOnCopyIdentifier()) {
+                        CopyRecipeIdentifierToast.addToast(I18n.get("msg.rei.copied_recipe_id"), I18n.get("msg.rei.recipe_id_details", displaySupplier.get().getRecipeLocation().get().toString()));
+                    }
+                    return true;
+                }
+                return super.keyPressed(keyCode, scanCode, modifiers);
+            }
+            
+            @Override
+            public boolean mouseClicked(double mouseX, double mouseY, int button) {
+                if (displaySupplier.get().getRecipeLocation().isPresent() && ConfigObject.getInstance().getCopyRecipeIdentifierKeybind().matchesMouse(button) && containsMouse(PointHelper.ofMouse())) {
+                    minecraft.keyboardHandler.setClipboard(displaySupplier.get().getRecipeLocation().get().toString());
+                    if (ConfigObject.getInstance().isToastDisplayedOnCopyIdentifier()) {
+                        CopyRecipeIdentifierToast.addToast(I18n.get("msg.rei.copied_recipe_id"), I18n.get("msg.rei.recipe_id_details", displaySupplier.get().getRecipeLocation().get().toString()));
+                    }
+                    return true;
+                }
+                return super.mouseClicked(mouseX, mouseY, button);
+            }
+        };
+    }
+    
+    public static WidgetWithBounds wrapLateRenderable(WidgetWithBounds widget) {
+        return new LateRenderableWidgetWithBounds(widget);
+    }
+    
+    public static WidgetWithBounds wrapTranslate(WidgetWithBounds widget, float x, float y, float z) {
+        return new WidgetWithBoundsWithTranslate(widget, x, y, z);
+    }
+    
+    public static Widget 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(PoseStack matrices, int mouseX, int mouseY, float delta) {
+            for (Widget widget : widgets) {
+                widget.setZ(getZ());
+                widget.render(matrices, mouseX, mouseY, delta);
+            }
+        }
+        
+        @Override
+        public List<? extends GuiEventListener> children() {
+            return widgets;
+        }
+        
+        @Override
+        public boolean mouseScrolled(double mouseX, double mouseY, double amount) {
+            for (Widget widget : this.widgets) {
+                if (widget.mouseScrolled(mouseX, mouseY, amount))
+                    return true;
+            }
+            return false;
+        }
+    }
+    
+    private static class LateRenderableWidget extends Widget implements LateRenderable {
+        private final Widget widget;
+        
+        private LateRenderableWidget(Widget widget) {
+            this.widget = widget;
+        }
+        
+        @Override
+        public void render(PoseStack matrices, int mouseX, int mouseY, float delta) {
+            this.widget.setZ(getZ());
+            this.widget.render(matrices, mouseX, mouseY, delta);
+        }
+        
+        @Override
+        public List<? extends GuiEventListener> children() {
+            return Collections.singletonList(this.widget);
+        }
+        
+        @Override
+        public boolean mouseScrolled(double mouseX, double mouseY, double amount) {
+            return this.widget.mouseScrolled(mouseX, mouseY, amount);
+        }
+    }
+    
+    private static class LateRenderableWidgetWithBounds extends WidgetWithBounds implements LateRenderable {
+        private final WidgetWithBounds widget;
+        
+        private LateRenderableWidgetWithBounds(WidgetWithBounds widget) {
+            this.widget = widget;
+        }
+        
+        @Override
+        public @NotNull Rectangle getBounds() {
+            return this.widget.getBounds();
+        }
+        
+        @Override
+        public void render(PoseStack matrices, int mouseX, int mouseY, float delta) {
+            this.widget.setZ(getZ());
+            this.widget.render(matrices, mouseX, mouseY, delta);
+        }
+        
+        @Override
+        public boolean containsMouse(double mouseX, double mouseY) {
+            return this.widget.containsMouse(mouseX, mouseY);
+        }
+        
+        @Override
+        public List<? extends GuiEventListener> children() {
+            return Collections.singletonList(this.widget);
+        }
+        
+        @Override
+        public boolean mouseScrolled(double mouseX, double mouseY, double amount) {
+            return this.widget.mouseScrolled(mouseX, mouseY, amount);
+        }
+    }
+    
+    private static class WidgetWithBoundsWithTranslate extends WidgetWithBounds implements LateRenderable {
+        private final WidgetWithBounds widget;
+        private final float x, y, z;
+        
+        public WidgetWithBoundsWithTranslate(WidgetWithBounds widget, float x, float y, float z) {
+            this.widget = widget;
+            this.x = x;
+            this.y = y;
+            this.z = z;
+        }
+        
+        @Override
+        public @NotNull Rectangle getBounds() {
+            return this.widget.getBounds();
+        }
+        
+        @Override
+        public void render(PoseStack matrices, int mouseX, int mouseY, float delta) {
+            matrices.pushPose();
+            matrices.translate(x, y, z);
+            this.widget.setZ(getZ());
+            this.widget.render(matrices, mouseX, mouseY, delta);
+            matrices.popPose();
+        }
+        
+        @Override
+        public boolean containsMouse(double mouseX, double mouseY) {
+            return this.widget.containsMouse(mouseX, mouseY);
+        }
+        
+        @Override
+        public List<? extends GuiEventListener> children() {
+            return Collections.singletonList(this.widget);
+        }
+        
+        @Override
+        public boolean mouseScrolled(double mouseX, double mouseY, double amount) {
+            return this.widget.mouseScrolled(mouseX, mouseY, amount);
+        }
+    }
+}