Skip to content

Commit

Permalink
Fix #37
Browse files Browse the repository at this point in the history
  • Loading branch information
@senseiwells
senseiwells committed Sep 24, 2024
1 parent 1f44123 commit 9a7885b
Show file tree
Hide file tree
Showing 5 changed files with 84 additions and 30 deletions.
11 changes: 11 additions & 0 deletions src/main/java/eu/pb4/sgui/api/gui/HotbarGui.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -107,6 +107,17 @@ public boolean click(int index, ClickType type, SlotActionType action) {
return super.click(index, type, action);
}

@Override
public int getHotbarSlotIndex(int index) {
// We add the offhand before the inventory, so we need to shift by -1
return super.getHotbarSlotIndex(index) - 1;
}

@Override
public int getOffhandSlotIndex() {
return 9;
}

@Override
public boolean open() {
if (this.player.isDisconnected() || this.isOpen()) {
Expand Down
21 changes: 21 additions & 0 deletions src/main/java/eu/pb4/sgui/api/gui/SlotGuiInterface.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ public interface SlotGuiInterface extends SlotHolder, GuiInterface {
int getSize();

boolean getLockPlayerInventory();

void setLockPlayerInventory(boolean value);

/**
Expand Down Expand Up @@ -64,6 +65,26 @@ default boolean onClick(int index, ClickType type, SlotActionType action, GuiEle
return false;
}


/**
* Maps a hotbar index into a slot index.
*
* @param index The hotbar index, this should be [0-8]
* @return The mapped slot index
*/
default int getHotbarSlotIndex(int index) {
return this.getSize() + index - 9;
}

/**
* Gets the offhand slot index
*
* @return The offhand slot index
*/
default int getOffhandSlotIndex() {
return -1;
}

@Nullable
default Slot getSlotRedirectOrPlayer(int index) {
if (index < this.getSize()) {
Expand Down
57 changes: 27 additions & 30 deletions src/main/java/eu/pb4/sgui/mixin/ServerPlayNetworkHandlerMixin.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
import eu.pb4.sgui.virtual.inventory.VirtualScreenHandler;
import eu.pb4.sgui.virtual.merchant.VirtualMerchantScreenHandler;
import io.netty.buffer.Unpooled;
import net.minecraft.entity.player.PlayerInventory;
import net.minecraft.item.ItemStack;
import net.minecraft.network.ClientConnection;
import net.minecraft.network.PacketByteBuf;
Expand Down Expand Up @@ -81,8 +82,12 @@ public ServerPlayNetworkHandlerMixin(MinecraftServer server, ClientConnection co
GuiHelpers.sendSlotUpdate(this.player, -1, -1, this.player.currentScreenHandler.getCursorStack(), handler.getRevision());

if (type.numKey) {
int x = type.value + handler.slots.size() - 10;
GuiHelpers.sendSlotUpdate(player, handler.syncId, x, handler.getSlot(x).getStack(), handler.nextRevision());
int index = handler.getGui().getHotbarSlotIndex(type.value - 1);
GuiHelpers.sendSlotUpdate(this.player, handler.syncId, index, handler.getSlot(index).getStack(), handler.nextRevision());
} else if (type == ClickType.OFFHAND_SWAP) {
int index = handler.getGui().getOffhandSlotIndex();
ItemStack updated = index >= 0 ? handler.getSlot(index).getStack() : ItemStack.EMPTY;
GuiHelpers.sendSlotUpdate(this.player, -2, PlayerInventory.OFF_HAND_SLOT, updated, handler.getRevision());
} else if (type == ClickType.MOUSE_DOUBLE_CLICK || type == ClickType.MOUSE_LEFT_SHIFT || type == ClickType.MOUSE_RIGHT_SHIFT || (type.isDragging && type.value == 2)) {
GuiHelpers.sendPlayerScreenHandler(this.player);
}
Expand Down Expand Up @@ -121,8 +126,8 @@ public ServerPlayNetworkHandlerMixin(MinecraftServer server, ClientConnection co
@Inject(method = "onCloseHandledScreen", at = @At(value = "INVOKE", target = "Lnet/minecraft/network/NetworkThreadUtils;forceMainThread(Lnet/minecraft/network/packet/Packet;Lnet/minecraft/network/listener/PacketListener;Lnet/minecraft/server/world/ServerWorld;)V", shift = At.Shift.AFTER), cancellable = true)
private void sgui$storeScreenHandler(CloseHandledScreenC2SPacket packet, CallbackInfo info) {
if (this.player.currentScreenHandler instanceof VirtualScreenHandlerInterface handler) {
if (sgui$bookIgnoreClose && this.player.currentScreenHandler instanceof BookScreenHandler) {
sgui$bookIgnoreClose = false;
if (this.sgui$bookIgnoreClose && this.player.currentScreenHandler instanceof BookScreenHandler) {
this.sgui$bookIgnoreClose = false;
info.cancel();
return;
}
Expand Down Expand Up @@ -177,7 +182,7 @@ public ServerPlayNetworkHandlerMixin(MinecraftServer server, ClientConnection co
}
}

@Inject(method = "onCraftRequest", at = @At(value = "INVOKE", target = "Lnet/minecraft/server/network/ServerPlayerEntity;updateLastActionTime()V", shift = At.Shift.BEFORE), cancellable = true)
@Inject(method = "onCraftRequest", at = @At(value = "INVOKE", target = "Lnet/minecraft/server/network/ServerPlayerEntity;updateLastActionTime()V", shift = At.Shift.BEFORE))
private void sgui$catchRecipeRequests(CraftRequestC2SPacket packet, CallbackInfo ci) {
if (this.player.currentScreenHandler instanceof VirtualScreenHandler handler && handler.getGui() instanceof SimpleGui gui) {
try {
Expand All @@ -199,7 +204,7 @@ public ServerPlayNetworkHandlerMixin(MinecraftServer server, ClientConnection co
ci.cancel();
}
} catch (Throwable e) {
if (this.player.currentScreenHandler instanceof VirtualScreenHandlerInterface handler ) {
if (this.player.currentScreenHandler instanceof VirtualScreenHandlerInterface handler) {
handler.getGui().handleException(e);
} else {
e.printStackTrace();
Expand Down Expand Up @@ -245,28 +250,22 @@ public ServerPlayNetworkHandlerMixin(MinecraftServer server, ClientConnection co

@Inject(method = "onPlayerInteractItem", at = @At(value = "INVOKE", shift = At.Shift.AFTER, target = "Lnet/minecraft/network/NetworkThreadUtils;forceMainThread(Lnet/minecraft/network/packet/Packet;Lnet/minecraft/network/listener/PacketListener;Lnet/minecraft/server/world/ServerWorld;)V"), cancellable = true)
private void sgui$clickWithItem(PlayerInteractItemC2SPacket packet, CallbackInfo ci) {
if (this.player.currentScreenHandler instanceof HotbarScreenHandler screenHandler) {
var gui = screenHandler.getGui();
if (screenHandler.slotsOld != null) {
screenHandler.slotsOld.set(gui.getSelectedSlot() + 36, ItemStack.EMPTY);
screenHandler.slotsOld.set(45, ItemStack.EMPTY);
}
if (this.player.currentScreenHandler instanceof HotbarScreenHandler handler) {
var gui = handler.getGui();
gui.onClickItem();
handler.syncSelectedSlot();
ci.cancel();
}
}

@Inject(method = "onPlayerInteractBlock", at = @At(value = "INVOKE", shift = At.Shift.AFTER, target = "Lnet/minecraft/network/NetworkThreadUtils;forceMainThread(Lnet/minecraft/network/packet/Packet;Lnet/minecraft/network/listener/PacketListener;Lnet/minecraft/server/world/ServerWorld;)V"), cancellable = true)
private void sgui$clickOnBlock(PlayerInteractBlockC2SPacket packet, CallbackInfo ci) {
if (this.player.currentScreenHandler instanceof HotbarScreenHandler screenHandler) {
var gui = screenHandler.getGui();
if (this.player.currentScreenHandler instanceof HotbarScreenHandler handler) {
var gui = handler.getGui();

if (!gui.onClickBlock(packet.getBlockHitResult())) {
var pos = packet.getBlockHitResult().getBlockPos();
if (screenHandler.slotsOld != null) {
screenHandler.slotsOld.set(gui.getSelectedSlot() + 36, ItemStack.EMPTY);
screenHandler.slotsOld.set(45, ItemStack.EMPTY);
}
handler.syncSelectedSlot();

this.sendPacket(new BlockUpdateS2CPacket(pos, this.player. getServerWorld().getBlockState(pos)));
pos = pos.offset(packet.getBlockHitResult().getSide());
Expand All @@ -280,15 +279,16 @@ public ServerPlayNetworkHandlerMixin(MinecraftServer server, ClientConnection co

@Inject(method = "onPlayerAction", at = @At(value = "INVOKE", shift = At.Shift.AFTER, target = "Lnet/minecraft/network/NetworkThreadUtils;forceMainThread(Lnet/minecraft/network/packet/Packet;Lnet/minecraft/network/listener/PacketListener;Lnet/minecraft/server/world/ServerWorld;)V"), cancellable = true)
private void sgui$onPlayerAction(PlayerActionC2SPacket packet, CallbackInfo ci) {
if (this.player.currentScreenHandler instanceof HotbarScreenHandler screenHandler) {
var gui = screenHandler.getGui();
if (this.player.currentScreenHandler instanceof HotbarScreenHandler handler) {
var gui = handler.getGui();

if (!gui.onPlayerAction(packet.getAction(), packet.getDirection())) {
var pos = packet.getPos();
if (screenHandler.slotsOld != null) {
screenHandler.slotsOld.set(gui.getSelectedSlot() + 36, ItemStack.EMPTY);
screenHandler.slotsOld.set(45, ItemStack.EMPTY);
handler.syncSelectedSlot();
if (packet.getAction() == PlayerActionC2SPacket.Action.SWAP_ITEM_WITH_OFFHAND) {
handler.syncOffhandSlot();
}

this.sendPacket(new BlockUpdateS2CPacket(pos, this.player.getServerWorld().getBlockState(pos)));
pos = pos.offset(packet.getDirection());
this.sendPacket(new BlockUpdateS2CPacket(pos, this.player.getServerWorld().getBlockState(pos)));
Expand All @@ -300,8 +300,8 @@ public ServerPlayNetworkHandlerMixin(MinecraftServer server, ClientConnection co

@Inject(method = "onPlayerInteractEntity", at = @At(value = "INVOKE", shift = At.Shift.AFTER, target = "Lnet/minecraft/network/NetworkThreadUtils;forceMainThread(Lnet/minecraft/network/packet/Packet;Lnet/minecraft/network/listener/PacketListener;Lnet/minecraft/server/world/ServerWorld;)V"), cancellable = true)
private void sgui$clickOnEntity(PlayerInteractEntityC2SPacket packet, CallbackInfo ci) {
if (this.player.currentScreenHandler instanceof HotbarScreenHandler screenHandler) {
var gui = screenHandler.getGui();
if (this.player.currentScreenHandler instanceof HotbarScreenHandler handler) {
var gui = handler.getGui();
var buf = new PacketByteBuf(Unpooled.buffer());
((PlayerInteractEntityC2SPacketAccessor)packet).invokeWrite(buf);

Expand All @@ -322,10 +322,7 @@ public ServerPlayNetworkHandlerMixin(MinecraftServer server, ClientConnection co
var isSneaking = buf.readBoolean();

if (!gui.onClickEntity(entityId, type, isSneaking, interactionPos)) {
if (screenHandler.slotsOld != null) {
screenHandler.slotsOld.set(gui.getSelectedSlot() + 36, ItemStack.EMPTY);
screenHandler.slotsOld.set(45, ItemStack.EMPTY);
}
handler.syncSelectedSlot();
ci.cancel();
}
}
Expand All @@ -348,7 +345,7 @@ public ServerPlayNetworkHandlerMixin(MinecraftServer server, ClientConnection co
private void sgui$onCommand(CommandExecutionC2SPacket packet, CallbackInfo ci) {
if (this.player.currentScreenHandler instanceof BookScreenHandler handler) {
try {
sgui$bookIgnoreClose = true;
this.sgui$bookIgnoreClose = true;
if (handler.getGui().onCommand("/" + packet.command())) {
ci.cancel();
}
Expand Down
15 changes: 15 additions & 0 deletions src/main/java/eu/pb4/sgui/virtual/hotbar/HotbarScreenHandler.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
import net.minecraft.screen.ScreenHandlerType;
import net.minecraft.screen.slot.Slot;
import net.minecraft.util.collection.DefaultedList;
import org.jetbrains.annotations.ApiStatus;
import org.jetbrains.annotations.Nullable;

public class HotbarScreenHandler extends VirtualScreenHandler {
Expand Down Expand Up @@ -76,4 +77,18 @@ public void sendContentUpdates() {
this.getGui().handleException(e);
}
}

@ApiStatus.Internal
public void syncSelectedSlot() {
var gui = this.getGui();
int index = gui.getHotbarSlotIndex(gui.getSelectedSlot());
GuiHelpers.sendSlotUpdate(gui.getPlayer(), this.syncId, index, this.getSlot(index).getStack(), this.nextRevision());
}

@ApiStatus.Internal
public void syncOffhandSlot() {
var gui = this.getGui();
int index = gui.getOffhandSlotIndex();
GuiHelpers.sendSlotUpdate(gui.getPlayer(), this.syncId, index, this.getSlot(index).getStack(), this.nextRevision());
}
}
10 changes: 10 additions & 0 deletions src/main/java/eu/pb4/sgui/virtual/inventory/VirtualScreenHandler.java
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
package eu.pb4.sgui.virtual.inventory;

import eu.pb4.sgui.api.GuiHelpers;
import eu.pb4.sgui.api.gui.SlotGuiInterface;
import eu.pb4.sgui.virtual.VirtualScreenHandlerInterface;
import net.minecraft.entity.player.PlayerEntity;
Expand Down Expand Up @@ -61,6 +62,15 @@ public void addListener(ScreenHandlerListener listener) {
this.gui.afterOpen();
}

@Override
public void syncState() {
super.syncState();
// We have to manually sync offhand state
int index = this.getGui().getOffhandSlotIndex();
ItemStack updated = index >= 0 ? this.getSlot(index).getStack() : ItemStack.EMPTY;
GuiHelpers.sendSlotUpdate(this.gui.getPlayer(), -2, PlayerInventory.OFF_HAND_SLOT, updated, this.getRevision());
}

@Override
public SlotGuiInterface getGui() {
return this.gui;
Expand Down

0 comments on commit 9a7885b

Please sign in to comment.