[全流程完成]

2026-08-17 15:15:33 +08:00
parent 13b981a916
commit da04b67ef0
5 changed files with 457 additions and 0 deletions
@@ -0,0 +1,118 @@
+package thaumicenergistics.common.container;
+
+import net.minecraft.network.chat.Component;
+import net.minecraft.world.entity.player.Inventory;
+import net.minecraft.world.item.ItemStack;
+
+import org.jetbrains.annotations.Nullable;
+
+import thaumcraft.api.aspects.AspectList;
+import thaumcraft.api.aspects.IEssentiaContainerItem;
+
+import appeng.client.gui.me.common.MEStorageScreen;
+import appeng.client.gui.style.ScreenStyle;
+
+import net.neoforged.neoforge.network.PacketDistributor;
+import thaumicenergistics.common.network.EssentiaTerminalDepositC2SPacket;
+import thaumicenergistics.common.network.EssentiaTerminalFillC2SPacket;
+
+/**
+ * 源质终端 GUI：在 AE2 通用终端基础上扩展右键交互。
+ * <ul>
+ *   <li>Shift+右键玩家背包格：格内是源质容器 → 源质存入网络（空容器留原格）。</li>
+ *   <li>右键玩家物品栏/背包槽 → 放行（AE2 原版物品交互）。</li>
+ *   <li>右键网络区域（源质条目或空白），手持装有源质的罐子/安瓿 → 源质存入 ME 网络（空容器回背包）。</li>
+ *   <li>手持空罐/空安瓿 + 右键源质条目 → 从网络取出源质装罐/装安瓿。</li>
+ *   <li>左键仍是正常物品操作（不受影响）。</li>
+ * </ul>
+ */
+public class GuiEssentiaTerminal extends MEStorageScreen<EssentiaTerminalMenu> {
+
+    public GuiEssentiaTerminal(EssentiaTerminalMenu menu, Inventory playerInventory,
+                               Component title, ScreenStyle style) {
+        super(menu, playerInventory, title, style);
+    }
+
+    @Override
+    public boolean mouseClicked(double xCoord, double yCoord, int btn) {
+        if (btn == 1) {
+            var carried = this.menu.getCarried();
+            boolean shiftDown = net.minecraft.client.gui.screens.inventory.AbstractContainerScreen.hasShiftDown();
+            // 源质物品来源：优先菜单 carried；为空时回退玩家主手物品
+            ItemStack sourceStack = !carried.isEmpty() ? carried
+                    : this.menu.getPlayerInventory().player.getMainHandItem();
+            boolean isContainer = isSourceContainer(sourceStack);
+            boolean containerHasEssentia = isContainer && getContainerEssentia(sourceStack) != null;
+
+            // Shift+右键玩家物品栏/背包格：格内是源质容器 → 存入源质（空容器留原格）
+            if (shiftDown && this.hoveredSlot != null && this.menu.isPlayerSideSlot(this.hoveredSlot)) {
+                var slotStack = this.hoveredSlot.getItem();
+                if (slotStack.getItem() instanceof IEssentiaContainerItem) {
+                    PacketDistributor.sendToServer(EssentiaTerminalDepositC2SPacket.fromSlot(
+                            this.menu.containerId, this.hoveredSlot.getSlotIndex(), slotStack));
+                    return true;
+                }
+                // 非源质容器 → 放行（AE2 原版 Shift+右键快速移动）
+                return super.mouseClicked(xCoord, yCoord, btn);
+            }
+
+            // 普通右键点击玩家物品栏/背包槽 → 放行（AE2 原版物品交互）
+            if (!shiftDown && this.hoveredSlot != null && this.menu.isPlayerSideSlot(this.hoveredSlot)) {
+                return super.mouseClicked(xCoord, yCoord, btn);
+            }
+
+            // Shift+右键：手持源质容器（满或空）→ 拦截。满→存入源质；空→阻止被当普通物品放入网络
+            if (shiftDown && isContainer) {
+                if (containerHasEssentia) {
+                    this.menu.setCarried(net.minecraft.world.item.ItemStack.EMPTY);
+                    PacketDistributor.sendToServer(EssentiaTerminalDepositC2SPacket.fromClientStack(
+                            this.menu.containerId, sourceStack));
+                }
+                return true;
+            }
+
+            // 普通右键网络区域：手持满容器 → 存入源质
+            if (containerHasEssentia) {
+                this.menu.setCarried(net.minecraft.world.item.ItemStack.EMPTY);
+                PacketDistributor.sendToServer(EssentiaTerminalDepositC2SPacket.fromClientStack(
+                        this.menu.containerId, sourceStack));
+                return true;
+            }
+
+            // 手持空容器：右键源质条目 → 取出装罐；右键网络其他 → 拦截（防被放入网络）
+            if (isContainer) {
+                if (this.hoveredSlot instanceof appeng.client.gui.me.common.RepoSlot repoSlot) {
+                    var entry = repoSlot.getEntry();
+                    if (entry != null && entry.getWhat() instanceof thaumicenergistics.common.integration.appeng.AEssentiaKey key) {
+                        PacketDistributor.sendToServer(EssentiaTerminalFillC2SPacket.create(
+                                this.menu.containerId, key.getId(), sourceStack));
+                        return true;
+                    }
+                }
+                return true;
+            }
+        }
+        return super.mouseClicked(xCoord, yCoord, btn);
+    }
+
+    /** 源质容器：实现 IEssentiaContainerItem 的物品，或空玻璃安瓿（glass_phial 不实现该接口但可装罐）。 */
+    private static boolean isSourceContainer(ItemStack stack) {
+        if (stack.getItem() instanceof IEssentiaContainerItem) {
+            return true;
+        }
+        return stack.is(net.minecraft.core.registries.BuiltInRegistries.ITEM.get(
+                net.minecraft.resources.ResourceLocation.fromNamespaceAndPath("thaumcraft", "glass_phial")));
+    }
+
+    /** 若手持物品是源质容器且非空，返回其 AspectList；否则返回 null。 */
+    @Nullable
+    private static AspectList getContainerEssentia(ItemStack stack) {
+        if (stack.getItem() instanceof IEssentiaContainerItem container) {
+            AspectList aspects = container.getAspects(stack);
+            if (aspects != null && !aspects.isEmpty()) {
+                return aspects;
+            }
+        }
+        return null;
+    }
+}