Editing Module:Life Guide

local p = {}
local lib = require('Module:Feature')
local checkExists = require('Module:Exists').checkExists
local getTemplateArgs = require('Module:Parser').getTemplateArgs

-- [[Template:Life Guide]]
function p.LifeGuide(frame, args)
	if args == nil then
		args = require('Module:Arguments').getArgs(frame, { parentFirst = true })
	end
	
	local html = mw.html.create()
	if args[1] and type(args[1]) == 'string' then
		for page in lib.gsplit(args[1], ';', { removeEmpty = true }) do
			local lifeguideArgs = getTemplateArgs('Life Guide/' .. page, { only = 'Life Guide' })
			if lib.isEmpty(lifeguideArgs) then
				html:tag('strong'):addClass('error'):wikitext('The page [[Life Guide/', page, ']] is not a valid lifeguide.')
			else
				html
					:tag('div')
						:addClass('lifeguide')
						:addClass(not args.table and title.rootText == 'Life Guide' and 'lifeguide-standalone' or '')
						:tag('dl')
							:tag('dt'):wikitext(lifeguideArgs.title or page)
							:tag('dd'):node(p._LifeGuideFormat(lifeguideArgs))
			end
		end
	else
		html
			:tag('div')
				:addClass('lifeguide')
				:addClass(not args.table and title.rootText == 'Life Guide' and 'lifeguide-standalone' or '')
				:node(p._LifeGuideFormat(args))
	end
	return frame:preprocess(tostring(html))
end

function p._LifeGuideFormat(args)
	local out = mw.html.create('')
	local title = mw.title.getCurrentTitle()
	if args.about then
		args.about = lib.split(args.about, ';')
		if title.rootText == 'Life Guide' then
			out:tag('i'):wikitext(require('Module:Hatnote').main(args.about))
		end
	end
	
	local size = args.size or '200px'
	if (title.rootText == 'Life Guide') then
		size = '400px'
	end
	
	-- create format
	local I = 1
	while (args['image' .. tostring(I)] or args['text' .. tostring(I)]) do
		local image = args['image' .. tostring(I)]
		local text = args['text' .. tostring(I)]
		if (image or text) then
			local row = out:tag('div'):addClass('lifeguide-detail')
			row:wikitext('[[File:', (image or 'Empty.png'), '|', size, ']]')
			row:tag('div'):wikitext((text or '<wbr>'))
			if (args['image' .. tostring(I+1)] or args['text' .. tostring(I+1)]) then
				row:tag('hr'):addClass('mobile-only')
			end
		end
		I = I + 1
	end
	
	-- add categories if correct page
	local subpage = args.title or title.subpageText
	if (not args['table'] and title.baseText == 'Life Guide' and (subpage ~= nil and not lib.inArray({ 'Elements', 'Enemies', 'System', 'Adventure', 'Event', 'Quest', 'Uncategorized' }, subpage))) then
		local cat = mw.html.create()
		cat:wikitext('[[Category:Life Guides|', subpage, ']]')
		
		if args['sort'] then
			cat:wikitext('[[Category:Life Guide|', string.format('%05d', args['sort']), ']]')
		else
			cat:wikitext('[[Category:Life Guide]]')
		end
		
		if lib.inArray({ 'Elements', 'Enemies', 'System', 'Adventure' }, args['type']) then
			cat:wikitext('[[Category:', args['type'], ' Life Guides|', subpage, ']]')
			if not args['sort'] then
				cat:wikitext('[[Category:Life Guide Missing Sort]]')
			end
		elseif args['type'] == 'Event' or args['type'] == 'Quest' then
			cat:wikitext('[[Category:', args['type'], ' Life Guides|', subpage, ']]')
		else
			cat:wikitext('[[Category:Uncategorized Life Guides|', subpage, ']]')
		end
		
		if args.about then
			for _, v in pairs(args.about) do
				cat:wikitext('[[Category:', v, ' Life Guides]]')
			end
		end
		
		local I = 1
		local exists = true
		while (args['image' .. tostring(I)] and exists) do
			if not checkExists('File:' .. args['image' .. tostring(I)]) then
				cat:wikitext('[[Category:Life Guide Missing Image]]')
				exists = false;
			end
			I = I + 1
		end
			
		if not args.nocat then
			out:node(require('Module:Namespace detect')._main{main=cat, module=cat})
		end
	end
	return out
end

-- [[Template:Life Guides by Category Table]]
function p.Table(frame)
	local args = require('Module:Arguments').getArgs(frame, { parentFirst = true })
	local result = mw.html.create('div'):addClass('lifeguide-table')
	local title = mw.title.getCurrentTitle().text
	
	if lib.isEmpty(args.pages) then
		result:tag('span'):wikitext('No [[Life Guide]]s ', (args[1] and 'that match the category selection' or 'are in ' .. title), '.')
	else
		local resultTable = mw.html.create('table'):addClass('article-table')
		local count = 0
		for lifeguidePage in lib.gsplit(args.pages, ';;;', { noTrim = true, removeEmpty = true }) do
			local lifeguideArgs = getTemplateArgs(lifeguidePage, { only = 'Life Guide' })
			for key, value in pairs(lifeguideArgs) do
				if value == '' then lifeguideArgs[key] = nil end	
			end
			lifeguideArgs.PAGENAME = lifeguidePage
			resultTable:node(p._TableRow(frame, lifeguideArgs))
			count = count + 1
		end
		
		local header = result:tag('div'):addClass('lifeguide-header')
		if lib.isNotEmpty(args.header) then
			header:wikitext(tostring(tostring(args.header):gsub('%%PAGES%%', count)))
		else
			header:wikitext(
				'There ',
				(count == 1 and 'is ' or 'are '),
				"'''",
				count,
				"'''",
				' [[Life Guide]]',
				(count == 1 and ' ' or 's '),
				(args[1] and ('that match' .. (count == 1 and 'es' or '') .. ' the category selection:') or ' are in ' .. title)
			)
		end
		result:node(resultTable)
	end
	
	return frame:preprocess(tostring(result))
end

function p._TableRow(frame, args)
	local out = mw.html.create('')
	
	if args.title then
		local header = out:tag('tr'):tag('th')
		header:wikitext('[[', args.PAGENAME, '|', args.title, ']]')
		if args.subtitle then
			header:wikitext(' - ', args.subtitle)
		end
		if args.icon then
			header:wikitext(' [[File:', args.icon, '|25px|link=]]')
		end
		
		local row = out:tag('tr'):tag('td')
		args['table'] = true
		row:node(p.LifeGuide(frame, args))
	end
	return out
end

return p