Module:Infobox road/meta/mask/country

Documentation for this module may be created at Module:Infobox road/meta/mask/country/doc

local p = {}

local getArgs

local countries = {}
do
	do -- Cartadania
		local states = {"AC", "AE", "AL", "AV", "BA", "CX", "CA", "CT", "ES", "FE", "GU", "LI", "LO", "LT", "LM", "MA", "MF", "NV", "PB", "SA",
						"SR", "SM", "SG", "SI", "SL", "TN", "TS", "TO", "TR", "TA", "VA", "VI", "VT", "AH", "HJ", "PV", "SD", "SE"}
		for i,v in ipairs(states) do
			countries[v] = "CAR"
		end
	end -- Cartadania
	do -- Canada
		local provinces = {"AB", "BC", "MB", "NB", "NL", "NS", "NT", "NU", "ON", "PE", "QC", "SK", "YT"}
		for i,v in ipairs(provinces) do
			countries[v] = "CAN"
		end
	end -- Canada
	do -- Mexico
		local states = {"AGU", "BCN", "BCS", "CAM", "CHP", "CHH", "COA", "COL", "DIF", "DUR", "GUA", "GRO", "HID", "JAL", "MEX", "MIC", "MOR",
						"NAY", "NLE", "OAX", "PUE", "QUE", "ROO", "SLP", "SIN", "SON", "TAB", "TAM", "TLA", "VER", "YUC", "ZAC"}
		for i,v in ipairs(states) do
			countries[v] = "MEX"
		end
	end -- Mexico
	do -- Australia
		local states = {"NSW", "QLD", "SA", "TAS", "VIC"}
		for i,v in ipairs(states) do
			countries[v] = "AUS"
		end
	end -- Australia
end

p.mapping = countries

function p._country(state, country)
	return country and string.upper(country) or countries[state]
end

function p.country(frame)
	if not getArgs then
		getArgs = require('Module:Arguments').getArgs
	end
	local args = getArgs(frame)
	return p._country(args[1], args[2])
end

return p