From c1298d39acb4e5cf6a57840eaace7d51635d421c Mon Sep 17 00:00:00 2001
From: Alexander Ross <ross@standout.se>
Date: Sun, 24 Mar 2024 19:11:40 +0100
Subject: [PATCH] Add "Third Wind" Potion to the list of potions (#39)

This potion is only usable in unrated battlegrounds. It is currently
healing the user for 50% of their health instantly so should be of high
priority to use.

Fix https://github.com/ollidiemaus/AutoPotion/issues/36
---
 Core/Potions.lua | 16 ++++++++++++++++
 code.lua         |  1 +
 2 files changed, 17 insertions(+)

diff --git a/Core/Potions.lua b/Core/Potions.lua
index c4c3c10..d4e829f 100644
--- a/Core/Potions.lua
+++ b/Core/Potions.lua
@@ -6,6 +6,7 @@ local isWrath = (WOW_PROJECT_ID == WOW_PROJECT_WRATH_CLASSIC)
 --ham.leywine = ham.Item.new(194684,"Azure Leywine")
 --ham.healthstone = ham.Item.new(117, "Healthstone")
 ham.healthstone = ham.Item.new(5512, "Healthstone")
+ham.thirdWind = ham.Item.new(138486, "\"Third Wind\" Potion")
 ham.witheringDreamsR3 = ham.Item.new(207041, "Potion of Withering Dreams")
 ham.witheringDreamsR2 = ham.Item.new(207040, "Potion of Withering Dreams")
 ham.witheringDreamsR1 = ham.Item.new(207039, "Potion of Withering Dreams")
@@ -75,9 +76,18 @@ ham.fel0 = ham.Item.new(36892, "Fel Healthstone")
 ham.fel1 = ham.Item.new(36893, "Fel Healthstone")
 ham.fel2 = ham.Item.new(36894, "Fel Healthstone")
 
+function RemoveFromList(list, itemToRemove)
+  for i = #list, 1, -1 do
+    if list[i] == itemToRemove then
+      table.remove(list, i)
+    end
+  end
+end
+
 function ham.getPots()
   if isRetail then
     local pots = {
+      ham.thirdWind,
       ham.dreamR3,
       ham.dreamsR2,
       ham.dreamR1,
@@ -107,6 +117,12 @@ function ham.getPots()
       ham.healingPotion
     }
 
+
+    local isUnratedBattleground = C_PvP.IsBattleground() and not C_PvP.IsRatedBattleground()
+    if not isUnratedBattleground then
+      RemoveFromList(pots, ham.thirdWind)
+    end
+
     if HAMDB.witheringPotion then
       table.insert(pots, 1, ham.witheringR1)
       table.insert(pots, 1, ham.witheringR2)
diff --git a/code.lua b/code.lua
index 5c181d3..c845d68 100755
--- a/code.lua
+++ b/code.lua
@@ -137,6 +137,7 @@ local onCombat = true
 local updateFrame = CreateFrame("Frame")
 updateFrame:RegisterEvent("BAG_UPDATE")
 updateFrame:RegisterEvent("PLAYER_LOGIN")
+updateFrame:RegisterEvent("PLAYER_ENTERING_WORLD") -- Initial login and UI reload
 if isClassic == false then
   updateFrame:RegisterEvent("TRAIT_CONFIG_UPDATED")
 end