package com.kodari.heartmod.screen; import com.kodari.heartmod.item.ModItems; import com.kodari.heartmod.network.ModPackets; import net.fabricmc.api.EnvType; import net.fabricmc.api.Environment; import net.fabricmc.fabric.api.client.networking.v1.ClientPlayNetworking; import net.fabricmc.fabric.api.networking.v1.PacketByteBufs; import net.minecraft.client.MinecraftClient; import net.minecraft.client.gui.DrawContext; import net.minecraft.client.gui.screen.Screen; import net.minecraft.client.gui.widget.ButtonWidget; import net.minecraft.entity.player.PlayerInventory; import net.minecraft.item.ItemStack; import net.minecraft.network.PacketByteBuf; import net.minecraft.text.Text; @Environment(EnvType.CLIENT) public class HeartSelectionScreen extends Screen { // Panel dimensions private static final int PANEL_W = 270; private static final int PANEL_H = 195; private static final int SLOT_SIZE = 22; private static final int SLOT_GAP = 2; // Colors private static final int COLOR_PANEL_BG = 0xD0101010; private static final int COLOR_PANEL_BORDER = 0xFFAAAAAA; private static final int COLOR_SLOT_NORMAL = 0xFF2A2A2A; private static final int COLOR_SLOT_HEART = 0xFF4A1A1A; private static final int COLOR_SLOT_SELECT = 0xFF1A3A6A; private static final int COLOR_SLOT_BORDER = 0xFF555555; private static final int COLOR_HEART_BORDER = 0xFFCC4444; private static final int COLOR_SEL_BORDER = 0xFF4488FF; private static final int COLOR_TITLE = 0xFFFFD0D0; private static final int COLOR_SEPARATOR = 0xFF666666; // Context menu private static final int CTX_W = 140; private static final int CTX_ENTRY_H = 22; private static final String[] CTX_LABELS = { "\u2764 Refill Hearts", "\u2750 Copy Heart", " Cancel" }; private static final int[] CTX_COLORS = { 0xFFFF8888, 0xFF88CCFF, 0xFFAAAAAA }; private static final int[] CTX_ACTIONS = { ModPackets.ACTION_REFILL, ModPackets.ACTION_COPY, -1 }; private int panelX, panelY; private int selectedSlot = -1; private int quantity = 1; private boolean showCtx = false; private int ctxX, ctxY, ctxSlot; private ButtonWidget btnMinus; private ButtonWidget btnPlus; private ButtonWidget btnRefill; private ButtonWidget btnCopy; public HeartSelectionScreen() { super(Text.literal("Heart Selector")); } @Override protected void init() { panelX = (width - PANEL_W) / 2; panelY = (height - PANEL_H) / 2; int qRowY = panelY + 125; int actRowY = panelY + 160; btnMinus = ButtonWidget.builder(Text.literal(" \u2212 "), b -> { if (quantity > 1) quantity--; }).dimensions(panelX + 85, qRowY, 24, 20).build(); btnPlus = ButtonWidget.builder(Text.literal(" + "), b -> { if (selectedSlot >= 0) { int max = getHotbarStack(selectedSlot).getCount(); if (quantity < max) quantity++; } }).dimensions(panelX + 155, qRowY, 24, 20).build(); btnRefill = ButtonWidget.builder(Text.literal("\u2764 Refill"), b -> { if (selectedSlot >= 0 && isHeartSlot(selectedSlot)) { sendAction(ModPackets.ACTION_REFILL, selectedSlot, quantity); close(); } }).dimensions(panelX + 20, actRowY, 100, 20).build(); btnCopy = ButtonWidget.builder(Text.literal("\u2750 Copy"), b -> { if (selectedSlot >= 0 && isHeartSlot(selectedSlot)) { sendAction(ModPackets.ACTION_COPY, selectedSlot, quantity); close(); } }).dimensions(panelX + 150, actRowY, 100, 20).build(); addDrawableChild(btnMinus); addDrawableChild(btnPlus); addDrawableChild(btnRefill); addDrawableChild(btnCopy); refreshButtonStates(); } private void refreshButtonStates() { boolean hasSelection = selectedSlot >= 0 && isHeartSlot(selectedSlot); btnMinus.active = hasSelection && quantity > 1; btnPlus.active = hasSelection; btnRefill.active = hasSelection; btnCopy.active = hasSelection; } @Override public void render(DrawContext ctx, int mx, int my, float delta) { // Dark background dim ctx.fillGradient(0, 0, width, height, 0xB0080808, 0xC0080808); // Panel ctx.fill(panelX, panelY, panelX + PANEL_W, panelY + PANEL_H, COLOR_PANEL_BG); ctx.drawBorder(panelX, panelY, PANEL_W, PANEL_H, COLOR_PANEL_BORDER); // Title bar ctx.fill(panelX + 1, panelY + 1, panelX + PANEL_W - 1, panelY + 18, 0xFF1A0A0A); ctx.drawCenteredTextWithShadow(textRenderer, Text.literal("\u2764 Heart Selector \u2764"), panelX + PANEL_W / 2, panelY + 5, COLOR_TITLE); // Separator ctx.fill(panelX + 5, panelY + 19, panelX + PANEL_W - 5, panelY + 20, COLOR_SEPARATOR); // "Hotbar" label ctx.drawTextWithShadow(textRenderer, Text.literal("Hotbar:"), panelX + 10, panelY + 27, 0xFF999999); // Hotbar slots renderHotbarSlots(ctx, mx, my); // Selection area renderSelectionInfo(ctx); // Buttons (drawn by super) super.render(ctx, mx, my, delta); // Context menu drawn last (on top) if (showCtx) { renderContextMenu(ctx, mx, my); } } private void renderHotbarSlots(DrawContext ctx, int mx, int my) { PlayerInventory inv = MinecraftClient.getInstance().player.getInventory(); int startX = panelX + 10; int startY = panelY + 40; int stride = SLOT_SIZE + SLOT_GAP; for (int i = 0; i < 9; i++) { int sx = startX + i * stride; int sy = startY; ItemStack stack = inv.getStack(i); boolean isHeart = !stack.isEmpty() && stack.getItem() == ModItems.HEART_ITEM; boolean isHover = mx >= sx && mx < sx + SLOT_SIZE && my >= sy && my < sy + SLOT_SIZE; boolean isSelect = selectedSlot == i; int bg = isSelect ? COLOR_SLOT_SELECT : (isHeart ? COLOR_SLOT_HEART : COLOR_SLOT_NORMAL); int border = isSelect ? COLOR_SEL_BORDER : (isHeart ? COLOR_HEART_BORDER : COLOR_SLOT_BORDER); if (isHover && isHeart && !isSelect) { bg = 0xFF2A2A3A; } ctx.fill(sx, sy, sx + SLOT_SIZE, sy + SLOT_SIZE, bg); ctx.drawBorder(sx, sy, SLOT_SIZE, SLOT_SIZE, border); if (!stack.isEmpty()) { ctx.drawItem(stack, sx + 3, sy + 3); ctx.drawItemInSlot(textRenderer, stack, sx + 3, sy + 3); } // Slot number indicator ctx.drawTextWithShadow(textRenderer, Text.literal(String.valueOf(i + 1)), sx + 2, sy + SLOT_SIZE - 8, isHeart ? 0xFFFF7777 : 0xFF666666); } } private void renderSelectionInfo(DrawContext ctx) { int infoY = panelY + 75; ctx.fill(panelX + 5, panelY + 70, panelX + PANEL_W - 5, panelY + 155, 0x33FFFFFF); ctx.drawBorder(panelX + 5, panelY + 70, PANEL_W - 10, 85, 0xFF444444); if (selectedSlot < 0 || !isHeartSlot(selectedSlot)) { ctx.drawCenteredTextWithShadow(textRenderer, Text.literal("Click a \u2764 slot to select"), panelX + PANEL_W / 2, infoY + 10, 0xFF888888); return; } ItemStack stack = getHotbarStack(selectedSlot); String name = stack.getName().getString(); // Item icon + name ctx.drawItem(stack, panelX + 12, infoY); ctx.drawTextWithShadow(textRenderer, Text.literal(name), panelX + 34, infoY + 4, 0xFFFFFFFF); // HP info int hp = quantity * 2; ctx.drawTextWithShadow(textRenderer, Text.literal("Restores: \u2764 " + hp + " HP (" + quantity + " heart" + (quantity > 1 ? "s" : "") + ")"), panelX + 12, infoY + 22, 0xFFFF9999); // Quantity row label ctx.drawTextWithShadow(textRenderer, Text.literal("Amount:"), panelX + 12, panelY + 131, 0xFFCCCCCC); ctx.drawCenteredTextWithShadow(textRenderer, Text.literal(String.valueOf(quantity)), panelX + 135, panelY + 131, 0xFFFFFF44); ctx.drawTextWithShadow(textRenderer, Text.literal("/ " + stack.getCount()), panelX + 185, panelY + 131, 0xFF888888); } private void renderContextMenu(DrawContext ctx, int mx, int my) { int entries = CTX_LABELS.length; int menuH = entries * CTX_ENTRY_H + 6; // Clamp to screen int rx = Math.min(ctxX, width - CTX_W - 4); int ry = Math.min(ctxY, height - menuH - 4); ctx.fill(rx, ry, rx + CTX_W, ry + menuH, 0xFF1C1C1C); ctx.drawBorder(rx, ry, CTX_W, menuH, 0xFF888888); for (int i = 0; i < entries; i++) { int ey = ry + 3 + i * CTX_ENTRY_H; boolean hover = mx >= rx + 2 && mx < rx + CTX_W - 2 && my >= ey && my < ey + CTX_ENTRY_H; if (hover) ctx.fill(rx + 2, ey, rx + CTX_W - 2, ey + CTX_ENTRY_H, 0xFF2A2A4A); // Divider before Cancel if (i == entries - 1) { ctx.fill(rx + 4, ey - 1, rx + CTX_W - 4, ey, 0xFF555555); } ctx.drawTextWithShadow(textRenderer, Text.literal(CTX_LABELS[i]), rx + 8, ey + 6, CTX_COLORS[i]); } } @Override public boolean mouseClicked(double mx, double my, int button) { if (showCtx) { return handleContextClick((int) mx, (int) my); } int startX = panelX + 10; int startY = panelY + 40; int stride = SLOT_SIZE + SLOT_GAP; for (int i = 0; i < 9; i++) { int sx = startX + i * stride; if (mx >= sx && mx < sx + SLOT_SIZE && my >= startY && my < startY + SLOT_SIZE) { if (isHeartSlot(i)) { selectedSlot = i; quantity = 1; refreshButtonStates(); if (button == 1) { ctxSlot = i; ctxX = (int) mx; ctxY = (int) my; showCtx = true; } } return true; } } return super.mouseClicked(mx, my, button); } private boolean handleContextClick(int mx, int my) { int entries = CTX_LABELS.length; int menuH = entries * CTX_ENTRY_H + 6; int rx = Math.min(ctxX, width - CTX_W - 4); int ry = Math.min(ctxY, height - menuH - 4); for (int i = 0; i < entries; i++) { int ey = ry + 3 + i * CTX_ENTRY_H; if (mx >= rx + 2 && mx < rx + CTX_W - 2 && my >= ey && my < ey + CTX_ENTRY_H) { int action = CTX_ACTIONS[i]; showCtx = false; if (action >= 0 && isHeartSlot(ctxSlot)) { sendAction(action, ctxSlot, quantity); close(); } return true; } } showCtx = false; return true; } private void sendAction(int action, int slot, int qty) { PacketByteBuf buf = PacketByteBufs.create(); buf.writeInt(action); buf.writeInt(slot); buf.writeInt(qty); ClientPlayNetworking.send(ModPackets.HEART_ACTION_ID, buf); } private boolean isHeartSlot(int slot) { ItemStack s = getHotbarStack(slot); return !s.isEmpty() && s.getItem() == ModItems.HEART_ITEM; } private ItemStack getHotbarStack(int slot) { return MinecraftClient.getInstance().player.getInventory().getStack(slot); } @Override public boolean shouldPause() { return false; } @Override public boolean shouldCloseOnEsc() { return true; } }