Editing Module:Sandbox/User:Californ1a/T/Professions

require('strict')
require('Module:Mw.html extension')
local pt = require('Module:Paramtest')
local yn = require('Module:Yesno')
local lang = mw.language.getContentLanguage()
local p = {}

local icon_size = 20
local ep_icon_size = 16

-- List episodes with their professions
local episodes = {
	hopeport = {"guard", "chef", "fisher", "forager", "alchemist"},
	hopeforest = {"scout", "gatherer", "woodcutter", "carpenter"},
	mine = {"minefighter", "bonewright", "miner", "blacksmith", "stonemason"},
	crenopolis = {"watchperson", "detective", "leatherworker", "merchant"}
}

local ep_dict = {
	hopeport = 'Hopeport',
	hopeforest = 'Hopeforest',
	mine = 'Mine of Mantuban',
	crenopolis = 'Crenopolis'
}

local ep_color = {
	hopeport = '#b99f46',
	hopeforest = '#84b876',
	mine = '#68ABF1',
	crenopolis = '#979797'
}

-- Get professions levels given an episode name
function p.filter_by_episode(professions, episodeName, goals)
	goals = goals or false
	local filtered = {}
	local episodeProfessions = episodes[episodeName]

	if not episodeProfessions then
		error('Invalid episode name: ' .. tostring(episodeName))
	end

	-- Add the professions and the total for the episode
	for _, profession in ipairs(episodeProfessions) do
		local val = profession
		if goals then
			val = profession..' goal'
		end
		if professions[val] ~= nil then
			filtered[val] = professions[val]
		end
	end

	-- Add the total for the episode
	if professions[episodeName] ~= nil then
		filtered[episodeName] = professions[episodeName]
	end

	return filtered
end

-- Template entrypoint
function p.main(frame)
	return p._main(frame:getParent().args)
end

function p._main(args)
	-- Create main profession and goals lists
	local professions = {
		total = 0
	}
	local goals = {}
	
	-- Build the professions + goals lists
	for episode, keys in pairs(episodes) do
		professions[episode] = 0
		for _, key in ipairs(keys) do
			-- Add the profession and goal to their respective lists
			professions[key] = pt.default_to(tonumber(args[key]), 0)
			goals[key] = pt.default_to(tonumber(args[key..' goal']), 0)
			
			-- Add this profession's level to the episode's total
			local value = professions[key] or 0
			professions[episode] = professions[episode] + value
		end
		-- Add this episode's total to the overall total
		professions.total = professions.total + professions[episode]
	end
	
	-- Non-profession config params
	local other = {
		vertical = yn(pt.default_to(args.vertical, true), true),
		free = yn(pt.default_to(args.free, false), false),
		date = pt.default_to(args.date, '')
	}
	
	--local debug_str = '<pre>'..mw.text.jsonEncode(professions, mw.text.JSON_PRETTY)..', '..mw.text.jsonEncode(goals, mw.text.JSON_PRETTY)..'</pre>'

	return tostring(p.create_table(professions, goals, other))--..debug_str
end

function p.create_row(tbl, episode, professions, goals, vertical)
	if vertical then
		tbl:tag('tr')
	end
	
	-- Create icons
	local cache = {}
	for _,profession in ipairs(episodes[episode]) do
		local PRO = pt.ucflc(profession)
		local size = icon_size
		if profession == 'alchemist' then -- Alchemist icon is a bit larger than others
			size = icon_size - 5
		end
		local icon = mw.ustring.format('[[File:%s small icon.png|link=%s|%spx]]', PRO, PRO, size)
		cache[profession] = {
			name = PRO,
			icon = icon
		}
		if vertical then
			tbl:tag('td')
				:css({border='none'})
				:wikitext(icon)
			:done()
		end
	end
	
	if vertical then
		tbl:done() -- /tr
	end
	tbl:tag('tr')
	
	-- Create levels
	for _,profession in ipairs(episodes[episode]) do
		local PRO = cache[profession].name
		local lvl = tostring(professions[profession])
		if goals[profession] ~= 0 then
			lvl = lvl..'/'..tostring(goals[profession])
		end
		
		if vertical then
			tbl:tag('td')
				:css({border='none'})
				:wikitext(lvl)
			:done()
		else
			tbl:tag('td')
				:wikitext(cache[profession].icon..' '..lvl)
			:done()
		end
	end
	
	return tbl:done()
end

function p.create_episode_table(episode, professions, goals, other)
	local profession_count = #episodes[episode]
	local EP = ep_dict[episode]
	local out = mw.html.create('tr'):tag('td'):tag('table')
		:addClass('wikitable')
		:addClass('center')
		:css({margin='0', padding='0'})
		:tag('tr')
			:tag('th')
				:css({['background-color']=ep_color[episode]})
				:attr({colspan = profession_count})
				:wikitext(mw.ustring.format('[[File:%s episode icon.png|link=%s|%spx]] [[%s]] (%s)', EP, EP, ep_icon_size, EP, professions[episode]))
			:done()
		:done()
		
	out = p.create_row(out, episode, professions, goals, other.vertical)
	
	return out:done():done():done() -- /table /td /tr
end

function p.create_table(professions, goals, other)
	local ep_tbl = {}
	for episode in pairs(episodes) do
		local ep_lvls = p.filter_by_episode(professions, episode)
		local ep_goals = p.filter_by_episode(goals, episode)
		ep_tbl[episode] = p.create_episode_table(episode, ep_lvls, ep_goals, other)
	end
	
	local out = mw.html.create('table')
		:tag('caption')
			:css({['font-weight']='bold'})
			:wikitext(mw.ustring.format('[[File:Professions icon.png|link=Professions|%spx]] [[Professions]] (%s)', ep_icon_size, professions.total))
		:done()
		
		-- Inner episode tables
		:node(ep_tbl.hopeport)
		:node(ep_tbl.hopeforest)
		:IF(not other.free)
			:node(ep_tbl.mine)
			:node(ep_tbl.crenopolis)
		:END()
		
		:IF(pt.has_content(other.date))
			:tag('tr')
				:tag('td')
					:css({['text-align']='center'})
					:tag('small')
						:tag('b')
							:wikitext('Last Updated:')
						:done()
						:tag('br'):done()
						:IF(other.date == 'now')
							:wikitext(lang:formatDate('j F Y h:i:s', mw.getCurrentFrame():preprocess('{{REVISIONTIMESTAMP}}')))
						:ELSE()
							:wikitext(other.date)
						:END()
					:done()
				:done()
			:done()
		:END()
		
	return out:allDone()
end

return p
@@ Line 11: / Line 11: @@
 -- List episodes with their professions
 local episodes = {
+	hopeport = {"guard", "chef", "fisher", "forager", "alchemist"},
-	{
+	hopeforest = {"scout", "gatherer", "woodcutter", "carpenter"},
-		free = true,
+	mine = {"minefighter", "bonewright", "miner", "blacksmith", "stonemason"},
-		name = 'Hopeport',
+	crenopolis = {"watchperson", "detective", "leatherworker", "merchant"}
-		color = '#b99f46',
-		professions = {'guard', 'chef', 'fisher', 'forager', 'alchemist'},
-	}, {
-		free = true,
-		name = 'Hopeforest',
-		color = '#84b876',
-		professions = {'scout', 'gatherer', 'woodcutter', 'carpenter'},
-	}, {
-		name = 'Mine of Mantuban',
-		color = '#68ABF1',
-		professions = {'minefighter', 'bonewright', 'miner', 'blacksmith', 'stonemason'},
-	}, {
-		name = 'Crenopolis',
-		color = '#979797',
-		professions = {'watchperson', 'detective', 'leatherworker', 'merchant'},
-	}, {
-		released = false,
-		name = 'Stonemaw Hill',
-		color = '#C57F41',
-		professions = {'shieldbearer', 'builder', 'armorer', 'delver'},
-	}, {
-		released = false,
-		name = 'Bleakholm Crags',
-		color = '#68ABF1',
-		professions = {},
-	},
 }
+local ep_dict = {
--- Get professions levels given an episode order number
+	hopeport = 'Hopeport',
-function p.filter_by_episode(professions, episodeOrder)
+	hopeforest = 'Hopeforest',
+	mine = 'Mine of Mantuban',
+	crenopolis = 'Crenopolis'
+}
+local ep_color = {
+	hopeport = '#b99f46',
+	hopeforest = '#84b876',
+	mine = '#68ABF1',
+	crenopolis = '#979797'
+}
+-- Get professions levels given an episode name
+function p.filter_by_episode(professions, episodeName, goals)
+	goals = goals or false
 	local filtered = {}
-	local episodeProfessions = episodes[episodeOrder].professions
+	local episodeProfessions = episodes[episodeName]
+	if not episodeProfessions then
+		error('Invalid episode name: ' .. tostring(episodeName))
+	end
 	-- Add the professions and the total for the episode
 	for _, profession in ipairs(episodeProfessions) do
+		local val = profession
-		if professions[profession] ~= nil then
+		if goals then
-			filtered[profession] = professions[profession]
+			val = profession..' goal'
+		end
+		if professions[val] ~= nil then
+			filtered[val] = professions[val]
 		end
 	end
 	-- Add the total for the episode
+	if professions[episodeName] ~= nil then
-	local ep_name = episodes[episodeOrder].name:lower()
+		filtered[episodeName] = professions[episodeName]
-	if professions[ep_name] ~= nil then
-		filtered[ep_name] = professions[ep_name]
 	end
@@ Line 76: / Line 73: @@
 	-- Build the professions + goals lists
-	for i, episode in ipairs(episodes) do
+	for episode, keys in pairs(episodes) do
+		professions[episode] = 0
-		local ep_name = episode.name:lower()
+		for _, key in ipairs(keys) do
-		professions[ep_name] = 0
-		for _, profession in ipairs(episode.professions) do
 			-- Add the profession and goal to their respective lists
-			professions[profession] = pt.default_to(tonumber(args[profession]), 0)
+			professions[key] = pt.default_to(tonumber(args[key]), 0)
-			goals[profession] = pt.default_to(tonumber(args[profession..' goal']), 0)
+			goals[key] = pt.default_to(tonumber(args[key..' goal']), 0)
 			-- Add this profession's level to the episode's total
-			local value = professions[profession] or 0
+			local value = professions[key] or 0
-			professions[ep_name] = professions[ep_name] + value
+			professions[episode] = professions[episode] + value
 		end
 		-- Add this episode's total to the overall total
-		professions.total = professions.total + professions[ep_name]
+		professions.total = professions.total + professions[episode]
 	end
@@ Line 96: / Line 92: @@
 		vertical = yn(pt.default_to(args.vertical, true), true),
 		free = yn(pt.default_to(args.free, false), false),
-		date = pt.default_to(args.date, ''),
+		date = pt.default_to(args.date, '')
-		right = yn(pt.default_to(args.right, false), false),
-		infobox = yn(pt.default_to(args.infobox, false), false),
 	}
@@ Line 106: / Line 100: @@
 end
-function p.create_row(tbl, episodeOrder, professions, goals, vertical)
+function p.create_row(tbl, episode, professions, goals, vertical)
 	if vertical then
 		tbl:tag('tr')
@@ Line 113: / Line 107: @@
 	-- Create icons
 	local cache = {}
-	for _,profession in ipairs(episodes[episodeOrder].professions) do
+	for _,profession in ipairs(episodes[episode]) do
 		local PRO = pt.ucflc(profession)
 		local size = icon_size
@@ Line 138: / Line 132: @@
 	-- Create levels
-	for _, profession in ipairs(episodes[episodeOrder].professions) do
+	for _,profession in ipairs(episodes[episode]) do
 		local PRO = cache[profession].name
 		local lvl = tostring(professions[profession])
@@ Line 160: / Line 154: @@
 end
-function p.create_episode_table(episodeOrder, professions, goals, other)
+function p.create_episode_table(episode, professions, goals, other)
-	local ep_professions = episodes[episodeOrder].professions
+	local profession_count = #episodes[episode]
-	local profession_count = #ep_professions
+	local EP = ep_dict[episode]
-	local EP = episodes[episodeOrder].name
-	local ep_name = episodes[episodeOrder].name:lower()
 	local out = mw.html.create('tr'):tag('td'):tag('table')
+		:addClass('wikitable')
 		:addClass('center')
+		:css({margin='0', padding='0'})
-		:IF(other.infobox)
-			:addClass('infobox')
-			:addClass('infobox-'..EP:gsub(' ', '_'))
-			:css({
-				margin = '0',
-				['table-layout'] = 'auto',
-				width = '100%',
-				float = 'unset',
-			})
-		:ELSE()
-			:addClass('wikitable')
-			:css({margin='0', padding='0'})
-		:END()
 		:tag('tr')
 			:tag('th')
+				:css({['background-color']=ep_color[episode]})
-				:IF(other.infobox)
-					:addClass('infobox-header')
-					:css({padding='0.5em'})
-				:ELSE()
-					:css({
-						['background-color'] = episodes[episodeOrder].color,
-						padding='0 0.5em',
-					})
-				:END()
 				:attr({colspan = profession_count})
-				:wikitext(mw.ustring.format('[[File:%s episode icon.png|link=%s|%spx]] [[%s]] (%s)', EP, EP, ep_icon_size, EP, professions[ep_name]))
+				:wikitext(mw.ustring.format('[[File:%s episode icon.png|link=%s|%spx]] [[%s]] (%s)', EP, EP, ep_icon_size, EP, professions[episode]))
 			:done()
 		:done()
-	out = p.create_row(out, episodeOrder, professions, goals, other.vertical)
+	out = p.create_row(out, episode, professions, goals, other.vertical)
 	return out:done():done():done() -- /table /td /tr
-end
-local function getReleasedEpisodes()
-	local filtered = {}
-	for i, episode in ipairs(episodes) do
-		if episode.released ~= false then
-			table.insert(filtered, episode)
-		end
-	end
-	return filtered
 end
 function p.create_table(professions, goals, other)
 	local ep_tbl = {}
+	for episode in pairs(episodes) do
-	local released_episodes = getReleasedEpisodes()
+		local ep_lvls = p.filter_by_episode(professions, episode)
-	for i in ipairs(released_episodes) do
-		local ep_lvls = p.filter_by_episode(professions, i)
+		local ep_goals = p.filter_by_episode(goals, episode)
+		ep_tbl[episode] = p.create_episode_table(episode, ep_lvls, ep_goals, other)
-		local ep_goals = p.filter_by_episode(goals, i)
-		table.insert(ep_tbl, p.create_episode_table(i, ep_lvls, ep_goals, other))
 	end
@@ Line 226: / Line 188: @@
 		:done()
+		-- Inner episode tables
+		:node(ep_tbl.hopeport)
+		:node(ep_tbl.hopeforest)
+		:IF(not other.free)
+			:node(ep_tbl.mine)
+			:node(ep_tbl.crenopolis)
+		:END()
+		:IF(pt.has_content(other.date))
-	for i, tbl in ipairs(ep_tbl) do
+			:tag('tr')
-		if (not other.free and episodes[i].free ~= true)
+				:tag('td')
-			or episodes[i].free == true then
+					:css({['text-align']='center'})
-			out:node(tbl)
+					:tag('small')
-		end
+						:tag('b')
-	end
+							:wikitext('Last Updated:')
+						:done()
-	if pt.has_content(other.date) then
-		out:tag('tr')
+						:tag('br'):done()
-			:tag('td')
+						:IF(other.date == 'now')
+							:wikitext(lang:formatDate('j F Y h:i:s', mw.getCurrentFrame():preprocess('{{REVISIONTIMESTAMP}}')))
-				:css({['text-align']='center'})
-				:tag('small')
+						:ELSE()
-					:tag('b')
+							:wikitext(other.date)
-						:wikitext('Last Updated:')
+						:END()
 					:done()
-					:tag('br'):done()
-					:IF(other.date == 'now')
-						:wikitext(lang:formatDate('j F Y', mw.getCurrentFrame():preprocess('{{REVISIONTIMESTAMP}}')))
-					:ELSE()
-						:wikitext(other.date)
-					:END()
 				:done()
 			:done()
-		:done()
+		:END()
-	end
-	out:allDone()
+	return out:allDone()
-	local div = out
-	if other.right then
-		div = mw.html.create('div')
-			:css({float='right'})
-			:node(out)
-		:done()
-	end
-	return div
 end