1 files changed, 61 insertions, 0 deletions
diff --git a/src/main/java/com/thatgravyboat/skyblockhud/api/KillTracking.java b/src/main/java/com/thatgravyboat/skyblockhud/api/KillTracking.java
new file mode 100644
index 0000000..9aad4c6
--- /dev/null
+++ b/src/main/java/com/thatgravyboat/skyblockhud/api/KillTracking.java
@@ -0,0 +1,61 @@
+package com.thatgravyboat.skyblockhud.api;
+
+import com.thatgravyboat.skyblockhud.DevModeConstants;
+import com.thatgravyboat.skyblockhud.api.events.SkyBlockEntityKilled;
+import com.thatgravyboat.skyblockhud.api.sbentities.EntityTypeRegistry;
+import net.minecraft.client.Minecraft;
+import net.minecraft.util.DamageSource;
+import net.minecraftforge.common.MinecraftForge;
+import net.minecraftforge.event.entity.EntityJoinWorldEvent;
+import net.minecraftforge.event.entity.living.LivingDeathEvent;
+import net.minecraftforge.event.entity.player.AttackEntityEvent;
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
+
+import java.util.HashSet;
+import java.util.Set;
+import java.util.UUID;
+
+public class KillTracking {
+
+    public static final Set<UUID> attackedEntities = new HashSet<>();
+
+    @SubscribeEvent
+    public void onAttack(AttackEntityEvent event) {
+        if (event.target != null) {
+            attackedEntities.add(event.target.getUniqueID());
+        }
+    }
+
+    @SubscribeEvent
+    public void onDeath(LivingDeathEvent event) {
+        if (DevModeConstants.mobDeathLogging) {
+            //Used for testing
+            System.out.println("----------------------------------------------------------------------------------------------------------------");
+            System.out.println("Name : " + event.entity.getName());
+            System.out.println("UUID : " + event.entity.getUniqueID());
+            System.out.println("Tag : " + event.entity.serializeNBT());
+            System.out.println("Damage : " + getDamageSourceString(event.source));
+            System.out.println("SBH Entity ID: " + EntityTypeRegistry.getEntityId(event.entity));
+            System.out.println("----------------------------------------------------------------------------------------------------------------");
+        }
+        if (attackedEntities.contains(event.entity.getUniqueID())) {
+            if (EntityTypeRegistry.getEntityId(event.entity) != null) {
+                MinecraftForge.EVENT_BUS.post(new SkyBlockEntityKilled(EntityTypeRegistry.getEntityId(event.entity), event.entity));
+            }
+            attackedEntities.remove(event.entity.getUniqueID());
+        }
+    }
+
+    public static String getDamageSourceString(DamageSource source) {
+        return ("{ " + source.getDamageType() + ", " + source.isDamageAbsolute() + ", " + source.isDifficultyScaled() + ", " + source.isFireDamage() + ", " + source.isProjectile() + ", " + source.isUnblockable() + ", " + source.isExplosion() + ", " + source.isMagicDamage() + ", " + source.isCreativePlayer() + ", " + source.getSourceOfDamage() + " }");
+    }
+
+    @SubscribeEvent
+    public void onWorldChange(EntityJoinWorldEvent event) {
+        if (event.entity != null) {
+            if (event.entity.getUniqueID().equals(Minecraft.getMinecraft().thePlayer.getUniqueID())) {
+                attackedEntities.clear();
+            }
+        }
+    }
+}