HealCommand.java

package net.aufdemrand.denizen.scripts.commands.entity;

import java.util.ArrayList;
import java.util.List;

import net.aufdemrand.denizen.exceptions.CommandExecutionException;
import net.aufdemrand.denizen.exceptions.InvalidArgumentsException;
import net.aufdemrand.denizen.objects.Element;
import net.aufdemrand.denizen.objects.aH;
import net.aufdemrand.denizen.objects.dEntity;
import net.aufdemrand.denizen.objects.dList;
import net.aufdemrand.denizen.scripts.ScriptEntry;
import net.aufdemrand.denizen.scripts.commands.AbstractCommand;

/**
 * Heals an entity.
 *
 * @author Jeremy Schroeder, Mason Adkins, Morphan1
 */

public class HealCommand extends AbstractCommand {

    @Override
    public void parseArgs(ScriptEntry scriptEntry) throws InvalidArgumentsException {

        for (aH.Argument arg : aH.interpret(scriptEntry.getArguments())) {

            if (!scriptEntry.hasObject("amount")
            		&& (arg.matchesPrimitive(aH.PrimitiveType.Double)
            		|| arg.matchesPrimitive(aH.PrimitiveType.Integer)))
            	scriptEntry.addObject("amount", arg.asElement());

        	else if (!scriptEntry.hasObject("entities")
                	&& arg.matchesArgumentList(dEntity.class)) {
                scriptEntry.addObject("entities", ((dList) arg.asType(dList.class)).filter(dEntity.class));
            }
        }
        
        if (!scriptEntry.hasObject("amount"))
        	scriptEntry.addObject("amount", Integer.MAX_VALUE);
        
        if (!scriptEntry.hasObject("entities")) {
        	List<dEntity> entities = new ArrayList<dEntity>();
        	if (scriptEntry.getPlayer() != null)
        		entities.add(scriptEntry.getPlayer().getDenizenEntity());
        	else if (scriptEntry.getNPC() != null)
        		entities.add(scriptEntry.getNPC().getDenizenEntity());
        	else
        		throw new InvalidArgumentsException("No valid target entities found.");
        	scriptEntry.addObject("entities", entities);
        }
        
    }

    @SuppressWarnings("unchecked")
	@Override
    public void execute(ScriptEntry scriptEntry) throws CommandExecutionException {
    	
    	List<dEntity> entities = (List<dEntity>) scriptEntry.getObject("entities");
    		
    	if (scriptEntry.getObject("amount").equals(Integer.MAX_VALUE))
    		for (dEntity entity : entities)
    			entity.getLivingEntity().setHealth(entity.getLivingEntity().getMaxHealth());
    	else {
    		Double amount = ((Element) scriptEntry.getObject("amount")).asDouble();
    		for (dEntity entity : entities)
    			entity.getLivingEntity().setHealth(entity.getLivingEntity().getHealth() + amount);
    	}
    }
}