Clean up the disguise runnable a bit for some more sanity, it's so ugly.

src/main/java/me/libraryaddict/disguise/LibsDisguises.java

@@ -171,6 +171,10 @@ public class LibsDisguises extends JavaPlugin {
         new MetricsInitalizer();
     }
 
+    private void startTask() {
+
+    }
+
     public void unregisterCommands(boolean force) {
         CommandMap map = ReflectionManager.getCommandMap();
         Map<String, Command> commands = ReflectionManager.getCommands(map);

src/main/java/me/libraryaddict/disguise/disguisetypes/Disguise.java

@@ -36,6 +36,7 @@ import org.bukkit.boss.BarStyle;
 import org.bukkit.boss.BossBar;
 import org.bukkit.entity.*;
 import org.bukkit.metadata.FixedMetadataValue;
+import org.bukkit.scheduler.BukkitRunnable;
 import org.bukkit.scheduler.BukkitTask;
 import org.bukkit.util.Vector;
 
@@ -57,6 +58,7 @@ public abstract class Disguise {
 
     private transient boolean disguiseInUse;
     private DisguiseType disguiseType;
+    private transient BukkitRunnable runnable;
     private transient Entity entity;
     private boolean hearSelfDisguise = DisguiseConfig.isSelfDisguisesSoundsReplaced();
     private boolean hideArmorFromSelf = DisguiseConfig.isHidingArmorFromSelf();
@@ -66,8 +68,6 @@ public abstract class Disguise {
     private boolean playerHiddenFromTab = DisguiseConfig.isHideDisguisedPlayers();
     private boolean replaceSounds = DisguiseConfig.isSoundEnabled();
     private boolean mobsIgnoreDisguise;
-    private transient BukkitTask task;
-    private Runnable velocityRunnable;
     private boolean velocitySent = DisguiseConfig.isVelocitySent();
     private boolean viewSelfDisguise = DisguiseConfig.isViewDisguises();
     @Getter
@@ -350,7 +350,36 @@ public abstract class Disguise {
         return this;
     }
 
+    private void doActionBar() {
+        if (getNotifyBar() == DisguiseConfig.NotifyBar.ACTION_BAR && getEntity() instanceof Player &&
+                !getEntity().hasPermission("libsdisguises.noactionbar") &&
+                DisguiseAPI.getDisguise(getEntity()) == Disguise.this) {
+            ((Player) getEntity()).spigot().sendMessage(ChatMessageType.ACTION_BAR,
+                    new ComponentBuilder(LibsMsg.ACTION_BAR_MESSAGE.get(getDisguiseName())).create());
+        }
+
+        if (isDynamicName()) {
+            String name = getEntity().getCustomName();
+
+            if (name == null) {
+                name = "";
+            }
+
+            if (isPlayerDisguise()) {
+                if (!((PlayerDisguise) Disguise.this).getName().equals(name)) {
+                    ((PlayerDisguise) Disguise.this).setName(name);
+                }
+            } else {
+                getWatcher().setCustomName(name);
+            }
+        }
+    }
+
     private void createRunnable() {
+        if (runnable != null) {
+            runnable.cancel();
+        }
+
         final boolean alwaysSendVelocity;
 
         switch (getType()) {
@@ -382,39 +411,24 @@ public abstract class Disguise {
         final TargetedDisguise disguise = (TargetedDisguise) this;
 
         // A scheduler to clean up any unused disguises.
-        velocityRunnable = new Runnable() {
+        runnable = new BukkitRunnable() {
             private int blockX, blockY, blockZ, facing;
             private int deadTicks = 0;
-            private int refreshDisguise = 0;
             private int actionBarTicks = -1;
+            private long lastRefreshed;
 
             @Override
             public void run() {
+                if (!isDisguiseInUse() || getEntity() == null) {
+                    runnable.cancel();
+                    runnable = null;
+                    return;
+                }
+
                 if (++actionBarTicks % 15 == 0) {
                     actionBarTicks = 0;
 
-                    if (getNotifyBar() == DisguiseConfig.NotifyBar.ACTION_BAR && getEntity() instanceof Player &&
+                    doActionBar();
-                            !getEntity().hasPermission("libsdisguises.noactionbar") &&
-                            DisguiseAPI.getDisguise(getEntity()) == Disguise.this) {
-                        ((Player) getEntity()).spigot().sendMessage(ChatMessageType.ACTION_BAR,
-                                new ComponentBuilder(LibsMsg.ACTION_BAR_MESSAGE.get(getDisguiseName())).create());
-                    }
-
-                    if (isDynamicName()) {
-                        String name = getEntity().getCustomName();
-
-                        if (name == null) {
-                            name = "";
-                        }
-
-                        if (isPlayerDisguise()) {
-                            if (!((PlayerDisguise) Disguise.this).getName().equals(name)) {
-                                ((PlayerDisguise) Disguise.this).setName(name);
-                            }
-                        } else {
-                            getWatcher().setCustomName(name);
-                        }
-                    }
                 }
 
                 // If entity is no longer valid. Remove it.
@@ -429,6 +443,8 @@ public abstract class Disguise {
                     if (getEntity() instanceof Player && expired.length() > 0) {
                         getEntity().sendMessage(expired);
                     }
+
+                    return;
                 } else if (!getEntity().isValid()) {
                     // If it has been dead for 30+ ticks
                     // This is to ensure that this disguise isn't removed while clients think its the real entity
@@ -436,43 +452,21 @@ public abstract class Disguise {
                     // death animation
                     // This is probably still a problem for wither and enderdragon deaths.
                     if (deadTicks++ > (getType() == DisguiseType.ENDER_DRAGON ? 200 : 20)) {
-                        deadTicks = 0;
-
                         if (isRemoveDisguiseOnDeath()) {
                             removeDisguise();
                         }
                     }
-                } else {
+
+                    return;
+                }
+
                 deadTicks = 0;
 
                 // If the disguise type is tnt, we need to resend the entity packet else it will turn invisible
-                    if (getType() == DisguiseType.FIREWORK) {
+                if (getType() == DisguiseType.FIREWORK || getType() == DisguiseType.EVOKER_FANGS) {
-                        refreshDisguise++;
+                    if (lastRefreshed < System.currentTimeMillis()) {
-
+                        lastRefreshed =
-                        if (refreshDisguise == 40) {
+                                System.currentTimeMillis() + ((getType() == DisguiseType.FIREWORK ? 40 : 23) * 50);
-                            refreshDisguise = 0;
-
-                            DisguiseUtilities.refreshTrackers(disguise);
-                        }
-                    } else if (getType() == DisguiseType.EVOKER_FANGS) {
-                        refreshDisguise++;
-
-                        if (refreshDisguise == 23) {
-                            refreshDisguise = 0;
-
-                            DisguiseUtilities.refreshTrackers(disguise);
-                        }
-                    } else if (getType() == DisguiseType.ITEM_FRAME) {
-                        Location loc = getEntity().getLocation();
-
-                        int newFacing = (((int) loc.getYaw() + 720 + 45) / 90) % 4;
-
-                        if (loc.getBlockX() != blockX || loc.getBlockY() != blockY || loc.getBlockZ() != blockZ ||
-                                newFacing != facing) {
-                            blockX = loc.getBlockX();
-                            blockY = loc.getBlockY();
-                            blockZ = loc.getBlockZ();
-                            facing = newFacing;
 
                         DisguiseUtilities.refreshTrackers(disguise);
                     }
@@ -486,6 +480,44 @@ public abstract class Disguise {
                     return;
                 }
 
+                doVelocity(vectorY, alwaysSendVelocity);
+
+                if (getType() == DisguiseType.EXPERIENCE_ORB) {
+                    PacketContainer packet = new PacketContainer(Server.REL_ENTITY_MOVE);
+
+                    packet.getIntegers().write(0, getEntity().getEntityId());
+                    try {
+                        for (Player player : DisguiseUtilities.getPerverts(disguise)) {
+                            if (getEntity() != player) {
+                                ProtocolLibrary.getProtocolManager().sendServerPacket(player, packet, false);
+                            } else if (!isSelfDisguiseVisible()) {
+                                continue;
+                            }
+
+                            PacketContainer selfPacket = packet.shallowClone();
+
+                            selfPacket.getModifier().write(0, DisguiseAPI.getSelfDisguiseId());
+
+                            try {
+                                ProtocolLibrary.getProtocolManager()
+                                        .sendServerPacket((Player) getEntity(), selfPacket, false);
+                            }
+                            catch (InvocationTargetException e) {
+                                e.printStackTrace();
+                            }
+                        }
+                    }
+                    catch (InvocationTargetException e) {
+                        e.printStackTrace();
+                    }
+                }
+            }
+        };
+
+        runnable.runTaskTimer(LibsDisguises.getInstance(), 1, 1);
+    }
+
+    private void doVelocity(Double vectorY, boolean alwaysSendVelocity) {
         // If the vectorY isn't 0. Cos if it is. Then it doesn't want to send any vectors.
         // If this disguise has velocity sending enabled and the entity is flying.
         if (isVelocitySent() && vectorY != null && (alwaysSendVelocity || !getEntity().isOnGround())) {
@@ -495,8 +527,7 @@ public abstract class Disguise {
             // head about the
             // if statement.
             // But it doesn't seem to do anything wrong..
-                        if (vector.getY() != 0 &&
+            if (vector.getY() != 0 && !(vector.getY() < 0 && alwaysSendVelocity && getEntity().isOnGround())) {
-                                !(vector.getY() < 0 && alwaysSendVelocity && getEntity().isOnGround())) {
                 return;
             }
 
@@ -504,8 +535,7 @@ public abstract class Disguise {
             if (getType() != DisguiseType.EXPERIENCE_ORB || !getEntity().isOnGround()) {
                 PacketContainer lookPacket = null;
 
-                            if (getType() == DisguiseType.WITHER_SKULL &&
+                if (getType() == DisguiseType.WITHER_SKULL && DisguiseConfig.isWitherSkullPacketsEnabled()) {
-                                    DisguiseConfig.isWitherSkullPacketsEnabled()) {
                     lookPacket = new PacketContainer(Server.ENTITY_LOOK);
 
                     StructureModifier<Object> mods = lookPacket.getModifier();
@@ -542,7 +572,7 @@ public abstract class Disguise {
                     mods.write(1, (int) (vector.getX() * 8000));
                     mods.write(3, (int) (vector.getZ() * 8000));
 
-                                for (Player player : DisguiseUtilities.getPerverts(disguise)) {
+                    for (Player player : DisguiseUtilities.getPerverts(this)) {
                         PacketContainer tempVelocityPacket = velocityPacket.shallowClone();
                         mods = tempVelocityPacket.getIntegers();
 
@@ -557,16 +587,13 @@ public abstract class Disguise {
                             mods.write(0, DisguiseAPI.getSelfDisguiseId());
                         }
 
-                                    mods.write(2,
+                        mods.write(2, (int) (8000D * (vectorY * ReflectionManager.getPing(player)) * 0.069D));
-                                            (int) (8000D * (vectorY * ReflectionManager.getPing(player)) * 0.069D));
 
                         if (lookPacket != null && player != getEntity()) {
-                                        ProtocolLibrary.getProtocolManager()
+                            ProtocolLibrary.getProtocolManager().sendServerPacket(player, lookPacket, false);
-                                                .sendServerPacket(player, lookPacket, false);
                         }
 
-                                    ProtocolLibrary.getProtocolManager()
+                        ProtocolLibrary.getProtocolManager().sendServerPacket(player, tempVelocityPacket, false);
-                                            .sendServerPacket(player, tempVelocityPacket, false);
                     }
                 }
                 catch (Exception e) {
@@ -577,36 +604,6 @@ public abstract class Disguise {
             // sided to
             // players.
         }
-                    if (getType() == DisguiseType.EXPERIENCE_ORB) {
-                        PacketContainer packet = new PacketContainer(Server.REL_ENTITY_MOVE);
-
-                        packet.getIntegers().write(0, getEntity().getEntityId());
-                        try {
-                            for (Player player : DisguiseUtilities.getPerverts(disguise)) {
-                                if (getEntity() != player) {
-                                    ProtocolLibrary.getProtocolManager().sendServerPacket(player, packet, false);
-                                } else if (isSelfDisguiseVisible()) {
-                                    PacketContainer selfPacket = packet.shallowClone();
-
-                                    selfPacket.getModifier().write(0, DisguiseAPI.getSelfDisguiseId());
-
-                                    try {
-                                        ProtocolLibrary.getProtocolManager()
-                                                .sendServerPacket((Player) getEntity(), selfPacket, false);
-                                    }
-                                    catch (InvocationTargetException e) {
-                                        e.printStackTrace();
-                                    }
-                                }
-                            }
-                        }
-                        catch (InvocationTargetException e) {
-                            e.printStackTrace();
-                        }
-                    }
-                }
-            }
-        };
     }
 
     /**
@@ -821,8 +818,8 @@ public abstract class Disguise {
      * Internal use
      */
     public boolean isRemoveDisguiseOnDeath() {
-        return getEntity() == null ||
+        return getEntity() == null || (getEntity() instanceof Player ? !isKeepDisguiseOnPlayerDeath() :
-                (getEntity() instanceof Player ? !isKeepDisguiseOnPlayerDeath() : getEntity().isDead());
+                getEntity().isDead() || !getEntity().isValid());
     }
 
     @Deprecated
@@ -886,9 +883,9 @@ public abstract class Disguise {
 
         disguiseInUse = false;
 
-        if (task != null) {
+        if (runnable != null) {
-            task.cancel();
+            runnable.cancel();
-            task = null;
+            runnable = null;
         }
 
         // If this disguise hasn't a entity set
@@ -1122,13 +1119,8 @@ public abstract class Disguise {
 
         disguiseInUse = true;
 
-        if (velocityRunnable == null) {
-            createRunnable();
-        }
-
         if (!DisguiseUtilities.isInvalidFile()) {
-            task = Bukkit.getScheduler().
+            createRunnable();
-                    runTaskTimer(LibsDisguises.getInstance(), velocityRunnable, 1, 1);
         }
 
         if (this instanceof PlayerDisguise) {

src/main/java/me/libraryaddict/disguise/utilities/DisguiseUtilities.java

@@ -1280,6 +1280,10 @@ public class DisguiseUtilities {
         file.delete();
     }
 
+    public static void doDisguiseTick(Disguise disguise) {
+
+    }
+
     public static void removeSelfDisguise(Disguise disguise) {
         if (!Bukkit.isPrimaryThread())
             throw new IllegalStateException("Cannot modify disguises on an async thread");