Module:Ru Paul's Drag Race tables

local p = {}

- -- Exports -- -

-- Used by Template:Drag Race contestant table function p.contestant_table( frame ) return p._contestant_table( frame ) end

-- Used by Template:Drag Race progress table function p.progressTable( frame ) return p._progressTable( frame ) end

-- Used by Template:Drag Race progress key function p.key( frame ) return p._key( frame ) end

--- -- Utility Functions -- --- function p._inTable( t, k ) return (t[k] ~= nil) end

function p._sortRows(a,b) return a[1] > b[1] end

function p._revSortRows(a,b) return a[1] < b[1] end

p.colorMap = { -- See Manual of Style/Accessibility/Colors ["red"] = "#FF7B7B", ['orange'] = "#FFA7AF", ["yellow"] = "#FAFA00", ["chartreuse"] = "#A8FF4F", ["green"] = "#7BFF7B", ["spring"] = "#4FFFA8", ["cyan"] = "#00FAFA", ["dodger"] = "#4FA8FF", ["blue"] = "#8888FF", ["indigo"] = "#BB76FF", ["magenta"] = "#FF29FF", ["pink"] = "#FF52A9", ["brown"] = "#E97500", ["grey"] = "#808080", ["gray"] = "#808080", ["silver"] = "#F8F9FA" -- Actually just the usual background table color }

-- -- Contestant table -- -- function p._contestant_table( frame ) local templateFrame = frame:getParent local contestantData = p._getContestantData( templateFrame ) ret = [=[ {| class="wikitable sortable" border="2" style="text-align:center;" |+ Contestants of All Stars 5 and their backgrounds ! scope="col"| Contestant ! scope="col"| Age ! scope="col"| Hometown ! scope="col"| Original season(s) ! scope="col"| Original placement(s) ! scope="col"| Outcome ]=]	for k, v in pairs( contestantData ) do ret = ret .. p._bio_makeRow(contestantData[k]) end return ret .. "|}" end

function p._bio_makeRow( contestant ) local rowTemplate = [=[ |-	! scope="row" rowspan="${NROWS}"|${NAME} |rowspan="${NROWS}"|${AGE} |rowspan="${NROWS}"|${HOMETOWN} ]=]	if string.find(contestant['season'], 'All Stars') ~= nil then rowTemplate = rowTemplate .. "|All Stars ${SEASON-NUM}\n" contestant['season-num'] = string.match(contestant['season'], 'All Stars (%d+)') contestant['season'] = nil else rowTemplate = rowTemplate .. "|Season ${SEASON}\n" end rowTemplate = rowTemplate .. [=[	|${PLACE} ]=]	if contestant['outcome'] ~= nil then rowTemplate = rowTemplate .. '|rowspan="${NROWS}"|${OUTCOME}\n' if #tostring(contestant['outcome']) < 3 then contestant['outcome'] = p._makePlace(contestant['outcome']) end else rowTemplate = rowTemplate .. '|rowspan="${NROWS}" style="background: #DDF; color: #2C2C2C; vertical-align: middle; text-align: center;" class="no table-no2"|TBA\n' end

if tonumber(contestant['nrows']) > 1 then rowTemplate = rowTemplate .. "|-\n" if string.find(contestant['season2'], 'All Stars') ~= nil then rowTemplate = rowTemplate .. "|All Stars ${SEASON2-NUM}\n" contestant['season2-num'] = tostring(string.match(contestant['season2'], '%d+')) contestant['season2'] = nil else rowTemplate = rowTemplate .. "|Season ${SEASON2}\n" end rowTemplate = rowTemplate .. '|${PLACE2} \n' if contestant['place2-sort'] == nil then contestant['place2-sort'] = p._makePlaceSort(contestant['place2']) end local place if #tostring(contestant['place2']) > 2 then place = string.match(contestant['place2'],'(%d+)%D%D') else place = contestant['place2'] end contestant['place2'] = p._makePlace(place) end for k, v in pairs( contestant ) do		mw.log(k:upper) rowTemplate = string.gsub(rowTemplate,"${"..k:upper:gsub('%-','%%-').."}",contestant[k]) end return rowTemplate end

function p._getContestant( k ) return string.match( k, "contestant%-(%d+)" ) end

function p._getField( k ) return string.match( k, "contestant%-%d+%-(.*)") end

function p._getContestantData( frame ) local contestantData = {} for k, v in pairs( frame.args ) do   	-- Read inputs and organize them by contestant if not p._inTable(contestantData, p._getContestant(k)) then contestantData[p._getContestant(k)] = {} end if p._getField(k) ~= nil then contestantData[p._getContestant(k)][p._getField(k)] = v   	else contestantData[p._getContestant(k)]["name"] = v   	end end for k, v in pairs( contestantData ) do   	-- Final cleanup of the input before rendering table if not p._inTable(contestantData[k],"nrows") then contestantData[k]["nrows"] = 1 end if not p._inTable(contestantData[k],"place-sort") then if #tostring(contestantData[k]['place']) > 2 then place = string.match(contestantData[k]['place'],'(%d+)%D%D') else place = contestantData[k]['place'] end contestantData[k]["place-sort"] = p._makePlaceSort(place) end if #tostring(contestantData[k]['place']) < 3 then contestantData[k]['place'] = p._makePlace(contestantData[k]['place']) end end return contestantData end

function p._makePlaceSort( place ) if #tostring(place) < 2 then return '0'..place else return place end end

function p._makePlace( place ) place = tonumber(place) if place == 1 then return '1st Place' elseif place == 2 then return '2nd Place' elseif place == 3 then return'3rd Place' else return place .. 'th Place' end end

--- -- Contestant progress table -- --- function p._progressTable( frame ) local templateFrame = frame:getParent local contestantData = {} ret = "" for i=1,50 do		arg = templateFrame.args[i] if arg == nil then break elseif i % 3 == 1 then -- First in triplet is contestant name contestantData[arg] = – elseif i % 3 == 2 then -- Second in triple is color codes contestantData[templateFrame.args[i-1]][1] = arg else -- Third in triplet is text contestantData[templateFrame.args[i-2]][2] = arg end end contestantData, width = p._parseRanks( contestantData ) ret = ret .. [=[{| class="wikitable" style="text-align:center;" |+Progress of contests including rank/position in each episode ! scope="col"| Contestant ]=]	for i=1,width do ret = ret .. "! scope='col'| " .. i .."\n" end rowList = {} for k,v in pairs(contestantData) do		table.insert(rowList,p._prog_makeRow(k,v,width)) end table.sort(rowList,p._sortRows) for i=1,20 do		if rowList[i] == nil then break end ret = ret .. rowList[i][2] end return ret .. "|}" end

function p._parseRanks( contestantData ) local data = {} local high = 0 for k,v in pairs(contestantData) do		data[k] = {} data[k][1] = mw.text.split(contestantData[k][1],',%s*') data[k][2] = mw.text.split(contestantData[k][2],',%s*') if #data[k][1] > high then high = #data[k][1] end end return data, high end function p._prog_makeRow( contestant, tableData, width ) row = "|-\n! scope='row'| " .. contestant .. "\n" colors = tableData[1] labels = tableData[2] final = 0 for i=1,20 do -- If we've reached the end of the list... if colors[i] == nil then final = i			-- ...and the entries span the entire width of the table, then finish if i > width then break end -- ...otherwise, fill the rest of the columns with darkgray row = row .. "| colspan='" .. width + 1 - i .. "' bgcolor='darkgray' |\n" break end text = labels[i] color = p.colorMap[colors[i]:gsub("%s+","")] row = row .. "| style='background:" .. color .. ";' |" .. text .. "\n" end retRow = { final, row }	return retRow end

- -- Make key for colors -- - function p._key( frame ) local templateFrame = frame:getParent local args = templateFrame.args local ret = '' local order = mw.text.split(args.order,',%s*') args.order = nil local reverseIndex = {} for i,v in ipairs(order) do		reverseIndex[v] = i	end local rowTable = {} for k,v in pairs(args) do		if k ~= 'order' then box = p._makeColorBox( k ) text = v .. "\n" row = { reverseIndex[k], box .. text } table.insert(rowTable,row) end end table.sort(rowTable,p._revSortRows) for i=1,20 do		if rowTable[i] == nil then break end ret = ret .. rowTable[i][2] end return ret end

function p._makeColorBox( key ) local template = ':   ' color = p.colorMap[key] return string.gsub(template, 'HEXCODE', color) end

return p