add tag server.list_traits

DenizenScript · Aug 20, 2019 · afa3083 · afa3083
1 parent 840a6cd
commit afa3083
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/plugin/src/main/java/com/denizenscript/denizen/tags/core/ServerTagBase.java b/plugin/src/main/java/com/denizenscript/denizen/tags/core/ServerTagBase.java
@@ -40,6 +40,7 @@
 import net.citizensnpcs.api.CitizensAPI;
 import net.citizensnpcs.api.command.CommandContext;
 import net.citizensnpcs.api.npc.NPC;
+import net.citizensnpcs.api.trait.TraitInfo;
 import org.bukkit.*;
 import org.bukkit.block.Biome;
 import org.bukkit.block.banner.PatternType;
@@ -359,6 +360,21 @@ public void serverTag(ReplaceableTagEvent event) {
             return;
         }
 
+        // <--[tag]
+        // @attribute <server.list_traits>
+        // @Plugin Citizens
+        // @returns ListTag
+        // @description
+        // Returns a list of all available NPC traits on the server.
+        // -->
+        if (attribute.startsWith("list_traits") && Depends.citizens != null) {
+            ListTag allTraits = new ListTag();
+            for (TraitInfo trait : CitizensAPI.getTraitFactory().getRegisteredTraits()) {
+                allTraits.add(trait.getTraitName());
+            }
+            event.setReplaced(allTraits.getAttribute(attribute.fulfill(1)));
+        }
+
         // <--[tag]
         // @attribute <server.list_biomes>
         // @returns ListTag