Module:RarityTier

-- local getArgs = require('Module:Arguments').getArgs local string, table, yesno = require('Module:String'), require('Module:Table'), require('Module:Yesno') local onmain = require('Module:Mainonly').on_main

local p = {}

local aliases = require('Module:RarityTier/Aliases') local tiers = require('Module:RarityTier/Data') local hexColors = require("Module:Color/Data").hexColors

function p._getRarity(tierid) if not tierid then return '' end tierid = tierid:lower:gsub(" ", "_") tier = tiers[aliases[tierid] or tierid] return tier and tier.name or tierid end

-- Get function for a tier in the data -- Accepts: tierid (& all aliases), order number (note: 'common' is indexed 0, not 1) function p._getTier(tierid) if not tierid then return '' end if type(tonumber(tierid)) == 'number' then for k,v in pairs(tiers) do			if v.order == tonumber(tierid) then return v, k			end end return "Rarity tier out of index " else tierid = tierid:lower:gsub(" ", "_") local key = aliases[tierid] or tierid local tier = tiers[key] if not tier then return "Unknown rarity tier '"..tierid.."' " end return tier, key end end

-- Returns an iterable version of p._getTier -- This is for use when the order is important, as using -- pairs on RarityTier/Data will not retain the order function p._getTierIterable(start, tail) local index, count = tonumber(start) or 0, tonumber(tail) or table.length(tiers) return function if index < count then local v, k = p._getTier(index) index = index + 1 return k, v		end end end

function p.getHexColor(r) -- gets hex color of a rarity return hexColors[p._getTier(r).color] end

function hideAlways(data) text = '' for i=1,data do		text = text..'​' end return ' '..text..' ' end

function p.link(frame) local args = getArgs(frame) return p._link(args[1], args[2], args["addcategory"], args["nolink"]) end

function p._link(tierid, linkText, addcategory, nolink) addcategory = yesno(addcategory, false) local tier = p._getTier(tierid) local name = tier.name if not tierid then return '' end if tier.name then return table.concat({		hideAlways(tier.order or 0),		(nolink and "" or ""),			" ",		(nolink and "" or ""),		-- Add category		(addcategory and onmain) and "" or "",	},"") else return tierid end end

-- Colors `text` based on the color of the rarity tierid specific function p._colorText(tierid, text, withFormat, link, nolink) withFormat = withFormat~=false -- default to true if nil return table.concat({		(link and "" or ""),			(withFormat and " " or ""),		(link and "" or ""),	},"") end

-- returns the value from a lua table based on the rarity passed in -- aliases are accounted for (ex: tierid="c" will trigger the "common" key in the table) -- ex: _switch("rare", { common="a", uncommon="b", rare="c" }) would return "c" function p._switch(tierid, cases) tierid = string.lower(tierid) tierid = aliases[tierid] or tierid return cases[tierid] or cases["default"] or nil end

return p