Add a /twa team command to add existing entities to the TWA_Golems team

Author: eccentricdevotion

src/main/java/me/eccentric_nz/tardisweepingangels/commands/TARDISWeepingAngelsCommand.java

@@ -70,6 +70,9 @@ public boolean onCommand(CommandSender sender, Command cmd, String label, String
                 case "give" -> {
                     return new GiveCommand(plugin).give(sender, args);
                 }
+                case "team" -> {
+                    return new TeamCommand(plugin).join(sender);
+                }
                 case "teleport" -> {
                     return new TeleportCommand(plugin).add(sender, args);
                 }

src/main/java/me/eccentric_nz/tardisweepingangels/commands/TabComplete.java

@@ -33,7 +33,7 @@
  */
 public class TabComplete extends TARDISCompleter implements TabCompleter {
 
-    final ImmutableList<String> CMD_SUBS = ImmutableList.of("spawn", "equip", "disguise", "kill", "count", "follow", "stay", "remove", "set", "give", "teleport");
+    final ImmutableList<String> CMD_SUBS = ImmutableList.of("spawn", "equip", "disguise", "kill", "count", "follow", "stay", "remove", "set", "give", "team", "teleport");
     private final ImmutableList<String> ONOFF_SUBS = ImmutableList.of("on", "off");
     private final ImmutableList<String> TP_SUBS = ImmutableList.of("replace", "true", "false");
     private final ImmutableList<String> WORLD_SUBS;

src/main/java/me/eccentric_nz/tardisweepingangels/commands/TeamCommand.java

@@ -0,0 +1,60 @@
+package me.eccentric_nz.tardisweepingangels.commands;
+
+import me.eccentric_nz.TARDIS.TARDIS;
+import me.eccentric_nz.TARDIS.enumeration.TardisModule;
+import me.eccentric_nz.TARDIS.sonic.actions.TARDISSonicFreeze;
+import me.eccentric_nz.TARDIS.utility.TARDISVector3D;
+import me.eccentric_nz.tardisweepingangels.utils.TeamAdder;
+import org.bukkit.Location;
+import org.bukkit.command.CommandSender;
+import org.bukkit.entity.ArmorStand;
+import org.bukkit.entity.Entity;
+import org.bukkit.entity.EntityType;
+import org.bukkit.entity.Player;
+
+public class TeamCommand {
+
+    private final TARDIS plugin;
+
+    public TeamCommand(TARDIS plugin) {
+        this.plugin = plugin;
+    }
+
+    public boolean join(CommandSender sender) {
+        Player player = null;
+        if (sender instanceof Player) {
+            player = (Player) sender;
+        }
+        if (player == null) {
+            plugin.getMessenger().send(sender, TardisModule.MONSTERS, "CMD_PLAYER");
+            return true;
+        }
+        // get the entity the player is targeting
+        Location observerPos = player.getEyeLocation();
+        TARDISVector3D observerDir = new TARDISVector3D(observerPos.getDirection());
+        TARDISVector3D observerStart = new TARDISVector3D(observerPos);
+        TARDISVector3D observerEnd = observerStart.add(observerDir.multiply(16));
+        Entity entity = null;
+        // get nearby entities
+        for (Entity target : player.getNearbyEntities(8.0d, 8.0d, 8.0d)) {
+            // bounding box of the given player
+            TARDISVector3D targetPos = new TARDISVector3D(target.getLocation());
+            TARDISVector3D minimum = targetPos.add(-0.5, 0, -0.5);
+            TARDISVector3D maximum = targetPos.add(0.5, 1.67, 0.5);
+            if (TARDISSonicFreeze.hasIntersection(observerStart, observerEnd, minimum, maximum)) {
+                if (entity == null || entity.getLocation().distanceSquared(observerPos) > target.getLocation().distanceSquared(observerPos)) {
+                    entity = target;
+                }
+            }
+        }
+        if (entity != null) {
+            // add entity to the team
+            TeamAdder.joinTeam(entity);
+            plugin.getMessenger().send(player, TardisModule.MONSTERS, "WA_TEAM");
+        } else {
+            plugin.getMessenger().send(sender, TardisModule.MONSTERS, "WA_TEAM_LOOK");
+            return true;
+        }
+        return true;
+    }
+}

src/main/resources/en.yml

@@ -1275,6 +1275,8 @@ WA_STAND: "You are not looking at an armour stand within 8 blocks!"
 WA_STAY: "Please tell your follower to stay before removing it! /twa stay"
 WA_STOLE: "A Weeping Angel stole your TARDIS Key"
 WA_STRAX: "Strax is not lactating right now, try again later."
+WA_TEAM: "Entity added to TWA_Golems team."
+WA_TEAM_LOOK: "You are not looking at an entity within 8 blocks!"
 WA_UUID: "Command can only be used by a player, or a player UUID must be supplied!"
 WEATHER_CLEAR: "clear"
 WEATHER_COLD: "clear, but cold"