/src/main/java/com/gmail/nossr50/commands/experience/SkillresetCommand.java

https://gitlab.com/Slind/mcMMO · Java · 171 lines · 134 code · 32 blank · 5 comment · 28 complexity · f1e7aa131582210660c28d3312ef7da4 MD5 · raw file

  1. package com.gmail.nossr50.commands.experience;
  2. import java.util.ArrayList;
  3. import java.util.List;
  4. import java.util.Set;
  5. import org.bukkit.command.Command;
  6. import org.bukkit.command.CommandSender;
  7. import org.bukkit.command.TabExecutor;
  8. import org.bukkit.entity.Player;
  9. import org.bukkit.util.StringUtil;
  10. import com.gmail.nossr50.mcMMO;
  11. import com.gmail.nossr50.datatypes.player.McMMOPlayer;
  12. import com.gmail.nossr50.datatypes.player.PlayerProfile;
  13. import com.gmail.nossr50.datatypes.skills.SkillType;
  14. import com.gmail.nossr50.locale.LocaleLoader;
  15. import com.gmail.nossr50.util.EventUtils;
  16. import com.gmail.nossr50.util.Permissions;
  17. import com.gmail.nossr50.util.commands.CommandUtils;
  18. import com.gmail.nossr50.util.player.UserManager;
  19. import com.google.common.collect.ImmutableList;
  20. /**
  21. * This class mirrors the structure of ExperienceCommand, except the
  22. * value/quantity argument is removed.
  23. */
  24. public class SkillresetCommand implements TabExecutor {
  25. @Override
  26. public boolean onCommand(CommandSender sender, Command command, String label, String[] args) {
  27. switch (args.length) {
  28. case 1:
  29. if (CommandUtils.noConsoleUsage(sender)) {
  30. return true;
  31. }
  32. if (!permissionsCheckSelf(sender)) {
  33. sender.sendMessage(command.getPermissionMessage());
  34. return true;
  35. }
  36. if (!validateArguments(sender, args[0])) {
  37. return true;
  38. }
  39. editValues((Player) sender, UserManager.getPlayer(sender.getName()).getProfile(), SkillType.getSkill(args[0]));
  40. return true;
  41. case 2:
  42. if (!permissionsCheckOthers(sender)) {
  43. sender.sendMessage(command.getPermissionMessage());
  44. return true;
  45. }
  46. if (!validateArguments(sender, args[1])) {
  47. return true;
  48. }
  49. SkillType skill;
  50. if (args[1].equalsIgnoreCase("all")) {
  51. skill = null;
  52. }
  53. else {
  54. skill = SkillType.getSkill(args[1]);
  55. }
  56. String playerName = CommandUtils.getMatchedPlayerName(args[0]);
  57. McMMOPlayer mcMMOPlayer = UserManager.getPlayer(playerName, true);
  58. // If the mcMMOPlayer doesn't exist, create a temporary profile and check if it's present in the database. If it's not, abort the process.
  59. if (mcMMOPlayer == null) {
  60. PlayerProfile profile = mcMMO.getDatabaseManager().loadPlayerProfile(playerName, false);
  61. if (CommandUtils.unloadedProfile(sender, profile)) {
  62. return true;
  63. }
  64. editValues(null, profile, skill);
  65. }
  66. else {
  67. editValues(mcMMOPlayer.getPlayer(), mcMMOPlayer.getProfile(), skill);
  68. }
  69. handleSenderMessage(sender, playerName, skill);
  70. return true;
  71. default:
  72. return false;
  73. }
  74. }
  75. @Override
  76. public List<String> onTabComplete(CommandSender sender, Command command, String alias, String[] args) {
  77. switch (args.length) {
  78. case 1:
  79. Set<String> playerNames = UserManager.getPlayerNames();
  80. return StringUtil.copyPartialMatches(args[0], playerNames, new ArrayList<String>(playerNames.size()));
  81. case 2:
  82. return StringUtil.copyPartialMatches(args[1], SkillType.SKILL_NAMES, new ArrayList<String>(SkillType.SKILL_NAMES.size()));
  83. default:
  84. return ImmutableList.of();
  85. }
  86. }
  87. protected void handleCommand(Player player, PlayerProfile profile, SkillType skill) {
  88. int levelsRemoved = profile.getSkillLevel(skill);
  89. float xpRemoved = profile.getSkillXpLevelRaw(skill);
  90. profile.modifySkill(skill, 0);
  91. if (player == null) {
  92. profile.save();
  93. return;
  94. }
  95. EventUtils.handleLevelChangeEvent(player, skill, levelsRemoved, xpRemoved, false);
  96. }
  97. protected boolean permissionsCheckSelf(CommandSender sender) {
  98. return Permissions.skillreset(sender);
  99. }
  100. protected boolean permissionsCheckOthers(CommandSender sender) {
  101. return Permissions.skillresetOthers(sender);
  102. }
  103. protected void handlePlayerMessageAll(Player player) {
  104. player.sendMessage(LocaleLoader.getString("Commands.Reset.All"));
  105. }
  106. protected void handlePlayerMessageSkill(Player player, SkillType skill) {
  107. player.sendMessage(LocaleLoader.getString("Commands.Reset.Single", skill.getName()));
  108. }
  109. private boolean validateArguments(CommandSender sender, String skillName) {
  110. if (CommandUtils.isInvalidSkill(sender, skillName) && !skillName.equalsIgnoreCase("all")) {
  111. return false;
  112. }
  113. return true;
  114. }
  115. protected static void handleSenderMessage(CommandSender sender, String playerName, SkillType skill) {
  116. if (skill == null) {
  117. sender.sendMessage(LocaleLoader.getString("Commands.addlevels.AwardAll.2", playerName));
  118. }
  119. else {
  120. sender.sendMessage(LocaleLoader.getString("Commands.addlevels.AwardSkill.2", skill.getName(), playerName));
  121. }
  122. }
  123. protected void editValues(Player player, PlayerProfile profile, SkillType skill) {
  124. if (skill == null) {
  125. for (SkillType skillType : SkillType.NON_CHILD_SKILLS) {
  126. handleCommand(player, profile, skillType);
  127. }
  128. if (player != null) {
  129. handlePlayerMessageAll(player);
  130. }
  131. }
  132. else {
  133. handleCommand(player, profile, skill);
  134. if (player != null) {
  135. handlePlayerMessageSkill(player, skill);
  136. }
  137. }
  138. }
  139. }