Add target block/entity expressions

Author: ShaneBeee

src/main/java/com/github/skriptdev/skript/plugin/elements/expressions/ExpressionHandler.java

@@ -6,10 +6,12 @@
 import com.github.skriptdev.skript.plugin.elements.expressions.block.ExprBlockTypeAtLocation;
 import com.github.skriptdev.skript.plugin.elements.expressions.block.ExprBlockTypeOfBlock;
 import com.github.skriptdev.skript.plugin.elements.expressions.block.ExprBlockFluidLevel;
+import com.github.skriptdev.skript.plugin.elements.expressions.block.ExprTargetBlockOfPlayer;
 import com.github.skriptdev.skript.plugin.elements.expressions.entity.ExprEntityHealth;
 import com.github.skriptdev.skript.plugin.elements.expressions.entity.ExprEntityStat;
 import com.github.skriptdev.skript.plugin.elements.expressions.entity.ExprNPCType;
 import com.github.skriptdev.skript.plugin.elements.expressions.entity.ExprName;
+import com.github.skriptdev.skript.plugin.elements.expressions.entity.ExprTargetEntityOfEntity;
 import com.github.skriptdev.skript.plugin.elements.expressions.item.ExprInventory;
 import com.github.skriptdev.skript.plugin.elements.expressions.item.ExprItemContainer;
 import com.github.skriptdev.skript.plugin.elements.expressions.item.ExprItemStack;
@@ -44,12 +46,14 @@ public static void register(SkriptRegistration registration) {
         ExprBlockFluidLevel.register(registration);
         ExprBlockTypeAtLocation.register(registration);
         ExprBlockTypeOfBlock.register(registration);
+        ExprTargetBlockOfPlayer.register(registration);
 
         // ENTITY
         ExprEntityHealth.register(registration);
         ExprEntityStat.register(registration);
         ExprName.register(registration);
         ExprNPCType.register(registration);
+        ExprTargetEntityOfEntity.register(registration);
 
         // ITEM
         ExprInventory.register(registration);

src/main/java/com/github/skriptdev/skript/plugin/elements/expressions/block/ExprTargetBlockOfPlayer.java

@@ -0,0 +1,63 @@
+package com.github.skriptdev.skript.plugin.elements.expressions.block;
+
+import com.github.skriptdev.skript.api.hytale.Block;
+import com.github.skriptdev.skript.api.skript.registration.SkriptRegistration;
+import com.hypixel.hytale.component.Ref;
+import com.hypixel.hytale.component.Store;
+import com.hypixel.hytale.math.vector.Vector3i;
+import com.hypixel.hytale.server.core.entity.entities.Player;
+import com.hypixel.hytale.server.core.universe.world.World;
+import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
+import com.hypixel.hytale.server.core.util.TargetUtil;
+import io.github.syst3ms.skriptparser.lang.Expression;
+import io.github.syst3ms.skriptparser.lang.TriggerContext;
+import io.github.syst3ms.skriptparser.parsing.ParseContext;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.Optional;
+
+public class ExprTargetBlockOfPlayer implements Expression<Block> {
+
+    public static void register(SkriptRegistration reg) {
+        reg.newExpression(ExprTargetBlockOfPlayer.class, Block.class, true,
+                "target block of %player%")
+            .name("Target Block of Player")
+            .description("Returns the block the player is looking at.")
+            .examples("set {_block} to target block of player")
+            .since("INSERT VERSION")
+            .register();
+    }
+
+    private Expression<Player> player;
+
+    @SuppressWarnings("unchecked")
+    @Override
+    public boolean init(Expression<?>[] expressions, int matchedPattern, @NotNull ParseContext parseContext) {
+        this.player = (Expression<Player>) expressions[0];
+        return true;
+    }
+
+    @Override
+    public Block[] getValues(@NotNull TriggerContext ctx) {
+        Optional<? extends Player> single = this.player.getSingle(ctx);
+        if (single.isEmpty()) return null;
+
+        Player player = single.get();
+        Ref<EntityStore> ref = player.getReference();
+        World world = player.getWorld();
+        if (world == null || ref == null) return null;
+
+        Store<EntityStore> store = world.getEntityStore().getStore();
+
+        // TODO configurable maxDistance
+        Vector3i targetBlock = TargetUtil.getTargetBlock(ref, 50, store);
+        if (targetBlock == null) return null;
+        return new Block[]{new Block(world, targetBlock)};
+    }
+
+    @Override
+    public String toString(@NotNull TriggerContext ctx, boolean debug) {
+        return "target block of " + this.player.toString(ctx, debug);
+    }
+
+}

src/main/java/com/github/skriptdev/skript/plugin/elements/expressions/entity/ExprTargetEntityOfEntity.java

@@ -0,0 +1,65 @@
+package com.github.skriptdev.skript.plugin.elements.expressions.entity;
+
+import com.github.skriptdev.skript.api.skript.registration.SkriptRegistration;
+import com.hypixel.hytale.component.Ref;
+import com.hypixel.hytale.component.Store;
+import com.hypixel.hytale.server.core.entity.Entity;
+import com.hypixel.hytale.server.core.entity.EntityUtils;
+import com.hypixel.hytale.server.core.universe.world.World;
+import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
+import com.hypixel.hytale.server.core.util.TargetUtil;
+import io.github.syst3ms.skriptparser.lang.Expression;
+import io.github.syst3ms.skriptparser.lang.TriggerContext;
+import io.github.syst3ms.skriptparser.parsing.ParseContext;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.Optional;
+
+public class ExprTargetEntityOfEntity implements Expression<Entity> {
+
+    public static void register(SkriptRegistration reg) {
+        reg.newExpression(ExprTargetEntityOfEntity.class, Entity.class, true,
+                "target entity of %entity%")
+            .name("Target Entity of Entity")
+            .description("Returns the target entity of the given entity.")
+            .examples("set {_target} to target entity of player")
+            .since("INSERT VERSION")
+            .register();
+    }
+
+    private Expression<Entity> entity;
+
+    @SuppressWarnings("unchecked")
+    @Override
+    public boolean init(Expression<?>[] expressions, int matchedPattern, @NotNull ParseContext parseContext) {
+        this.entity = (Expression<Entity>) expressions[0];
+        return true;
+    }
+
+    @SuppressWarnings("deprecation")
+    @Override
+    public Entity[] getValues(@NotNull TriggerContext ctx) {
+        Optional<? extends Entity> single = this.entity.getSingle(ctx);
+        if (single.isEmpty()) return null;
+
+        Entity entity = single.get();
+        Ref<EntityStore> ref = entity.getReference();
+        World world = entity.getWorld();
+        if (world == null || ref == null) return null;
+
+        Store<EntityStore> store = world.getEntityStore().getStore();
+        Ref<EntityStore> targetEntity = TargetUtil.getTargetEntity(ref, store);
+        if (targetEntity == null || !targetEntity.isValid()) return null;
+
+        // TODO better handling of this deprecation
+        Entity target = EntityUtils.getEntity(targetEntity, store);
+
+        return new Entity[]{target};
+    }
+
+    @Override
+    public String toString(@NotNull TriggerContext ctx, boolean debug) {
+        return "target entity of " + this.entity.toString(ctx, debug);
+    }
+
+}