From 53b0906d2e556cbce5a94cdf8908bbeeb8086ee2 Mon Sep 17 00:00:00 2001 From: Li Junyu Date: Sat, 1 Feb 2025 16:50:00 +0800 Subject: [PATCH] Add util methods --- .../kubejs/client/highlight/HighlightKubeEvent.java | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/src/main/java/dev/latvian/mods/kubejs/client/highlight/HighlightKubeEvent.java b/src/main/java/dev/latvian/mods/kubejs/client/highlight/HighlightKubeEvent.java index 79d9e0daa..be7585550 100644 --- a/src/main/java/dev/latvian/mods/kubejs/client/highlight/HighlightKubeEvent.java +++ b/src/main/java/dev/latvian/mods/kubejs/client/highlight/HighlightKubeEvent.java @@ -2,6 +2,7 @@ import dev.latvian.mods.kubejs.client.ClientPlayerKubeEvent; import dev.latvian.mods.kubejs.color.KubeColor; +import dev.latvian.mods.kubejs.level.LevelBlock; import dev.latvian.mods.kubejs.typings.Info; import net.minecraft.client.Minecraft; import net.minecraft.commands.arguments.selector.EntitySelector; @@ -132,4 +133,15 @@ public void addTarget(KubeColor color) { addBlock(hit.getBlockPos(), color); } } + + public LevelBlock getTargetBlock() { + if (mc.hitResult instanceof BlockHitResult block && mc.hitResult.getType() == HitResult.Type.BLOCK) { + return getLevel().kjs$getBlock(block.getBlockPos()); + } + return null; + } + + public Entity getTargetEntity() { + return mc.hitResult instanceof EntityHitResult hit ? hit.getEntity() : null; + } } \ No newline at end of file