Add Icon API to WItemSlot (#119)

* Add Icon API to WItemSlot

* Fix checkstyle

* Clean up test description

1 files changed, 28 insertions, 0 deletions

diff --git a/src/main/java/io/github/cottonmc/cotton/gui/widget/WItemSlot.java b/src/main/java/io/github/cottonmc/cotton/gui/widget/WItemSlot.java
index 1667bee..e1f0bce 100644
--- a/src/main/java/io/github/cottonmc/cotton/gui/widget/WItemSlot.java
+++ b/src/main/java/io/github/cottonmc/cotton/gui/widget/WItemSlot.java
@@ -12,6 +12,7 @@ import net.minecraft.screen.slot.SlotActionType;
 import io.github.cottonmc.cotton.gui.GuiDescription;
 import io.github.cottonmc.cotton.gui.ValidatedSlot;
 import io.github.cottonmc.cotton.gui.client.BackgroundPainter;
+import io.github.cottonmc.cotton.gui.widget.icon.Icon;
 import org.jetbrains.annotations.Nullable;
 
 import java.util.ArrayList;
@@ -62,6 +63,8 @@ public class WItemSlot extends WWidget {
 	@Nullable
 	@Environment(EnvType.CLIENT)
 	private BackgroundPainter backgroundPainter = null;
+	@Nullable
+	private Icon icon = null;
 	private Inventory inventory;
 	private int startIndex = 0;
 	private int slotsWide = 1;
@@ -149,6 +152,27 @@ public class WItemSlot extends WWidget {
 	}
 
 	/**
+	 * Sets the icon to this slot. Can be used for labeling slots for certain activities.
+	 *
+	 * @param icon the icon
+	 * @since 4.0.0
+	 */
+	public WItemSlot setIcon(@Nullable Icon icon) {
+		this.icon = icon;
+		return this;
+	}
+
+	/**
+	 *
+	 * @return icon if set, otherwise null
+	 * @since 4.0.0
+	 */
+	@Nullable
+	public Icon getIcon() {
+		return this.icon;
+	}
+
+	/**
 	 * Returns true if the contents of this {@code WItemSlot} can be modified by players.
 	 *
 	 * @return true if items can be inserted into or taken from this slot widget, false otherwise
@@ -330,6 +354,10 @@ public class WItemSlot extends WWidget {
 		if (backgroundPainter != null) {
 			backgroundPainter.paintBackground(matrices, x, y, this);
 		}
+
+		if (icon != null) {
+			icon.paint(matrices, x + 1, y + 1, 16);
+		}
 	}
 
 	@Nullable