From 0b5e20180d31993de1ff7912a986b103af2a2c3c Mon Sep 17 00:00:00 2001
From: libraryaddict <libraryaddict115@yahoo.co.nz>
Date: Wed, 14 May 2014 10:54:44 +1200
Subject: [PATCH] Fixed disguise entity command throwing errors. Cleaned up the
 messages moar

---
 .../disguise/DisguiseListener.java            | 26 ++++++++++++++++---
 1 file changed, 22 insertions(+), 4 deletions(-)

diff --git a/src/me/libraryaddict/disguise/DisguiseListener.java b/src/me/libraryaddict/disguise/DisguiseListener.java
index bcb84755..8787ac5d 100644
--- a/src/me/libraryaddict/disguise/DisguiseListener.java
+++ b/src/me/libraryaddict/disguise/DisguiseListener.java
@@ -3,6 +3,7 @@ package me.libraryaddict.disguise;
 import java.util.HashMap;
 
 import me.libraryaddict.disguise.disguisetypes.Disguise;
+import me.libraryaddict.disguise.disguisetypes.PlayerDisguise;
 import me.libraryaddict.disguise.disguisetypes.TargetedDisguise;
 import me.libraryaddict.disguise.disguisetypes.watchers.LivingWatcher;
 import me.libraryaddict.disguise.utilities.DisguiseUtilities;
@@ -123,12 +124,17 @@ public class DisguiseListener implements Listener {
             Disguise disguise = disguiseSlap.remove(event.getPlayer().getName());
             disguiseRunnable.remove(event.getPlayer().getName()).cancel();
             Entity entity = event.getRightClicked();
-            String entityName = entity.getType().name().toLowerCase().replace("_", " ");
+            String entityName = "";
             if (entity instanceof Player) {
                 entityName = ((Player) entity).getName();
             } else {
                 String[] split = entity.getType().name().split("_");
-                entityName = split[0].substring(1).toLowerCase() + " " + split[1].substring(1).toLowerCase();
+                for (int i = 0; i < split.length; i++) {
+                    entityName += split[0].substring(0, 1) + split[0].substring(1).toLowerCase();
+                    if (i + 1 < split.length) {
+                        entityName += " ";
+                    }
+                }
             }
             if (disguise != null) {
                 if (entity instanceof Player && DisguiseConfig.isNameOfPlayerShownAboveDisguise()) {
@@ -140,9 +146,21 @@ public class DisguiseListener implements Listener {
                     }
                 }
                 DisguiseAPI.disguiseToAll(entity, disguise);
+                String disguiseName = "a ";
+                if (disguise instanceof PlayerDisguise) {
+                    disguiseName = "the player " + ((PlayerDisguise) disguise).getName();
+                } else {
+                    String[] split = disguise.getType().name().split("_");
+                    for (int i = 0; i < split.length; i++) {
+                        disguiseName += split[0].substring(0, 1) + split[0].substring(1).toLowerCase();
+                        if (i + 1 < split.length) {
+                            disguiseName += " ";
+                        }
+                    }
+                }
                 event.getPlayer().sendMessage(
-                        ChatColor.RED + "Disguised " + (entity instanceof Player ? "" : "the ") + entityName + " as a "
-                                + disguise.getType().name().toLowerCase().replace("_", " ") + "!");
+                        ChatColor.RED + "Disguised " + (entity instanceof Player ? "" : "the ") + entityName + " as "
+                                + disguiseName + "!");
             } else {
                 if (DisguiseAPI.isDisguised(entity)) {
                     DisguiseAPI.undisguiseToAll(entity);