Modification de Module:Capacité

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

function p.liste ( frame )
	local pokemon = frame.args[1]
	if pokemon ~= nil then pokemon = mw.text.split( pokemon, "/" ) else pokemon = {} end
	local type = mw.smw.ask('[[' .. frame:getParent():getTitle() .. ']]|?Type')
	if type ~= nil then
		type = type[1]['Type']
		if type ~= nil then type = mw.ustring.lower( string.gsub( type, "É", "é" ) ) else type = '' end
	else type = '' end

	local resultat = {}
	table.insert(resultat, '<table class="tableaustandard ' .. type .. ' capacité-liste"><tr><th colspan="5">Pokémon</th></tr><tr>')
	local i = 1
	local j = 1
	while pokemon[i] do
		if mw.text.trim( pokemon[i] ) ~= "" then
			table.insert(resultat, '<td>' .. ressources.infosPokemon( pokemon[i] ) .. '</td>')
			if j == 4 then
				j = 1
				table.insert(resultat, '</tr><tr>')
			else j = j + 1
			end
		end
		i = i + 1
	end
	table.insert(resultat, '</tr></table>')
	return table.concat(resultat, '')
end

function p.tous( frame )
	local type = mw.smw.ask('[[' .. frame:getParent():getTitle() .. ']]|?Type')
	if type ~= nil then
		type = type[1]['Type']
		if type ~= nil then type = mw.ustring.lower( string.gsub( type, "É", "é" ) ) else type = '' end
	else type = '' end

	local resultat = {}
	table.insert(resultat, '<table class="tableaustandard ' .. type .. '"><tr><th colspan="5">Pokémon</th></tr><tr><td colspan="5">Tout Pokémon')
	if frame.args[1] == "sexués" then table.insert(resultat, ' sexué') end
	table.insert(resultat, ' pouvant apprendre une capacité par capsule peut apprendre cette capacité')
	if frame.args['exceptions'] or frame.args['inclus'] then table.insert(resultat, ' sauf pour les cas particuliers notés ci-dessous') end
	table.insert(resultat, '.</td></tr>')
	
	if frame.args['exceptions'] then
		exceptions = mw.text.split(frame.args['exceptions'], "/")
		table.insert(resultat, '<tr><th colspan="5">Les Pokémon ci-dessous sont inaptes à utiliser la capsule</th></tr>')
		local i = 1
		local j = 1
		while exceptions[i] do
			if mw.text.trim( exceptions[i] ) ~= "" then
				table.insert(resultat, '<td>' .. ressources.infosPokemon( exceptions[i] ) .. '</td>')
				if j == 4 then
					j = 1
					table.insert(resultat, '</tr><tr>')
				else j = j + 1
				end
			end
			i = i + 1
		end
	end
	
	if frame.args['inclus'] then
		inclus = mw.text.split(frame.args['inclus'], "/")
		table.insert(resultat, '<tr><th colspan="5">Les Pokémon ci-dessous peuvent utiliser la capsule, bien qu\'ils soient asexués</th></tr>')
		local i = 1
		local j = 1
		while inclus[i] do
			if mw.text.trim( inclus[i] ) ~= "" then
				table.insert(resultat, '<td>' .. ressources.infosPokemon( inclus[i] ) .. '</td>')
				if j == 4 then
					j = 1
					table.insert(resultat, '</tr><tr>')
				else j = j + 1
				end
			end
			i = i + 1
		end
	end
	
	table.insert(resultat, '</table>')
	return table.concat(resultat, '')
end

function p.niveau ( frame )
	local pokemon = frame.args[1]
	if pokemon ~= nil then pokemon = mw.text.split( pokemon, "\n" ) else pokemon = {} end
	local type = mw.smw.ask('[[' .. frame:getParent():getTitle() .. ']]|?Type')
	if type ~= nil then
		type = type[1]['Type']
		if type ~= nil then type = mw.ustring.lower( string.gsub( type, "É", "é" ) ) else type = '' end
	else type = '' end

	local resultat = {}
	table.insert(resultat, '<table class="tableaustandard ' .. type .. ' capacité-niveau"><tr><th colspan="6">Pokémon</th></tr><tr>')
	local i = 1
	local j = 1
	local y = 1
	local poke = {}
	local infos = ""
	while pokemon[i] do
		poke = mw.text.split( pokemon[i], "/" )
		if poke[1] and poke[2] then
			y = 3
			infos = ""
			while poke[y] do
				poke[y] = mw.text.trim( poke[y] )
				if y % 2 == 0 then
					local maitrise = false
					if mw.ustring.find(poke[y], "M") == 1
					then maitrise = true
						poke[y] = mw.ustring.sub(poke[y], 2)
					end
					if tonumber(poke[y]) ~= nil then poke[y] = "N." .. poke[y] end
					infos = infos .. "<br />"
					if maitrise then infos = infos .. "[[Fichier:Icône Maîtrise LPA.png|x20px|Maîtrise|link=Maîtrise]]" end
					infos = infos .. poke[y]
				else infos = infos .. frame:expandTemplate{ title = 'Sup', args = { poke[y] } }
				end
				y = y + 1
			end
			local prefix_poke2 = ""
			if mw.ustring.find(poke[2], "M") == 1
			then prefix_poke2 = "[[Fichier:Icône Maîtrise LPA.png|x20px|Maîtrise|link=Maîtrise]]"
				poke[y] = mw.ustring.sub(poke[2], 2)
			end
			if tonumber(poke[2]) ~= nil then poke[2] = "N." .. mw.text.trim( poke[2] ) else poke[2] = mw.text.trim( poke[2] ) end
			table.insert(resultat, '<td>' .. ressources.infosPokemon( poke[1] ) .. '</td><td>' .. prefix_poke2 .. poke[2] .. infos .. '</td>')
			if j == 3 then
				j = 1
				table.insert(resultat, '</tr><tr>')
			else j = j + 1
			end
		end
		i = i + 1
	end
	table.insert(resultat, '</tr></table>')
	return table.concat(resultat, "")
end

function p.reproduction ( frame )
	local pokemon = frame.args[1]
	if pokemon ~= nil then pokemon = mw.text.split( pokemon, "\n" ) else pokemon = {} end
	local type = mw.smw.ask('[[' .. frame:getParent():getTitle() .. ']]|?Type')
	if type ~= nil then
		type = type[1]['Type']
		if type ~= nil then type = mw.ustring.lower( string.gsub( type, "É", "é" ) ) else type = '' end
	else type = '' end

	local resultat = {}
	table.insert(resultat, '<table class="tableaustandard centre ' .. type .. '" style="max-width: 100%;"><tr><th>Pokémon</th><th>Parent(s) possible(s)</th></tr><tr>')
	local i = 1
	local j = 1
	local y = 2
	local poke = {}
	local parents = {}
	local enfants = {}
	local miniaturesParents = ""
	local miniaturesEnfants = ""
	while pokemon[i] do
		poke = mw.text.split( pokemon[i], "/" )
		if poke[2] and mw.text.trim( poke[1] ) ~= "" and mw.text.trim( poke[2] ) ~= "" then
			j = 1
			parents = mw.text.split( poke[2], "," )
			miniaturesParents = ""
			while parents[j] do
				miniaturesParents = miniaturesParents .. ressources.infosPokemon( parents[j], "miniature" )
				j = j + 1
			end

			y = 2
			enfants = mw.text.split( poke[1], "," )
			miniaturesEnfants = ressources.infosPokemon( enfants[1] )
			while enfants[y] do
				miniaturesEnfants = miniaturesEnfants .. '<br />' .. ressources.infosPokemon( enfants[y] )
				y = y + 1
			end
			table.insert(resultat, '<tr><td style="text-align: left;">' .. miniaturesEnfants .. '</td><td>' .. miniaturesParents .. '</td></tr>')
		end
		i = i + 1
	end
	table.insert(resultat, '</tr></table>')
	return table.concat(resultat, '')
end

function p.preevolution ( frame )
	local pokemon = frame.args[1]
	if pokemon ~= nil then pokemon = mw.text.split( pokemon, "\n" ) else pokemon = {} end
	local type = mw.smw.ask('[[' .. frame:getParent():getTitle() .. ']]|?Type')
	if type ~= nil then
		type = type[1]['Type']
		if type ~= nil then type = mw.ustring.lower( string.gsub( type, "É", "é" ) ) else type = '' end
	else type = '' end

	local resultat = {}
	table.insert(resultat, '<table class="tableaustandard centre ' .. type .. '"><tr><th>Pokémon</th><th>Pré-évolution</th><th>Méthode d\'apprentissage</th></tr>')
	local i = 1
	while pokemon[i] do
		pokemon[i] = mw.text.split( pokemon[i], "/" )
		if pokemon[i][2] and mw.text.trim( pokemon[i][1] ) ~= "" and mw.text.trim( pokemon[i][2] ) ~= "" then
			if not pokemon[i][3] then pokemon[i][3] = "-" end
			table.insert(resultat, '<tr><td style="text-align: left;">' .. ressources.infosPokemon( pokemon[i][1] ) .. '</td><td style="text-align: left;">' .. ressources.infosPokemon( pokemon[i][2] ) .. '</td><td>' .. pokemon[i][3] .. '</td></tr>')
		end
		i = i + 1
	end
	table.insert(resultat, '</table>')
	return table.concat(resultat, '')
end

function p.evenement ( frame )
	local pokemon = frame.args[1]
	if pokemon ~= nil then pokemon = mw.text.split( pokemon, "\n" ) else pokemon = {} end
	local type = mw.smw.ask('[[' .. frame:getParent():getTitle() .. ']]|?Type')
	if type ~= nil then
		type = type[1]['Type']
		if type ~= nil then type = mw.ustring.lower( string.gsub( type, "É", "é" ) ) else type = '' end
	else type = '' end

	local resultat = {}
	table.insert(resultat, '<table class="tableaustandard centre ' .. type .. '"><tr><th>Pokémon</th><th>Événement(s)</th></tr>')
	local i = 1
	while pokemon[i] do
		pokemon[i] = mw.text.split( pokemon[i], "/" )
		if mw.text.trim( pokemon[i][1] ) ~= "" then
			if not pokemon[i][2] then pokemon[i][2] = 'Inconnu' end
			table.insert(resultat, '<tr><td style="text-align: left;">' .. ressources.infosPokemon( pokemon[i][1] ) .. '</td><td>' .. pokemon[i][2] .. '</td></tr>')
		end
		i = i + 1
	end
	table.insert(resultat, '</table>')
	return table.concat(resultat, '')
end

return p