EntityFramed.java

package net.aufdemrand.denizen.objects.properties.entity;

import net.aufdemrand.denizen.objects.dEntity;
import net.aufdemrand.denizen.objects.dItem;
import net.aufdemrand.denizen.utilities.debugging.dB;
import net.aufdemrand.denizencore.objects.Element;
import net.aufdemrand.denizencore.objects.Mechanism;
import net.aufdemrand.denizencore.objects.dList;
import net.aufdemrand.denizencore.objects.dObject;
import net.aufdemrand.denizencore.objects.properties.Property;
import net.aufdemrand.denizencore.tags.Attribute;
import net.aufdemrand.denizencore.utilities.CoreUtilities;
import org.bukkit.Material;
import org.bukkit.Rotation;
import org.bukkit.entity.EntityType;
import org.bukkit.entity.ItemFrame;

public class EntityFramed implements Property {

    // TODO: Possibly merge class with EntityItem?
    public static boolean describes(dObject entity) {
        return entity instanceof dEntity && ((dEntity) entity).getBukkitEntityType() == EntityType.ITEM_FRAME;
    }

    public static EntityFramed getFrom(dObject entity) {
        if (!describes(entity)) {
            return null;
        }
        else {
            return new EntityFramed((dEntity) entity);
        }
    }


    ///////////////////
    // Instance Fields and Methods
    /////////////

    private EntityFramed(dEntity item) {
        item_frame = item;
    }

    dEntity item_frame;

    public boolean hasItem() {
        return getItemFrameEntity().getItem() != null
                && getItemFrameEntity().getItem().getType() != Material.AIR;
    }

    public ItemFrame getItemFrameEntity() {
        return (ItemFrame) item_frame.getBukkitEntity();
    }

    public void setItem(dItem item) {
        getItemFrameEntity().setItem(item.getItemStack());
    }

    public dItem getItem() {
        return new dItem(getItemFrameEntity().getItem());
    }


    /////////
    // Property Methods
    ///////

    @Override
    public String getPropertyString() {
        if (hasItem()) {
            return getItem().identify()
                    + (getItemFrameEntity().getRotation() == Rotation.NONE ? ""
                    : '|' + CoreUtilities.toLowerCase(getItemFrameEntity().getRotation().name()));
        }
        else {
            return null;
        }
    }

    @Override
    public String getPropertyId() {
        return "framed";
    }


    ///////////
    // dObject Attributes
    ////////

    @Override
    public String getAttribute(Attribute attribute) {

        if (attribute == null) {
            return null;
        }

        // <--[tag]
        // @attribute <e@entity.framed_item_rotation>
        // @returns Element
        // @mechanism dEntity.framed
        // @group properties
        // @description
        // If the entity is an item frame, returns the rotation of the material currently framed.
        // -->
        if (attribute.startsWith("framed_item_rotation")) {
            return new Element(CoreUtilities.toLowerCase(getItemFrameEntity().getRotation().name()))
                    .getAttribute(attribute.fulfill(1));
        }

        // <--[tag]
        // @attribute <e@entity.framed_item>
        // @returns dItem
        // @mechanism dEntity.framed
        // @group properties
        // @description
        // If the entity is an item frame, returns the material currently framed.
        // -->
        if (attribute.startsWith("framed_item")) {
            return getItem()
                    .getAttribute(attribute.fulfill(1));
        }

        // <--[tag]
        // @attribute <e@entity.has_framed_item>
        // @returns Element(Boolean)
        // @mechanism dEntity.framed
        // @group properties
        // @description
        // If the entity is an item frame, returns whether the frame has an item in it.
        // -->
        if (attribute.startsWith("has_framed_item")) {
            return new Element(hasItem())
                    .getAttribute(attribute.fulfill(1));
        }

        return null;
    }

    @Override
    public void adjust(Mechanism mechanism) {

        // <--[mechanism]
        // @object dEntity
        // @name framed
        // @input dItem(|Element)
        // @description
        // Sets the entity's framed item and optionally the rotation as well.
        // Valid rotations: NONE, CLOCKWISE, FLIPPED, COUNTER_CLOCKWISE
        // For example: framed:i@diamond_sword|clockwise
        // @tags
        // <e@entity.is_frame>
        // <e@entity.has_framed_item>
        // <e@entity.framed_item>
        // <e@entity.framed_item_rotation>
        // -->

        if (mechanism.matches("framed")) {
            dList list = mechanism.getValue().asType(dList.class);
            if (list.size() == 0) {
                dB.echoError("Missing value for 'framed' mechanism!");
                return;
            }
            if (new Element(list.get(0)).matchesType(dItem.class)) {
                setItem(new Element(list.get(0)).asType(dItem.class));
            }
            else {
                dB.echoError("Invalid item '" + list.get(0) + "'");
            }
            if (list.size() > 1 && new Element(list.get(1)).matchesEnum(Rotation.values())) {
                getItemFrameEntity().setRotation(Rotation.valueOf(list.get(1).toUpperCase()));
            }
        }

    }
}