Module:RarityHandler: Difference between revisions

← Older edit

@@ Line 1: / Line 1: @@
 local p = {}
--- Clean: trim and strip leading ~
+-- Clean up the rarity string: strip ~, commas, trim
 local function clean(rarity)
-    if not rarity or type(rarity) ~= "string" then return nil end
+	if not rarity or type(rarity) ~= "string" then return nil end
-    rarity = mw.text.trim(rarity)
+	rarity = mw.text.trim(rarity)
-    -- Strip leading ~
+	rarity = rarity:gsub("^~", "") -- remove ~
-    if rarity:sub(1, 1) == "~" then
+	rarity = rarity:gsub(",", "") -- remove commas
-        rarity = rarity:sub(2)
+	return rarity
-    end
-    -- Strip commas
-    rarity = rarity:gsub(",", "")
-    return rarity
 end
 function p.getSortValue(frame)
-    local rarity = frame.args[1]
+	local raw = frame.args[1]
-    if not rarity or type(rarity) ~= "string" then
+	if not raw or type(raw) ~= "string" then return 999999999 end
-        return 999999999
-    end
-    -- Trim whitespace and strip leading ~
+	local rarity = clean(raw)
-    rarity = mw.text.trim(rarity)
+	if not rarity then return 999999999 end
-    if rarity:sub(1, 1) == "~" then
-        rarity = rarity:sub(2)
-    end
-    -- Normalize lowercase
+	local lowered = rarity:lower()
-    local lowered = rarity:lower()
-    -- Handle special values
+	if lowered == "1" or lowered == "always" then
-    if lowered == "1" or lowered == "always" then
+		return 1
-        return 1
+	elseif lowered == "rare" then
-    elseif lowered == "rare" then
+		return 999999998
-        return 999999998
+	elseif lowered == "never" then
-    elseif lowered == "never" then
+		return 0
-        return 0
+	end
-    end
-    -- Match and convert fractional chance like "1/2.5", "1/128"
+	-- Match 1/X, return raw number (not string)
-    local num = lowered:match("^1/(%d+%.?%d*)$")
+	local numStr = lowered:match("^1/(%d+%.?%d*)$")
-    if num then
+	if numStr then
-        local n = tonumber(num)
+		local num = tonumber(numStr)
-        if n then
+		if num then return math.floor(num) end
-            return n
+	end
-        end
-    end
-    -- Unknown or badly formatted
+	return 999999999
-    return 999999999
 end
--- For class logic: cleaned string, lowercase
+-- For logic styling
 function p.getCleanValue(frame)
-    local rarity = clean(frame.args[1])
+	local rarity = clean(frame.args[1])
-    if not rarity then return "Unknown" end
+	if not rarity then return "Unknown" end
+	if rarity == "1" then return "Always" end
-    if rarity == "1" then
+	return rarity
-        return "Always"
-    end
-    return rarity
 end
 -- For display
 function p.normalizeDisplay(frame)
-    local original = frame.args[1]
+	local original = frame.args[1]
-    if not original or type(original) ~= "string" then return "Unknown" end
+	if not original or type(original) ~= "string" then return "Unknown" end
+	local trimmed = mw.text.trim(original)
-    local trimmed = mw.text.trim(original)
+	local hasTilde = trimmed:sub(1, 1) == "~"
-    local hasTilde = trimmed:sub(1, 1) == "~"
+	local cleaned = hasTilde and trimmed:sub(2) or trimmed
-    local cleaned = hasTilde and trimmed:sub(2) or trimmed
-    if cleaned == "1" then
+	if cleaned == "1" then return "Always" end
-        return "Always"
-    end
-    -- Format numbers like 1/10000 → 1/10,000
+	local prefix, num = cleaned:match("^(1/)(%d+%.?%d*)$")
-    local prefix, num = cleaned:match("^(1/)(%d+%.?%d*)$")
+	if prefix and num then
-    if prefix and num then
+		local withCommas = mw.language.getContentLanguage():formatNum(tonumber(num))
-        local withCommas = mw.language.getContentLanguage():formatNum(tonumber(num))
+		return (hasTilde and "~" or "") .. prefix .. withCommas
-        return (hasTilde and "~" or "") .. prefix .. withCommas
+	end
-    end
-    return hasTilde and ("~" .. cleaned) or cleaned
+	return hasTilde and "~" .. cleaned or cleaned
 end
 return p