Add *Player::golem()

diasurgical · Mar 12, 2024 · 5d933d7 · 5d933d7
1 parent 1a3dade
commit 5d933d7
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/Source/player.cpp b/Source/player.cpp
@@ -2096,6 +2096,11 @@ bool Player::isLevelOwnedByLocalClient() const
 	return false;
 }
 
+Monster *Player::golem() const
+{
+	return &Monsters[getId()];
+}
+
 Player *PlayerAtPosition(Point position, bool ignoreMovingPlayers /*= false*/)
 {
 	if (!InDungeonBounds(position))

diff --git a/Source/player.h b/Source/player.h
@@ -894,6 +894,8 @@ struct Player {
 
 	/** @brief Checks if the player level is owned by local client. */
 	bool isLevelOwnedByLocalClient() const;
+
+	Monster *golem() const;
 };
 
 extern DVL_API_FOR_TEST uint8_t MyPlayerId;

diff --git a/Source/spells.cpp b/Source/spells.cpp
@@ -147,10 +147,12 @@ int GetManaAmount(const Player &player, SpellID sn)
 
 void ConsumeSpell(Player &player, SpellID sn)
 {
-	Monster &golem = Monsters[player.getId()];
+	if (sn == SpellID::Golem && &player == MyPlayer) {
+		Monster *golem = player.golem();
 
-	if (golem.hitPoints <= 0 && &player == MyPlayer)
-		return;
+		if (golem->hitPoints <= 0)
+			return;
+	}
 	switch (player.executedSpell.spellType) {
 	case SpellType::Skill:
 	case SpellType::Invalid: