feat: introduce EffectInstance.saveNBT()

AllayMC · Jun 27, 2024 · 4fa2e40 · 4fa2e40
1 parent 4c886a9
commit 4fa2e40
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/Allay-API/src/main/java/org/allaymc/api/entity/effect/EffectInstance.java b/Allay-API/src/main/java/org/allaymc/api/entity/effect/EffectInstance.java
@@ -1,5 +1,7 @@
 package org.allaymc.api.entity.effect;
 
+import org.cloudburstmc.nbt.NbtMap;
+
 /**
  * Allay Project 2023/10/27
  *
@@ -20,4 +22,6 @@ public interface EffectInstance {
     int getDuration();
 
     void setDuration(int duration);
+
+    NbtMap saveNBT();
 }
diff --git a/Allay-API/src/main/java/org/allaymc/api/entity/effect/SimpleEffectInstance.java b/Allay-API/src/main/java/org/allaymc/api/entity/effect/SimpleEffectInstance.java
@@ -3,6 +3,7 @@
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.Setter;
+import org.cloudburstmc.nbt.NbtMap;
 
 /**
  * Allay Project 2023/10/27
@@ -17,4 +18,17 @@ public final class SimpleEffectInstance implements EffectInstance {
     private int amplifier;
     private int duration;
     private boolean visible;
+
+    @Override
+    public NbtMap saveNBT() {
+        return NbtMap.builder()
+                .putByte("Id", (byte) type.getId())
+                .putByte("Amplifier", (byte) amplifier)
+                // TODO: DurationEasy, DurationNormal, DurationHard
+                .putInt("Duration", duration)
+                // TODO: Ambient
+                // TODO: DisplayOnScreenTextureAnimation
+                .putBoolean("ShowParticles", visible)
+                .build();
+    }
 }