Files
ToolStats/src/main/java/lol/hyper/toolstats/events/EntityDamage.java
2025-05-07 17:58:44 -04:00

287 lines
12 KiB
Java

/*
* This file is part of ToolStats.
*
* ToolStats is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* ToolStats is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with ToolStats. If not, see <https://www.gnu.org/licenses/>.
*/
package lol.hyper.toolstats.events;
import lol.hyper.toolstats.ToolStats;
import org.bukkit.Bukkit;
import org.bukkit.GameMode;
import org.bukkit.Material;
import org.bukkit.entity.*;
import org.bukkit.event.EventHandler;
import org.bukkit.event.EventPriority;
import org.bukkit.event.Listener;
import org.bukkit.event.entity.EntityDamageByEntityEvent;
import org.bukkit.event.entity.EntityDamageEvent;
import org.bukkit.inventory.ItemStack;
import org.bukkit.inventory.PlayerInventory;
import org.bukkit.inventory.meta.ItemMeta;
import org.bukkit.projectiles.ProjectileSource;
import java.util.*;
public class EntityDamage implements Listener {
private final ToolStats toolStats;
// track mobs that are killed by a player
public final Set<UUID> trackedMobs = new HashSet<>();
private final List<EntityDamageEvent.DamageCause> ignoredCauses = Arrays.asList(EntityDamageEvent.DamageCause.SUICIDE, EntityDamageEvent.DamageCause.VOID, EntityDamageEvent.DamageCause.CUSTOM, EntityDamageEvent.DamageCause.KILL);
public EntityDamage(ToolStats toolStats) {
this.toolStats = toolStats;
}
@EventHandler(priority = EventPriority.MONITOR)
public void onDamage(EntityDamageByEntityEvent event) {
if (event.isCancelled()) {
return;
}
if (!(event.getEntity() instanceof LivingEntity mobBeingAttacked)) {
return;
}
// ignore void and /kill damage
EntityDamageEvent.DamageCause cause = event.getCause();
if (ignoredCauses.contains(cause)) {
return;
}
Entity damager = event.getDamager();
boolean playerAttacking = damager instanceof Player;
boolean playerBeingAttacked = mobBeingAttacked instanceof Player;
double finalDamage = event.getFinalDamage();
boolean modDied = mobBeingAttacked.getHealth() - finalDamage <= 0;
// player attacks something
if (playerAttacking) {
PlayerInventory playerAttackingInventory = ((Player) damager).getInventory();
// make sure the item the player used is an item we want
if (!toolStats.itemChecker.isMeleeWeapon(playerAttackingInventory.getItemInMainHand().getType())) {
return;
}
// update their weapon's damage
updateWeaponDamage(playerAttackingInventory, event.getFinalDamage());
// the mob the player attacked died
if (modDied) {
// player killed another player
if (playerBeingAttacked) {
updateWeaponKills(playerAttackingInventory, "player");
} else {
// player kills a regular mob
updateWeaponKills(playerAttackingInventory, "mob");
}
}
trackedMobs.add(mobBeingAttacked.getUniqueId());
Bukkit.getGlobalRegionScheduler().runDelayed(toolStats, scheduledTask -> trackedMobs.remove(mobBeingAttacked.getUniqueId()), 20);
}
// something was hit by a trident
if (damager instanceof Trident trident) {
ProjectileSource source = trident.getShooter();
if (source instanceof Player) {
// update the trident's tracked damage
updateTridentDamage(trident, finalDamage);
// if the mob died from the trident
if (modDied) {
// if the trident killed a player, update the kills
if (playerBeingAttacked) {
updateTridentKills(trident, "player");
} else {
// the trident killed a mob, update the kills
updateTridentKills(trident, "mob");
}
}
trackedMobs.add(mobBeingAttacked.getUniqueId());
Bukkit.getGlobalRegionScheduler().runDelayed(toolStats, scheduledTask -> trackedMobs.remove(mobBeingAttacked.getUniqueId()), 20);
}
}
// something was hit by an arrow
if (damager instanceof Arrow arrow) {
ProjectileSource source = arrow.getShooter();
// a player shot the arrow
if (source instanceof Player shootingPlayer) {
// update the player's bow damage
updateBowDamage(shootingPlayer.getInventory(), finalDamage);
// if the mob died from the arrow
if (modDied) {
if (playerBeingAttacked) {
// player killed another player with an arrow
updateBowKills(shootingPlayer.getInventory(), "player");
} else {
// player killed mob with an arrow
updateBowKills(shootingPlayer.getInventory(), "mob");
}
}
trackedMobs.add(mobBeingAttacked.getUniqueId());
Bukkit.getGlobalRegionScheduler().runDelayed(toolStats, scheduledTask -> trackedMobs.remove(mobBeingAttacked.getUniqueId()), 20);
}
}
}
@EventHandler(priority = EventPriority.MONITOR)
public void onDamage(EntityDamageEvent event) {
if (!(event.getEntity() instanceof LivingEntity mobBeingAttacked)) {
return;
}
// ignore void and /kill damage
EntityDamageEvent.DamageCause cause = event.getCause();
if (ignoredCauses.contains(cause)) {
return;
}
// player is taking damage
if (mobBeingAttacked instanceof Player playerTakingDamage) {
if (playerTakingDamage.getGameMode() == GameMode.CREATIVE && !toolStats.config.getBoolean("allow-creative")) {
return;
}
updateArmorDamage(playerTakingDamage.getInventory(), event.getFinalDamage());
}
}
private void updateArmorDamage(PlayerInventory playerInventory, double damage) {
ItemStack[] armorContents = playerInventory.getArmorContents();
for (ItemStack armorPiece : armorContents) {
if (armorPiece != null) {
if (toolStats.itemChecker.isArmor(armorPiece.getType())) {
ItemMeta newItem = toolStats.itemLore.updateArmorDamage(armorPiece, damage, false);
if (newItem != null) {
armorPiece.setItemMeta(newItem);
}
}
}
}
// apply the new armor
playerInventory.setArmorContents(armorContents);
}
private void updateBowDamage(PlayerInventory playerInventory, double damage) {
ItemStack heldBow = toolStats.itemChecker.getBow(playerInventory);
if (heldBow == null) {
return;
}
boolean isMain = playerInventory.getItemInMainHand().getType() == Material.BOW || playerInventory.getItemInMainHand().getType() == Material.CROSSBOW;
boolean isOffHand = playerInventory.getItemInOffHand().getType() == Material.BOW || playerInventory.getItemInOffHand().getType() == Material.CROSSBOW;
ItemMeta newBowDamage = toolStats.itemLore.updateWeaponDamage(heldBow, damage, false);
//toolStats.logger.info(newBowDamage.toString());
// player is shooting another player
if (newBowDamage != null) {
if (isMain && isOffHand) {
playerInventory.getItemInMainHand().setItemMeta(newBowDamage);
} else if (isMain) {
playerInventory.getItemInMainHand().setItemMeta(newBowDamage);
} else if (isOffHand) {
playerInventory.getItemInOffHand().setItemMeta(newBowDamage);
}
}
}
private void updateBowKills(PlayerInventory playerInventory, String type) {
ItemStack heldBow = toolStats.itemChecker.getBow(playerInventory);
if (heldBow == null) {
return;
}
boolean isMain = playerInventory.getItemInMainHand().getType() == Material.BOW || playerInventory.getItemInMainHand().getType() == Material.CROSSBOW;
boolean isOffHand = playerInventory.getItemInOffHand().getType() == Material.BOW || playerInventory.getItemInOffHand().getType() == Material.CROSSBOW;
if (type.equalsIgnoreCase("mob")) {
// player is shooting a mob
ItemMeta newBow = toolStats.itemLore.updateMobKills(heldBow, 1);
if (newBow != null) {
if (isMain && isOffHand) {
playerInventory.getItemInMainHand().setItemMeta(newBow);
} else if (isMain) {
playerInventory.getItemInMainHand().setItemMeta(newBow);
} else if (isOffHand) {
playerInventory.getItemInOffHand().setItemMeta(newBow);
}
}
}
if (type.equalsIgnoreCase("player")) {
ItemMeta newBowKills = toolStats.itemLore.updatePlayerKills(heldBow, 1);
if (newBowKills != null) {
if (isMain && isOffHand) {
playerInventory.getItemInMainHand().setItemMeta(newBowKills);
} else if (isMain) {
playerInventory.getItemInMainHand().setItemMeta(newBowKills);
} else if (isOffHand) {
playerInventory.getItemInOffHand().setItemMeta(newBowKills);
}
}
}
}
private void updateTridentKills(Trident trident, String type) {
ItemStack newTrident = trident.getItemStack();
ItemMeta newKills;
if (type.equalsIgnoreCase("player")) {
newKills = toolStats.itemLore.updatePlayerKills(trident.getItemStack(), 1);
} else {
newKills = toolStats.itemLore.updateMobKills(trident.getItemStack(), 1);
}
if (newKills != null) {
newTrident.setItemMeta(newKills);
trident.setItemStack(newTrident);
}
}
private void updateTridentDamage(Trident trident, double damage) {
ItemStack newTrident = trident.getItemStack();
ItemMeta newDamage = toolStats.itemLore.updateWeaponDamage(trident.getItemStack(), damage, false);
if (newDamage != null) {
newTrident.setItemMeta(newDamage);
trident.setItemStack(newTrident);
}
}
private void updateWeaponDamage(PlayerInventory playerInventory, double damage) {
ItemStack heldWeapon = playerInventory.getItemInMainHand();
ItemMeta newHeldWeaponMeta = toolStats.itemLore.updateWeaponDamage(heldWeapon, damage, false);
if (newHeldWeaponMeta != null) {
playerInventory.getItemInMainHand().setItemMeta(newHeldWeaponMeta);
}
}
private void updateWeaponKills(PlayerInventory playerInventory, String type) {
ItemStack heldWeapon = playerInventory.getItemInMainHand();
ItemMeta newHeldWeaponMeta = null;
if (type.equalsIgnoreCase("player")) {
newHeldWeaponMeta = toolStats.itemLore.updatePlayerKills(heldWeapon, 1);
}
if (type.equalsIgnoreCase("mob")) {
newHeldWeaponMeta = toolStats.itemLore.updateMobKills(heldWeapon, 1);
}
if (newHeldWeaponMeta != null) {
playerInventory.getItemInMainHand().setItemMeta(newHeldWeaponMeta);
}
}
}