Módulo:Partido político

local function getPartyParams(country)
	mw.logObject(country, "country")
	local moduleprefix = "Módulo:Partido político/Dados/"
	local moduletitle = mw.title.new(country)
	if not moduletitle or (moduletitle.namespace ~= 828) then
		moduletitle = mw.title.new(moduleprefix..country)
	end
	if not moduletitle then
		mw.log("„"..country.."” não é um nome válido com os dados dos partidos políticos")
	elseif moduletitle.exists then
		local status, partiesData = pcall(mw.loadData, moduletitle.fullText)
		if status then
			return partiesData
		end
	else
		mw.log("Não consigo encontrar definições de partidos políticos. Não existe „"..moduleprefix..country.."”.")
	end
end

local function selectPartyData(args)
	local abbr = args[1]
	if not abbr then
		return
	end
	
	local group = args[2]
	mw.logObject({abbr, group}, "abbr, group")
	local data = getPartyParams(group or "Polônia") or {}
	if data[abbr] then
		mw.logObject({abbr, group or false, data[abbr]}, "selected key")
		return abbr, data[abbr]
	end
	
	local result = {}
	-- look for name
	for k, v in pairs(data) do
		local key = type(k) == "string" and k or false
		if (abbr == v.descricao) or (abbr == v.link) then
			table.insert(result, {key, v})
		end
	end

	if #result == 1 then
		mw.logObject({abbr, group or false, result[1]}, "selected name")
		return result[1][1], result[1][2]
	end
	
	mw.logObject({abbr, group or false}, "not selected")
end

local function Name(party)
	if not party then
		return
	end
	
	local desc = party.descricao
	local link = party.link
	mw.logObject({desc, link}, "desc, link")
	if link and desc and (link ~= desc) then
		mw.logObject("[["..link.."|"..desc.."]]", "return nome A")
		return "[["..link.."|"..desc.."]]"
	elseif link then
		mw.logObject("[["..link.."]]", "return nome B")
		return "[["..link.."]]"
	elseif desc then
		mw.logObject(desc, "return nome C")
		return desc
	end
end

return {
	
["sigla"] = function(frame)
	local args = require('Module:Arguments').getArgs(frame)
	local key, party = selectPartyData(args)
	mw.logObject(party, "party")
	return key or args[1]
end,

["cor"] = function(frame)
	local args = require('Module:Arguments').getArgs(frame)
	local key, party = selectPartyData(args)
	mw.logObject(party, "party")
	if party and party.cor then
		return party.cor
	end
	
	return args.cor or "E6E6E6"
end,

["link"] = function(frame)
	local args = require('Module:Arguments').getArgs(frame)
	local key, party = selectPartyData(args)
	mw.logObject(party, "party")
	if party then
		return party.link
	end
end,

["descrição"] = function(frame)
	local args = require('Module:Arguments').getArgs(frame)
	local key, party = selectPartyData(args)
	mw.logObject(party, "party")
	if party then
		return party.descricao
	end
end,

["nome"] = function(frame)
	local args = require('Module:Arguments').getArgs(frame)
	local key, party = selectPartyData(args)
	mw.logObject(party, "party")
	return Name(party) or args[1]
end,

["curto"] = function(frame)
	local args = require('Module:Arguments').getArgs(frame)
	local abbr, party = selectPartyData(args)
	mw.logObject(party, "party")
	if party then
		local link = party.link
		if link and abbr then
			return link == abbr and ("[["..link.."]]") or ("[["..link.."|"..abbr.."]]")
		elseif abbr then
			return abbr
		else
			return Name(party) or args[1]
		end
	end

	return args[1]
end,

["doc"] = function(frame)
	local moduleName = mw.title.getCurrentTitle().fullText
	local status, partiesData = pcall(mw.loadData, moduleName)
	if not status then
		-- registrar uma mensagem de erro pode ser útil
		mw.log(partiesData)
		return
	end
	
	local data = {}
	local lang = mw.getContentLanguage()
	for k, v in pairs(partiesData) do
		table.insert(data, {
			_sk = v.cor and "1" or "0",
			_ss = type(k) == "string" and "1" or "0",
			_sn = lang:caseFold(v.descricao or v.link or (type(k) == "string" and k or "")),
			key = type(k) == "string" and k or "''brak''",
			color = v.cor and ("#"..v.cor) or "none",
			cor = v.cor or "''brak''",
			name = Name(v) or "''brak''"
		})
	end
	
	table.sort(data, function(a, b) return a._sn < b._sn end)
	
	local builder = mw.html.create()
	builder
		:wikitext(
			'Dados que são usados por modelos para partidos políticos.',
			'\n{| class="wikitable collapsible collapsed" width="100%"',
			'\n! estrutura do registro descrevendo um partido',
			'\n|-',
			'\n| '
		)
		:tag("pre"):wikitext(
			'["..."] = {\t-- sigla opcional do nome do partido',
			'\n\tcor = "...",\t-- 6 dígitos hexadecimais que codificam a cor em RGB',
			'\n\tlink = "...",\t-- nome do artigo que descreve o partido',
			'\n\tdescrição = "...",\t-- nome do partido quando não há artigo ou o link que contém um colchete de desambiguação',
			'\n}'
		):done()
		:wikitext(
			'\n|}',
			'\n{| class="wikitable sortable"',
			'\n|+ Lista de partidos políticos',
			'\n|-',
			'\n! class="unsortable ordinal" | lp.',
			'\n!Cor',
			'\n!Sigla',
			'\n!Nome'
		)
	for i, v in ipairs(data) do
		builder
			:wikitext(
				'\n|-',
				'\n| ', i,
				'\n| '
			)
		builder
			:tag("span"):css({ display = "none"}):wikitext(v._sk):done()
			:tag("span"):css({ border = "none", background = "none", ["text-aling"] = "left", margin = "0px!important"})
				:tag("span"):css({border = "solid 1px silver", background = v.color, ["font-size"] = "80%!important", ["vertical-align"] = "middle", })
					:wikitext("&nbsp;&nbsp;&nbsp;&nbsp;")
				:done()
				:wikitext('&nbsp;', v.cor)
			:done()
			:wikitext('\n| '):tag("span"):css({ display = "none"}):wikitext(v._ss):done():wikitext(v.key)
			:wikitext('\n| '):tag("span"):css({ display = "none"}):wikitext(#v._sn > 0 and "1" or "0", v._sn):done():wikitext(v.name)
	end
	
	builder:wikitext("\n|}")
	local result = tostring(builder)
	mw.log(result, "result")
	return result
end,
}