Modification de Module:Capsule

local p = {}
local ressources = {
	["infosPokemon"] = require("Module:Ressources/infosPokemon"),
	["donneesCapacite"] = require("Module:Ressources/donneesCapacite"),
	["indexTableau"] = require("Module:Ressources/indexTableau")
}

function p.tableau( frame )
	local generation = tonumber( mw.text.trim( frame.args['génération'] ) )
	local capacite = mw.text.trim( frame.args['capacité'] )
	local jeux = frame.args['jeux']
	local poke_aptes = frame.args['pokémon']
	if generation == nil then generation = "" end
	if capacite == nil then return "Pas de nom de capacité renseigné" end
	if jeux ~= nil then jeux = mw.text.split( string.lower(jeux), "/" ) else jeux = {} end
	for i,jeu in ipairs(jeux) do
		jeux[i] = mw.text.trim( jeu )
	end
	if poke_aptes ~= nil then poke_aptes = mw.text.split( poke_aptes, "/" ) else poke_aptes = {} end
	local generations = {"Première", "Deuxième", "Troisième", "Quatrième", "Cinquième", "Sixième", "Septième", "Huitième", "Neuvième"}
	local titre
	if generations[generation] then titre = generations[generation] end
	titre = titre .. ' génération'
	if frame.args['jeu'] then titre = titre .. ' - <i>' .. frame.args['jeu'] .. '</i>' end
	
	local colspan = 6
	if generation == 3 or generation == 6 then colspan = 9
	elseif generation == 4 then colspan = 8
	end

	local i = 1
	local infos_jeux = {}
	local emplacement, achat, vente
	while jeux[i] ~= nil do
		jeux[i] = mw.text.trim( jeux[i] )
		if frame.args[jeux[i] .. '-emplacement'] == nil then
			if jeux[i] == "ev" and frame.args["ev-recette-pl"] ~= nil then emplacement = "—"
			else emplacement = "Inconnu"
			end
		else emplacement = mw.text.trim( frame.args[jeux[i] .. '-emplacement'] )
		end
		if frame.args[jeux[i] .. '-achat'] == nil or frame.args[jeux[i] .. '-achat'] == "-" or frame.args[jeux[i] .. '-achat'] == "—" then achat = "—"
		else 
			if frame.args[jeux[i] .. '-achat-unité'] then achat = mw.text.trim( frame.args[jeux[i] .. '-achat'] ) .. " " .. mw.text.trim( frame.args[jeux[i] .. '-achat-unité'] )
			else achat = mw.text.trim( frame.args[jeux[i] .. '-achat'] ) .. " [[Fichier:Miniature Pokédollar EB.png|10px|class=imagenoire|link=Pokédollar|alt=Pokédollar]]"
			end
		end
		if frame.args[jeux[i] .. '-vente'] == nil or frame.args[jeux[i] .. '-vente'] == "-" or frame.args[jeux[i] .. '-vente'] == "—" then vente = "—"
		else vente = mw.text.trim( frame.args[jeux[i] .. '-vente'] ) .. " [[Fichier:Miniature Pokédollar EB.png|10px|class=imagenoire|link=Pokédollar|alt=Pokédollar]]"
		end
		if jeux[i] == "ev" and frame.args["ev-recette-échantillons"] ~= nil and frame.args["ev-recette-pl"] ~= nil then
			if emplacement == "—" then table.insert(infos_jeux, '<tr class="ligne-jeu-recette"><td>' .. frame:expandTemplate{ title = 'Abréviation', args = { jeux[i] } } .. '</td><td colspan="'.. colspan - 2 .. '">Fabricable à la [[Machine à CT]]')
			else table.insert(infos_jeux, '<tr class="ligne-jeu"><td rowspan="2">' .. frame:expandTemplate{ title = 'Abréviation', args = { jeux[i] } } .. '</td><td colspan="' .. colspan - 3 .. '">' .. emplacement .. '</td><td>' .. achat .. '</td><td rowspan="2">' .. vente .. '</td></tr><tr class="ligne-jeu-recette-2"><td colspan="'.. colspan - 2 .. '">Fabricable à la [[Machine à CT]]')
			end
			if frame.args["ev-recette-condition"] ~= nil then table.insert(infos_jeux, ' après ' .. mw.text.trim(frame.args["ev-recette-condition"])) end
			table.insert(infos_jeux, ' pour ' .. mw.text.trim(frame.args["ev-recette-pl"]) .. ' [[PL]] et avec les échantillons suivants :<br />' .. mw.text.trim(frame.args["ev-recette-échantillons"]) .. '</td>')
			if emplacement == "—" then table.insert(infos_jeux, '<td>' .. vente .. '</td>') end
			table.insert(infos_jeux, '</tr>')
		else
			table.insert(infos_jeux, '<tr class="ligne-jeu"><td>' .. frame:expandTemplate{ title = 'Abréviation', args = { jeux[i] } } .. '</td><td colspan="' .. colspan - 3 .. '">' .. emplacement .. '</td><td>' .. achat .. '</td><td>' .. vente .. '</td></tr>')
		end
		i = i + 1
	end

	local poke = {}
	local aucun = "non"
	table.insert(poke, '<tr class="tableau-capsule-pokémon tableau-capsule-generation' .. generation)
	if frame.args['jeu'] then table.insert(poke, '-' .. frame.args['jeu'] .. '') end
	table.insert(poke, '-pokémon">')
	if frame.args['tous'] == "oui" or frame.args['tous'] == "sexués" then
		table.insert(poke, '<td class="cellule-unique" colspan="' .. colspan + 1 .. '">Tout Pokémon')
		if frame.args['tous'] == "sexués" then table.insert(poke, ' sexué') end
		table.insert(poke, ' pouvant apprendre une capacité par capsule peut apprendre ' .. capacite)
		if frame.args['exceptions'] then table.insert(poke, ' sauf ' .. frame.args['exceptions']) end
		table.insert(poke, '.</td>')
	else
		local y = 1
		local z = 1
		local colspan_poke
		if poke_aptes[1] == "" then
			aucun = "oui"
			table.insert(poke, '<td class="cellule-unique" colspan="' .. colspan + 1 .. '">\'\'Incomplet\'\'</td>')
		end
		while poke_aptes[y] and aucun == "non" do
			if generation == 3 and (z == 1 or z == 3 or z == 5) then colspan_poke = 2
			elseif generation == 4 and (z == 3 or z == 4) then colspan_poke = 2
			elseif generation == 9 and z == 6 then colspan_poke = 2
			else colspan_poke = 1
			end
			table.insert(poke, '<td colspan="' .. colspan_poke .. '" style="text-align: left;">' .. ressources.infosPokemon( poke_aptes[y] ) .. '</td>')
			y = y + 1
			 if z == 6 then
			 	table.insert(poke, '</tr><tr class="tableau-capsule-pokémon tableau-capsule-generation' .. generation)
				if frame.args['jeu'] then table.insert(poke, '-' ..frame.args['jeu'] .. '') end
				table.insert(poke, '-pokémon">')
			 	z = 1
			 else z = z + 1 end
		end
	end
	table.insert(poke, '</tr>')

	local donnees = ressources.donneesCapacite(generation, capacite, jeux)
	local resultat = {}
	table.insert(resultat, '<table class="tableaustandard centre ' .. donnees["type"] .. ' tableau-capsule tableau-capsule-generation' .. generation .. '" style="width: 90%;"><caption><h3>' .. titre.. '</h3></caption><tr class="ligne-1">')
	if ressources.indexTableau(jeux, "ev") then table.insert(resultat, '<th>Vignette</th>') end
	table.insert(resultat, '<th>Capacité</th><th>Type</th><th>Catégorie</th><th>Puissance</th><th>Précision</th><th>PP</th>')
	if generation == 3 or generation == 6 then table.insert(resultat, '<th>[[Concours Pokémon|Type concours]]</th><th>[[Charme (concours)|Charme]]</th><th>[[Blocage (concours)|Blocage]]</th>')
	elseif generation == 4 then table.insert(resultat, '<th>[[Concours Pokémon|Type concours]]</th><th>[[Charme (concours)|Charme]]</th>')
	end
	table.insert(resultat, '</tr><tr class="ligne-2">')
	if ressources.indexTableau(jeux, "ev") then
		rowspan = #jeux + 2
		if frame.args["ev-recette-échantillons"] ~= nil and frame.args["ev-recette-pl"] ~= nil then rowspan = rowspan + 1 end
		table.insert(resultat, '<td rowspan="' .. rowspan .. '">[[Fichier:Vignette ' .. frame:getParent():getTitle() .. ' EV.png|150px]]</td>')
	end
	table.insert(resultat, '</tr><tr>')
	for j = 1, colspan do table.insert(resultat, '<td>' .. donnees[j] .. '</td>') end
	table.insert(resultat, '</tr><tr class="ligne-3"><th>Jeu</th><th colspan="' .. colspan - 3 .. '">Emplacement</th><th>Achat</th><th>Vente</th></tr>' .. table.concat(infos_jeux, ""))
	
	local j = 1
	local galerie = {}
	if frame.args['image1'] then table.insert(galerie, '<gallery widths="180px" heights="120px">') end
	while frame.args['image' .. j] do
		if frame.args['image' .. j .. '-description'] == nil then frame.args['image' .. j .. '-description'] = "" end
		table.insert(galerie, frame.args['image' .. j] .. '|' .. frame.args['image' .. j .. '-description'] .. '\n')
		j = j + 1
	end
	if frame.args['image1'] then
		table.insert(galerie, '</gallery>')
		table.insert(resultat, '<tr><td colspan="' .. colspan .. '">' .. frame:preprocess( table.concat(galerie, "") .. '</td></tr>' ) )
	end
	
	table.insert(resultat, '<tr><th colspan=' .. colspan + 1 .. '>Pokémon aptes</th></tr>')
	if frame.args['tous'] ~= "oui" and frame.args['tous'] ~= "sexués" and aucun ~= "oui" then 
		table.insert(resultat, '<tr><td colspan="' .. colspan + 1 .. '"><button type="button" name="tableau-capsule-generation' .. generation .. '')
		if frame.args['jeu'] then table.insert(resultat, '-' .. frame.args['jeu']) end
		table.insert(resultat, '-pokémon" value="afficher" style="font-size:80%" title="[-] Masquer les Pokémon aptes">[+] Afficher les Pokémon aptes</button></td></tr>')
	end
	table.insert(resultat, table.concat(poke, "") .. '</table>')

	return table.concat(resultat, "")
end

return p