Add some new placeholders

CitizensDev · Jan 2, 2022 · 4abea2f · 4abea2f
1 parent 7eb7494
commit 4abea2f
Showing 1 changed file with 82 additions and 9 deletions.
diff --git a/src/main/java/net/citizensnpcs/api/util/Placeholders.java b/src/main/java/net/citizensnpcs/api/util/Placeholders.java
@@ -1,10 +1,25 @@
 package net.citizensnpcs.api.util;
 
+import java.util.Collection;
+import java.util.List;
+import java.util.Random;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import org.bukkit.Bukkit;
+import org.bukkit.Location;
 import org.bukkit.OfflinePlayer;
 import org.bukkit.command.CommandSender;
 import org.bukkit.entity.Entity;
+import org.bukkit.entity.EntityType;
+import org.bukkit.entity.Player;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.Iterables;
+import com.google.common.collect.Lists;
 
 import me.clip.placeholderapi.PlaceholderAPI;
+import net.citizensnpcs.api.CitizensAPI;
 import net.citizensnpcs.api.npc.NPC;
 import net.citizensnpcs.api.trait.trait.Owner;
 
@@ -14,22 +29,78 @@ public static String replace(String text, CommandSender sender, NPC npc) {
         if (npc == null) {
             return text;
         }
-        for (int i = 0; i < CITIZENS_PLACEHOLDERS.length; i++) {
-            text = text.replace(CITIZENS_PLACEHOLDERS[i], i == 0 ? Integer.toString(npc.getId())
-                    : i == 1 ? npc.getName() : npc.getOrAddTrait(Owner.class).getOwner());
+        StringBuffer out = new StringBuffer();
+        Matcher matcher = PLACEHOLDER_MATCHER.matcher(text);
+        while (matcher.find()) {
+            String replacement = "";
+            String group = matcher.group(1);
+            if (group.equals("id")) {
+                replacement = Integer.toString(npc.getId());
+            } else if (group.equals("npc")) {
+                replacement = npc.getName();
+            } else if (group.equals("owner")) {
+                replacement = npc.getOrAddTrait(Owner.class).getOwner();
+            }
+            matcher.appendReplacement(out, "");
+            out.append(replacement);
         }
-        return text;
+        matcher.appendTail(out);
+        return out.toString();
     }
 
     public static String replace(String text, OfflinePlayer player) {
         if (player == null) {
             return setPlaceholderAPIPlaceholders(text, player);
         }
         if (player instanceof Entity && ((Entity) player).isValid()) {
-            for (int i = 0; i < PLAYER_WORLD_PLACEHOLDERS.length; i++) {
-                text = text.replace(PLAYER_WORLD_PLACEHOLDERS[i],
-                        i < 4 ? player.getName() : ((Entity) player).getWorld().getName());
+            StringBuffer out = new StringBuffer();
+            Matcher matcher = PLAYER_PLACEHOLDER_MATCHER.matcher(text);
+            while (matcher.find()) {
+                String replacement = "";
+                String group = matcher.group(1);
+                if (PLAYER_VARIABLES.contains(group)) {
+                    replacement = player.getName();
+                } else if (group.equals("<random_player>")) {
+                    Collection<? extends Player> players = Bukkit.getServer().getOnlinePlayers();
+                    Player possible = Iterables.get(players, new Random().nextInt(players.size()), null);
+                    if (possible != null) {
+                        replacement = possible.getName();
+                    }
+                } else if (group.equals("<random_npc>")) {
+                    List<NPC> all = Lists.newArrayList(CitizensAPI.getNPCRegistry());
+                    if (all.size() > 0) {
+                        replacement = all.get(new Random().nextInt(all.size())).getName();
+                    }
+                } else if (group.equals("<random_npc_id>")) {
+                    List<NPC> all = Lists.newArrayList(CitizensAPI.getNPCRegistry());
+                    if (all.size() > 0) {
+                        replacement = Integer.toString(all.get(new Random().nextInt(all.size())).getId());
+                    }
+                } else if (group.equals("<nearest_player>")) {
+                    double min = Double.MAX_VALUE;
+                    Player closest = null;
+                    for (Entity entity : ((Player) player).getNearbyEntities(25, 25, 25)) {
+                        if (entity == player || !(entity instanceof Player) || entity.getType() != EntityType.PLAYER
+                                || CitizensAPI.getNPCRegistry().isNPC(entity))
+                            continue;
+                        Location location = entity.getLocation();
+                        double dist = location.distanceSquared(((Player) player).getLocation());
+                        if (dist > min)
+                            continue;
+                        min = dist;
+                        closest = (Player) entity;
+                    }
+                    if (closest != null) {
+                        replacement = closest.getName();
+                    }
+                } else if (group.equals("<world>")) {
+                    replacement = ((Entity) player).getWorld().getName();
+                }
+                matcher.appendReplacement(out, "");
+                out.append(replacement);
             }
+            matcher.appendTail(out);
+            text = out.toString();
         } else {
             for (int i = 0; i < PLAYER_PLACEHOLDERS.length; i++) {
                 text = text.replace(PLAYER_PLACEHOLDERS[i], player.getName());
@@ -50,8 +121,10 @@ private static String setPlaceholderAPIPlaceholders(String text, OfflinePlayer p
         }
     }
 
-    private static final String[] CITIZENS_PLACEHOLDERS = { "<id>", "<npc>", "<owner>" };
+    private static final Pattern PLACEHOLDER_MATCHER = Pattern.compile("<(id|npc|owner)>");
     private static boolean PLACEHOLDERAPI_ENABLED = true;
+    private static final Pattern PLAYER_PLACEHOLDER_MATCHER = Pattern.compile(
+            "(<player>|<p>|@p|%player%|<random_player>|<random_npc>|<random_npc_id>|<nearest_player>|<world>)");
     private static final String[] PLAYER_PLACEHOLDERS = { "<player>", "<p>", "@p", "%player%" };
-    private static final String[] PLAYER_WORLD_PLACEHOLDERS = { "<player>", "<p>", "@p", "%player%", "<world>" };
+    private static final Collection<String> PLAYER_VARIABLES = ImmutableList.of("<player>", "<p>", "@p", "%player%");
 }