Add warning for going over augment limits

Author: vaisest

src/Classes/Item.lua

@@ -546,7 +546,7 @@ function ItemClass:ParseRaw(raw, rarity, highQuality)
 					local runeData = data.itemMods.Runes[specVal]
 					if runeData then
 						for _, slotData in pairs(runeData) do
-							runeLevel = math.max(runeLevel, slotData.rank[1])
+							runeLevel = m_max(runeLevel, slotData.rank)
 						end
 					end
 					if runeLevel > 0 and (not self.requirements.runeLevel or runeLevel > self.requirements.runeLevel) then

src/Modules/Build.lua

@@ -2201,6 +2201,11 @@ function buildMode:InsertItemWarnings()
 			InsertIfNew(self.controls.warnings.lines, "You are exceeding jewel limit with the jewel "..warning)
 		end
 	end
+	if self.calcsTab.mainEnv.itemWarnings.augmentLimitWarning then
+		for _, warning in ipairs(self.calcsTab.mainEnv.itemWarnings.augmentLimitWarning) do
+			InsertIfNew(self.controls.warnings.lines, "You are exceeding augment limit with: "..warning)
+		end
+	end
 	if self.calcsTab.mainEnv.itemWarnings.socketLimitWarning then
 		for _, warning in ipairs(self.calcsTab.mainEnv.itemWarnings.socketLimitWarning) do
 			InsertIfNew(self.controls.warnings.lines, "You have too many gems in your "..warning.." socket group")

src/Modules/CalcSetup.lua

@@ -808,11 +808,16 @@ function calcs.initEnv(build, mode, override, specEnv)
 
 	-- Build and merge item modifiers, and create list of radius jewels
 	if not accelerate.requirementsItems then
+		-- save augment counts so we can track going over count limits
+		local augmentCounts = {}
 		for runeSlotName, slot in pairs(build.itemsTab.runeSlots) do
 			local rune = slot:GetSelValue()
 			if runeSlotName == override.repSlotName then
 				rune = override.repItem
 			end
+			if rune.name then
+				augmentCounts[rune.name] = (augmentCounts[rune.name] or 0) + 1
+			end
 			for _, mod in ipairs(rune.mods) do
 				env.itemModDB:AddMod(mod)
 			end
@@ -1101,6 +1106,7 @@ function calcs.initEnv(build, mode, override, specEnv)
 				for i = 1, item.itemSocketCount do
 					if item.runes[i] ~= "None" then
 						socketed = socketed + 1
+						augmentCounts[item.runes[i]] = (augmentCounts[item.runes[i]] or 0) + 1
 					end
 				end
 				env.itemModDB.multipliers["RunesSocketedIn"..slotName] = socketed
@@ -1319,6 +1325,20 @@ function calcs.initEnv(build, mode, override, specEnv)
 				env.charms[override.toggleCharm] = true
 			end
 		end
+
+		for augmentName, count in pairs(augmentCounts) do
+			local dbAugment = data.itemMods.Runes[augmentName]
+			local _, dbMod = next(dbAugment)
+			if dbMod and dbMod.limit and count > dbMod.limit then
+				-- warn for going over augment limits
+				if env.build.calcsTab.mainEnv then
+					env.build.calcsTab.mainEnv.itemWarnings.augmentLimitWarning = env.build.calcsTab.mainEnv.itemWarnings.augmentLimitWarning or { }
+					t_insert(env.build.calcsTab.mainEnv.itemWarnings.augmentLimitWarning, augmentName)
+				end
+			end
+		end
+
+		
 	end
 
 	-- Merge env.itemModDB with env.ModDB