Add API to set Arrow pickup status

1 files changed, 13 insertions, 1 deletions

diff --git a/src/main/java/org/bukkit/craftbukkit/entity/CraftArrow.java b/src/main/java/org/bukkit/craftbukkit/entity/CraftArrow.java
index dca85204..0e1d9343 100644
--- a/src/main/java/org/bukkit/craftbukkit/entity/CraftArrow.java
+++ b/src/main/java/org/bukkit/craftbukkit/entity/CraftArrow.java
@@ -1,5 +1,6 @@
 package org.bukkit.craftbukkit.entity;
 
+import com.google.common.base.Preconditions;
 import net.minecraft.server.EntityArrow;
 
 import org.apache.commons.lang.Validate;
@@ -33,7 +34,7 @@ public class CraftArrow extends AbstractProjectile implements Arrow {
     }
 
     public ProjectileSource getShooter() {
-        return getHandle().projectileSource;   
+        return getHandle().projectileSource;
     }
 
     public void setShooter(ProjectileSource shooter) {
@@ -46,6 +47,17 @@ public class CraftArrow extends AbstractProjectile implements Arrow {
     }
 
     @Override
+    public PickupStatus getPickupStatus() {
+        return PickupStatus.values()[getHandle().fromPlayer.ordinal()];
+    }
+
+    @Override
+    public void setPickupStatus(PickupStatus status) {
+        Preconditions.checkNotNull(status, "status");
+        getHandle().fromPlayer = EntityArrow.PickupStatus.a(status.ordinal());
+    }
+
+    @Override
     public EntityArrow getHandle() {
         return (EntityArrow) entity;
     }