TARDISArtronStorageCommand.java

/*
 * Copyright (C) 2016 eccentric_nz
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package me.eccentric_nz.TARDIS.artron;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import me.eccentric_nz.TARDIS.TARDIS;
import me.eccentric_nz.TARDIS.database.QueryFactory;
import me.eccentric_nz.TARDIS.database.ResultSetPlayerPrefs;
import me.eccentric_nz.TARDIS.database.ResultSetTardisArtron;
import me.eccentric_nz.TARDIS.enumeration.INVENTORY_MANAGER;
import me.eccentric_nz.TARDIS.utility.TARDISMessage;
import me.eccentric_nz.TARDIS.utility.TARDISNumberParsers;
import org.bukkit.command.Command;
import org.bukkit.command.CommandExecutor;
import org.bukkit.command.CommandSender;
import org.bukkit.enchantments.Enchantment;
import org.bukkit.entity.Player;
import org.bukkit.inventory.ItemFlag;
import org.bukkit.inventory.ItemStack;
import org.bukkit.inventory.meta.ItemMeta;

/**
 *
 * @author eccentric_nz
 */
public class TARDISArtronStorageCommand implements CommandExecutor {

    private final TARDIS plugin;
    private final List<String> firstArgs = new ArrayList<String>();

    public TARDISArtronStorageCommand(TARDIS plugin) {
        this.plugin = plugin;
        this.firstArgs.add("tardis");
        this.firstArgs.add("timelord");
    }

    @Override
    public boolean onCommand(CommandSender sender, Command cmd, String label, String[] args) {
        // If the player typed /tardisartron then do the following...
        // check there is the right number of arguments
        if (cmd.getName().equalsIgnoreCase("tardisartron")) {
            if (!sender.hasPermission("tardis.store")) {
                TARDISMessage.send(sender, "NO_PERMS");
                return true;
            }
            if (args.length < 2) {
                return false;
            }
            Player player = null;
            if (sender instanceof Player) {
                player = (Player) sender;
            }
            if (player == null) {
                TARDISMessage.send(sender, "CMD_PLAYER");
                return true;
            }
            ItemStack is = player.getInventory().getItemInMainHand();
            if (is == null || !is.hasItemMeta()) {
                TARDISMessage.send(player, "CELL_IN_HAND");
                return true;
            }
            if (is.getAmount() > 1) {
                TARDISMessage.send(player, "CELL_ONE");
                return true;
            }
            ItemMeta im = is.getItemMeta();
            String name = im.getDisplayName();
            if (name == null || !name.equals("Artron Storage Cell")) {
                TARDISMessage.send(player, "CELL_IN_HAND");
                return true;
            }
            String which = args[0].toLowerCase(Locale.ENGLISH);
            if (!firstArgs.contains(which)) {
                TARDISMessage.send(player, "CELL_WHICH");
                return false;
            }
            // must be a timelord
            String playerUUID = player.getUniqueId().toString();
            int current_level;
            if (which.equals("tardis")) {
                ResultSetTardisArtron rs = new ResultSetTardisArtron(plugin);
                if (!rs.fromUUID(playerUUID)) {
                    TARDISMessage.send(player, "NO_TARDIS");
                    return true;
                }
                current_level = rs.getArtron_level();
            } else {
                HashMap<String, Object> wheret = new HashMap<String, Object>();
                wheret.put("uuid", playerUUID);
                ResultSetPlayerPrefs rs = new ResultSetPlayerPrefs(plugin, wheret);
                if (!rs.resultSet()) {
                    TARDISMessage.send(player, "NO_TARDIS");
                    return true;
                }
                current_level = rs.getArtronLevel();
            }
            int amount;
            try {
                amount = Integer.parseInt(args[1]);
                if (amount < 0) {
                    TARDISMessage.send(player, "ENERGY_NOT_NEG");
                    return true;
                }
            } catch (NumberFormatException n) {
                TARDISMessage.send(player, "ARG_SEC_NUMBER");
                return false;
            }
            // must have sufficient energy
            if (which.equals("tardis")) {
                if (current_level - amount < plugin.getArtronConfig().getInt("comehere")) {
                    TARDISMessage.send(player, "CELL_NO_TRANSFER");
                    return true;
                }
            } else if (current_level - amount < 0) {
                TARDISMessage.send(player, "CELL_NOT_ENOUGH");
                return true;
            }
            List<String> lore = im.getLore();
            int level = TARDISNumberParsers.parseInt(lore.get(1));
            int new_amount = amount + level;
            int max = plugin.getArtronConfig().getInt("full_charge");
            if (new_amount > max) {
                TARDISMessage.send(player, "CELL_NO_CHARGE", String.format("%d", (max - level)));
                return false;
            }
            lore.set(1, "" + new_amount);
            im.setLore(lore);
            im.addEnchant(Enchantment.DURABILITY, 1, true);
            if (!plugin.getInvManager().equals(INVENTORY_MANAGER.MULTIVERSE)) {
                im.addItemFlags(ItemFlag.values());
            }
            is.setItemMeta(im);
            // remove the energy from the tardis/timelord
            HashMap<String, Object> where = new HashMap<String, Object>();
            String table;
            if (which.equals("tardis")) {
                where.put("uuid", playerUUID);
                table = "tardis";
            } else {
                where.put("uuid", playerUUID);
                table = "player_prefs";
            }
            new QueryFactory(plugin).alterEnergyLevel(table, -amount, where, player);
            TARDISMessage.send(player, "CELL_CHARGED", String.format("%d", new_amount));
            return true;
        }
        return false;
    }
}