Editing Module:Sandbox/Ahecht/Gridiron color

--
-- This module implements
-- {{Gridiron primary color}} -- {{Gridiron primary color raw}} -- {{Gridiron primary style}}
-- {{Gridiron secondary color}} -- {{Gridiron secondary color raw}}
-- {{Gridiron tertiary color raw}}
-- {{Gridiron alt primary color}} -- {{Gridiron alt primary style}} -- {{Gridiron alt secondary color}}
--

local p = {}

local color_data = mw.loadData("Module:Gridiron color/data/sandbox")

local yesno = require('Module:Yesno')

local prefixes = {
	"background: ",
	"color: ",
	"",
	"background: ",
	"color: "
	}

local function get_year(colors, year)
	year = tonumber(year)
	if year and (year > 0) and colors and colors[6] and type(colors[6] == 'table') then
		local end_year = nil
		for k in pairs(colors[6]) do
			if (year <= k) and (k < (end_year or 32767)) then end_year = k end
		end
		if end_year then return colors[6][end_year] end
	end
	
	return colors
end

local function get_colors(team, unknown, year)
	team = (team or ''):match("^%s*(.-)%s*$")
	year = tonumber(year)
	unknown = unknown or color_data["#default"] or {"#DCDCDC", "", "none", "", ""}
	
	local use_default = {
		[""] = 1,
		["retired"] = 1,
		["free agent"] = 1,
	}
	
	local colors = nil
	
	if ( team and use_default[team:lower()] ) then
		colors = unknown
	else
		if mw.ustring.find(team, "%d?%d?%d%dthru%d?%d?%d%d$") then
			if (not year or year <= 0) then
				team, year = mw.ustring.match(team, "^(.-)(%d?%d?%d%d)thru%d?%d?%d%d$")
				team, year = team:match("^%s*(.-)%s*$"), tonumber(year)
				
				if year >= 20 and year < 100 then -- Two-digit years were deprecated in 2018
					year = year + 1900
				elseif year < 20 then
					year = year + 2000
				elseif year < 1000 then
					year = nil
				end
			else
				team = mw.ustring.match(team, "^(.-) -%d?%d?%d%dthru%d?%d?%d%d$")
			end
		end
		
		colors = get_year(color_data[team], year)
		
		if ( colors and type(colors) == 'string' ) then
			-- follow alias recursively
			return  get_colors (colors, unknown, year)
		end
	end
	
	return colors or unknown
end

local function bordercss(c, w)
	local s = 'inset ' .. w .. 'px ' .. w .. 'px 0 ' .. c 
		.. ', inset -' .. w .. 'px -' .. w .. 'px 0 ' .. c
	return '-moz-box-shadow: ' .. s .. '; -webkit-box-shadow: ' .. s .. '; box-shadow: ' .. s .. ';'
end
	
function p.test(frame)
	local args = frame.args.team and frame.args or frame:getParent().args
	local colors = get_colors((args.team or args[1]), nil, args.year)
	return '["' .. args.team .. '"] = { "' .. colors[1] .. '", '.. colors[2] ..
		'", '.. colors[3] .. '", '.. colors[4] .. '", '.. colors[5] .. '"}'
end

function p.color(frame, column, altcolumn)
	local args = frame.args.team and frame.args or frame:getParent().args
	local colors = get_colors((args.team or args[1]), nil, args.year)

	column = (column or tonumber(frame.args.column)) or 1
	altcolumn = altcolumn or tonumber(frame.args.altcolumn)
	if ((not colors[column]) or (colors[column] == '')) and altcolumn then
		column = altcolumn
	end
	
	return (yesno(frame.args.raw) and "" or prefixes[column]) .. 
		mw.ustring.gsub(colors[column],"#",frame:extensionTag('nowiki', '#'))
end

function p.style(frame)
	local team = frame.args.team or frame:getParent().args.team
	local border = frame.args.border or frame:getParent().args.border
	local alt = frame.args.alt or frame:getParent().args.alt
	local s = yesno(alt) and (p.color(frame, 4, 1) .. "; " .. p.color(frame, 5, 2) .. "; ") or (p.color(frame, 1) .. "; " .. p.color(frame, 2) .. "; ")
	if tonumber(border) or yesno(border) then
		border = (tonumber(border) and (tonumber(border) > 0)) and tostring(border) or "2"
		s = s .. bordercss(p.color(frame, 3), border)
	end
	return s
end

return p