feat(player): handle no more hearts

1 files changed, 9 insertions, 2 deletions

diff --git a/src/main/java/world/anhgelus/lifesteal/LifeStealer.java b/src/main/java/world/anhgelus/lifesteal/LifeStealer.java
index 2942055..b8d774a 100644
--- a/src/main/java/world/anhgelus/lifesteal/LifeStealer.java
+++ b/src/main/java/world/anhgelus/lifesteal/LifeStealer.java
@@ -9,6 +9,7 @@ import net.minecraft.entity.attribute.EntityAttributes;
 import net.minecraft.item.ItemStack;
 import net.minecraft.item.Items;
 import net.minecraft.nbt.NbtCompound;
+import net.minecraft.server.BannedPlayerEntry;
 import net.minecraft.server.network.ServerPlayerEntity;
 import net.minecraft.util.Identifier;
 import net.minecraft.util.math.MathHelper;
@@ -66,8 +67,14 @@ public class LifeStealer {
 
     public void death() {
         healthModifier = MathHelper.clamp(healthModifier - 2, -20, 20);
-        if (healthModifier <= 0) {
-            //TODO: ban player
+        if (healthModifier <= -20) {
+            player.getEntityWorld()
+                    .getServer()
+                    .getPlayerManager()
+                    .getUserBanList()
+                    .add(new BannedPlayerEntry(
+                            player.getPlayerConfigEntry(), null, "LifeSteal", null, "You have lose all your hearts..."
+                    ));
             return;
         }
         updateHealth();