add enderdragon phase tag/mechanism

DenizenScript · Nov 25, 2016 · eebe9b4 · eebe9b4
1 parent d805164
commit eebe9b4
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/plugin/src/main/java/net/aufdemrand/denizen/objects/dEntity.java b/plugin/src/main/java/net/aufdemrand/denizen/objects/dEntity.java
@@ -2429,6 +2429,21 @@ && getBukkitEntity() instanceof Item) {
                     .getAttribute(attribute.fulfill(1));
         }
 
+        // <--[tag]
+        // @attribute <e@entity.dragon_phase>
+        // @returns Element(Number)
+        // @group properties
+        // @description
+        // Returns the phase an EnderDragon is currently in.
+        // Valid phases: CIRCLING, STRAFING, FLY_TO_PORTAL, LAND_ON_PORTAL, LEAVE_PORTAL,
+        // BREATH_ATTACK, SEARCH_FOR_BREATH_ATTACK_TARGET, ROAR_BEFORE_ATTACK,
+        // CHARGE_PLAYER, DYING, HOVER;
+        // -->
+        if (attribute.startsWith("experience") && getBukkitEntity() instanceof ExperienceOrb) {
+            return new Element(((EnderDragon) getLivingEntity()).getPhase().name())
+                    .getAttribute(attribute.fulfill(1));
+        }
+
         // <--[tag]
         // @attribute <e@entity.describe>
         // @returns Element(Boolean)
@@ -2759,6 +2774,20 @@ public void adjust(Mechanism mechanism) {
             }
         }
 
+        // <--[mechanism]
+        // @object dEntity
+        // @name dragon_phase
+        // @input Element
+        // @description
+        // Sets an EnderDragon's combat phase.
+        // @tags
+        // <e@entity.dragon_phase>
+        // -->
+        if (mechanism.matches("dragon_phase")) {
+            EnderDragon ed = (EnderDragon) getLivingEntity();
+            ed.setPhase(EnderDragon.Phase.valueOf(value.asString().toUpperCase()));
+        }
+
         // <--[mechanism]
         // @object dEntity
         // @name experience