Add pick block handling for Infinite Spray Can color selection

Author: WhitePhant0m

src/main/java/net/neganote/gtutilities/client/event/ClientTickHandler.java

@@ -1,9 +1,17 @@
 package net.neganote.gtutilities.client.event;
 
+import appeng.api.implementations.blockentities.IColorableBlockEntity;
+import appeng.api.util.AEColor;
+import com.gregtechceu.gtceu.GTCEu;
+import com.gregtechceu.gtceu.api.blockentity.IPaintable;
+import com.gregtechceu.gtceu.api.pipenet.IPipeNode;
 import net.minecraft.client.Minecraft;
 import net.minecraft.sounds.SoundEvents;
 import net.minecraft.world.InteractionHand;
+import net.minecraft.world.item.DyeColor;
 import net.minecraft.world.item.ItemStack;
+import net.minecraft.world.level.block.entity.ShulkerBoxBlockEntity;
+import net.minecraft.world.phys.BlockHitResult;
 import net.minecraft.world.phys.HitResult;
 import net.minecraftforge.api.distmarker.Dist;
 import net.minecraftforge.client.event.InputEvent;
@@ -81,4 +89,69 @@ public static void onRightClick(InputEvent.InteractionKeyMappingTriggered event)
         event.setCanceled(true);
         mc.setScreen(new ColorRadialMenuScreen(InteractionHand.MAIN_HAND));
     }
+
+    @SubscribeEvent
+    public static void onKeyInput(InputEvent.InteractionKeyMappingTriggered event) {
+        if (!event.isPickBlock()) return;
+
+        var mc = Minecraft.getInstance();
+        if (mc.player == null) return;
+
+        var player = mc.player;
+        var stack = player.getMainHandItem();
+        if (!(stack.getItem() instanceof InfiniteSprayCanItem)) return;
+        event.setCanceled(true);
+
+        var target = mc.hitResult;
+        if (target == null || target.getType() != HitResult.Type.BLOCK) return;
+
+        var level = player.level();
+        var blockHit = (BlockHitResult) target;
+        var pos = blockHit.getBlockPos();
+        var be = level.getBlockEntity(pos);
+
+        if (GTCEu.Mods.isAE2Loaded() && be instanceof IColorableBlockEntity colorable) {
+            if (colorable.getColor().equals(AEColor.TRANSPARENT)) {
+                UtilsNetwork.CHANNEL.sendToServer(new SelectColorPacket(InteractionHand.MAIN_HAND, -1));
+                return;
+            }
+
+            for (AEColor color : AEColor.values()) {
+                if (color.equals(colorable.getColor())) {
+                    UtilsNetwork.CHANNEL.sendToServer(new SelectColorPacket(InteractionHand.MAIN_HAND, color.ordinal()));
+                    return;
+                }
+            }
+        } else if (be instanceof IPipeNode pipe) {
+            if (!pipe.isPainted()) {
+                UtilsNetwork.CHANNEL.sendToServer(new SelectColorPacket(InteractionHand.MAIN_HAND, -1));
+            } else {
+                for (int i = 0; i < DyeColor.values().length; i++) {
+                    DyeColor color = DyeColor.byId(i);
+                    if (color.getMapColor().col == pipe.getPaintingColor()) {
+                        UtilsNetwork.CHANNEL.sendToServer(new SelectColorPacket(InteractionHand.MAIN_HAND, i));
+                        return;
+                    }
+                }
+            }
+        } else if (be instanceof IPaintable paintable) {
+            if (!paintable.isPainted()) {
+                UtilsNetwork.CHANNEL.sendToServer(new SelectColorPacket(InteractionHand.MAIN_HAND, -1));
+            } else {
+                for (int i = 0; i < DyeColor.values().length; i++) {
+                    DyeColor color = DyeColor.byId(i);
+                    if (color.getMapColor().col == paintable.getRealColor()) {
+                        UtilsNetwork.CHANNEL.sendToServer(new SelectColorPacket(InteractionHand.MAIN_HAND, i));
+                        return;
+                    }
+                }
+            }
+        } else if (be instanceof ShulkerBoxBlockEntity shulkerBox) {
+            if (shulkerBox.getColor() == null) {
+                UtilsNetwork.CHANNEL.sendToServer(new SelectColorPacket(InteractionHand.MAIN_HAND, -1));
+            } else {
+                UtilsNetwork.CHANNEL.sendToServer(new SelectColorPacket(InteractionHand.MAIN_HAND, shulkerBox.getColor().ordinal()));
+            }
+        }
+    }
 }