From b624a6526d698605310c3d6d27372b61aa48b698 Mon Sep 17 00:00:00 2001
From: Tamion <70228790+notTamion@users.noreply.github.com>
Date: Sat, 27 Jul 2024 16:37:51 +0200
Subject: [PATCH] Fix PickupStatus getting reset (#11154)

---
 .../api/Fix-PickupStatus-getting-reset.patch  | 25 +++++++
 .../Fix-PickupStatus-getting-reset.patch      | 65 +++++++++++++++++++
 2 files changed, 90 insertions(+)
 create mode 100644 patches/api/Fix-PickupStatus-getting-reset.patch
 create mode 100644 patches/server/Fix-PickupStatus-getting-reset.patch

diff --git a/patches/api/Fix-PickupStatus-getting-reset.patch b/patches/api/Fix-PickupStatus-getting-reset.patch
new file mode 100644
index 0000000000..e7353cd9fc
--- /dev/null
+++ b/patches/api/Fix-PickupStatus-getting-reset.patch
@@ -0,0 +1,25 @@
+From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
+From: Tamion <70228790+notTamion@users.noreply.github.com>
+Date: Tue, 23 Jul 2024 18:37:12 +0200
+Subject: [PATCH] Fix PickupStatus getting reset
+
+
+diff --git a/src/main/java/org/bukkit/entity/AbstractArrow.java b/src/main/java/org/bukkit/entity/AbstractArrow.java
+index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644
+--- a/src/main/java/org/bukkit/entity/AbstractArrow.java
++++ b/src/main/java/org/bukkit/entity/AbstractArrow.java
+@@ -0,0 +0,0 @@ public interface AbstractArrow extends Projectile {
+      */
+     void setHitSound(@NotNull org.bukkit.Sound sound);
+     // Paper end - more projectile API
++
++    // Paper start - Fix PickupStatus getting reset
++    /**
++     * Set the shooter of this projectile.
++     *
++     * @param source the {@link org.bukkit.projectiles.ProjectileSource} that shot this projectile
++     * @param resetPickupStatus whether the {@link org.bukkit.entity.AbstractArrow.PickupStatus} should be reset
++     */
++    void setShooter(@Nullable org.bukkit.projectiles.ProjectileSource source, boolean resetPickupStatus);
++    // Paper end - Fix PickupStatus getting reset
+ }
diff --git a/patches/server/Fix-PickupStatus-getting-reset.patch b/patches/server/Fix-PickupStatus-getting-reset.patch
new file mode 100644
index 0000000000..4d67fea087
--- /dev/null
+++ b/patches/server/Fix-PickupStatus-getting-reset.patch
@@ -0,0 +1,65 @@
+From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
+From: Tamion <70228790+notTamion@users.noreply.github.com>
+Date: Sun, 21 Jul 2024 19:11:22 +0200
+Subject: [PATCH] Fix PickupStatus getting reset
+
+
+diff --git a/src/main/java/net/minecraft/world/entity/projectile/AbstractArrow.java b/src/main/java/net/minecraft/world/entity/projectile/AbstractArrow.java
+index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644
+--- a/src/main/java/net/minecraft/world/entity/projectile/AbstractArrow.java
++++ b/src/main/java/net/minecraft/world/entity/projectile/AbstractArrow.java
+@@ -0,0 +0,0 @@ public abstract class AbstractArrow extends Projectile {
+ 
+     @Override
+     public void setOwner(@Nullable Entity entity) {
++        // Paper start - Fix PickupStatus getting reset
++        this.setOwner(entity, true);
++    }
++
++    public void setOwner(@Nullable Entity entity, boolean resetPickup) {
++        // Paper end - Fix PickupStatus getting reset
+         super.setOwner(entity);
++        if (!resetPickup) return; // Paper - Fix PickupStatus getting reset
+         Entity entity1 = entity;
+         byte b0 = 0;
+ 
+diff --git a/src/main/java/net/minecraft/world/entity/projectile/Projectile.java b/src/main/java/net/minecraft/world/entity/projectile/Projectile.java
+index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644
+--- a/src/main/java/net/minecraft/world/entity/projectile/Projectile.java
++++ b/src/main/java/net/minecraft/world/entity/projectile/Projectile.java
+@@ -0,0 +0,0 @@ public abstract class Projectile extends Entity implements TraceableEntity {
+     public boolean deflect(ProjectileDeflection deflection, @Nullable Entity deflector, @Nullable Entity owner, boolean fromAttack) {
+         if (!this.level().isClientSide) {
+             deflection.deflect(this, deflector, this.random);
+-            this.setOwner(owner);
++            // Paper start - Fix PickupStatus getting reset
++            if (this instanceof AbstractArrow arrow) {
++                arrow.setOwner(owner, false);
++            } else {
++                this.setOwner(owner);
++            }
++            // Paper end - Fix PickupStatus getting reset
+             this.onDeflection(deflector, fromAttack);
+         }
+ 
+diff --git a/src/main/java/org/bukkit/craftbukkit/entity/CraftAbstractArrow.java b/src/main/java/org/bukkit/craftbukkit/entity/CraftAbstractArrow.java
+index 0000000000000000000000000000000000000000..0000000000000000000000000000000000000000 100644
+--- a/src/main/java/org/bukkit/craftbukkit/entity/CraftAbstractArrow.java
++++ b/src/main/java/org/bukkit/craftbukkit/entity/CraftAbstractArrow.java
+@@ -0,0 +0,0 @@ public class CraftAbstractArrow extends AbstractProjectile implements AbstractAr
+         this.getHandle().setSoundEvent(org.bukkit.craftbukkit.CraftSound.bukkitToMinecraft(sound));
+     }
+     // Paper end
++
++    // Paper start - Fix PickupStatus getting reset - Copy of CraftProjectile#setShooter, calling setOwner(Entity,boolean)
++    @Override
++    public void setShooter(org.bukkit.projectiles.ProjectileSource shooter, boolean resetPickupStatus) {
++        if (shooter instanceof CraftEntity craftEntity) {
++            this.getHandle().setOwner(craftEntity.getHandle(), resetPickupStatus);
++        } else {
++            this.getHandle().setOwner(null, resetPickupStatus);
++        }
++        this.getHandle().projectileSource = shooter;
++    }
++    // Paper end - Fix PickupStatus getting reset
+ }