Module:BrandSeries

Documentation for this module may be created at Module:BrandSeries/doc

-- Module:BrandSeries
-- Generates a series breakdown table for a brand's CRTs
-- Shows only series that have CRTs for the specified brand

local p = {}

local function trim(s)
	if s == nil then return nil end
	s = tostring(s)
	s = s:gsub('^%s+', ''):gsub('%s+$', '')
	if s == '' then return nil end
	return s
end

function p.seriesTable(frame)
	-- Try to get brand from frame.args first (direct #invoke call)
	-- then from parent.args (called through a template)
	local brand = trim(frame.args.brand) or trim(frame.args[1])
	
	if not brand then
		local parent = frame:getParent()
		if parent then
			brand = trim(parent.args.brand) or trim(parent.args[1])
		end
	end
	
	if not brand then
		return '<span class="error">Error: brand parameter required</span>'
	end
	
	-- Query all CRTs for this brand with their series using broadtable
	local crtQuery = string.format(
		'{{#ask:[[Category:CRT models]][[Has brand::%s]][[Has series::+]]|?Has series|format=broadtable|link=none|headers=hide|limit=1000|searchlabel=}}',
		brand
	)
	local crtResult = frame:preprocess(crtQuery)
	
	-- Count occurrences of each series by parsing HTML table
	local seriesCounts = {}
	if crtResult and crtResult ~= '' then
		-- Parse each table row
		for rowContent in crtResult:gmatch('<tr[^>]*>(.-)</tr>') do
			local cells = {}
			for cellContent in rowContent:gmatch('<td[^>]*>(.-)</td>') do
				-- Strip HTML tags
				local cleanContent = cellContent:gsub('<[^>]+>', '')
				table.insert(cells, trim(cleanContent))
			end
			-- cells[1] = page name, cells[2] = series
			if #cells >= 2 and cells[2] then
				local series = cells[2]
				-- Filter out invalid series values like "+"
				if series and series ~= '' and series ~= '+' then
					seriesCounts[series] = (seriesCounts[series] or 0) + 1
				end
			end
		end
	end
	
	-- Convert to sorted array
	local rows = {}
	for series, count in pairs(seriesCounts) do
		table.insert(rows, {
			series = series,
			count = count
		})
	end
	
	-- Sort by series name
	table.sort(rows, function(a, b) return a.series < b.series end)
	
	-- If no series found for this brand
	if #rows == 0 then
		return ""
	end
	
	-- Build the table as HTML
	local tbl = mw.html.create('table')
	tbl:addClass('wikitable')
	
	-- Header row
	local headerRow = tbl:tag('tr')
	headerRow:tag('th'):wikitext('Series')
	headerRow:tag('th'):wikitext('Count')
	headerRow:tag('th'):wikitext('Browse')
	
	-- Data rows
	for _, row in ipairs(rows) do
		local tr = tbl:tag('tr')
		
		-- Series link - build as HTML anchor
		local seriesLink = mw.html.create('a')
		seriesLink:attr('href', '/wiki/' .. mw.uri.encode(row.series, 'WIKI'))
		seriesLink:attr('title', row.series)
		seriesLink:wikitext(row.series)
		tr:tag('td'):node(seriesLink)
		
		-- Count
		tr:tag('td'):wikitext(tostring(row.count))
		
		-- Browse link
		local browseQuery = string.format(
			'{{#queryformlink:form=CRT query|query string=CRT query[brand]=%s&CRT query[series]=%s&_run|link text=Browse}}',
			brand, row.series
		)
		local browseHtml = frame:preprocess(browseQuery)
		tr:tag('td'):wikitext(browseHtml)
	end
	
	return tostring(tbl)
end

return p