MagicSpellReward.java

package com.elmakers.mine.bukkit.magicquests;

import com.elmakers.mine.bukkit.api.magic.Mage;
import com.elmakers.mine.bukkit.api.magic.MageClass;
import com.elmakers.mine.bukkit.api.magic.MagicAPI;

import me.pikamug.quests.module.BukkitCustomReward;
import org.apache.commons.lang.StringUtils;
import org.bukkit.Bukkit;
import org.bukkit.Server;
import org.bukkit.entity.Player;
import org.bukkit.inventory.ItemStack;
import org.bukkit.plugin.Plugin;

import java.util.Map;
import java.util.UUID;

public class MagicSpellReward extends BukkitCustomReward {

    // This boilerplate cannot be abstracted due to the class-scanning that Quests does
    private static MagicAPI api;
    protected static MagicAPI getAPI(Server server) {
        if (api == null) {
            Plugin magicPlugin = server.getPluginManager().getPlugin("Magic");
            if (magicPlugin != null && magicPlugin instanceof MagicAPI) {
                api = (MagicAPI)magicPlugin;
            }
        }

        return api;
    }

    public MagicSpellReward() {
        this.setName("Magic Spell");
        this.setAuthor("NathanWolf");
        this.setDisplay("Spell");
        this.addStringPrompt("Spell", "The KEY name of the Magic spell to give to the player.", null);
        this.addStringPrompt("Class", "The KEY name of the class (or a list of classes) to apply the spell to. If not set will be given to the player as an item.", null);
    }

    @Override
    public void giveReward(UUID uuid, Map<String, Object> stringObjectMap) {
        Player player = Bukkit.getPlayer(uuid);
        MagicAPI api = getAPI(player.getServer());
        String spellKey = (String)stringObjectMap.get("Spell");
        String classKeyList = (String)stringObjectMap.get("Class");
        if (spellKey != null) {
            if (classKeyList != null) {
                Mage mage = api.getController().getMage(player);
                String[] classKeys = StringUtils.split(classKeyList, ",");
                for (String classKey : classKeys) {
                    if (mage != null) {
                        MageClass mageClass = mage.getClass(classKey);
                        if (mageClass != null) {
                            mageClass.addSpell(spellKey);
                        }
                    }
                }
            } else {
                ItemStack item = api.createSpellItem(spellKey);
                if (item != null) {
                    api.giveItemToPlayer(player, item);
                } else {
                    Bukkit.getLogger().warning("Invalid spell given as Quests reward: " + spellKey);
                }
            }
        }
    }
}