--------------------------------------------------------------------------------

function CAghanim:OnAbilityUpgradeButtonClicked( eventSourceIndex, data )
	-- Grants the
	local nPlayerID = data["PlayerID"]
	local szAbilityName = data["AbilityName"]
	local bIsLevelReward = data["LevelReward"]
	--print ("OnAbilityUpgradeButtonClicked", nPlayerID, szAbilityName, bIsLevelReward, self.bTestingAbilityUpgrades)
	if bIsLevelReward == true or self.bTestingAbilityUpgrades == true then
		if nPlayerID ~= nil and szAbilityName ~= nil then
			local hPlayerHero = PlayerResource:GetSelectedHeroEntity( nPlayerID )
			if hPlayerHero ~= nil then
				CustomGameEventManager:Send_ServerToAllClients( "special_ability_upgrades_button_clicked", data )
				if ( hPlayerHero:FindAbilityByName(szAbilityName) ~= nil ) then 
					-- Only grant an ability once (can be changed if necessary)
					hPlayerHero:RemoveAbility(szAbilityName)
					--print ("Removing ability:", szAbilityName)
				else 				
					local hNewAbility = hPlayerHero:AddAbility( szAbilityName )
					if ( hNewAbility ) then
						-- Grants and upgrades the ability
						hNewAbility:UpgradeAbility( false )
						--print ("Adding new ability", szAbilityName)
					end
				end
			end
		end
	end
end

--------------------------------------------------------------------------------

function CAghanim:TestAbilityUpgradesUICC ( cmdName, bVisible )
	-- Enables the button to open the dev upgrade UI
	print ("TestAbilityUpgradesUI Enabled ")

	self.bTestingAbilityUpgrades = true
	CustomGameEventManager:Send_ServerToAllClients( "special_ability_upgrades_enabled", {} )
    
end