Change the use of the disguiseclone command to use references instead

config.yml

@@ -62,6 +62,8 @@ DisguiseEntityExpire: 10
 
 # Another option to choose the same thing for DisguiseClone command
 DisguiseCloneExpire: 10
+# Max disguises to store at a time with the DisguiseClone command
+DisguiseCloneSize: 3
 
 # This I don't really recommend turning on as it can make a memory leak..
 # These disguises, as normal will not persist after a server restart.

src/me/libraryaddict/disguise/DisguiseConfig.java

@@ -19,6 +19,7 @@ public class DisguiseConfig {
     private static boolean keepDisguiseEntityDespawn;
     private static boolean keepDisguisePlayerDeath;
     private static boolean keepDisguisePlayerLogout;
+    private static int maxClonedDisguises;
     private static boolean maxHealthIsDisguisedEntity;
     private static boolean miscDisguisesForLivingEnabled;
     private static boolean modifyBoundingBox;
@@ -44,6 +45,10 @@ public class DisguiseConfig {
         return disguiseEntityExpire;
     }
 
+    public static int getMaxClonedDisguises() {
+        return maxClonedDisguises;
+    }
+
     public static boolean isAnimationPacketsEnabled() {
         return animationEnabled;
     }
@@ -263,6 +268,10 @@ public class DisguiseConfig {
         keepDisguisePlayerLogout = keepDisguise;
     }
 
+    public static void setMaxClonedDisguises(int newMax) {
+        maxClonedDisguises = newMax;
+    }
+
     public static void setMaxHealthDeterminedByDisguisedEntity(boolean isDetermined) {
         maxHealthIsDisguisedEntity = isDetermined;
     }

src/me/libraryaddict/disguise/DisguiseListener.java

@@ -1,6 +1,7 @@
 package me.libraryaddict.disguise;
 
 import java.util.HashMap;
+import java.util.Random;
 
 import me.libraryaddict.disguise.disguisetypes.Disguise;
 import me.libraryaddict.disguise.disguisetypes.PlayerDisguise;
@@ -127,7 +128,7 @@ public class DisguiseListener implements Listener {
             disguiseRunnable.remove(p.getName()).cancel();
             Entity entity = event.getRightClicked();
             String entityName = "";
-            if (entity instanceof Player) {
+            if (entity instanceof Player && !disguiseClone.containsKey(p.getName())) {
                 entityName = ((Player) entity).getName();
             } else {
                 String[] split = entity.getType().name().split("_");
@@ -138,36 +139,51 @@ public class DisguiseListener implements Listener {
                     }
                 }
             }
-            Disguise disguise = null;
-            Entity disguiseTarget = null;
             if (disguiseClone.containsKey(p.getName())) {
                 Boolean[] options = disguiseClone.remove(p.getName());
-                disguiseTarget = p;
+                Disguise disguise = DisguiseAPI.getDisguise(p, entity);
-                disguise = DisguiseAPI.getDisguise(p, entity);
                 if (disguise == null) {
                     disguise = DisguiseAPI.constructDisguise(entity, options[0], options[1], options[2]);
                 } else {
                     disguise = disguise.clone();
                 }
-            } else if (disguiseEntity.containsKey(p.getName())) {
+                char[] alphabet = "abcdefghijklmnopqrstuvwxyz".toCharArray();
-                disguiseTarget = entity;
+                String reference = null;
-                disguise = disguiseEntity.remove(p.getName());
+                int referenceLength = Math.max(2, (int) Math.ceil((0.1D + DisguiseConfig.getMaxClonedDisguises()) / 26D));
+                int attempts = 0;
+                while (reference == null && attempts++ < 1000) {
+                    reference = "@";
+                    for (int i = 0; i < referenceLength; i++) {
+                        reference += alphabet[new Random().nextInt(alphabet.length)];
                     }
+                    if (DisguiseUtilities.getClonedDisguise(reference) != null) {
+                        reference = null;
+                    }
+                }
+                if (reference != null && DisguiseUtilities.addClonedDisguise(reference, disguise)) {
+                    p.sendMessage(ChatColor.RED + "Constructed a " + entityName + " disguise! Your reference is " + reference);
+                    p.sendMessage(ChatColor.RED + "Example usage: /disguise " + reference);
+                } else {
+                    p.sendMessage(ChatColor.RED
+                            + "Failed to store the reference due to lack of size. Please set this in the config");
+                }
+            } else if (disguiseEntity.containsKey(p.getName())) {
+                Disguise disguise = disguiseEntity.remove(p.getName());
                 if (disguise != null) {
                     if (disguise.isMiscDisguise() && !DisguiseConfig.isMiscDisguisesForLivingEnabled()
-                        && disguiseTarget instanceof LivingEntity) {
+                            && entity instanceof LivingEntity) {
                         p.sendMessage(ChatColor.RED
                                 + "Can't disguise a living entity as a misc disguise. This has been disabled in the config!");
                     } else {
-                    if (disguiseTarget instanceof Player && DisguiseConfig.isNameOfPlayerShownAboveDisguise()) {
+                        if (entity instanceof Player && DisguiseConfig.isNameOfPlayerShownAboveDisguise()) {
                             if (disguise.getWatcher() instanceof LivingWatcher) {
-                            ((LivingWatcher) disguise.getWatcher()).setCustomName(((Player) disguiseTarget).getDisplayName());
+                                ((LivingWatcher) disguise.getWatcher()).setCustomName(((Player) entity).getDisplayName());
                                 if (DisguiseConfig.isNameAboveHeadAlwaysVisible()) {
                                     ((LivingWatcher) disguise.getWatcher()).setCustomNameVisible(true);
                                 }
                             }
                         }
-                    DisguiseAPI.disguiseToAll(disguiseTarget, disguise);
+                        DisguiseAPI.disguiseToAll(entity, disguise);
                         String disguiseName = "a ";
                         if (disguise instanceof PlayerDisguise) {
                             disguiseName = "the player " + ((PlayerDisguise) disguise).getName();
@@ -180,14 +196,9 @@ public class DisguiseListener implements Listener {
                                 }
                             }
                         }
-                    if (disguiseTarget == p) {
-                        p.sendMessage(ChatColor.RED + "Disguised yourself" + " as " + (entity instanceof Player ? "" : "a ")
-                                + entityName + "!");
-                    } else {
                         p.sendMessage(ChatColor.RED + "Disguised " + (entity instanceof Player ? "" : "the ") + entityName
                                 + " as " + disguiseName + "!");
                     }
-                }
                 } else {
                     if (DisguiseAPI.isDisguised(entity)) {
                         DisguiseAPI.undisguiseToAll(entity);
@@ -197,6 +208,7 @@ public class DisguiseListener implements Listener {
                 }
             }
         }
+    }
 
     @EventHandler
     public void onTarget(EntityTargetEvent event) {

src/me/libraryaddict/disguise/LibsDisguises.java

@@ -119,6 +119,7 @@ public class LibsDisguises extends JavaPlugin {
         DisguiseConfig.setMaxHealthDeterminedByDisguisedEntity(getConfig().getBoolean("MaxHealthDeterminedByEntity"));
         DisguiseConfig.setDisguiseEntityExpire(getConfig().getInt("DisguiseEntityExpire"));
         DisguiseConfig.setDisguiseCloneExpire(getConfig().getInt("DisguiseCloneExpire"));
+        DisguiseConfig.setMaxClonedDisguises(getConfig().getInt("DisguiseCloneSize"));
         try {
             // Here I use reflection to set the plugin for Disguise..
             // Kind of stupid but I don't want open API calls for a commonly used object.

src/me/libraryaddict/disguise/commands/DisguiseCloneCommand.java

@@ -45,7 +45,7 @@ public class DisguiseCloneCommand extends BaseDisguiseCommand {
             }
             listener.setDisguiseClone(sender.getName(), new Boolean[] { doEnquipment, doSneak, doSprint });
             sender.sendMessage(ChatColor.RED + "Right click a entity in the next " + DisguiseConfig.getDisguiseCloneExpire()
-                    + " seconds to disguise as it!");
+                    + " seconds to grab the disguise reference!");
         } else {
             sender.sendMessage(ChatColor.RED + "You are forbidden to use this command.");
         }
@@ -56,7 +56,9 @@ public class DisguiseCloneCommand extends BaseDisguiseCommand {
      * Send the player the information
      */
     protected void sendCommandUsage(CommandSender sender) {
-        sender.sendMessage(ChatColor.DARK_GREEN + "Disguise as the entity you right click! Or as their disguise!");
+        sender.sendMessage(ChatColor.DARK_GREEN
+                + "Right click a entity to get a disguise reference you can pass to other disguise commands!");
+        sender.sendMessage(ChatColor.DARK_GREEN + "Beware however, the reference bypasses all permissions checks");
         sender.sendMessage(ChatColor.DARK_GREEN + "/disguiseclone IgnoreEnquipment" + ChatColor.DARK_GREEN + "("
                 + ChatColor.GREEN + "Optional" + ChatColor.DARK_GREEN + ")");
     }

src/me/libraryaddict/disguise/utilities/BaseDisguiseCommand.java

@@ -166,6 +166,24 @@ public abstract class BaseDisguiseCommand implements CommandExecutor {
             sendCommandUsage(sender);
             throw new Exception();
         }
+        // How many args to skip due to the disugise being constructed
+        // Time to start constructing the disguise.
+        // We will need to check between all 3 kinds of disguises
+        int toSkip = 1;
+        ArrayList<String> usedOptions = new ArrayList<String>();
+        Disguise disguise = null;
+        HashSet<HashSet<String>> optionPermissions;
+        if (args[0].startsWith("@")) {
+            if (sender.hasPermission("libsdisguises.disguise.disguiseclone")) {
+                disguise = DisguiseUtilities.getClonedDisguise(args[0].toLowerCase());
+                if (disguise == null) {
+                    throw new Exception(ChatColor.RED + "Cannot find a disguise under the reference " + args[0]);
+                }
+            } else {
+                throw new Exception(ChatColor.RED + "You do not have perimssion to use disguise references!");
+            }
+            optionPermissions = this.getPermissions(sender, disguise.getType().name().toLowerCase());
+        } else {
             DisguiseType disguiseType = null;
             if (args[0].equalsIgnoreCase("p")) {
                 disguiseType = DisguiseType.PLAYER;
@@ -187,14 +205,7 @@ public abstract class BaseDisguiseCommand implements CommandExecutor {
             if (!allowedDisguises.contains(disguiseType.name().toLowerCase())) {
                 throw new Exception(ChatColor.RED + "You are forbidden to use this disguise.");
             }
-        ArrayList<String> usedOptions = new ArrayList<String>();
+            optionPermissions = this.getPermissions(sender, disguiseType.name().toLowerCase());
-        Disguise disguise = null;
-        // How many args to skip due to the disugise being constructed
-        int toSkip = 1;
-        // Time to start constructing the disguise.
-        // We will need to check between all 3 kinds of disguises
-
-        HashSet<HashSet<String>> optionPermissions = this.getPermissions(sender, disguiseType.name().toLowerCase());
             if (disguiseType.isPlayer()) {// If he is doing a player disguise
                 if (args.length == 1) {
                     // He needs to give the player name
@@ -251,6 +262,7 @@ public abstract class BaseDisguiseCommand implements CommandExecutor {
                     disguise = new MiscDisguise(disguiseType, miscId, miscData);
                 }
             }
+        }
         // Copy strings to their new range
         String[] newArgs = new String[args.length - toSkip];
         for (int i = toSkip; i < args.length; i++) {

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

@@ -7,10 +7,12 @@ import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.HashSet;
 import java.util.Iterator;
+import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.UUID;
 
 import me.libraryaddict.disguise.DisguiseAPI;
+import me.libraryaddict.disguise.DisguiseConfig;
 import me.libraryaddict.disguise.LibsDisguises;
 import me.libraryaddict.disguise.disguisetypes.Disguise;
 import me.libraryaddict.disguise.disguisetypes.DisguiseType;
@@ -19,6 +21,7 @@ import me.libraryaddict.disguise.disguisetypes.TargetedDisguise;
 import me.libraryaddict.disguise.disguisetypes.watchers.AgeableWatcher;
 import me.libraryaddict.disguise.disguisetypes.watchers.ZombieWatcher;
 import me.libraryaddict.disguise.disguisetypes.TargetedDisguise.TargetType;
+
 import org.bukkit.Bukkit;
 import org.bukkit.Location;
 import org.bukkit.Material;
@@ -42,6 +45,7 @@ public class DisguiseUtilities {
      * the plugin to do that.
      */
     private static HashSet<String> addedByPlugins = new HashSet<String>();
+    private static LinkedHashMap<String, Disguise> clonedDisguises = new LinkedHashMap<String, Disguise>();
     /**
      * A hashmap of the uuid's of entitys, alive and dead. And their disguises in use
      **/
@@ -63,6 +67,21 @@ public class DisguiseUtilities {
      **/
     private static HashMap<UUID, Integer> selfDisguisesIds = new HashMap<UUID, Integer>();
 
+    public static boolean addClonedDisguise(String key, Disguise disguise) {
+        if (DisguiseConfig.getMaxClonedDisguises() > 0) {
+            if (clonedDisguises.containsKey(key)) {
+                clonedDisguises.remove(key);
+            } else if (DisguiseConfig.getMaxClonedDisguises() == clonedDisguises.size()) {
+                clonedDisguises.remove(clonedDisguises.keySet().iterator().next());
+            }
+            if (DisguiseConfig.getMaxClonedDisguises() > clonedDisguises.size()) {
+                clonedDisguises.put(key, disguise);
+                return true;
+            }
+        }
+        return false;
+    }
+
     public static void addDisguise(UUID entityId, TargetedDisguise disguise) {
         if (!getDisguises().containsKey(entityId)) {
             getDisguises().put(entityId, new HashSet<TargetedDisguise>());
@@ -217,6 +236,13 @@ public class DisguiseUtilities {
         return addedByPlugins;
     }
 
+    public static Disguise getClonedDisguise(String key) {
+        if (clonedDisguises.containsKey(key)) {
+            return clonedDisguises.get(key);
+        }
+        return null;
+    }
+
     public static TargetedDisguise getDisguise(Player observer, Entity entity) {
         UUID entityId = entity.getUniqueId();
         if (futureDisguises.containsKey(entity.getEntityId())) {