From b2eddbed84d55978337d1b52f6a9a2eabb658017 Mon Sep 17 00:00:00 2001
From: Nate <nate@natedpalm.com>
Date: Thu, 8 Apr 2021 06:17:24 -0400
Subject: Teleport off of mount when throwing enderpearl (#5179)

* Teleport off of mount when throwing enderpearl

* Simplify returning boolean

* Oops forgot a semicolon

* Fix code conventions in Player.h
---
 src/Entities/Player.cpp                 | 16 ++++++++++++++++
 src/Entities/Player.h                   |  1 +
 src/Entities/ThrownEnderPearlEntity.cpp | 24 +++++++++++++++++++++---
 3 files changed, 38 insertions(+), 3 deletions(-)

diff --git a/src/Entities/Player.cpp b/src/Entities/Player.cpp
index a727429d4..d9c51fcef 100644
--- a/src/Entities/Player.cpp
+++ b/src/Entities/Player.cpp
@@ -2590,6 +2590,15 @@ void cPlayer::AttachTo(cEntity * a_AttachTo)
 
 
 void cPlayer::Detach()
+{
+	Detach(false);
+}
+
+
+
+
+
+void cPlayer::Detach(bool a_IsTeleporting)
 {
 	if (m_AttachedTo == nullptr)
 	{
@@ -2607,6 +2616,13 @@ void cPlayer::Detach()
 	}
 
 	Super::Detach();
+
+	// If they are teleporting, no need to figure out position
+	if (a_IsTeleporting)
+	{
+		return;
+	}
+
 	int PosX = POSX_TOINT;
 	int PosY = POSY_TOINT;
 	int PosZ = POSZ_TOINT;
diff --git a/src/Entities/Player.h b/src/Entities/Player.h
index 5a17ca783..65dbfaed7 100644
--- a/src/Entities/Player.h
+++ b/src/Entities/Player.h
@@ -604,6 +604,7 @@ public:
 	// cEntity overrides:
 	virtual void AttachTo(cEntity * a_AttachTo) override;
 	virtual void Detach(void) override;
+	virtual void Detach(bool a_IsTeleporting);
 	virtual cItem GetEquippedWeapon(void) const override { return m_Inventory.GetEquippedItem(); }
 	virtual cItem GetEquippedHelmet(void) const override { return m_Inventory.GetEquippedHelmet(); }
 	virtual cItem GetEquippedChestplate(void) const override { return m_Inventory.GetEquippedChestplate(); }
diff --git a/src/Entities/ThrownEnderPearlEntity.cpp b/src/Entities/ThrownEnderPearlEntity.cpp
index 4f1029b27..ec4af7a7c 100644
--- a/src/Entities/ThrownEnderPearlEntity.cpp
+++ b/src/Entities/ThrownEnderPearlEntity.cpp
@@ -34,12 +34,29 @@ void cThrownEnderPearlEntity::OnHitSolidBlock(Vector3d a_HitPos, eBlockFace a_Hi
 void cThrownEnderPearlEntity::OnHitEntity(cEntity & a_EntityHit, Vector3d a_HitPos)
 {
 	int TotalDamage = 0;
+
+	bool isAttachedEntity = GetWorld()->FindAndDoWithPlayer(m_CreatorData.m_Name, [& a_EntityHit](cPlayer & a_Entity)
+		{
+			const cEntity * attachedEntity = a_Entity.GetAttached();
+			if (attachedEntity == nullptr)
+			{
+				// nothing attached
+				return false;
+			}
+
+			return attachedEntity->GetUniqueID() == a_EntityHit.GetUniqueID();
+		}
+	);
 	// TODO: If entity is Ender Crystal, destroy it
 
-	TeleportCreator(a_HitPos);
-	a_EntityHit.TakeDamage(dtRangedAttack, this, TotalDamage, 1);
 
-	m_DestroyTimer = 5;
+	if (!isAttachedEntity)
+	{
+		TeleportCreator(a_HitPos);
+		a_EntityHit.TakeDamage(dtRangedAttack, this, TotalDamage, 1);
+		m_DestroyTimer = 5;
+	}
+
 }
 
 
@@ -79,6 +96,7 @@ void cThrownEnderPearlEntity::TeleportCreator(Vector3d a_HitPos)
 			// Teleport the creator here, make them take 5 damage:
 			a_Entity.TeleportToCoords(a_HitPos.x, a_HitPos.y + 0.2, a_HitPos.z);
 			a_Entity.TakeDamage(dtEnderPearl, this, 5, 0);
+			a_Entity.Detach(true);
 			return true;
 		}
 	);
-- 
cgit v1.2.3