ExtendedAE_Plus/src/main/java/com/extendedae_plus/client/InputEvents.java

package com.extendedae_plus.client;

import appeng.api.stacks.GenericStack;
import appeng.client.gui.me.common.MEStorageScreen;
import appeng.integration.modules.jei.GenericEntryStackHelper;
import com.extendedae_plus.init.ModNetwork;
import com.extendedae_plus.integration.jei.JeiRuntimeProxy;
import com.extendedae_plus.mixin.ae2.accessor.MEStorageScreenAccessor;
import com.extendedae_plus.mixin.extendedae.accessor.GuiExPatternTerminalAccessor;
import com.extendedae_plus.network.PullFromJeiOrCraftC2SPacket;
import com.extendedae_plus.network.crafting.OpenCraftFromJeiC2SPacket;
import com.glodblock.github.extendedae.client.gui.GuiExPatternTerminal;
import mezz.jei.api.ingredients.ITypedIngredient;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.screens.Screen;
import net.minecraftforge.client.event.ScreenEvent;
import net.minecraftforge.eventbus.api.SubscribeEvent;
import net.minecraftforge.fml.ModList;
import org.lwjgl.glfw.GLFW;

import java.util.Optional;

public final class InputEvents {
    private InputEvents() {}

    @SubscribeEvent
    public static void onMouseButtonPre(ScreenEvent.MouseButtonPressed.Pre event) {
        // 若未安装 JEI，直接跳过，避免触发 JEI 类加载导致的 NoClassDefFoundError
        if (!ModList.get().isLoaded("jei")) {
            return;
        }
        // 注意：不要在 try/catch 之外直接访问 JEI 运行时，避免类加载崩溃
        // 优先处理：Shift + 左键（拉取或下单）
        if (event.getButton() == GLFW.GLFW_MOUSE_BUTTON_LEFT && Screen.hasShiftDown()) {
            try {
                double mouseX = event.getMouseX();
                double mouseY = event.getMouseY();
                Optional<ITypedIngredient<?>> hovered = JeiRuntimeProxy.getIngredientUnderMouse(mouseX, mouseY);
                if (hovered.isEmpty()) {
                    hovered = JeiRuntimeProxy.getIngredientUnderMouse();
                }
                if (hovered.isPresent()) {
                    // 若 JEI 作弊模式开启，则放行给 JEI 处理（Shift+左键=一组）
                    if (JeiRuntimeProxy.isJeiCheatModeEnabled()) {
                        return;
                    }
                    ITypedIngredient<?> typed = hovered.get();
                    GenericStack stack = GenericEntryStackHelper.ingredientToStack(typed);
                    if (stack != null) {
                        // 发送到服务端：若网络有库存则拉取一组到空槽，否则若可合成则打开下单界面
                        ModNetwork.CHANNEL.sendToServer(new PullFromJeiOrCraftC2SPacket(stack));
                        // 消费此次点击，避免 JEI/原版对左键的其它处理
                        event.setCanceled(true);
                        return;
                    }
                }
            } catch (Throwable ignored) {
                // 兼容 JEI 版本差异或运行时异常
            }
        }

        // 中键：打开 AE 下单界面（保持原有功能）
        if (event.getButton() == GLFW.GLFW_MOUSE_BUTTON_MIDDLE) {
            try {
                // 优先在 JEI 配方界面基于坐标获取；若无，再从覆盖层/书签获取
                double mouseX = event.getMouseX();
                double mouseY = event.getMouseY();
                Optional<ITypedIngredient<?>> hovered = JeiRuntimeProxy.getIngredientUnderMouse(mouseX, mouseY);
                if (hovered.isEmpty()) {
                    hovered = JeiRuntimeProxy.getIngredientUnderMouse();
                }
                if (hovered.isEmpty()) return;

                ITypedIngredient<?> typed = hovered.get();
                // 若 JEI 作弊模式开启，则放行给 JEI 处理（中键=一组）
                if (JeiRuntimeProxy.isJeiCheatModeEnabled()) {
                    return;
                }
                GenericStack stack = GenericEntryStackHelper.ingredientToStack(typed);
                if (stack == null) return;

                // 发送到服务端，让其验证并打开 CraftAmountMenu
                ModNetwork.CHANNEL.sendToServer(new OpenCraftFromJeiC2SPacket(stack));

                // 消费此次点击，避免 JEI/原版对中键的其它处理
                event.setCanceled(true);
            } catch (Throwable ignored) {
                // 兼容 JEI 版本差异或运行时异常
            }
        }
    }

    @SubscribeEvent
    public static void onKeyPressedPre(ScreenEvent.KeyPressed.Pre event) {
        // 若未安装 JEI，直接跳过
        if (!ModList.get().isLoaded("jei")) {
            return;
        }
        // 注意：不要在 try/catch 之外直接访问 JEI 运行时，避免类加载崩溃
        // 检查是否按下了填充搜索框的快捷键
        if (!ModKeybindings.FILL_SEARCH_KEY.matches(event.getKeyCode(), event.getScanCode())) {
            return;
        }

        // 仅当鼠标确实悬停在 JEI 配料上时触发
        try {
            Optional<ITypedIngredient<?>> hovered = JeiRuntimeProxy.getIngredientUnderMouse();
            if (hovered.isEmpty()) return;

            ITypedIngredient<?> typed = hovered.get();

            // 通用获取显示名称（兼容物品/流体等）
            String name = JeiRuntimeProxy.getTypedIngredientDisplayName(typed);
            if (name == null || name.isEmpty()) return;

            // 写入 AE2 终端的搜索框
            var screen = Minecraft.getInstance().screen;
            if (screen instanceof MEStorageScreen<?> me) {
                try {
                    MEStorageScreenAccessor acc = (MEStorageScreenAccessor) me;
                    acc.eap$getSearchField().setValue(name);
                    acc.eap$setSearchText(name); // 同步到 Repo 并刷新
                    event.setCanceled(true);
                } catch (Throwable ignored) {
                }
            }else if (screen instanceof GuiExPatternTerminal<?> gpt) {
                try {
                    GuiExPatternTerminalAccessor acc = (GuiExPatternTerminalAccessor) gpt;
                    acc.getSearchOutField().setValue(name);
                    event.setCanceled(true);
                }catch (Throwable ignored) {}
            }
        } catch (Throwable ignored) {
            // 兼容 JEI 版本差异或运行时异常
        }
    }
}