Module:Rescue Team HM Moves table

Documentation for this module may be created at Module:Rescue Team HM Moves table/doc

--------------------------------------------------------------------------------
--
--						Module:Rescue Team HM Moves table
--
--------------------------------------------------------------------------------

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

function p.main(frame)
	local data = require("Module:Rescue Team Move 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')
			:tag('th'):attr("colspan", "2"):wikitext(frame:expandTemplate{title = "RescueTeamSprite", args = {"HM", size = "32px"}} .. "<br>" .. frame:expandTemplate{title = "RescueTeam", args = {"HM"}}):done()
			:tag('th'):attr("rowspan", "2"):cssText("width: 10%"):wikitext(frame:expandTemplate{title = "RescueTeam", args = {"Type"}}):done()
			:tag('th'):attr("rowspan", "2"):cssText("width: 20%"):wikitext("Base " .. frame:expandTemplate{title = "RescueTeam", args = {"PP"}}):done()
			:tag('th'):attr("rowspan", "2"):cssText("width: 20%"):wikitext("Range"):done()
		:tag('tr')
			:tag('th'):cssText("width: 25%"):wikitext("English"):done()
			:tag('th'):cssText("width: 25%"):wikitext("Japanese"):done()

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

	while true do
		local row = mw.html.create()
		local moveArg = args["move_" .. i]

		if not moveArg then
			break -- exit loop when no more rows are found
		end
		
		local moveTableArg = moveArg:gsub(" HM", "")

		row
			:tag('tr'):done()
				:tag('td'):wikitext(frame:expandTemplate{title = "RescueTeam", args = {(moveArg or "")}}):done()
				:tag('td'):wikitext(data[moveTableArg]["ja"]):done()
				:tag('th'):wikitext(frame:expandTemplate{title = "RescueTeamType", args = {data[moveTableArg]["type"]}}):done()
				:tag('td'):wikitext(data[moveTableArg]["pp"]):done()
				:tag('td'):wikitext(data[moveTableArg]["range"]):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