Implement new method

CitizensDev · Aug 20, 2019 · 6f4ec9c · 6f4ec9c
1 parent 950ab24
commit 6f4ec9c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/main/src/main/java/net/citizensnpcs/npc/CitizensTraitFactory.java b/main/src/main/java/net/citizensnpcs/npc/CitizensTraitFactory.java
@@ -1,5 +1,6 @@
 package net.citizensnpcs.npc;
 
+import java.util.Collection;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
@@ -132,6 +133,11 @@ public void deregisterTrait(TraitInfo info) {
         registered.remove(info.getTraitName());
     }
 
+    @Override
+    public Collection<TraitInfo> getRegisteredTraits() {
+        return registered.values();
+    }
+
     @Override
     public <T extends Trait> T getTrait(Class<T> clazz) {
         for (TraitInfo entry : registered.values()) {