Module:RarityHandler: Difference between revisions

← Older edit Newer edit →

@@ Line 1: / Line 1: @@
 local p = {}
--- Clean: trim and strip leading ~
+-- Clean: trim, strip ~, remove commas
 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 ~
+	if rarity:sub(1, 1) == "~" then
-    if rarity:sub(1, 1) == "~" then
+		rarity = rarity:sub(2)
-        rarity = rarity:sub(2)
+	end
-    end
+	rarity = rarity:gsub(",", "") -- strip commas
-    -- Strip commas
+	return rarity
-    rarity = rarity:gsub(",", "")
-    return rarity
 end
+-- RaritySort (raw integer value)
 function p.getSortValue(frame)
-    local raw = frame.args[1]
+	local raw = frame.args[1]
-    if not raw or type(raw) ~= "string" then
+	local rarity = clean(raw)
-        return 999999999
+	if not rarity then return 999999999 end
-    end
-    local rarity = clean(raw)
+	local lowered = rarity:lower()
-    if not rarity then return 999999999 end
-    local lowered = rarity:lower()
+	if lowered == "1" or lowered == "always" then
+		return 1
+	elseif lowered == "rare" then
+		return 999999998
+	elseif lowered == "never" then
+		return 0
+	end
-    -- Handle special values
+	local num = lowered:match("^1/(%d+%.?%d*)$")
-    if lowered == "1" or lowered == "always" then
+	if num then
-        return 1
+		local n = tonumber(num)
-    elseif lowered == "rare" then
+		if n then
-        return 999999998
+			return math.floor(n)
-    elseif lowered == "never" then
+		end
-        return 0
+	end
-    end
-    -- Match and convert fractional chance like "1/2.5", "1/10000"
+	return 999999999
-    local num = lowered:match("^1/(%d+%.?%d*)$")
-    if num then
-        local n = tonumber(num)
-        if n then
-            return math.floor(n)
-        end
-    end
-    return 999999999
 end
--- For class logic: cleaned string, lowercase
+-- Used for bg-class logic
 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
+-- For display (add ~ and formatted commas back)
 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
+	-- Format 1/#### with commas
-    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 formatted = mw.language.getContentLanguage():formatNum(tonumber(num))
-        return (hasTilde and "~" or "") .. prefix .. withCommas
+		return (hasTilde and "~" or "") .. prefix .. formatted
-    end
+	end
-    return hasTilde and ("~" .. cleaned) or cleaned
+	return hasTilde and ("~" .. cleaned) or cleaned
 end
 return p