Add /buddha command

src/main/java/net/serble/estools/Commands/Buddha.java

@@ -0,0 +1,160 @@
+package net.serble.estools.Commands;
+
+import net.serble.estools.ConfigManager;
+import net.serble.estools.EntityCommand;
+import net.serble.estools.Main;
+import org.bukkit.Bukkit;
+import org.bukkit.command.Command;
+import org.bukkit.command.CommandSender;
+import org.bukkit.configuration.file.FileConfiguration;
+import org.bukkit.configuration.file.YamlConfiguration;
+import org.bukkit.entity.LivingEntity;
+import org.bukkit.event.EventHandler;
+import org.bukkit.event.Listener;
+import org.bukkit.event.entity.EntityDamageEvent;
+
+import java.util.*;
+
+public class Buddha extends EntityCommand implements Listener {
+	private static final HashMap<UUID, Integer> currentPlayers = new HashMap<>();
+	private static final String usage = genUsage("/buddha <entity> <time>");
+
+	@Override
+	public void onEnable() {
+		Bukkit.getServer().getPluginManager().registerEvents(this, Main.plugin);
+
+		FileConfiguration f = ConfigManager.load("gods.yml");
+		List<String> buddhas = f.getStringList("buddhas");
+		buddhas.forEach(w -> currentPlayers.put(UUID.fromString(w), -1));
+	}
+
+	@Override
+	public boolean onCommand(CommandSender sender, Command command, String label, String[] args) {
+		LivingEntity p;
+		int timer = -1;
+
+		if (args.length == 0) {
+			if (isNotEntity(sender, usage)) {
+                return false;
+            }
+
+			p = (LivingEntity) sender;
+		} else {
+			p = getEntity(sender, args[0]);
+
+			if (p == null) {
+                return false;
+            }
+
+			if (args.length > 1) {
+				try {
+					timer = Math.max((int)(Double.parseDouble(args[1]) * 20), -1);
+				} catch (NumberFormatException x) {
+					send(sender, usage);
+					return false;
+				}
+			}
+		}
+
+		UUID uid = p.getUniqueId();
+
+		if (currentPlayers.containsKey(uid)) {
+			int taskId = currentPlayers.get(uid);
+			currentPlayers.remove(uid);
+
+			if (taskId == -1) {
+				save();
+			} else {
+				Bukkit.getScheduler().cancelTask(taskId);
+			}
+
+			send(sender, "&aBuddha mode &6disabled&a for &6%s", getEntityName(p));
+		}
+		else {
+			int taskId = -1;
+
+			String timerStr = "forever";
+			if (timer >= 0) {
+				timerStr = timer / 20d + " seconds";
+
+				taskId = Bukkit.getScheduler().scheduleSyncDelayedTask(Main.plugin, () -> currentPlayers.remove(uid), timer);
+			}
+
+			currentPlayers.put(uid, taskId);
+			if (timer < 0) {
+				save();
+			}
+
+			send(sender, "&aBuddha mode &6enabled&a for &6%s&a for &6%s", getEntityName(p), timerStr);
+		}
+
+		return true;
+	}
+
+	private double getDamageFromEvent(EntityDamageEvent e) {
+		if (Main.majorVersion > 5) {
+			return e.getDamage();
+		}
+
+		try {
+			return (double)(int)EntityDamageEvent.class.getMethod("getDamage").invoke(e);
+		} catch (Exception ex) {
+			Bukkit.getLogger().severe(ex.toString());
+			return 0d;
+		}
+	}
+
+	private void setDamageFromEvent(EntityDamageEvent e, @SuppressWarnings("SameParameterValue") double d) {
+		if (Main.majorVersion > 5) {
+			e.setDamage(d);
+			return;
+		}
+
+		try {
+            //noinspection JavaReflectionMemberAccess, It's an int in older versions
+            EntityDamageEvent.class.getMethod("setDamage", int.class).invoke(e, (int) d);
+		} catch (Exception ex) {
+			Bukkit.getLogger().severe(ex.toString());
+		}
+	}
+
+	@EventHandler
+	public void damage(EntityDamageEvent e) {
+		if (!(e.getEntity() instanceof LivingEntity) || !currentPlayers.containsKey(e.getEntity().getUniqueId())) {
+			return;
+		}
+
+		LivingEntity entity = (LivingEntity) e.getEntity();
+
+		// Get all our vars since Minecraft broke everything in 1.6
+		double damage = getDamageFromEvent(e);
+		double health = getHealth(entity);
+		double maxHealth = getMaxHealth(entity);
+
+		if (damage < health) {  // Not lethal
+			return;
+		}
+
+		// Lethal hit (lethal company reference)
+		double extraDamage = damage - health;
+		double resultingDamageTaken = extraDamage % maxHealth;
+
+		setDamageFromEvent(e, 0);
+		setHealth(entity, maxHealth - resultingDamageTaken);
+	}
+
+	private static void save() {
+		FileConfiguration f = new YamlConfiguration();
+
+		List<String> buddhas = new ArrayList<>();
+		for (Map.Entry<UUID, Integer> kv : currentPlayers.entrySet()) {
+			// only save if there isn't a time limit!
+			if (kv.getValue() == -1) {
+				buddhas.add(kv.getKey().toString());
+			}
+		}
+
+		f.set("buddhas", buddhas);
+		ConfigManager.save("gods.yml", f);
+	}
+}

src/main/java/net/serble/estools/Main.java

@@ -113,6 +113,7 @@ public void onEnable() {
 		sc("editsign", "editsign", new EditSign());
 
 		sc("god", "god", new God(), 1);
+		sc("buddha", "god", new Buddha(), 1);
 
 		sc("music", "music", new Music(), 9);
 		sc("potion", "potion", new Potion(), 4);

src/main/resources/plugin.yml

@@ -133,6 +133,8 @@ commands:
     description: Make entities ride another entity.
   dismount:
     description: Make entities dismount other entities.
+  buddha:
+    description: Stop entities from dying from damage, their health will overflow back to full.
 
 permissions:
   estools.teleport: