Module:Adventure Squad Dungeon Pokémon table

Documentation for this module may be created at Module:Adventure Squad Dungeon Pokémon table/doc

--------------------------------------------------------------------------------
--
--				Module:Adventure Squad Dungeon Pokémon table
--
--------------------------------------------------------------------------------

local p = {}
local mw = require('mw')

function p.main(frame)
	local data = require("Module:Adventure Squad Pokémon Data Cell")
	local metatable = {__index = function () return "" end}
	setmetatable(data, metatable)
	local args = frame.args

	local output = mw.html.create()
	local header = mw.html.create()
	-- Header row
	header
		:tag('tr'):done()
			:tag('th'):attr("rowspan", "2"):cssText("width: 10%"):wikitext("Model"):done()
			:tag('th'):attr("colspan", "2"):wikitext(frame:expandTemplate{title = "AdventureSquad", args = {"Pokémon"}}):done()
			:tag('th'):attr("colspan", "2"):attr("rowspan", "2"):cssText("width: 10%"):wikitext(frame:expandTemplate{title = "AdventureSquad", args = {"Type"}}):done()
			:tag('th'):attr("rowspan", "2"):cssText("width: 10%"):wikitext(frame:expandTemplate{title = "AdventureSquad", args = {"Floor", "Floors"}}):done()
			:tag('th'):attr("rowspan", "2"):cssText("width: 10%"):wikitext(frame:expandTemplate{title = "AdventureSquad", args = {"Team Size"}}):done()
			:tag('th'):attr("rowspan", "2"):cssText("width: 10%"):wikitext(frame:expandTemplate{title = "AdventureSquad", args = {"Recruiting", "Base Recruit %"}}):done()
			:tag('th'):attr("rowspan", "2"):cssText("width: 10%"):wikitext(frame:expandTemplate{title = "AdventureSquad", args = {"Exclusive"}}):done()
		:tag('tr'):done()
			:tag('th'):cssText("width: 15%"):wikitext("English"):done()
			:tag('th'):cssText("width: 15%"):wikitext("Japanese"):done()

	-- Data rows
	local rows = mw.html.create()
	local i = 1

	while true do
		local row = mw.html.create()
		local monsterArg = args["pokémon_" .. i]
		local floorsArg = args["floors_" .. i]
		local recruitableArg = args["recruitable_" .. i]

		if not monsterArg and not floorsArg and not recruitableArg then
			break -- exit loop when no more rows are found
		end

		local typeOne = data[monsterArg]["type1"] or ""
		local typeTwo = data[monsterArg]["type2"] or ""
		local exclusive = data[monsterArg]["exclusive"] or ""
		local verColor = ""

		if string.lower(exclusive) == "No" then
			verColor = 'background-color: red'
		elseif string.lower(exclusive) == "yes" then
			verColor = 'background-color: cornflowerblue'
		end

		row
			:tag('tr'):done()
				:tag('th'):wikitext(frame:expandTemplate{title = "AdventureSquadModel", args = {(monsterArg or ""), male = frame:preprocess("{{#switch:{{lc:" .. monsterArg .. "}} |nidoran|ambipom|aipom|alakazam|abomasnow|butterfree|cacturne|camerupt|doduo|mamoswine|magikarp|luxio|luxray|lumineon|raticate|donphan|dustox|hippowdon|heracross|hippopotas|houndoom|hypno|xatu|zubat|wooper|weavile|combusken|dodrio|gulpin|gyarados|pikachu|piloswine|politoed|quagsire|pachirisu|raichu|finneon|floatzel|gabite|garchomp|kadabra|kricketot|kricketune|ledian|ledyba|milotic|medicham|meditite|croagunk|gloom|gligar|girafarig|gible|golbat|goldeen|numel|nuzleaf|octillery|murkrow|meganium|scizor|seaking|staraptor|staravia|starly|steelix|sudowoodo|rhydon|rhyhorn|rhyperior|roselia|sneasel|snover|rattata|relicanth|shinx|buizel|bidoof|bibarel|blaziken|beautifly|venusaur|vileplume|toxicroak|swalot|tangrowth|ursaring|wobbuffet|scyther|roserade|shiftry|ludicolo|vileplume=true}}"), female = frame:preprocess("{{#switch:{{lc:" .. monsterArg .. "}}|combee=true}}"), size = "64x64px"}}):done()
				:tag('td'):wikitext(frame:expandTemplate{title = "AdventureSquad", args = {(monsterArg or "")}}):done()
				:tag('td'):wikitext(data[monsterArg]["ja"]):done()
	
		if typeTwo == nil or typeTwo == '' then
			row
				:tag('th'):attr("colspan", "2"):wikitext(frame:expandTemplate{title = "AdventureSquadType", args = {typeOne}}):done()
		else
			row
				:tag('th'):wikitext(frame:expandTemplate{title = "AdventureSquadType", args = {typeOne}}):done()
				:tag('th'):wikitext(frame:expandTemplate{title = "AdventureSquadType", args = {typeTwo}}):done()
		end

		row
			:tag('td'):wikitext(floorsArg or frame:expandTemplate{title = "Research"}):done()
			:tag('td'):cssText(verColor):wikitext(data[monsterArg]["teamSize"]):done()

		if recruitableArg == nil or recruitableArg == '' or string.lower(recruitableArg) == "false" or string.lower(recruitableArg) == "no" then
			row
				:tag('td'):wikitext("Unrecruitable"):done()
		else
			row
				:tag('td'):wikitext(data[monsterArg]["recruit"]):done()
		end

		row
			:tag('td'):cssText(verColor):wikitext(exclusive):done()

		rows
			:node(row)
			:allDone()

		i = i + 1
	end

	output
		:tag('table'):addClass("wikitable"):cssText("text-align: center; margin: auto; width: 50%;")
			:node(header)
			:node(rows)
	return tostring(output)

end

return p