LibItemStatsPlus.lua

--[[
	Method:
	stats = lib:GetItemStats(itemlink);
	Returns: A table of item stats, as returns from API GetItemStats(itemlink).
	Including item's enchant, gems, reforged stats by scanning item's tooltip BUT EXCLUDING GEMS SCOKET.
	
	===Additional Func===
	
	ilevel = lib:GetUpgradeLevel(itemlink)
	Returns: upgraded itemlevel of itemlink
	------
	ratings = lib:GetRatingsFromStat(value, level, statName [, classid, specid] )
	Returns: Rating of statName from level in level
		-- TODO: add formula for resilience with DR in other levels
	------
	MeleeCritRating = lib:GetCritFromAgi(value, level, classid)
	SpellCritRating = lib:GetCritFromInt(value, level, classid)
	
	lib.SpellCritBase = lib:GetSpellCritBase(classid)
	lib.MeleeCritBase = lib:GetMeleeCritBase(classid)
	
	value = lib:GetAPFromAgi(value, classid)
	value = lib:GetAPFromStr(value, classid)
	value = lib:GetRAPFromAgi(value, classid)
	value = lib:GetMPFromSpt(value, classid)
	value = lib:GetHPFromSta(value, level)
	
	Struct for each gear stats table:
	stats = {	[statName1] = value,
				[statName2] = value,
				... 
			}
]]
local MAJOR, MINOR = "LibItemStatsPlus",10;
local debugmode = false

if not LibStub then error(MAJOR.." requires LibStub") end
local lib = LibStub:NewLibrary(MAJOR, MINOR);
if not lib then return end

SLASH_LIBITEMSTATSPLUS1 = "/LISP";

local cache = {}
setmetatable(cache, {__mode = "kv"}) -- weak table to enable garbage collection

local tip = lib.tip
if not tip then
	-- Create a custom tooltip for scanning
	tip = CreateFrame("GameTooltip", MAJOR.."Tooltip", nil, "GameTooltipTemplate")
	lib.tip = tip
	tip:SetOwner(UIParent, "ANCHOR_NONE")
	for i = 1, 40 do
		tip[i] = _G[MAJOR.."TooltipTextLeft"..i]
		if not tip[i] then
			tip[i] = tip:CreateFontString()
			tip:AddFontStrings(tip[i], tip:CreateFontString())
			_G[MAJOR.."TooltipTextLeft"..i] = tip[i]
		end
	end
elseif not _G[MAJOR.."TooltipTextLeft40"] then
	for i = 1, 40 do
		_G[MAJOR.."TooltipTextLeft"..i] = tip[i]
	end
end

local function CheckClassID(classid)
	if type(classid) ~= "number" then classid = lib.GetClassIDFromClassName [classid] end
	if not classid then
		return
	else
		return classid
	end
end

local function GetStatIndex(statName)
	statIndex = lib.CombatRatingsFromNameToIndex[statName] or nil
	if statIndex == nil then statIndex = lib.CombatRatingsFromRatingIDToIndex[statName] end
	return statIndex
end

local Locale = GetLocale()
do
	if (Locale=="enUS" or Locale=="zhCN" or Locale=="zhTW" or Locale=="koKR") then
		LOCALE_STHOUSAND = "," --Character used to separate groups of digits
		LOCALE_SDECIMAL = "." --Character(s) used for the decimal separator
	elseif (Locale=="deDE" or Locale=="esES" or Locale=="frFR" or Locale=="itIT" or Locale=="ruRU") then
		LOCALE_STHOUSAND = "."
		LOCALE_SDECIMAL = ","
	elseif (Locale=="ptBR") then
		LOCALE_STHOUSAND = "%p"
		LOCALE_SDECIMAL = "%p"
	else
		print("unknow locale")
		LOCALE_STHOUSAND = "%p"
		LOCALE_SDECIMAL = "%p"
	end
end

local patDecimal = "%d-[%"..LOCALE_STHOUSAND.."?%d]+%"..LOCALE_SDECIMAL.."?%d*"; --regex to find a localized decimal number e.g. 

--[[local Greycolor ={128/255, 128/255, 128/255}
local Whitecolor ={1, 1, 1}
local Greedcolor ={0, 1, 0}
local Yellocolor ={1, 210/255, 0}]]

local function AddStats(stats, statName, value)
	value = string.gsub( value , LOCALE_STHOUSAND , "" )
	value = tonumber(value)
	if stats[statName] == nil then
		stats[statName] = value
	else
		stats[statName] = stats[statName] + value
	end
	return stats
end

function ParseLine(stats, text, r, g, b)
	text = strtrim(text)
	if (ceil(r*255)==128) then return stats end --if color is grey then do nothing
	if strsub(text, -2) == "|r" then
		text = strsub(text, 1, -3)
	end
	if strfind(strsub(text, 1, 10), "|c%x%x%x%x%x%x%x%x") then
		text = strsub(text, 11)
	end

	--armor
	local found, _, value, statNameStr = string.find(text, "("..patDecimal..")(.*)");
	if found then
		found = string.find(string.upper(statNameStr), ".?"..string.upper(RESISTANCE0_NAME));
		if found then
			AddStats(stats, "RESISTANCE0_NAME", value)
			return stats
		end
	end
	
	--dual stats
	local found, _, value1, statNameStr1, value2, statNameStr2 = string.find(text, ".-%+("..patDecimal..")(.-)%+("..patDecimal..")(.*)");
	if found then
		for statName, statNameText in pairs(lib.StatsList) do
			if string.find(string.upper(statNameStr1), "^%s*"..string.upper(statNameText)) then
				AddStats(stats, statName, value1)
			end
			if string.find(string.upper(statNameStr2), "^%s*"..string.upper(statNameText)) then
				AddStats(stats, statName, value2)
			end
			
		end
		return stats
	end

	--single stats
	local found, _, value, statNameStr = string.find(text, ".-%+("..patDecimal..")(.*)");
	if found then
		for statName, statNameText in pairs(lib.StatsList) do
			found = string.find(string.upper(statNameStr), "^%s*"..string.upper(statNameText));
			if found then
				if statName=="SPELL_STATALL" then
					AddStats(stats, "ITEM_MOD_STAMINA_SHORT", value)
					AddStats(stats, "ITEM_MOD_AGILITY_SHORT", value)
					AddStats(stats, "ITEM_MOD_INTELLECT_SHORT", value)
					AddStats(stats, "ITEM_MOD_STRENGTH_SHORT", value)
					AddStats(stats, "ITEM_MOD_SPIRIT_SHORT", value)
				else
					AddStats(stats, statName, value)
				end
				return stats
			end
		end
	end
	
	--"Equip:" pvp power for weapon
	local found = string.find(text, ITEM_SPELL_TRIGGER_ONEQUIP);
	if found then
		local statName, statNameText = "ITEM_MOD_PVP_POWER_SHORT" , ITEM_MOD_PVP_POWER_SHORT
		found = string.find(string.upper(text), string.upper(statNameText));
		if found then
			found, _, value = string.find(text, "("..patDecimal..")");
			if found then
				AddStats(stats, statName, value)
				return stats
			end
		end
	end
	return stats
end

function lib:GetItemStats(itemlink, ...)
	if itemlink == nil then return 0 end
	if cache[itemlink] then
		return cache[itemlink]
	end
	local check, _, color, Ltype, Id, Enchant, Gem1, Gem2, Gem3, Gem4, Suffix, Unique, LinkLvl, Reforge, Upgrade, Name = string.find(itemlink, "|?c?f?f?(%x*)|?H?([^:]*):?(%d+):?(%d*):?(%d*):?(%d*):?(%d*):?(%d*):?(%-?%d*):?(%-?%d*):?(%d*):?(%d*):?(%d*)|?h?%[?([^%[%]c]*)%]?|?h?|?r?");
	if check == nil then return 0 end
	
	tip:ClearLines() -- this is required or SetX won't work the second time its called
	tip:SetHyperlink(itemlink)

	local stats = {}
	for i = 2, tip:NumLines() do
		local text = tip[i]:GetText();
		local r, g, b = tip[i]:GetTextColor()
		if debugmode then
			print(text)
		end
		ParseLine(stats, text, r, g, b);
	end --for each line in the tooltip
	
	if debugmode then
		print(Name)
		for i,v in pairs(stats) do
			print(i..","..v)
		end
		print("------")	
	end

	return stats;
end


function lib:GetRatingsFromStat(value, level, statName, classid, specid)
	local statIndex, ratings
	statIndex = GetStatIndex(statName)
	if statIndex == nil or statIndex == 0 or type(value) ~= "number" then return 0 end
	if not (level <= MAX_PLAYER_LEVEL) then return 0 end
	--[[if ( level <= 60 ) and ( statIndex ~= 9 ) then
		if level < 10 then 
			ratings = value / CombatRating10[statIndex]
		else
			ratings = value / ( level * CombatRating60[statIndex][1] + CombatRating60[statIndex][2] )
		end
		return ratings
	elseif ( level <= 60 ) and ( level >= 10 ) then
		classid = CheckClassID(classid)
		if specid > 4 then
			specid = specid - lib.SpecIDToSpecIndex[classid]
		end
		if not lib.MasteryCoefficients[classid][specid] then return 0 end
		ratings = value / ( level * CombatRating60[statIndex][1] + CombatRating60[statIndex][2] ) * lib.MasteryCoefficients[classid][specid]
		return ratings
	end]]
	if ( statIndex == 6 ) and  ( level == 90 ) then
		-- handle resilience with DR in lv 90
		-- TODO: add formula for resilience with DR in other levels
		-- more info here: http://www.icy-veins.com/forums/topic/303-combat-ratings-at-level-90-in-mists-of-pandaria/#pvp_resilience
		-- ratings = 100 - 100 * 0.99 ^ ( value / 310 )
		ratings = 35 * ( value / ( value + 23187 ) )
	elseif ( statIndex == 9 ) then 
		classid = CheckClassID(classid)
		if specid > 4 then
			specid = specid - lib.SpecIDToSpecIndex[classid]
		end
		if not lib.MasteryCoefficients[classid][specid] then return 0 end
		ratings = value / lib.CombatRatings[level][statIndex] * lib.MasteryCoefficients[classid][specid]
	else
		ratings = value / lib.CombatRatings[level][statIndex]
	end
	return ratings
end

function lib:GetSpellCritBase(classid)
	classid = CheckClassID(classid)
	if not (classid <= MAX_CLASSES) then return 0 end
	return lib.SpellCritBase[classid]
end

function lib:GetMeleeCritBase(classid)
	classid = CheckClassID(classid)
	if not (classid <= MAX_CLASSES) then return 0 end
	return lib.MeleeCritBase[classid]
end

function lib:GetCritFromAgi(value, level, classid)
	if type(level) ~= "number" or type(value) ~= "number" then return 0 end
	classid = CheckClassID(classid)
	if not (level <= MAX_PLAYER_LEVEL) or not (classid <= MAX_CLASSES) then return 0 end
	return value / lib.MeleeCritRatings[level][classid]
end

function lib:GetCritFromInt(value, level, classid)
	if type(level) ~= "number" or type(value) ~= "number" then return 0 end
	classid = CheckClassID(classid)
	if not (level <= MAX_PLAYER_LEVEL) or not (classid <= MAX_CLASSES) then return 0 end
	return value / lib.SpellCritRatings[level][classid]
end

function lib:GetAPFromAgi(value, classid)
	if type(value) ~= "number" then return 0 end
	classid = CheckClassID(classid)
	if not (classid <= MAX_CLASSES) then return 0 end
	return value * lib.APPerAgi[classid]
end

function lib:GetAPFromStr(value, classid)
	if type(value) ~= "number" then return 0 end
	classid = CheckClassID(classid)
	if not (classid <= MAX_CLASSES) then return 0 end
	return value * lib.APPerStr[classid]
end

function lib:GetRAPFromAgi(value, classid)
	if type(value) ~= "number" then return 0 end
	classid = CheckClassID(classid)
	if not (classid <= MAX_CLASSES) then return 0 end
	return value * lib.RAPPerAgi[classid]
end

function lib:GetSPFromInt(value, classid)
	if type(value) ~= "number" then return 0 end
	classid = CheckClassID(classid)
	if not (classid <= MAX_CLASSES) then return 0 end
	return value * lib.SPPerInt[classid]
end

function lib:GetMPFromSpt(value, classid)
	if type(value) ~= "number" then return 0 end
	classid = CheckClassID(classid)
	if not (classid <= MAX_CLASSES) then return 0 end
	return value * lib.MPPerSpt[classid]
end

function lib:GetHPFromSta(value, level)
	if type(level) ~= "number" or type(value) ~= "number" then return 0 end
	if not (level <= MAX_PLAYER_LEVEL) then return 0 end
	return value * lib.HpPerSta[level]
end

function lib:GetUpgradeLevel(link)
	if not link then return 0 end
	local upgrade = link:match(":(%d+)\124h%[")
	local itemLevel = select(4, GetItemInfo(link)) 
	if itemLevel ~= nil then
		itemLevel = itemLevel + (lib.UpgradeLevels[upgrade] or 0)
		return itemLevel
	end
end

function CommandHandler(msg)
	if (not msg) then msg=""; end
	if (strlen(msg)>0) then msg=strlower(msg); end
	
	if(msg == "debug") then
		if debugmode then
			debugmode = false;
			print(MAJOR.." "..MINOR.." : debug disabled")
		else
			debugmode = true;
			print(MAJOR.." "..MINOR.." : debug enabled")
		end;
		return;
	elseif(msg=="") then
		print(MAJOR.." slash list:")
		print(SLASH_LIBITEMSTATSPLUS1.." debug")
	end;
end
SlashCmdList["LIBITEMSTATSPLUS"]=function(msg) CommandHandler(msg) end;