initial commit

aghanim_singleplayer/scripts/vscripts/aghanim_ability_upgrade_interface.lua

@@ -0,0 +1,40 @@
+--------------------------------------------------------------------------------
+
+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