From f0e7ea830264faaff09f166e5e7f7be54aca5486 Mon Sep 17 00:00:00 2001
From: D3v1s0m <d3v1s0m@gmail.com>
Date: Mon, 3 Jun 2024 13:37:31 +0530
Subject: [PATCH] added spawned import from citizens

---
 .../model/CitizensTraitsRegistry.java         |  1 +
 .../citizens/model/traits/SpawnedTrait.java   | 20 +++++++++++++++++++
 2 files changed, 21 insertions(+)
 create mode 100644 plugin/src/main/java/lol/pyr/znpcsplus/conversion/citizens/model/traits/SpawnedTrait.java

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;
+    }
+}