Added an interface for solvers for minigames inside chests

3 files changed, 117 insertions, 0 deletions

diff --git a/src/main/java/me/xmrvizzy/skyblocker/container/ColorHighlight.java b/src/main/java/me/xmrvizzy/skyblocker/container/ColorHighlight.java
new file mode 100644
index 00000000..c4380eab
--- /dev/null
+++ b/src/main/java/me/xmrvizzy/skyblocker/container/ColorHighlight.java
@@ -0,0 +1,4 @@
+package me.xmrvizzy.skyblocker.container;
+
+public record ColorHighlight(int slot, int color) {
+}
\ No newline at end of file
diff --git a/src/main/java/me/xmrvizzy/skyblocker/container/ContainerSolver.java b/src/main/java/me/xmrvizzy/skyblocker/container/ContainerSolver.java
new file mode 100644
index 00000000..5c3e1e5e
--- /dev/null
+++ b/src/main/java/me/xmrvizzy/skyblocker/container/ContainerSolver.java
@@ -0,0 +1,35 @@
+package me.xmrvizzy.skyblocker.container;
+
+import net.minecraft.item.ItemStack;
+
+import java.util.List;
+import java.util.Map;
+import java.util.regex.Pattern;
+
+public abstract class ContainerSolver {
+    private final Pattern CONTAINER_NAME;
+    protected final static int GREEN_HIGHLIGHT = 128 << 24 | 64 << 16 | 196 << 8 | 64;
+
+    public ContainerSolver(String containerName) {
+        CONTAINER_NAME = Pattern.compile(containerName);
+    }
+
+    public abstract boolean isEnabled();
+
+    public Pattern getName() {
+        return CONTAINER_NAME;
+    }
+
+    public abstract List<ColorHighlight> getColors(String[] groups, Map<Integer, ItemStack> slots);
+
+    public void trimEdges(Map<Integer, ItemStack> slots, int rows) {
+        for (int i = 0; i < rows; i++) {
+            slots.remove(9 * i);
+            slots.remove(9 * i + 8);
+        }
+        for (int i = 1; i < 8; i++) {
+            slots.remove(i);
+            slots.remove((rows - 1) * 9 + i);
+        }
+    }
+}
diff --git a/src/main/java/me/xmrvizzy/skyblocker/container/ContainerSolverManager.java b/src/main/java/me/xmrvizzy/skyblocker/container/ContainerSolverManager.java
new file mode 100644
index 00000000..836d38a6
--- /dev/null
+++ b/src/main/java/me/xmrvizzy/skyblocker/container/ContainerSolverManager.java
@@ -0,0 +1,78 @@
+package me.xmrvizzy.skyblocker.container;
+
+import com.mojang.blaze3d.systems.RenderSystem;
+import net.minecraft.client.gui.DrawableHelper;
+import net.minecraft.client.gui.screen.ingame.GenericContainerScreen;
+import net.minecraft.client.util.math.MatrixStack;
+import net.minecraft.item.ItemStack;
+import net.minecraft.screen.slot.Slot;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.List;
+import java.util.Map;
+import java.util.TreeMap;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+public class ContainerSolverManager extends DrawableHelper {
+    private static final Pattern PLACEHOLDER_PATTERN = Pattern.compile("");
+    private final ContainerSolver[] solvers;
+    private ContainerSolver currentSolver = null;
+    private String[] groups;
+    private List<ColorHighlight> highlights;
+
+    public ContainerSolverManager() {
+        solvers = new ContainerSolver[]{
+        };
+    }
+
+    public void onSetScreen(@NotNull GenericContainerScreen screen) {
+        String screenName = screen.getTitle().getString();
+        Matcher matcher = PLACEHOLDER_PATTERN.matcher(screenName);
+        for (ContainerSolver solver : solvers) {
+            if (solver.isEnabled()) {
+                matcher.usePattern(solver.getName());
+                matcher.reset();
+                if (matcher.matches()) {
+                    currentSolver = solver;
+                    groups = new String[matcher.groupCount()];
+                    for (int i = 0; i < groups.length; i++)
+                        groups[i] = matcher.group(i + 1);
+                    return;
+                }
+            }
+        }
+        currentSolver = null;
+    }
+
+    public void clearScreen() {
+        currentSolver = null;
+    }
+
+    public void markDirty() {
+        highlights = null;
+    }
+
+    public void onDraw(MatrixStack matrices, List<Slot> slots) {
+        if (currentSolver == null)
+            return;
+        if (highlights == null)
+            highlights = currentSolver.getColors(groups, slotMap(slots));
+        RenderSystem.disableDepthTest();
+        RenderSystem.colorMask(true, true, true, false);
+        for (ColorHighlight highlight : highlights) {
+            Slot slot = slots.get(highlight.slot());
+            int color = highlight.color();
+            fillGradient(matrices, slot.x, slot.y, slot.x + 16, slot.y + 16, color, color);
+        }
+        RenderSystem.colorMask(true, true, true, true);
+        RenderSystem.enableDepthTest();
+    }
+
+    private Map<Integer, ItemStack> slotMap(List<Slot> slots) {
+        Map<Integer, ItemStack> slotMap = new TreeMap<>();
+        for (int i = 0; i < slots.size(); i++)
+            slotMap.put(i, slots.get(i).getStack());
+        return slotMap;
+    }
+}