Add slot text adders, which render arbitrary text on arbitrary slots

2 files changed, 96 insertions, 0 deletions

diff --git a/src/main/java/de/hysky/skyblocker/skyblock/item/slottext/SlotTextAdder.java b/src/main/java/de/hysky/skyblocker/skyblock/item/slottext/SlotTextAdder.java
new file mode 100644
index 00000000..5ef388bd
--- /dev/null
+++ b/src/main/java/de/hysky/skyblocker/skyblock/item/slottext/SlotTextAdder.java
@@ -0,0 +1,42 @@
+package de.hysky.skyblocker.skyblock.item.slottext;
+
+import net.minecraft.screen.slot.Slot;
+import net.minecraft.text.Text;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.regex.Pattern;
+
+/**
+ * Extend this class and add it to {@link SlotTextManager#adders} to add text to any arbitrary slot.
+ */
+public abstract class SlotTextAdder {
+	/**
+	 * The title of the screen must match this pattern for this adder to be applied. Null means it will be applied to all screens.
+	 */
+	public final Pattern titlePattern;
+
+	protected SlotTextAdder(String titlePattern) {
+		this(Pattern.compile(titlePattern));
+	}
+
+	protected SlotTextAdder(Pattern titlePattern) {
+		this.titlePattern = titlePattern;
+	}
+
+	/**
+	 * Creates a SlotTextRenderer that will be applied to all screens.
+	 */
+	protected SlotTextAdder() {
+		this.titlePattern = null;
+	}
+
+	/**
+	 * This method will be called for each rendered slot. Consider using a switch statement on {@link Slot#getIndex()} if you wish to add different text to different slots.
+	 *
+	 * @return The text to be rendered. Return null if no text should be rendered.
+	 * @implNote By minecraft's design, scaled text inexplicably moves around.
+	 * So, limit your text to 3 characters (or roughly less than 20 width) if you want it to not look horrible.
+	 */
+	@Nullable
+	public abstract Text getText(Slot slot);
+}
diff --git a/src/main/java/de/hysky/skyblocker/skyblock/item/slottext/SlotTextManager.java b/src/main/java/de/hysky/skyblocker/skyblock/item/slottext/SlotTextManager.java
new file mode 100644
index 00000000..b6dfa6cd
--- /dev/null
+++ b/src/main/java/de/hysky/skyblocker/skyblock/item/slottext/SlotTextManager.java
@@ -0,0 +1,54 @@
+package de.hysky.skyblocker.skyblock.item.slottext;
+
+import net.fabricmc.fabric.api.client.screen.v1.ScreenEvents;
+import net.minecraft.client.gui.screen.Screen;
+import net.minecraft.client.gui.screen.ingame.HandledScreen;
+import net.minecraft.screen.slot.Slot;
+import net.minecraft.text.Text;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.ArrayList;
+
+public class SlotTextManager {
+	private static final SlotTextAdder[] adders = new SlotTextAdder[]{
+	};
+	private static final ArrayList<SlotTextAdder> currentScreenAdders = new ArrayList<>();
+
+	private SlotTextManager() {
+	}
+
+	public static void init() {
+		ScreenEvents.AFTER_INIT.register((client, screen, width, height) -> {
+			if (screen instanceof HandledScreen<?>) {
+				onScreenChange(screen);
+				ScreenEvents.remove(screen).register(ignored -> currentScreenAdders.clear());
+			}
+		});
+	}
+
+	private static void onScreenChange(Screen screen) {
+		final String title = screen.getTitle().getString();
+		for (SlotTextAdder adder : adders) {
+			if (adder.titlePattern == null || adder.titlePattern.matcher(title).matches()) {
+				currentScreenAdders.add(adder);
+			}
+		}
+	}
+
+	/**
+	 * The returned text is rendered on top of the slot. The text will be scaled if it doesn't fit in the slot,
+	 * but 3 characters should be seen as the maximum to keep it readable and in place as it tends to move around when scaled.
+	 *
+	 * @implNote Only the first adder that returns a non-null text will be used.
+	 * The order of the adders remains the same as they were added to the {@link SlotTextManager#adders} array.
+	 */
+	@Nullable
+	public static Text getText(Slot slot) {
+		if (currentScreenAdders.isEmpty()) return null;
+		for (SlotTextAdder adder : currentScreenAdders) {
+			Text text = adder.getText(slot);
+			if (text != null) return text;
+		}
+		return null;
+	}
+}