From 0ad7b12ff33efd1c3f492027e1560f7ac5214a57 Mon Sep 17 00:00:00 2001
From: BillyGalbreath <Blake.Galbreath@GMail.com>
Date: Sat, 29 Sep 2018 16:08:23 -0500
Subject: [PATCH] Turtle API


diff --git a/src/main/java/net/minecraft/server/EntityTurtle.java b/src/main/java/net/minecraft/server/EntityTurtle.java
index 013b3a1ca8..0037d13806 100644
--- a/src/main/java/net/minecraft/server/EntityTurtle.java
+++ b/src/main/java/net/minecraft/server/EntityTurtle.java
@@ -26,51 +26,63 @@ public class EntityTurtle extends EntityAnimal {
         this.K = 1.0F;
     }
 
+    public void setHome(BlockPosition pos) { g(pos); } // Paper - OBFHELPER
     public void g(BlockPosition blockposition) {
         this.datawatcher.set(EntityTurtle.bA, blockposition);
     }
 
+    public BlockPosition getHome() { return dY(); } // Paper - OBFHELPER
     private BlockPosition dY() {
         return (BlockPosition) this.datawatcher.get(EntityTurtle.bA);
     }
 
+    public void setTravelPos(BlockPosition pos) { h(pos); } // Paper - OBFHELPER
     private void h(BlockPosition blockposition) {
         this.datawatcher.set(EntityTurtle.bD, blockposition);
     }
 
+    public BlockPosition getTravelPos() { return dZ(); } // Paper - OBFHELPER
     private BlockPosition dZ() {
         return (BlockPosition) this.datawatcher.get(EntityTurtle.bD);
     }
 
+    public boolean hasEgg() { return dW(); } // Paper - OBFHELPER
     public boolean dW() {
         return (Boolean) this.datawatcher.get(EntityTurtle.bB);
     }
 
+    public void setHasEgg(boolean hasEgg) { r(hasEgg); } // Paper - OBFHELPER
     private void r(boolean flag) {
         this.datawatcher.set(EntityTurtle.bB, flag);
     }
 
+    public boolean isDigging() { return dW(); } // Paper - OBFHELPER
     public boolean dX() {
         return (Boolean) this.datawatcher.get(EntityTurtle.bC);
     }
 
+    public void setDigging(boolean digging) { s(digging); } // Paper - OBFHELPER
     private void s(boolean flag) {
         this.bG = flag ? 1 : 0;
         this.datawatcher.set(EntityTurtle.bC, flag);
     }
 
+    public boolean isGoingHome() { return ea(); } // Paper - OBFHELPER
     private boolean ea() {
         return (Boolean) this.datawatcher.get(EntityTurtle.bE);
     }
 
+    public void setGoingHome(boolean goingHome) { t(goingHome); } // Paper - OBFHELPER
     private void t(boolean flag) {
         this.datawatcher.set(EntityTurtle.bE, flag);
     }
 
+    public boolean isTravelling() { return ef(); } // Paper - OBFHELPER
     private boolean ef() {
         return (Boolean) this.datawatcher.get(EntityTurtle.bF);
     }
 
+    public void setTravelling(boolean travelling) { u(travelling); } // Paper - OBFHELPER
     private void u(boolean flag) {
         this.datawatcher.set(EntityTurtle.bF, flag);
     }
@@ -441,14 +453,18 @@ public class EntityTurtle extends EntityAnimal {
 
             if (!this.g.isInWater() && this.k()) {
                 if (this.g.bG < 1) {
-                    this.g.s(true);
+                    this.g.setDigging(new com.destroystokyo.paper.event.entity.TurtleStartDiggingEvent((org.bukkit.entity.Turtle) this.g.getBukkitEntity(), MCUtil.toLocation(this.g.world, this.e)).callEvent()); // Paper
                 } else if (this.g.bG > 200) {
                     World world = this.g.world;
 
                     // CraftBukkit start
-                    if (!org.bukkit.craftbukkit.event.CraftEventFactory.callEntityChangeBlockEvent(this.g, this.e.up(), (IBlockData) Blocks.TURTLE_EGG.getBlockData().set(BlockTurtleEgg.b, this.g.random.nextInt(4) + 1)).isCancelled()) {
+                    // Paper start
+                    int eggCount = this.g.random.nextInt(4) + 1;
+                    com.destroystokyo.paper.event.entity.TurtleLayEggEvent layEggEvent = new com.destroystokyo.paper.event.entity.TurtleLayEggEvent((org.bukkit.entity.Turtle) this.g.getBukkitEntity(), MCUtil.toLocation(this.g.world, this.e.up()), eggCount);
+                    if (layEggEvent.callEvent() && !org.bukkit.craftbukkit.event.CraftEventFactory.callEntityChangeBlockEvent(this.g, this.e.up(), Blocks.TURTLE_EGG.getBlockData().set(BlockTurtleEgg.b, layEggEvent.getEggCount())).isCancelled()) {
                     world.a((EntityHuman) null, blockposition, SoundEffects.ENTITY_TURTLE_LAY_EGG, SoundCategory.BLOCKS, 0.3F, 0.9F + world.random.nextFloat() * 0.2F);
-                    world.setTypeAndData(this.e.up(), (IBlockData) Blocks.TURTLE_EGG.getBlockData().set(BlockTurtleEgg.b, this.g.random.nextInt(4) + 1), 3);
+                    world.setTypeAndData(this.e.up(), (IBlockData) Blocks.TURTLE_EGG.getBlockData().set(BlockTurtleEgg.b, layEggEvent.getEggCount()), 3);
+                    // Paper end
                     }
                     // CraftBukkit end
                     this.g.r(false);
@@ -583,7 +599,7 @@ public class EntityTurtle extends EntityAnimal {
 
         @Override
         public boolean a() {
-            return this.a.isBaby() ? false : (this.a.dW() ? true : (this.a.getRandom().nextInt(700) != 0 ? false : !this.a.dY().a((IPosition) this.a.ci(), 64.0D)));
+            return this.a.isBaby() ? false : (this.a.dW() ? true : (this.a.getRandom().nextInt(700) != 0 ? false : !this.a.dY().a((IPosition) this.a.ci(), 64.0D))) && new com.destroystokyo.paper.event.entity.TurtleGoHomeEvent((org.bukkit.entity.Turtle) this.a.getBukkitEntity()).callEvent(); // Paper
         }
 
         @Override
diff --git a/src/main/java/org/bukkit/craftbukkit/entity/CraftTurtle.java b/src/main/java/org/bukkit/craftbukkit/entity/CraftTurtle.java
index 123a2c75ca..8edcf7af65 100644
--- a/src/main/java/org/bukkit/craftbukkit/entity/CraftTurtle.java
+++ b/src/main/java/org/bukkit/craftbukkit/entity/CraftTurtle.java
@@ -1,6 +1,8 @@
 package org.bukkit.craftbukkit.entity;
 
 import net.minecraft.server.EntityTurtle;
+import net.minecraft.server.MCUtil;
+import org.bukkit.Location;
 import org.bukkit.craftbukkit.CraftServer;
 import org.bukkit.entity.EntityType;
 import org.bukkit.entity.Turtle;
@@ -25,4 +27,36 @@ public class CraftTurtle extends CraftAnimals implements Turtle {
     public EntityType getType() {
         return EntityType.TURTLE;
     }
+
+    // Paper start
+    @Override
+    public Location getHome() {
+        return MCUtil.toLocation(getHandle().world, getHandle().getHome());
+    }
+
+    @Override
+    public void setHome(Location location) {
+        getHandle().setHome(MCUtil.toBlockPosition(location));
+    }
+
+    @Override
+    public boolean isGoingHome() {
+        return getHandle().isGoingHome();
+    }
+
+    @Override
+    public boolean isDigging() {
+        return getHandle().isDigging();
+    }
+
+    @Override
+    public boolean hasEgg() {
+        return getHandle().hasEgg();
+    }
+
+    @Override
+    public void setHasEgg(boolean hasEgg) {
+        getHandle().setHasEgg(hasEgg);
+    }
+    // Paper end
 }
-- 
2.22.0