diff --git a/plugin/src/main/java/lol/pyr/znpcsplus/conversion/citizens/model/CitizensTraitsRegistry.java b/plugin/src/main/java/lol/pyr/znpcsplus/conversion/citizens/model/CitizensTraitsRegistry.java index 2297af3..81f6158 100644 --- a/plugin/src/main/java/lol/pyr/znpcsplus/conversion/citizens/model/CitizensTraitsRegistry.java +++ b/plugin/src/main/java/lol/pyr/znpcsplus/conversion/citizens/model/CitizensTraitsRegistry.java @@ -24,6 +24,7 @@ public class CitizensTraitsRegistry { register(new CommandTrait(taskScheduler)); register(new HologramTrait(textSerializer)); register(new EquipmentTrait(propertyRegistry)); + register(new SpawnedTrait()); } public CitizensTrait getByName(String name) { diff --git a/plugin/src/main/java/lol/pyr/znpcsplus/conversion/citizens/model/traits/SpawnedTrait.java b/plugin/src/main/java/lol/pyr/znpcsplus/conversion/citizens/model/traits/SpawnedTrait.java new file mode 100644 index 0000000..9f080c5 --- /dev/null +++ b/plugin/src/main/java/lol/pyr/znpcsplus/conversion/citizens/model/traits/SpawnedTrait.java @@ -0,0 +1,20 @@ +package lol.pyr.znpcsplus.conversion.citizens.model.traits; + +import lol.pyr.znpcsplus.conversion.citizens.model.CitizensTrait; +import lol.pyr.znpcsplus.npc.NpcImpl; +import org.jetbrains.annotations.NotNull; + +public class SpawnedTrait extends CitizensTrait { + + public SpawnedTrait() { + super("spawned"); + } + + @Override + public @NotNull NpcImpl apply(NpcImpl npc, Object value) { + if (value != null) { + npc.setEnabled((boolean) value); + } + return npc; + } +}