AgeCommand.java

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

import java.util.List;

import net.aufdemrand.denizen.objects.properties.EntityAge;
import org.bukkit.entity.Zombie;

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;
import net.aufdemrand.denizen.utilities.debugging.dB;


/**
 * Sets the ages of a list of entities, optionally locking them in those ages.
 *
 * @author David Cernat
 */
public class AgeCommand extends AbstractCommand {

    // <--[example]
    // @Title Age Command and Item Script/Item Event Example
    // @Description
    // This script shows a very small example that utilizes the Age Command with the use
    // of some items scripts and item events. Use /ex give i@aged_wand or /ex give i@baby_wand
    // and right click an entity.

    // @Code
    // # +--------------------
    // # | Age Command and Item Script/Item Event Example
    // # |
    // # | This script shows a very small example that utilizes the Age Command with the use
    // # | of some items scripts and item events. Use /ex give i@aged_wand or /ex give i@baby_wand
    // # | and right click an entity.
    //
    // Age Wands Handler:
    //   type: world
    //
    //   events:
    //
    //     # Check for the player right clicking entities with the baby_wand item
    //     on player right clicks entity with i@baby_wand:
    //
    //     # Play an effect, and run the age command on the entity
    //     - playeffect <c.entity.location> effect:mob_spell qty:100 data:1 offset:0.5
    //     - age <c.entity> baby
    //
    //     # ...and again for the aged_wand item
    //     on player right clicks entity with i@aged_wand:
    //     - playeffect <c.entity.location> effect:mob_spell qty:200 data:0 offset:0.5
    //     - age <c.entity>
    //
    //
    // # Build item script containers
    // baby_wand:
    //   type: item
    //
    //   material: blaze_rod
    //   display name: a baby wand
    //   lore:
    //   - "This wand is smooth as a baby's bottom."
    //
    // aged_wand:
    //   type: item
    //
    //   material: bone
    //   display name: an aged wand
    //   lore:
    //   - "This wand reeks of old age."
    //

    // -->


    private enum AgeType { ADULT, BABY }

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

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

            if (!scriptEntry.hasObject("entities")
                && arg.matchesArgumentList(dEntity.class)) {

                scriptEntry.addObject("entities", ((dList) arg.asType(dList.class)).filter(dEntity.class));
            }

            else if (!scriptEntry.hasObject("agetype")
                    && arg.matchesEnum(AgeType.values())) {
                // add Action
                scriptEntry.addObject("agetype", AgeType.valueOf(arg.getValue().toUpperCase()));
            }


            else if (!scriptEntry.hasObject("age")
                     && arg.matchesPrimitive(aH.PrimitiveType.Integer)) {

                scriptEntry.addObject("age", arg.asElement());
            }

            else if (!scriptEntry.hasObject("lock")
                     && arg.matches("lock")) {

                scriptEntry.addObject("lock", "");
            }

            else arg.reportUnhandled();
        }

        // Check to make sure required arguments have been filled

        if (!scriptEntry.hasObject("entities"))
            throw new InvalidArgumentsException("Must specify entity/entities!");

        // Use default duration if one is not specified

        scriptEntry.defaultObject("age", new Element(1));
    }

    @SuppressWarnings("unchecked")
    @Override
    public void execute(final ScriptEntry scriptEntry) throws CommandExecutionException {

        // Get objects
        List<dEntity> entities = (List<dEntity>) scriptEntry.getObject("entities");
        AgeType ageType = (AgeType) scriptEntry.getObject("agetype");
        int age = scriptEntry.getElement("age").asInt();
        boolean lock = scriptEntry.hasObject("lock");

        // Report to dB
        dB.report(scriptEntry, getName(), (lock ? aH.debugObj("lock", lock) : "") +
                             (ageType != null ? aH.debugObj("agetype", ageType)
                                              : aH.debugObj("age", age)) +
                             aH.debugObj("entities", entities.toString()));

        // Go through all the entities and set their ages
        for (dEntity entity : entities) {
            if (entity.isSpawned()) {

                // Check if entity specified can be described by 'EntityAge'
                if (EntityAge.describes(entity)) {

                    EntityAge property = EntityAge.getFrom(entity);

                    // Adjust 'ageType'
                    if (ageType != null) {
                        if (ageType.equals(AgeType.BABY))
                            property.setBaby(true);
                        else property.setBaby(false);

                    } else property.setAge(age);

                    // Adjust 'locked'
                    if (lock) property.setLock(true);
                }

                else dB.echoError(entity.identify() + " is not ageable!");

            }
        }

    }
}