Modification de Module:Bandeau de rubrique

local p = {}

--------------------------------------------------------
-----    Référentiel des bandeaux de rubrique      -----
--------------------------------------------------------

data = {
	["connexe"] = {
		warning  = false,
		text     = function(args)
			local link = args[1]
			if not link then return "Lien ?" end
			
			local text
			if args[2] then
				text = "Articles connexes&nbsp;: "
			else
				text = "Article connexe&nbsp;: "
			end
			
			local linkText = args["t1"] or link
			text = text .. "[[:" .. link .. "|" .. linkText .. "]]"
			
			for i = 2, 5 do
				link = args[i]
				if link then
					linkText = args["t" .. i] or link
					text = text .. ", [[:" .. link .. "|" .. linkText .. "]]"
				else
					break
				end
			end
			
			return text
		end
	},
	["extrait"] = {
		warning  = false,
		text     = function(args)
			local article = args[1]
			local title = mw.title.new(article)
			
			local inclusionType = ''
			local section = args['section']
			local heading = args['rubrique']
			
			if section then
				inclusionType = ' · Section ' .. section
			elseif heading then
				inclusionType = ' · Rubrique ' .. heading
			else
				inclusionType = ' · Introduction'
			end
			
			return 'Extrait de l\'article [[' .. article .. ']]'
					.. inclusionType
					.. ' · ([' .. title:fullUrl('action=edit') .. ' modifier])'
					.. ' · ([[Aide:Inclusion|Qu\'est ce que c\'est ?]])'
		end
	},
	["jeu"] = {
		warning  = false,
		image    = 'Portail jeux.png',
		text     = function(args)
			local abbModule = require( "Module:Abréviation" )
			
			-- Il doit y avoir un moins un jeu reconnu, dans le cas
			-- contraire, on ne retourne rien
			-- Le lien est par défaut le nom de la page mais il peut
			-- être surchargé
			
			local abb = args[1]
			local link = args["lien"] or mw.title.getCurrentTitle().text
			
			if not abb then return "" end
			
			local game = abbModule.name(abb)
			if not game then return "" end
			
			-- Au pluriel si plusieurs jeux

			local text = ''
			if args[2] then
				text = "Articles associés aux jeux&nbsp;: "
			else
				text = "Article associé au jeu&nbsp;: "
			end
			
			text = text .. "[[" .. (args["lien1"] or link) .. " (" .. game .. ")|''" .. game .. "'']]"
			
			-- On ajoute les jeux 2 à 10 mais on s'arrête si le
			-- n-ième est absent.
			
			for i = 2, 10 do
				abb = args[i]
				if abb then
					game = abbModule.name(abb)
					if game then
						local linkAbb = args["lien" .. i] or link
						text = text .. ", [[" .. linkAbb .. " (" .. game .. ")|''" .. game .. "'']]"
					end
				else
					break
				end
			end
			
			return text
		end
	},
	["principal"] = {
		warning  = false,
		text     = function(args)
			local link = args[1]
			local display = args['t1'] or link
			
			if not link then return '' end
			
			text = '[[' .. link .. '|' .. display .. ']]'
			
			local many = false
			for i = 2, 5 do
				local link = args[i]
				local display = args['t' .. i] or link
				if link then
					many = true
					text = text .. ', [[' .. link .. '|' .. display .. ']]'
				else
					break
				end
			end
			
			if many then
				text = 'Articles principaux&nbsp;: ' .. text
			else
				text = 'Article principal&nbsp;: ' .. text
			end
			
			return text
		end
	},
	["utile"] = {
		warning  = false,
		text     = function(args)
			local link = args[1]
			local display = args['t1'] or link
			
			if not link then return '' end
			
			text = '[' .. link .. ' ' .. display .. ' ]'
			
			local many = false
			for i = 2, 5 do
				local link = args[i]
				local display = args['t' .. i] or link
				if link then
					many = true
					text = text .. ', [' .. link .. ' ' .. display .. ']'
				else
					break
				end
			end
			
			if many then
				text = 'Liens utiles&nbsp;: ' .. text
			else
				text = 'Lien utile&nbsp;: ' .. text
			end
			
			return text
		end
	},
	["vide"] = {
		warning  = true,
		image    = 'Portail participation.png',
		category = 'Rubrique vide ou incomplète',
		text     = function(args)
			local text = 'Cette rubrique est vide, insuffisamment détaillée ou incomplète. ['
					.. mw.title:getCurrentTitle():fullUrl('action=edit') .. ' Votre aide] est la bienvenue'
			local todo = args['à faire']
			if todo then
				text = text .. ' pour ' .. todo
			end
			return text .. '.'
		end
	},
}

-----------------------------------------------------------

function p.build(frame)
	local args = {}
	args.nom = frame.args.nom
	
	local argsParent = frame:getParent().args
	for cle, val in pairs(argsParent) do
		if val then
			args[cle] = mw.text.trim(val)
		end
	end
	
	return p._build(args)
end

function p._build(args)
	local banner = data[args.nom:lower()]
	
	local image = banner.image
	local text = banner.text
	local category = banner.category
	
	if category and mw.title.getCurrentTitle().namespace == 0 then
		local game = args['jeu'] or args[1]  -- FIXME: args[1] conflit
		if game ~= '' then
			game = require( 'Module:Abréviation' ).name(game) or ''
			if game ~= '' then
				game = ' (' .. game .. ')'
			end
		end
		category = '[[Catégorie:' .. category .. game .. ']]'
	else
		category = ''
	end
	
	if image then
		image = mw.html.create( 'div' )
			:wikitext( '[[Fichier:' .. image .. '|x30px|link=|alt=]]' )
			:done()
	end
	
	if type(text) == 'function' then
		text = text(args)
	end
	
	local res = mw.html.create( 'div' )
		:addClass( 'avt-section-banner noexcerpt' )
		:node(image)
		:tag( 'div' )
			:wikitext(text)
			:done()
		:done()
	
	return tostring(res) .. category
end

return p