Editing Module:PageLocations

local cargo = mw.ext.cargo
local Flags = require('Module:Flags')

local PageLocations = {}

function PageLocations.main(frame)
	local currentTitle = mw.title.getCurrentTitle().text
	local empty = true

	-- Check if there are any interviews for this page
	local tables = 'Locations'
	local fields = '_pageName, franchise, country, city, closed'
	local cargoArgs = {
		where = 'franchise="' .. currentTitle .. '"',
		orderBy = 'country, city, closed'
	}
	local results = cargo.query(tables, fields, cargoArgs)

	local fullString = ''

	-- local container = mw.html.create()
	-- 	:node(mw.html.create('h2'):wikitext('Locations'))
	
	local sponsorshipsIconHtml = '[[File:sponsors-icon.png|30px|alt=Sponsorships Icon]]'
	local container = mw.html.create('div'):addClass('container')
	container
		:node(mw.html.create('div'):addClass('news-icon'):wikitext(sponsorshipsIconHtml .. ' <span>Sponsorships</span>'))
		:node(mw.html.create('h2'):wikitext('Locations'))
	
	if #results > 0 then
		empty = false
		local country = Flags.CountryName(results[1].country)
		fullString = fullString .. '<tabber>|-|' .. country .. '='
	
		local Table = mw.html.create('table')
			:addClass('striped-table')
	
		local titleRow = mw.html.create('tr')
			:addClass('headerRow')
		local nameCell = mw.html.create('th')
			:wikitext('Name')
		local locationCell = mw.html.create('th')
			:wikitext('City')
		local statusCell = mw.html.create('th')
			:wikitext('Status')
		Table:node(titleRow:node(nameCell):node(locationCell):node(statusCell))	
		
		for r = 1, #results do
			local result = results[r]
			
			if results[r-1] and results[r-1].country ~= result.country then 
				country = Flags.CountryName(result.country)
				fullString = fullString .. tostring(mw.clone(Table)) .. '|-|' .. country .. '='
				Table = mw.html.create('table')
					:addClass('striped-table')
			
				titleRow = mw.html.create('tr')
					:addClass('headerRow')
				nameCell = mw.html.create('th')
					:wikitext('Name')
				locationCell = mw.html.create('th')
					:wikitext('City')
				statusCell = mw.html.create('th')
					:wikitext('Status')
				Table:node(titleRow:node(nameCell):node(locationCell):node(statusCell))	
			end
		
			local eventRow = mw.html.create('tr'):addClass('bodyRow')
			local name = mw.html.create('td'):wikitext('[[' .. result._pageName .. ']]')
			-- local city = mw.html.create('td'):wikitext(result.city)
			local flagDiv = mw.html.create('div'):addClass('flag')
			local location = mw.html.create('td')
				:node(flagDiv)
			
			local flag = Flags.icon(result.country)
			flagDiv:wikitext(flag)
			flagDiv:wikitext(result.city)
			
			local status = mw.html.create('td'):wikitext(result.closed == '1' and 'Closed' or 'Open')
			
			-- Table:node(eventRow:node(name):node(city):node(status))
			Table:node(eventRow:node(name):node(location):node(status))
		end
		fullString = fullString .. tostring(Table)
	end
	
	fullString = fullString .. '</tabber>'
	
	if empty then
		return '<span class="d-none"></span>'
	else
		return container:wikitext(frame:preprocess(fullString))
	end
	
end

return PageLocations