feat: format entity name

TeamVastsea · Jul 11, 2024 · f750e42 · f750e42
1 parent 87d8800
commit f750e42
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/main/java/cc/vastsea/healthbar/EntityDamageListener.java b/src/main/java/cc/vastsea/healthbar/EntityDamageListener.java
@@ -69,7 +69,7 @@ private void setBossBar(BossBar bossBar, Entity entity, double damage) {
             health = 0;
         }
 
-        String name = entity.getCustomName() == null ? entity.getType().name().toLowerCase() : entity.getCustomName();
+        String name = entity.getCustomName() == null ? formatEntityName(entity.getType().name()) : entity.getCustomName();
 
         if (progress > 0.6) {
             bossBar.setColor(BarColor.GREEN);
@@ -102,6 +102,16 @@ private boolean isMonitoredEntity(Entity entity) {
         return !(entity instanceof Player) && !(entity instanceof Monster) && !(entity instanceof Animals) && !(entity instanceof EnderDragon);
     }
 
+    private String formatEntityName(String name) {
+        name = name.toLowerCase().replace("_", " ");
+        String[] words = name.split(" ");
+        StringBuilder formattedName = new StringBuilder();
+        for (String word : words) {
+            formattedName.append(Character.toUpperCase(word.charAt(0))).append(word.substring(1)).append(" ");
+        }
+        return formattedName.toString().trim();
+    }
+
     class BossBarRecord {
         BossBar bossBar;