org.bukkit.command.defaults.EnchantCommand Maven / Gradle / Ivy

Go to download
package org.bukkit.command.defaults;

import com.google.common.collect.ImmutableList;
import org.apache.commons.lang.Validate;
import org.apache.commons.lang.WordUtils;
import org.bukkit.Bukkit;
import org.bukkit.ChatColor;
import org.bukkit.Material;
import org.bukkit.command.Command;
import org.bukkit.command.CommandSender;
import org.bukkit.enchantments.Enchantment;
import org.bukkit.entity.Player;
import org.bukkit.inventory.ItemStack;
import org.bukkit.util.StringUtil;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;

@Deprecated
public class EnchantCommand extends VanillaCommand {
  private static final List ENCHANTMENT_NAMES = new ArrayList();

  public EnchantCommand() {
    super("enchant");
    this.description = "Adds enchantments to the item the player is currently holding. Specify 0 for the level to remove an enchantment. Specify force to ignore normal enchantment restrictions";
    this.usageMessage = "/enchant   [level|max|0] [force]";
    this.setPermission("bukkit.command.enchant");
  }

  public static void buildEnchantments() {
    if (!ENCHANTMENT_NAMES.isEmpty()) {
      throw new IllegalStateException("Enchantments have already been built!");
    }

    for (Enchantment enchantment : Enchantment.values()) {
      ENCHANTMENT_NAMES.add(enchantment.getName());
    }

    Collections.sort(ENCHANTMENT_NAMES);
  }

  @Override
  public boolean execute(CommandSender sender, String commandLabel, String[] args) {
    if (!testPermission(sender)) return true;
    if (args.length < 2) {
      sender.sendMessage(ChatColor.RED + "Usage: " + usageMessage);
      return false;
    }

    boolean force = false;
    if (args.length > 2) {
      force = args[args.length > 3 ? 3 : 2].equalsIgnoreCase("force");
    }

    Player player = Bukkit.getPlayerExact(args[0]);
    if (player == null) {
      sender.sendMessage("Can't find player " + args[0]);
    } else {
      ItemStack item = player.getItemInHand();
      if (item.getType() == Material.AIR) {
        sender.sendMessage("The player isn't holding an item");
      } else {
        String itemName = item.getType().toString().replaceAll("_", " ");
        itemName = WordUtils.capitalizeFully(itemName);

        Enchantment enchantment = getEnchantment(args[1].toUpperCase());
        if (enchantment == null) {
          sender.sendMessage(String.format("Enchantment does not exist: %s", args[1]));
        } else {
          String enchantmentName = enchantment.getName().replaceAll("_", " ");
          enchantmentName = WordUtils.capitalizeFully(enchantmentName);

          if (!force && !enchantment.canEnchantItem(item)) {
            sender.sendMessage(String.format("%s cannot be applied to %s", enchantmentName, itemName));
          } else {
            int level = 1;
            if (args.length > 2) {
              Integer integer = getInteger(args[2]);
              int minLevel = enchantment.getStartLevel();
              int maxLevel = force ? Short.MAX_VALUE : enchantment.getMaxLevel();

              if (integer != null) {
                if (integer == 0) {
                  item.removeEnchantment(enchantment);
                  Command.broadcastCommandMessage(sender, String.format("Removed %s on %s's %s", enchantmentName, player.getName(), itemName));
                  return true;
                }

                if (integer < minLevel || integer > maxLevel) {
                  sender.sendMessage(String.format("Level for enchantment %s must be between %d and %d", enchantmentName, minLevel, maxLevel));
                  sender.sendMessage("Specify 0 for level to remove an enchantment");
                  return true;
                }

                level = integer;
              }

              if ("max".equals(args[2])) {
                level = maxLevel;
              }
            }

            Map enchantments = item.getEnchantments();
            boolean conflicts = false;

            if (!force && !enchantments.isEmpty()) { // TODO: Improve this to use a "hasEnchantments" call
              for (Map.Entry entry : enchantments.entrySet()) {
                Enchantment enchant = entry.getKey();

                if (enchant.equals(enchantment)) continue;
                if (enchant.conflictsWith(enchantment)) {
                  sender.sendMessage(String.format("Can't apply the enchantment %s on an item with the enchantment %s", enchantmentName, WordUtils.capitalizeFully(enchant.getName().replaceAll("_", " "))));
                  conflicts = true;
                  break;
                }
              }
            }

            if (!conflicts) {
              item.addUnsafeEnchantment(enchantment, level);

              Command.broadcastCommandMessage(sender, String.format("Applied %s (Lvl %d) on %s's %s", enchantmentName, level, player.getName(), itemName), false);
              sender.sendMessage(String.format("Enchanting succeeded, applied %s (Lvl %d) onto your %s", enchantmentName, level, itemName));
            }
          }
        }
      }
    }
    return true;
  }

  @Override
  public List tabComplete(CommandSender sender, String alias, String[] args) throws IllegalArgumentException {
    Validate.notNull(sender, "Sender cannot be null");
    Validate.notNull(args, "Arguments cannot be null");
    Validate.notNull(alias, "Alias cannot be null");

    if (args.length == 1) {
      return super.tabComplete(sender, alias, args);
    }

    if (args.length == 2) {
      return StringUtil.copyPartialMatches(args[1], ENCHANTMENT_NAMES, new ArrayList(ENCHANTMENT_NAMES.size()));
    }

    if (args.length == 3 || args.length == 4) {
      if (!args[args.length - 2].equalsIgnoreCase("force")) {
        return ImmutableList.of("force");
      }
    }

    return ImmutableList.of();
  }

  private Enchantment getEnchantment(String lookup) {
    Enchantment enchantment = Enchantment.getByName(lookup);

    if (enchantment == null) {
      Integer id = getInteger(lookup);
      if (id != null) {
        enchantment = Enchantment.getById(id);
      }
    }

    return enchantment;
  }
}