Jump to content

Module:Sports roster/NHL

From Wikipedia, the free encyclopedia
local pp = {}

local ice_hockey_positions = {
	['G'] = '<span data-sort-value="1.0">[[Goaltender|G]]</span>',
	['D'] = '<span data-sort-value="2.0">[[Defenceman|D]]</span>',
	['C'] = '<span data-sort-value="3.0">[[Centre (ice hockey)|C]]</span>',
	['LW'] = '<span data-sort-value="4.0">[[Winger (ice hockey)|LW]]</span>',
	['RW'] = '<span data-sort-value="5.0">[[Winger (ice hockey)|RW]]</span>'
}
local status_called = {}

-- Roster format
function pp.headings(args, keys)
	local h = mw.html.create('tr')
	if keys['pos'] then
		h:tag('th'):tag('abbr'):attr('title', 'Position'):wikitext('Pos.')
	end
	if keys['num'] then
		h:tag('th'):tag('abbr'):attr('title', 'Number'):wikitext('No.')
	end
	if keys['nat'] then
		h:tag('th'):tag('abbr'):attr('title', 'Nationality'):wikitext('Nat.')
	end
	h:tag('th'):wikitext('Name')
	if keys['sg'] then
		h:tag('th'):tag('abbr'):attr('title', 'Shot/Glove'):wikitext('S/G')
	end
	if keys['ft'] or keys['in'] then
		h:tag('th'):wikitext('Height')
	end
	if keys['lbs'] then
		h:tag('th'):wikitext('Weight')
	end
	if keys['DOB'] or keys['birthyear'] then
		h:tag('th'):tag('abbr'):attr('title', 'Date of birth'):wikitext('DOB')
	end
	if keys['acq'] then
		h:tag('th'):wikitext('Acquired')
	end
	if keys['college'] or keys['school'] or keys['from'] or keys['birthplace'] then
		h:tag('th'):wikitext('From')
	end
	if keys['yrs'] then
		h:tag('th'):tag('abbr'):attr('title', 'Years'):wikitext('Yrs')
	end
	return tostring(h)
end

function pp.players_roster(args, players, keys, frame)
	local tracking = ''
	local res = mw.html.create('')
	
	for k,v in ipairs(players) do
		local r = res:tag('tr'):css('text-align', 'center')
		if keys['pos'] then
			local pos = v['pos']
			if ice_hockey_positions[pos] then
				pos = ice_hockey_positions[pos]
				if ice_hockey_positions[pos] then
					pos = ice_hockey_positions[pos]
				end
			end
			r:tag('td'):wikitext(pos or '')
		end
		if keys['num'] then
			r:tag('td'):wikitext(v['num'] or '')
		end
		if keys['nat'] then
			r:tag('td'):wikitext(v['nat'] or '')
		end
		
		local name_string = v['name'] or ''
		if (v['inj'] or 'no') == 'yes' then
			name_string = name_string .. '&nbsp;[[File:Cruz Roja.svg|8px|Injured|link=]]'
		end
		if v['note'] then
			name_string = name_string .. '&nbsp;' .. "'''(" .. v['note'] .. ")'''"
		end
		
		-- Insert status when needed
		local status_string = ''
		local status_local = v['note'] or nil
		local strTwo = require('Module:String2')
		
		local status_let_first = true
		-- Only if it is defined
		if status_local then
			local status_list = mw.text.split(status_local, ",")
			for i, s in ipairs(status_list) do
				if status_let_first then
					status_string = s
					table.insert(status_called, s)
					status_let_first = false
				else
					status_string = status_string..', '..s
					table.insert(status_called, s)
				end
				-- end
			end
			if not status_let_first then 
				status_string = ' <span style="font-weight:bold">('..status_string..')</span>' 
			end
		end
	
		r:tag('td'):css('text-align', 'left'):wikitext(name_string)
		
		if keys['sg'] then
			local sg = v['sg'] or ''
			r:tag('td'):wikitext(sg)
		end
		
		if keys['ft'] or keys['in'] then
			local h = ''
			if v['ft'] and tonumber(v['ft']) then
				local inches = 12*tonumber(v['ft']) + (tonumber(v['in']) or 0)
				local cm = math.floor(2.54*inches + 0.5)
				h = '<span data-sort-value="' .. inches .. '">'
					.. tonumber(v['ft']) .. '\' ' .. (tonumber(v['in']) or 0) .. '"'
					.. ' (' .. (cm / 100) .. 'm)</span>'
			end
			r:tag('td')
				:css('text-align', 'right')
				:css('white-space', 'nowrap')
				:wikitext(h)
		end
		
		if keys['lbs'] then
			local w = ''
			if v['lbs'] and tonumber(v['lbs']) then
				local kg = math.floor(0.45359237*tonumber(v['lbs']) + 0.5)
				w = tonumber(v['lbs']) .. ' lb '
					.. ' (' .. kg .. 'kg)'
			end
			r:tag('td')
				:css('text-align', 'right')
				:css('white-space', 'nowrap')
				:wikitext(w)
		end
		
		if keys['birthyear'] then
			local DOB = v['birthyear'].."-"..v['birthmonth'].."-"..v['birthday'] or ''
			DOB = mw.ustring.gsub(DOB, '^%s*(%d)%-(%d%d?)%-(%d%d%d%d)%s*$', '0%1-%2-%3')
			DOB = mw.ustring.gsub(DOB, '^%s*(%d%d?)%-(%d)%-(%d%d%d%d)%s*$', '%1-0%2-%3')
			DOB = mw.ustring.gsub(DOB, '^%s*(%d%d)%-(%d%d)%-(%d%d%d%d)%s*$', '%3-%1-%2')
			DOB = mw.ustring.gsub(DOB, '^%s*(%d%d%d%d)%-(%d)%-(%d%d?)%s*$', '%1-0%2-%3')
			DOB = mw.ustring.gsub(DOB, '^%s*(%d%d%d%d)%-(%d%d?)%-(%d)%s*$', '%1-%2-0%3')
			
			local age = frame:expandTemplate{title = 'age', args = { v['birthyear'], v['birthmonth'], v['birthday']}}
			DOB = DOB..' (age '..age..')'
			r:tag('td'):wikitext(DOB)
		end

		if keys['DOB'] then
			local DOB = v['DOB'] or ''
			DOB = mw.ustring.gsub(DOB, '^%s*(%d)%-(%d%d?)%-(%d%d%d%d)%s*$', '0%1-%2-%3')
			DOB = mw.ustring.gsub(DOB, '^%s*(%d%d?)%-(%d)%-(%d%d%d%d)%s*$', '%1-0%2-%3')
			DOB = mw.ustring.gsub(DOB, '^%s*(%d%d)%-(%d%d)%-(%d%d%d%d)%s*$', '%3-%1-%2')
			DOB = mw.ustring.gsub(DOB, '^%s*(%d%d%d%d)%-(%d)%-(%d%d?)%s*$', '%1-0%2-%3')
			DOB = mw.ustring.gsub(DOB, '^%s*(%d%d%d%d)%-(%d%d?)%-(%d)%s*$', '%1-%2-0%3')
			local age = frame:expandTemplate{title = 'age', args = { v['DOB']}}
			DOB = DOB..' (age '..age..')'
			r:tag('td'):wikitext(DOB)
		end
		
		if keys['acq'] then
			r:tag('td'):wikitext(v['acq'])
		end
		
		if keys['college'] or keys['school'] or keys['from'] or keys['birthplace'] then
			if v['college'] or v['school'] then
				r:tag('td'):css('text-align', 'left'):wikitext(v['college'] or v['school'])
			elseif v['from'] then
				r:tag('td'):css('text-align', 'left'):wikitext('[[' .. v['from'] .. ']]')
			elseif v['birthplace'] then
				r:tag('td'):css('text-align', 'left'):wikitext(v['birthplace'])
			else
				r:tag('td')
			end
		end
		
		if keys['yrs'] then
			if v['yrs'] == 'R' then
				r:tag('td'):wikitext('<span data-sort-value="0">R</span>')
			else
				r:tag('td'):wikitext(v['yrs'])
			end
		end
	end
		
	return tostring(res) .. tracking
end

function pp.coaches_roster(coaches)
	local res = ''
	local role = ''
	for k,v in ipairs(coaches) do
		if v['role'] ~= role then
			role = v['role']
			res = res .. '\n; ' .. role
		end
		res = res .. '\n: ' .. (v['nat'] or '') .. ' ' .. (v['name'] or '')
		if v['note'] then
			res = res .. '&nbsp;' .. "'''(" .. v['note'] .. ")'''"
		end
		if v['college'] then
			res = res .. ' (' .. v['college'] .. ')'
		elseif v['from'] then
			res = res .. ' (' .. v['from'] .. ')'
		end
	end
	return res
end

function pp.staff_roster(staff)
	local res = ''
	local role = ''
	for k,v in ipairs(staff) do
		if v['role'] ~= role then
			role = v['role']
			res = res .. '\n; ' .. role
		end
		res = res .. '\n: ' .. (v['nat'] or '') .. ' ' .. (v['name'] or '')
		if v['college'] then
			res = res .. ' (' .. v['college'] .. ')'
		elseif v['from'] then
			res = res .. ' (' .. v['from'] .. ')'
		end
	end
	return res
end

function pp.legend(args)
	local res = '\n----'
	res = res .. '\n;Legend'
	for k,v in ipairs({
		{'A', 'Assistant captain'},
		{'C', 'Team captain'},
		{'S', 'Suspended'},
		}) do
		res = res .. "\n*'''(" .. v[1] .. ")''' " .. v[2]
	end
	if args['otherlegend'] then
		res = res .. '\n*' .. args['otherlegend']
	end
	if args['otherlegend2'] then
		res = res .. '\n*' .. args['otherlegend2']
	end
	if args['otherlegend3'] then
		res = res .. '\n*' .. args['otherlegend3']
	end
	res = res .. '\n*[[File:Cruz Roja.svg|8px|Injured|link=]] Injured'
	local team = args['rosterteam']
	if not team then
		team = string.lower(args['team'])
		team = mw.ustring.gsub(team, '^.*%s(%S+)%s*$', '%1')
	end
	if args['roster_url'] then
		res = res .. '\n----\n&nbsp; <span class="plainlinks">[' .. args['roster_url'] .. ' NHL roster page]</span>'
	elseif args['season'] then
	elseif team then
		res = res .. '\n----\n&nbsp; <span class="plainlinks">[http://nhl.com/' .. team .. '/roster NHL roster page]</span>'
	end
	if args['notes'] then
		res = res .. ';Notes\n' .. args['notes']
	end
	return res
end

function pp.above(args, keys)
	return '[[' .. args['team'] .. ']] roster'
end

function pp.abovestyle(frame, args, keys)
	local sportscolor = require('Module:Sports color')
	return sportscolor.colorcell({['args'] = {args['team'], ['width'] = '5', ['sport'] = 'ice hockey'}})
end

function pp.headingstyle(frame, args, keys)
	local sportscolor = require('Module:Sports color')
	return sportscolor.colorcell2({['args'] = {args['team'], ['width'] = '5', ['sport'] = 'ice hockey'}})
end

function pp.footer(args, keys)
	return ""
end

function pp.footerstyle(frame, args, keys)
	return 'border:1px #aaa solid;'
end

-- Navbox format

function pp.players_list(args, players, keys)
	local r = ''
	for k,v in ipairs(players) do
		r = r .. '\n*' .. (v['num'] or '') .. ' ' .. (v['name'] or '')
	end
	return r
end

function pp.coaches_list(coaches)
	local res = ''
	local role = ''
	for k,v in ipairs(coaches) do
		if v['role'] ~= role then
			role = v['role']
			res = res .. '\n; ' .. role
		end
		res = res .. '\n: ' .. (v['name'] or '')
	end
	return res
end

function pp.title(args, keys)
	return '[[' .. args['team'] .. ']] current roster'
end

function pp.titlestyle(frame, args, keys)
	local sportscolor = require('Module:Sports color')
	return sportscolor.colorcell({['args'] = {args['team'], ['width'] = '5', ['sport'] = args['sport'] or 'ice hockey'}})
end

function pp.below(args, keys)
	return pp.footer(args, keys)
end

function pp.belowstyle(frame, args, keys)
	return 'background:transparent; border-top:1px #aaa solid;'
end

return pp