aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/me/xmrvizzy/skyblocker/gui/ContainerSolverManager.java
blob: 354e7e3aa0ff35359d4cf9d36888f8920e105922 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package me.xmrvizzy.skyblocker.gui;

import com.mojang.blaze3d.systems.RenderSystem;
import me.xmrvizzy.skyblocker.mixin.HandledScreenAccessor;
import me.xmrvizzy.skyblocker.skyblock.dungeon.CroesusHelper;
import me.xmrvizzy.skyblocker.skyblock.dungeon.terminal.ColorTerminal;
import me.xmrvizzy.skyblocker.skyblock.dungeon.terminal.OrderTerminal;
import me.xmrvizzy.skyblocker.skyblock.dungeon.terminal.StartsWithTerminal;
import me.xmrvizzy.skyblocker.utils.Utils;
import net.fabricmc.fabric.api.client.screen.v1.ScreenEvents;
import net.minecraft.client.gui.DrawContext;
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;

/**
 * Manager class for {@link ContainerSolver}s like terminal solvers and experiment solvers. To add a new gui solver, extend {@link ContainerSolver} and register it in {@link #ContainerSolverManager()}.
 */
public class ContainerSolverManager {
    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[]{
                new ColorTerminal(),
                new OrderTerminal(),
                new StartsWithTerminal(),
                new CroesusHelper()
        };
    }

    public void init() {
        ScreenEvents.BEFORE_INIT.register((client, screen, scaledWidth, scaledHeight) -> {
            if (Utils.isOnSkyblock() && screen instanceof GenericContainerScreen genericContainerScreen) {
                ScreenEvents.afterRender(screen).register((screen1, context, mouseX, mouseY, delta) -> {
                	MatrixStack matrices = context.getMatrices();
                    matrices.push();
                    matrices.translate(((HandledScreenAccessor) genericContainerScreen).getX(), ((HandledScreenAccessor) genericContainerScreen).getY(), 300);
                    onDraw(context, genericContainerScreen.getScreenHandler().slots.subList(0, genericContainerScreen.getScreenHandler().getRows() * 9));
                    matrices.pop();
                });
                onSetScreen(genericContainerScreen);
            } else {
                clearScreen();
            }
        });
    }

    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(DrawContext context, List<Slot> slots) {
        if (currentSolver == null)
            return;
        if (highlights == null)
            highlights = currentSolver.getColors(groups, slotMap(slots));
        RenderSystem.enableDepthTest();
        RenderSystem.colorMask(true, true, true, false);
        for (ColorHighlight highlight : highlights) {
            Slot slot = slots.get(highlight.slot());
            int color = highlight.color();
            context.fillGradient(slot.x, slot.y, slot.x + 16, slot.y + 16, color, color);
        }
        RenderSystem.colorMask(true, true, true, true);
    }

    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;
    }
}