Add dLocation.generate_tree mechanism

DenizenScript · Feb 18, 2018 · 82c34c4 · 82c34c4
1 parent 547e6cb
commit 82c34c4
Showing 1 changed file with 18 additions and 5 deletions.
diff --git a/plugin/src/main/java/net/aufdemrand/denizen/objects/dLocation.java b/plugin/src/main/java/net/aufdemrand/denizen/objects/dLocation.java
@@ -21,11 +21,7 @@
 import net.aufdemrand.denizencore.tags.TagContext;
 import net.aufdemrand.denizencore.tags.core.EscapeTags;
 import net.aufdemrand.denizencore.utilities.CoreUtilities;
-import org.bukkit.Bukkit;
-import org.bukkit.DyeColor;
-import org.bukkit.Location;
-import org.bukkit.Material;
-import org.bukkit.World;
+import org.bukkit.*;
 import org.bukkit.block.*;
 import org.bukkit.block.banner.PatternType;
 import org.bukkit.entity.Entity;
@@ -2436,6 +2432,23 @@ && getBlock().getState() instanceof CreatureSpawner) {
             getBlock().setData((byte) value.asInt());
         }
 
+        // <--[mechanism]
+        // @object dLocation
+        // @name generate_tree
+        // @input Element
+        // @description
+        // Generates a tree at this location if possible.
+        // For a list of valid tree types, see <@link url http://bit.ly/2o7m1je>
+        // @tags
+        // None
+        // -->
+        if (mechanism.matches("generate_tree") && mechanism.requireEnum(false, TreeType.values())) {
+            boolean generated = getWorld().generateTree(this, TreeType.valueOf(value.asString().toUpperCase()));
+            if (!generated) {
+                dB.echoError("Could not generate tree at " + identifySimple() + ". Make sure this location can naturally generate a tree!");
+            }
+        }
+
         if (!mechanism.fulfilled()) {
             mechanism.reportInvalid();
         }