MyPetBattle.lua

-----------------
-- MyPetBattle --
-----------------
MyPetBattle = {}
---------------
---- SETUP ----
---------------

-- INITIALIZATION -
mypetbattle_enabled = false
mypetbattle_join_pvp = false
mypetbattle_auto_forfeit = false
mypetbattle_wintrade_enabled = false

MPB_STATS_TABLE = {}

-- Variable used to make sure we only call the heal function 1 time after combat.
-- It will be set to "false" on events:PET_BATTLE_OPENING_DONE
-- and set back to "true" on events:UPDATE_SUMMONPETS_ACTION
mypetbattle_hasHealedAfterCombat = true

RegisterAddonMessagePrefix("MPB")

--------------------
---- SETUP DONE ----
--------------------

name, speciesName = C_PetBattles.GetName(LE_BATTLE_PET_ALLY, currentPetID)

---------------------
-- EVENTS HANDLING --
---------------------
local mypetbattle_frame, events = CreateFrame("Frame"), {};


--------------------------------------------------------
-- LOAD A LOT OF STUFF WHEN THE ADDON HAS BEEN LOADED --
function events:ADDON_LOADED(...)
--	print("ADDON_LOADED")
	local addonName = ...
	if addonName == "MyPetBattle" then
		---------------------------------------------------------------
		-- SET SAVEDVARIABLES TO DEFAULT VALUES IF THEY DO NOT EXIST --
		-- TEAM SETUP OPTIONS
		if MPB_CONFIG_TEAMSETUP_RANDOM_TEAM_PET_HEALTH_THRESHOLD == nil then
			MPB_CONFIG_TEAMSETUP_RANDOM_TEAM_PET_HEALTH_THRESHOLD = 0.80
		end
		
		if MPB_CONFIG_TEAMSETUP_PET1_LEVEL_ADJUSTMENT == nil then
			MPB_CONFIG_TEAMSETUP_PET1_LEVEL_ADJUSTMENT = -2
		end
		
		if MPB_CONFIG_TEAMSETUP_PET2_LEVEL_ADJUSTMENT == nil then
			MPB_CONFIG_TEAMSETUP_PET2_LEVEL_ADJUSTMENT = -2
		end
		
		if MPB_CONFIG_TEAMSETUP_PET3_LEVEL_ADJUSTMENT == nil then
			MPB_CONFIG_TEAMSETUP_PET3_LEVEL_ADJUSTMENT = 2
		end
		
		-- PRE-COMBAT OPTIONS
		if MPB_CONFIG_PRE_COMBAT_ATTEMPT_STRONGEST_RANDOM_TEAM == nil then
			MPB_CONFIG_PRE_COMBAT_ATTEMPT_STRONGEST_RANDOM_TEAM = false
		end
		
		-- COMBAT OPTIONS
		if MPB_CONFIG_COMBAT_SWAP_PET_HEALTH_THRESHOLD == nil then
			MPB_CONFIG_COMBAT_SWAP_PET_HEALTH_THRESHOLD = 0.35
		end
		
		-- POST-COMBAT OPTIONS
		if MPB_CONFIG_POST_COMBAT_USE_REVIVE_BATTLE_PETS_AFTER_COMBAT == nil then
			MPB_CONFIG_POST_COMBAT_USE_REVIVE_BATTLE_PETS_AFTER_COMBAT = false
		end

		if not MPB_CONFIG_POST_COMBAT_USE_BATTLE_PET_BANDAGE_AFTER_COMBAT == nil then
			MPB_CONFIG_POST_COMBAT_USE_BATTLE_PET_BANDAGE_AFTER_COMBAT = false
		end

		if not MPB_CONFIG_POST_COMBAT_AUTOMATIC_NEW_RANDOM_TEAM_AFTER_COMBAT == nil then
			MPB_CONFIG_POST_COMBAT_AUTOMATIC_NEW_RANDOM_TEAM_AFTER_COMBAT = false
		end

		-- MISC OPTIONS
		if MPB_CONFIG_MISC_AUTOMATIC_RELEASE_NON_RARES == nil then
			MPB_CONFIG_MISC_AUTOMATIC_RELEASE_NON_RARES = false
		end		

		-- CAPTURE PETS OPTIONS
		if MPB_CAPTURE_RARES == nil then
			MPB_CAPTURE_RARES = false
		end

		if MPB_CAPTURE_COMMON_UNCOMMON == nil then
			MPB_CAPTURE_COMMON_UNCOMMON = false
		end
		-- TEAM PET LEVEL
		if MPB_EDITBOX_DESIRED_PET_LEVEL == nil then
			MPB_EDITBOX_DESIRED_PET_LEVEL = 1
		end
		-- FORFEIT TIMER
		if MPB_FORFEIT_TIMER == nil then
			MPB_FORFEIT_TIMER = 55
		end

		---------------------------------------
		-- SET UI ELEMENTS TO SAVEDVARIABLES --
		-- TEAM SETUP
		EditBox_min_team_pet_health:SetText(tostring(MPB_CONFIG_TEAMSETUP_RANDOM_TEAM_PET_HEALTH_THRESHOLD * 100)) 
		Slider_pet1_level:SetValue(MPB_CONFIG_TEAMSETUP_PET1_LEVEL_ADJUSTMENT)
		Slider_pet2_level:SetValue(MPB_CONFIG_TEAMSETUP_PET2_LEVEL_ADJUSTMENT)
		Slider_pet3_level:SetValue(MPB_CONFIG_TEAMSETUP_PET3_LEVEL_ADJUSTMENT)
	
		-- PRE-COMBAT
		CheckButton15:SetChecked(MPB_CONFIG_PRE_COMBAT_ATTEMPT_STRONGEST_RANDOM_TEAM)
		-- COMBAT
		EditBox_swap_pet_health_threshold:SetText(tostring(MPB_CONFIG_COMBAT_SWAP_PET_HEALTH_THRESHOLD * 100)) 

		-- POST-COMBAT
		CheckButton12:SetChecked(MPB_CONFIG_POST_COMBAT_USE_REVIVE_BATTLE_PETS_AFTER_COMBAT)
		CheckButton14:SetChecked(MPB_CONFIG_POST_COMBAT_USE_BATTLE_PET_BANDAGE_AFTER_COMBAT)
		CheckButton11:SetChecked(MPB_CONFIG_POST_COMBAT_AUTOMATIC_NEW_RANDOM_TEAM_AFTER_COMBAT)

		-- MISC	
		CheckButton13:SetChecked(MPB_CONFIG_MISC_AUTOMATIC_RELEASE_NON_RARES)

		-- LOAD LAST USED DESIRED_PET_LEVEL FOR THE RANDOM TEAM GENERATION
		local loadLastUsedDesiredPetLevel = MPB_EDITBOX_DESIRED_PET_LEVEL
		EditBox_PetLevel:SetText(loadLastUsedDesiredPetLevel) 

		-- LOCK PETS FOR RANDOM TEAM GENERATION
		Check_lock_pet_1:SetChecked(MPB_LOCK_PET1)
		Check_lock_pet_2:SetChecked(MPB_LOCK_PET2)
		Check_lock_pet_3:SetChecked(MPB_LOCK_PET3)

		-- SET TEXTURE FOR CONFIG BUTTON (GEAR) MANUALLY AS THE XML FILE DO NOT WANT DO WHAT I WANT!
		MPB_Config_Button:SetNormalTexture("Interface\\Addons\\MyPetBattle\\Images\\icon-config")
		MPB_Config_Button:SetHighlightTexture("Interface\\Buttons\\UI-Panel-MinimizeButton-Highlight.blp")

		-- SET CAPTURE RARE/UNCOMMON/COMMON CHECKBOXES
		CheckButton3:SetChecked(MPB_CAPTURE_RARES)
		CheckButton4:SetChecked(MPB_CAPTURE_COMMON_UNCOMMON)

		-- SET FORFEIT TIMER
		EditBox_ForfeitTimer:SetText(MPB_FORFEIT_TIMER)
		
		-- CALL FUNCTION IN Frame.lua TO UPDATE THE POSITION OF THE MINIMAP BUTTON
		MPB_MMButton_UpdatePosition()

		-- SHOW/HIDE UI AT LOGIN ACCORDING TO CHARACTER SPECIFIC SAVEDVARIABLE
		if MPB_SHOW_UI == nil then MPB_SHOW_UI = true end -- Set variable to true the first time
		if MPB_SHOW_UI then MyPetBattleForm:Show() else MyPetBattleForm:Hide() end

		-- PRINT THAT WE ARE DONE LOADING
		print("Loaded |cffff8000MPB")
	end


end

function events:PLAYER_LOGIN(...)				-- 
--	print("PLAYER_LOGIN")
	print("\124cFFFF9933My Pet Battle ready for action")
	-- Set default text of textfield "PetLevel" for random team generation
end

function events:PLAYER_ENTERING_WORLD(...)				-- 
--	print("PLAYER_ENTERING_WORLD")
end

function events:PET_BATTLE_ABILITY_CHANGED(...)				-- 
--	print("PET_BATTLE_ABILITY_CHANGED")
end
function events:PET_BATTLE_ACTION_SELECTED(...)				-- Player selected a pet action
--	print("PET_BATTLE_ACTION_SELECTED")
end
function events:PET_BATTLE_AURA_APPLIED(...)				-- Aura applied e.g. Adrenaline for 3 turn (3 stacks)
--	print("PET_BATTLE_AURA_APPLIED")
end
function events:PET_BATTLE_AURA_CANCELED(...)				-- 
--	print("PET_BATTLE_AURA_CANCELED")
end
function events:PET_BATTLE_AURA_CHANGED(...)				-- Aura changed e.g. Adrenaline changed to 2 stacks
--	print("PET_BATTLE_AURA_CHANGED")
end
function events:PET_BATTLE_CAPTURED(...)					-- 
--	print("PET_BATTLE_CAPTURED")
--	print(...)
	local petOwner, petIndex = ...
	if (petOwner == LE_BATTLE_PET_ENEMY) then
		local name, speciesName = C_PetBattles.GetName(petOwner, petIndex)
--		local speciesId, petGUID = C_PetJournal.FindPetIDByName(speciesName)
--		local link = C_PetJournal.GetBattlePetLink(petGUID)

--		local petName = C_PetBattles.GetName(petOwner, petIndex);
--		local petIcon = C_PetBattles.GetIcon(petOwner, petIndex);
--		local quality = C_PetBattles.GetBreedQuality(petOwner, petIndex);
		-- ADD STATS TO STATS TABLE
		MyPetBattle.addStats(speciesName,1)
	end
end
function events:PET_BATTLE_CLOSE(...)						-- 
--	print("PET_BATTLE_CLOSE")
	-- Close the Stopwatch window after battle
	if ( StopwatchFrame:IsShown() ) then
		StopwatchFrame:Hide();
	end
	
	-- Resetting variable opening is done
	petBattleOpeningIsDone = false
end

function events:PET_BATTLE_FINAL_ROUND(...)					-- 
--	print("PET_BATTLE_FINAL_ROUND")
--	print(...)
	if mypetbattle_debug then
		-- SHOWS THE SAME RESULT ALL THE TIME FOR PVP BATTLES REGARDLESS OF WHO WINS?!?!
		if ... == LE_BATTLE_PET_ALLY then
			print("We won!")
			-- ADD STATS TO STATS TABLE
			MyPetBattle.addStats("Wins",1)
		elseif ... == LE_BATTLE_PET_ENEMY then
			print("We lost!")
			-- ADD STATS TO STATS TABLE
			MyPetBattle.addStats("Losses",1)
		else
			print("Not LE_BATTLE_PET_ALLY or LE_BATTLE_PET_ENEMY won?!")
		end
	end
end

function events:PET_BATTLE_HEALTH_CHANGED(...)				-- 
	-- petOwner, petIndex, healthChange
--	local myOutput1, myOutput2, myOutput3 = ...;
--	print("OUTPUT: " .. myOutput1 .. ", " .. myOutput2 .. ", " .. myOutput3)

--	print("PET_BATTLE_HEALTH_CHANGED")
	local petOwner, petIndex, healthChange = ...;
	local pet_name, pet_speciesName = C_PetBattles.GetName(petOwner, petIndex)
	local pet_icon = C_PetBattles.GetIcon(petOwner, petIndex)
	local pet_currentHealth = C_PetBattles.GetHealth(petOwner, petIndex)
	local pet_maxHealth = C_PetBattles.GetMaxHealth(petOwner, petIndex)
	local pet_healthPercentage = pet_currentHealth / pet_maxHealth * 100
	
	local textColor
	if pet_healthPercentage < 30 then
		textColor = "\124cFFFF0000" -- red
	elseif pet_healthPercentage < 75 then
		textColor = "\124cFFFFFF00" -- yellow
	elseif pet_healthPercentage <= 100 then
		textColor = "\124cFF00FF00" -- green
	end
	
	local textPetOwner
	if petOwner == 1 then
		textPetOwner = "Player: "
	else
		textPetOwner = "Enemy: "
	end
    if mypetbattle_debug then	
	    if healthChange > 0 then 
    --		print("gained " .. healthChange .. " health") 
		    print(textPetOwner.."\124T"..pet_icon..":0\124t |cFF0066FF[" .. pet_name .. "]|r gained \124cFF00FF00" .. healthChange .. "\124r health. Current health: " .. textColor .. pet_currentHealth .. "/" .. pet_maxHealth .. string.format(" (%2.0f%%)", pet_healthPercentage)) 
	    elseif healthChange < 0 then
    --		print("lost " .. abs(healthChange) .. " health") 
		    print(textPetOwner.."\124T"..pet_icon..":0\124t |cFF0066FF[" .. pet_name .. "]|r lost \124cFFFF0000" .. abs(healthChange) .. "\124r health. Current health: " .. textColor .. pet_currentHealth .. "/" .. pet_maxHealth .. string.format(" (%2.0f%%)", pet_healthPercentage)) 
	    end
    end
	
--	print(textPetOwner.."\124T"..pet_icon..":0\124t |cFF0066FF[" .. pet_name .. "]|r current health: " .. textColor .. pet_currentHealth .. "/" .. pet_maxHealth .. string.format(" (%2.0f%%)", pet_healthPercentage) .. " point") 

--	print("Pet number "..petIndex .. " health is: " .. C_PetBattles.GetHealth(petOwner, petIndex))

	currentPetID = C_PetBattles.GetActivePet(LE_BATTLE_PET_ALLY)
	currentPet_Health = C_PetBattles.GetHealth(LE_BATTLE_PET_ALLY, currentPetID)
	currentPet_MAX_Health = C_PetBattles.GetMaxHealth(LE_BATTLE_PET_ALLY, currentPetID)
	currentPet_Health_percentage = currentPet_Health / currentPet_MAX_Health * 100
	
--	currentPetID = C_PetBattles.GetActivePet(LE_BATTLE_PET_ALLY)
--	currentPet_Health = C_PetBattles.GetHealth(LE_BATTLE_PET_ALLY, currentPetID)
--	currentPet_MAX_Health = C_PetBattles.GetMaxHealth(LE_BATTLE_PET_ALLY, currentPetID)
--	currentPet_Health_percentage = currentPet_Health / currentPet_MAX_Health * 100
	
	
--	print("|cff00ff00 Current pets health: " .. currentPet_Health .. "/" .. currentPet_MAX_Health .. string.format(" (%2.0f%%)", currentPet_Health_percentage) .. " points")
end

function events:PET_BATTLE_LEVEL_CHANGED(...)				-- 
--	print("PET_BATTLE_LEVEL_CHANGED")
	local petOwner, petIndex = ...;
	if petOwner == LE_BATTLE_PET_ALLY then
		local pet_name, pet_speciesName = C_PetBattles.GetName(petOwner, petIndex)
		local pet_icon = C_PetBattles.GetIcon(petOwner, petIndex)
		local pet_level = C_PetBattles.GetLevel(petOwner, petIndex)
		print("|cFF0066FF\124T"..pet_icon..":16\124t [" .. pet_name .. "]\124r is now level: " .. pet_level .."!") 
	end
	--	print("|cffffff00 Your pet is now level: " .. C_PetBattles.GetLevel(LE_BATTLE_PET_ALLY,1) .. "!")
end

function events:PET_BATTLE_LOOT_RECEIVED(...)				-- 
--	print("PET_BATTLE_LOOT_RECEIVED")

	local typeIdentifier, itemLink, quantity = ...;
	print("|cffff8000MPB|r: \124cFF00FF00Item won:\124r " .. quantity .. " x " .. itemLink)
	
	if ( typeIdentifier == "item" ) then
		local itemName, itemLink, _, _, _, _, _, _, _, itemIcon, _ = GetItemInfo(itemLink);
--		local itemName, itemLink, itemRarity, itemLevel, itemMinLevel, itemType, itemSubType, itemStackCount, itemEquipLoc, itemTexture, itemSellPrice = GetItemInfo(itemLink);
	elseif ( typeIdentifier == "currency" ) then
		local itemName, _, itemIcon, _, _, _, _, _ = GetCurrencyInfo(itemLink);
--		local currencyName, currencyQuantity, currencyIcon, earnedThisWeek, weeklyMax, maxQuantity, discovered, rarity = GetCurrencyInfo(itemLink);
	end

	-- ADD STATS TO STATS TABLE
	MyPetBattle.addStats(itemLink,quantity)
end

function MyPetBattle.addStats(key,value)
	-- Loot received
		-- items
		-- currency
	-- Pets caught: common, uncommon, rare, type etc.
	-- Pets fought
	-- PvP win/loose, streak
	-- XP from WT and other
	if MPB_STATS_TABLE[key] ~= nil then 
		MPB_STATS_TABLE[key] = MPB_STATS_TABLE[key] + value
	else
		MPB_STATS_TABLE[key] = value
	end	
end

function MyPetBattle.printStats()
	-- PRINT STATISTICS OF OUR BATTLES
	print("|cffff8000MPB stats for current session:|r")
	for key,value in pairs(MPB_STATS_TABLE) do 
		print(" - "..value.."x",key) 
	end
end

function events:PET_BATTLE_MAX_HEALTH_CHANGED(...)			-- 
--	print("PET_BATTLE_MAX_HEALTH_CHANGED")
end

function events:PET_BATTLE_OPENING_DONE(...)				-- OPENING DONE AND READY TO BATTLE
	-- STOPWATCH FOR WT
	if mypetbattle_wintrade_enabled then
		if Stopwatch_IsPlaying() then Stopwatch_Clear() end
		Stopwatch_StartCountdown(0, 0, MPB_FORFEIT_TIMER+5) -- SET STOP WATCH TO COUNT DOWN FROM MPB_FORFEIT_TIMER + 5 SEC. +5 FOR MINOR ADJUSTMENT BECAUSE FORFEIT ANIMATION TAKES ABOUT 5 SEC
		Stopwatch_Play() -- STARTS THE STOP WATCH
		if mypetbattle_debug then  print("Started Stopwatch") end
	end
		
	--	print("PET_BATTLE_OPENING_DONE")
	print("|cffff8000MPB|r: We have to fight " .. C_PetBattles.GetNumPets(LE_BATTLE_PET_ENEMY) .. " enemy pets")
	-- Auto-select first available pet
	if C_PetBattles.ShouldShowPetSelect() == true and (mypetbattle_enabled or mypetbattle_wintrade_enabled) then
		for i=1,3 do
			if MyPetBattle.hp(i) > 0 then
				C_PetBattles.ChangePet(i)
			end
		end
	end

	-- Setting global variable to false so we will heal after combat if needed
	mypetbattle_hasHealedAfterCombat = false
	
	-- Setting variable that opening is done
	petBattleOpeningIsDone = true
	MPB_timerTotal = 0		-- Reset timer for automatic forfeit
end

function events:PET_BATTLE_OPENING_START(...)				-- 
--	print("PET_BATTLE_OPENING_START")
	print("|cffff8000MPB|r: |cFF00FFFFGame Starting!")

	-- Get pets level
	pet1_level_start = C_PetBattles.GetLevel(LE_BATTLE_PET_ALLY,1)
	pet2_level_start = C_PetBattles.GetLevel(LE_BATTLE_PET_ALLY,2)
	pet3_level_start = C_PetBattles.GetLevel(LE_BATTLE_PET_ALLY,3)

	-- Get pets current xp
	pet1_xp_start, pet1_maxXp_start = C_PetBattles.GetXP(1, 1)
	pet2_xp_start, pet2_maxXp_start = C_PetBattles.GetXP(1, 2)
	pet3_xp_start, pet3_maxXp_start = C_PetBattles.GetXP(1, 3)
	
end

function events:PET_BATTLE_OVER(...)						-- Pet battle over (someone won)
--	print("PET_BATTLE_OVER")
	
	-- Get pet names
	pet1_name, pet1_speciesName = C_PetBattles.GetName(1, 1)
	pet2_name, pet2_speciesName = C_PetBattles.GetName(1, 2)
	pet3_name, pet3_speciesName = C_PetBattles.GetName(1, 3)

	-- Get pet icons
	pet1_icon = C_PetBattles.GetIcon(1, 1)
	pet2_icon = C_PetBattles.GetIcon(1, 2)
	pet3_icon = C_PetBattles.GetIcon(1, 3)

	-- Get xp after combat has ended
	pet1_xp_end, pet1_maxXp_end = C_PetBattles.GetXP(1, 1)
	pet2_xp_end, pet2_maxXp_end = C_PetBattles.GetXP(1, 2)
	pet3_xp_end, pet3_maxXp_end = C_PetBattles.GetXP(1, 3)
	
	-- Calculate xp gain
	pet1_xp_gained = pet1_xp_end - pet1_xp_start
	pet2_xp_gained = pet2_xp_end - pet2_xp_start
	pet3_xp_gained = pet3_xp_end - pet3_xp_start

	-- Print xp change
	if pet1_xp_gained > 0 then print("|cFF0066FF\124T"..pet1_icon..":0\124t [" .. pet1_name .. "] |cFFFFFFFFgained " .. pet1_xp_gained .. " xp") end
	if pet2_xp_gained > 0 then print("|cFF0066FF\124T"..pet2_icon..":0\124t [" .. pet2_name .. "] |cFFFFFFFFgained " .. pet2_xp_gained .. " xp") end
	if pet3_xp_gained > 0 then print("|cFF0066FF\124T"..pet3_icon..":0\124t [" .. pet3_name .. "] |cFFFFFFFFgained " .. pet3_xp_gained .. " xp") end
	
	print("|cffff8000MPB|r: |cFF00FFFFGame Over!")
end

function events:UPDATE_SUMMONPETS_ACTION(...)					-- 
--	print("UPDATE_SUMMONPETS_ACTION")

	-- Set new team after combat if MPB_CONFIG_POST_COMBAT_AUTOMATIC_NEW_RANDOM_TEAM_AFTER_COMBAT is true
	if MPB_CONFIG_POST_COMBAT_AUTOMATIC_NEW_RANDOM_TEAM_AFTER_COMBAT then
		local desiredPetLevel = EditBox_PetLevel:GetText()  -- Get user input for desired pet level
		MyPetBattle.setTeam(desiredPetLevel)                -- Setup our team
		-- Save desired pet level for next time we log in
		MPB_EDITBOX_DESIRED_PET_LEVEL = desiredPetLevel
		-- Clear focus from the editbox
		EditBox_PetLevel:ClearFocus()
	end

	-- Setting global variable to false so we will heal after combat if needed
	if mypetbattle_hasHealedAfterCombat == false then
		MyPetBattle.revive_and_heal_Pets() -- function in MyPetBattle_data.lua
		mypetbattle_hasHealedAfterCombat = true
	end
end

function events:COMPANION_UPDATE(...)					-- 
--	print("COMPANION_UPDATE")
end

function events:PET_JOURNAL_LIST_UPDATE(...)
--	print(PET_JOURNAL_LIST_UPDATE)
end

function events:PET_BATTLE_PET_CHANGED(...)					-- 
--	print("PET_BATTLE_PET_CHANGED")
end

function events:PET_BATTLE_PET_ROUND_PLAYBACK_COMPLETE(...)	-- 
--	print("PET_BATTLE_PET_ROUND_PLAYBACK_COMPLETE")
	local round = ...
--	print(round)
	
	if mypetbattle_enabled or (mypetbattle_wintrade_enabled and round <= math.random(0,2)) then -- ATTACK A RANDOM NUMBER OF TIMES BETWEEN 1-3 IF WE ARE DOING WT (0 AND 2 SINCE ROUND 0 EXISTS)
		-------------------------------------
		-- AUTO-SELECT FIRST AVAILABLE PET --
		if C_PetBattles.ShouldShowPetSelect() == true then
			for i=1,3 do
				if MyPetBattle.hp(i) > 0 then
					C_PetBattles.ChangePet(i)
				end
			end
		end
		-------------------------------
		-- GET PETOWNER AND PET INFO --
		local petOwner = LE_BATTLE_PET_ALLY
		local petIndex = C_PetBattles.GetActivePet(petOwner)
		local petType = C_PetBattles.GetPetType(petOwner, petIndex)
		-----------------------------------------------------------------------------------------
		-- SKIP TURN IF POLYMORPHED, STUNNED ETC. BY ENEMY (MAYBE CHANGE PET IF LOW ON HEALTH) --
		if MyPetBattle.buff("Polymorphed") or MyPetBattle.buff("Asleep") or MyPetBattle.buff("Crystal Prison") or MyPetBattle.buff("Stunned") or MyPetBattle.buff("Drowsy") then 
			C_PetBattles.SkipTurn()
		end
		--------------------------------------------------------------------------------
		-- SWITCH PET AT HEALTH THRESHOLD BEFORE IT DIES. CAN BE SET FROM CONFIG MENU --
		if MyPetBattle.hp(petIndex) < MPB_CONFIG_COMBAT_SWAP_PET_HEALTH_THRESHOLD then
			for j=1,3 do
				if MyPetBattle.hp(j) >= MPB_CONFIG_COMBAT_SWAP_PET_HEALTH_THRESHOLD then
					C_PetBattles.ChangePet(j)
					print("|cFFFF3300.. Changing pet due to low health! ..")
				end
			end
		end
		------------------------------------------------------------------
		-- CHECK IF WE SHOULD AND CAN CAPTURE RARE PETS WE ARE FIGHTING --
		-- THE FUNCTION canCaptureRare() IN MyPetBattle_data.lua WILL CHECK IF WE CAN CAPTURE THE RARE WE ARE FIGHTING
		if MPB_CAPTURE_RARES and MyPetBattle.canCaptureRare() and mypetbattle_enabled then
			print("|cFF8A2BE2 We are trying to capture a rare!")
			C_PetBattles.UseTrap() -- USE THE TRAP
		end
		-------------------------------------------------------------------------
		-- CHECK IF WE SHOULD CAPTURE COMMON/UNCOMMON IF WE DO NOT OWN THE PET --
		if MPB_CAPTURE_COMMON_UNCOMMON and MyPetBattle.canCaptureCommon() and mypetbattle_enabled then
			print("|cFF00FF00 We do not have this pet (0/3), let us capture it (common/uncommon)!")
			C_PetBattles.UseTrap() -- USE THE TRAP
		end
		-----------------------------
		-- GETTING READY TO ATTACK --
		local spell = nil
		if petType == 1 then 		-- HUMANOID
			spell = humanoid()		
		elseif petType == 2 then 	-- DRAGONKIN
			spell = dragonkin()
		elseif petType == 3 then 	-- FLYING
			spell = flying()
		elseif petType == 4 then 	-- UNDEAD
			spell = undead()
		elseif petType == 5 then 	-- CRITTER
			spell = critter()
		elseif petType == 6 then 	-- MAGIC
			spell = magic()
		elseif petType == 7 then 	-- ELEMENTAL
			spell = elemental()
		elseif petType == 8 then 	-- BEAST
			spell = beast()
		elseif petType == 9 then 	-- WATER / AQUATIC
			spell = aquatic()
		elseif petType == 10 then 	-- MECHANICAL
			spell = mechanical()
		end
		----------------------------------------------
		-- IF WE HAVE SPELL WE CAN CAST, THEN CAST! --
		if spell ~= nil and spell ~= "UNKNOWN" then	
			actionIndex = MyPetBattle.getSpellSlotIndex(spell)
			
			local spellID, spellName, spellIcon, _, _, _, _, _ = C_PetBattles.GetAbilityInfo(petOwner, petIndex, actionIndex)
				
			print("|cffff8000MPB|r: |cffFF4500Casting\124r \124T"..spellIcon..":0\124t \124cff4e96f7\124HbattlePetAbil:"..spellID..":0:0:0\124h["..spell.."]\124h\124r");
--			print("actionIndex: ", actionIndex)
			C_PetBattles.UseAbility(actionIndex) -- USE PET ABILITY 
		-- IF THE PET IS UNKNOWN, THEN CAST THE FIRST AVAILABLE SPELL SO WE WILL AT LEAST ATTACK
		elseif spell == "UNKNOWN" then
			print("Pet is unknown, casting first spell available")
			C_PetBattles.UseAbility(1) -- USE PET ABILITY 1
		end
    elseif mypetbattle_debug then
        ------------------------------------------------------------------------------
		-- IF WE'RE NOT ENABLED BUT IN DEBUG MODE THEN SHOW WHAT WE WOULD HAVE CAST --
		print("|cffFF4500 Would be Casting: ", spell)

	end
end

function events:PET_BATTLE_PET_ROUND_RESULTS(...)			-- 
--	print("PET_BATTLE_PET_ROUND_RESULTS")
	local roundNumber = ...
	if roundNumber ~= 0 and mypetbattle_debug then
		print("Round "..roundNumber)
	end
end

function events:PET_BATTLE_PVP_DUEL_REQUESTED(...)			-- 
	print("PET_BATTLE_PVP_DUEL_REQUESTED")
end

function events:PET_BATTLE_PVP_DUEL_REQUEST_CANCEL(...)		-- 
	print("PET_BATTLE_PVP_DUEL_REQUEST_CANCEL")
end

function events:PET_BATTLE_QUEUE_PROPOSAL_ACCEPTED(...)		-- 
--	print("PET_BATTLE_QUEUE_PROPOSAL_ACCEPTED")
	print("|cffff8000MPB|r: Pet Battle PvP queue accepted")
end

function events:PET_BATTLE_QUEUE_PROPOSAL_DECLINED(...)		-- 
--	print("PET_BATTLE_QUEUE_PROPOSAL_DECLINED")
	print("|cffff8000MPB|r: Removed from pet battle PvP queue")
end

function events:PET_BATTLE_QUEUE_PROPOSE_MATCH(...)			-- 
--	print("PET_BATTLE_QUEUE_PROPOSE_MATCH")

	-- Sync setup for wt
	if mypetbattle_wintrade_enabled then
		MPB_SyncTimeSent = GetTime()
		if mypetbattle_debug then print("Queue popped, sending sync message: "..MPB_SyncTimeSent) end
		SendAddonMessage("MPB", MPB_SyncTimeSent, "PARTY")
	end
	
	-- Automatic accept PvP queue popup if enabled	
	if (mypetbattle_join_pvp and not mypetbattle_wintrade_enabled) then
		if mypetbattle_debug then print("Auto-accepting pet PvP match!") end
		C_PetBattles.AcceptQueuedPVPMatch()
	end
end

function events:PET_BATTLE_QUEUE_STATUS(...)				-- 
--	print("PET_BATTLE_QUEUE_STATUS")
end

function events:PET_BATTLE_TURN_STARTED(...)				-- 
	print("PET_BATTLE_TURN_STARTED")
end

function events:PET_BATTLE_XP_CHANGED(...)					-- 
--	print("PET_BATTLE_XP_CHANGED")
end

function events:CHAT_MSG_ADDON(...)							-- 
--	print("CHAT_MSG_ADDON")
	local prefix, message, channel, sender = ...
	-- CHECK IF MESSAGE COMES FROM MPB ADDON
	if prefix == "MPB" and channel == "PARTY" and sender ~= UnitName("player") then
		MPB_SyncTimeReceived = tonumber(message)
		if mypetbattle_debug then print("We received sync message: "..MPB_SyncTimeReceived.." from "..prefix) end
	end
end

mypetbattle_frame:SetScript("OnEvent", function(self, event, ...)
 events[event](self, ...); -- call one of the functions above
end);

-- REGISTER ALL EVENTS FOR WHICH HANDLERS HAVE BEEN DEFINED
for k, v in pairs(events) do
 mypetbattle_frame:RegisterEvent(k); 
end

--------------------
--- TIMER FRAME ----
--------------------
MPB_timerTotal = 0 -- TIMER INIT FOR AUTOMATIC FORFEIT
MPB_timerOneSec = 0 -- 1 SEC TIMER INIT FOR DIFFERENT MECHANICS E.G. AUTO RE-QUEUE PVP

if MPB_SyncTimeReceived == nil then MPB_SyncTimeReceived = 0 end
if MPB_SyncTimeSent == nil then MPB_SyncTimeSent = 0 end
MPB_syncCounter = 0

MPB_petguids = {0,0,0}

local function MPB_onUpdate(self,elapsed)
	-- AUTOMATIC FORFEIT TIMER
	if petBattleOpeningIsDone then
	    MPB_timerTotal = MPB_timerTotal + elapsed
		if MPB_timerTotal >= MPB_FORFEIT_TIMER then -- DEFAULT: 55 SECONDS
			if mypetbattle_debug then  print(MPB_FORFEIT_TIMER+5.." sec. almost up!") end -- +5 ADJUSTMENT
			MPB_timerTotal = 0
			petBattleOpeningIsDone = false
    	    -- FORFEIT
    	    if mypetbattle_auto_forfeit then
    	    	if mypetbattle_debug then  print("Forfeiting!") end
	    	    C_PetBattles.ForfeitGame()
	    	end
	    end
	else
		MPB_timerTotal = 0
    end

	-- 1 SEC TIMER CHECK
	MPB_timerOneSec = MPB_timerOneSec + elapsed
	if MPB_timerOneSec >= 1 then
		-- CHECK IF WE SHOULD BE IN THE PVP MATCHMAKING QUEUE, BUT WE ARE NOT
		if not C_PetBattles.IsInBattle() and (C_PetBattles.GetPVPMatchmakingInfo() == nil) and mypetbattle_join_pvp then
			C_PetBattles.StartPVPMatchmaking()
		end
		-- DISMISS PET SO WE DO NOT HAVE IT RUNNING AROUND
        local petGUID = C_PetJournal.GetSummonedPetGUID()
        if petGUID and petGUID == C_PetJournal.GetPetLoadOutInfo(1) then -- Check if we have a pet summoned already, or we will get an error
            C_PetJournal.SummonPetByGUID(petGUID) -- Dismiss pet
        end
		-- SET UI TEXTURES FOR CURRENT PET TEAM WHEN THE UI IS LOADED -- IS NOT NEEDED ANYMORE SINCE setTeam() IS CALLED ABOVE, AND THAT FUNCTION SETS THE TEXTURES
		-- MIGHT BE NEEDED ANYWAY, IF AUTO NEW TEAM IS ACTIVE, THEN THE TEXTURE WILL NOT BE LOADED ON LOGIN
		-- MOVED TEXTURE UPDATES TO THE TIMER, BECAUSE THE EVENTS USED BEFORE, WE CANNOT BE SURE THEY ARE ALWAYS CALLED WHEN WE WANT TO UPDATE
		for i_ = 1, 3 do
			local petGUID = C_PetJournal.GetPetLoadOutInfo(i_)
			if not petGUID then  break end -- BREAK THE LOOP IF THERE ARE NO PETS YET OR WE WILL GET AN ERROR

			local speciesID, customName, level, xp, maxXp, displayID, isFavorite, name, icon, petType, creatureID, sourceText, description, isWild, canBattle, tradable, unique, obtainable = C_PetJournal.GetPetInfoByPetID(petGUID) 

			if MPB_petguids[i_] ~= petGUID then -- CHECK IF WE ALREADY HAVE THE PET, THEN NO NEED TO SET THE TEXTURE
				MPB_petguids[i_] = petGUID
				-- SET PET 1 UI TEXTURE
				if i_ == 1 then 
					Pet1_texture:SetTexture(icon) 
					Pet1_level_string:SetText(level)
				end
				-- SET PET 2 UI TEXTURE
				if i_ == 2 then 
					Pet2_texture:SetTexture(icon) 
					Pet2_level_string:SetText(level)
				end
				-- SET PET 3 UI TEXTURE
				if i_ == 3 then 
					Pet3_texture:SetTexture(icon) 
					Pet3_level_string:SetText(level)
				end
			end
		end

		-- WT SYNC CHECK AND PVP ACCEPT
		local syncThreshold = 3 -- Seconds
		local syncDifference = abs(MPB_SyncTimeSent - MPB_SyncTimeReceived)
		-- GET PVP QUEUE INFORMATION
		local queueState, estimatedTime, queuedTime = C_PetBattles.GetPVPMatchmakingInfo()
		-- CHECK THAT WE ARE READY TO ACCEPT WHEN RECEIVING THE MESSAGE FROM OUR PARTNER
		if mypetbattle_wintrade_enabled and not C_PetBattles.IsInBattle() and mypetbattle_debug then print("|cffff8000MPB|r sync: " .. syncDifference) end
		-- WHEN QUEUE POPS CHECK FOR SYNC
		if queueState == "proposal" then
			MPB_syncCounter = MPB_syncCounter + 1
			if mypetbattle_debug then print("Sync count"..MPB_syncCounter) end				-- FOR DEBUGGING
			if mypetbattle_debug then print("Sync difference is: "..syncDifference) end		-- FOR DEBUGGING

			-- IF SYNC RECEIVED AND WITHING THRESHOLD, THEN ACCEPT
			if syncDifference < syncThreshold then 
				if mypetbattle_debug then print("Sync is ok, accepting battle")	end			-- FOR DEBUGGING
				C_PetBattles.AcceptQueuedPVPMatch()
				-- RESET COUNTER AFTER ACCEPTING QUEUE
				MPB_syncCounter = 0
			-- COUNT syncThreshold SECONDS BEFORE DECLINING
			elseif MPB_syncCounter >= syncThreshold then 
				if mypetbattle_debug then print("Not in sync, declining battle") end 		-- FOR DEBUGGING
				C_PetBattles.DeclineQueuedPVPMatch()
			end
		else
			-- IF NO QUEUE HAS POPPED YET MAKE SURE THE COUNTER IS 0
			MPB_syncCounter = 0
		end

		-- RESET THE TIMER
		MPB_timerOneSec = 0
	end

end
 
local f = CreateFrame("frame")
f:SetScript("OnUpdate", MPB_onUpdate)

--------------------
-- SLASH COMMANDS --
--------------------
SLASH_MYPETBATTLE1 = '/mypetbattle'
function SlashCmdList.MYPETBATTLE(msg, editbox)
	if msg == "" then
		mypetbattle_enabled = not mypetbattle_enabled
		if mypetbattle_enabled then status = "\124cFF00FF00Enabled" else status = "\124cFFFF0000Disabled" end
		print("|cffff8000MPB|r:", status)
		CheckButton1:SetChecked(mypetbattle_enabled)
	elseif msg == "join_pvp" then
		mypetbattle_join_pvp = not mypetbattle_join_pvp
		if mypetbattle_join_pvp then 
			status = "\124cFF00FF00PvP enabled"
			-- if C_PetBattles.GetPVPMatchmakingInfo == nil then
				C_PetBattles.StartPVPMatchmaking()
			-- end
		else 
			status = "\124cFFFF0000PvP disabled"
			-- if C_PetBattles.GetPVPMatchmakingInfo ~= nil then
				C_PetBattles.StopPVPMatchmaking()
			-- end
		end
		print("|cffff8000MPB|r:", status)
	elseif msg == "capture_rares" then
		MPB_CAPTURE_RARES = not MPB_CAPTURE_RARES
		if MPB_CAPTURE_RARES then status = "\124cFF00FF00Automatic capture rare pets enabled" else status = "\124cFFFF0000Automatic capture rare pets disabled" end
		print("|cffff8000MPB|r:", status)
		CheckButton3:SetChecked(MPB_CAPTURE_RARES)
	elseif msg == "capture_common_uncommon" then
		MPB_CAPTURE_COMMON_UNCOMMON = not MPB_CAPTURE_COMMON_UNCOMMON
		if MPB_CAPTURE_COMMON_UNCOMMON then status = "\124cFF00FF00Automatic capture common/uncommon (0/3 owned) pets enabled" else status = "\124cFFFF0000Automatic capture common/uncommon pets (0/3 owned) disabled" end
		print("|cffff8000MPB|r:", status)
	elseif msg == "auto_forfeit" then
		mypetbattle_auto_forfeit = not mypetbattle_auto_forfeit
		if mypetbattle_auto_forfeit then status = "\124cFF00FF00Automatic forfeit after "..MPB_FORFEIT_TIMER.." sec enabled" else status = "\124cFFFF0000Automatic forfeit disabled" end
		print("|cffff8000MPB|r:", status)
	elseif msg == "wintrade_enable" then
		if GetNumGroupMembers() == 0 then -- MAKE SURE WE ARE IN A GROUP TO WT
			print("|cffff8000MPB|r: \124cFFFF0000You need to be in a party to use wt")
			mypetbattle_wintrade_enabled = false
			CheckButton_wintrade_enable:SetChecked(mypetbattle_wintrade_enabled)
		else
			mypetbattle_wintrade_enabled = not mypetbattle_wintrade_enabled
			if mypetbattle_wintrade_enabled then status = "\124cFF00FF00Automatic wintrade enabled" else status = "\124cFFFF0000Automatic wintrade disabled" end
			print("|cffff8000MPB|r:", status)
		end
	elseif msg == "debug" then
		-- TURN OFF DEBUG MESSAGES AS REQUIRED
		mypetbattle_debug =  not mypetbattle_debug
		if mypetbattle_debug then status = "\124cFF00FF00Enabled" else status = "\124cFFFF0000Disabled" end
		print("Debugging:",status)
	elseif msg == "ui" then
		-- SHOW/HIDE UI
		MPB_SHOW_UI = not MPB_SHOW_UI
		if MPB_SHOW_UI then MyPetBattleForm:Show() else MyPetBattleForm:Hide() end
	elseif msg == "help" then
		-- SHOW HELP
		print("|==================================|")
		print("MPB command list:")
		print("  /MyPetBattle")
		print("  /MyPetBattle join_pvp")
		print("  /MyPetBattle capture_rares")
		print("  /MyPetBattle capture_common_uncommon")
		print("  /MyPetBattle wintrade_enable")
		print("  /MyPetBattle debug")
		print("  /MyPetBattle ui")
		print("  /MyPetBattle help")
	else
		-- UNKNOWN COMMAND
		print("|cffff8000MPB|r: Unknown command ("..msg..")")
    end
end