มอดูล:Series overview

-- This module implements .
require('strict')
local yesno = require('มอดูล:Yesno')
local HTMLcolor = mw.loadData( 'มอดูล:Color contrast/colors' )
--------------------------------------------------------------------------------
-- SeriesOverview class
-- The main class.
--------------------------------------------------------------------------------
local SeriesOverview = {}
function SeriesOverview.cellspan(SeasonEntries, SeasonEntries_ordered, key, cell, multipart, setspan)
	if setspan ~= nil then return setspan end
	
	local spanlength = 1
	
	local firstEntry = SeasonEntriesSeasonEntries_orderedcell
	if key == 'network' and firstEntry.networkA and not firstEntry.networkB then spanlength = 2 end
	
	for i = cell+1, #SeasonEntries_ordered do
		local entry = SeasonEntriesSeasonEntries_orderedi
		-- Split season, then regular season
		if entry.startA or entry.releasedA then
			if not entrykey..'A' then spanlength = spanlength + 1
			else break end
			if not entrykey..'B' then spanlength = spanlength + 1
			else break end
		else
			if not entrykey and (key == 'network' or ((string.sub(key,0,7) == 'postaux' or string.sub(key,0,3) == 'aux') and (not entry.special or entry.episodes)) or (string.sub(key,0,4) == 'info') and multipart) then
				spanlength = spanlength + 1
			else break end
		end
	end
	return spanlength
end
-- Sorting function
function SeriesOverview.series_sort(op1, op2)
	local n1,s1 = string.match(op1,"(%d+)(%a*)")
	local n2,s2 = string.match(op2,"(%d+)(%a*)")
	local n1N,n2N = tonumber(n1),tonumber(n2)
	if n1N == n2N then
		return s1 "
end
function SeriesOverview.new(frame, args)
	args = args or {}
	
	local initialArticle = args'1' or 
	local categories = 
	local title = mw.title.getCurrentTitle()
	
	local oldDateSupport = true
	-- Create series overview table
	local root = mw.html.create((args.multiseries or not args.series) and 'table' or )
	local cellPadding = '0 8px'
	local basePadding = '0.2em 0.4em'
	root
		:addClass('wikitable')
		:addClass('plainrowheaders')
		:css('text-align', 'center')
		:css('height', '1px')
		:css('display', 'table')
	
	-- Sortable
	if args.sortable then
		root:addClass('sortable');
	end
	
	-- "Series overview" ID
	if args.id then
		root:attr('id', 'Series overview')
	end
	
	-- Width
	if args.width then
		root:css('width', args.width)
	end
	-- Caption
	if args.caption then
		root:tag('caption'):wikitext(frame:expandTemplate{title='Screen reader-only',args={args.caption}})
	end
	-- Extract seasons info and place into a 3D array
	local SeasonEntries = {}
	for k,v in pairs(args) do
		local str, num, str2 = string.match(k, '(^%d*)(%d*)(%a*)')
		if tonumber(k) ~= 1 and num ~=  then 
			-- Special
			local special = false
			if string.sub(str2,1,1) == 'S' then
				special = true
				num = num .. str2
				str2 = 
			end
			-- Add to entries, create if necessary
			if not SeasonEntriesnum then
				SeasonEntriesnum = {}
			end
			SeasonEntriesnumstr .. str2 = v
			if special then
				SeasonEntriesnum'special' = 'y'
			end
		end
	end
	-- Order table by season number
	local SeasonEntries_ordered = {}
	for k in pairs(SeasonEntries) do
		table.insert(SeasonEntries_ordered, k)
	end
	table.sort(SeasonEntries_ordered,SeriesOverview.series_sort)
	
	local firstRow = args.multiseries and {} or SeasonEntriesSeasonEntries_ordered1
	
	-- Colspan calculation for information cells (0 = no info set)
	local numAuxCells = 0
	local numInfoCells = 0
	for i = string.byte('A'), string.byte('Z') do
		local param = 'info' .. string.char(i)
		if argsparam then numInfoCells = numInfoCells + 1 end
	end
	
	-- Use of colors and network
	local noColors = true
	local setNetwork = false
	local allReleased = true
	local noEndDates = true
	
	if (args.multiseries and args.network) then setNetwork = true end
	if (args.multiseries) then allReleased = false end
	
	for i = 1, #SeasonEntries_ordered do
		local season, entry = SeasonEntries_orderedi, SeasonEntriesSeasonEntries_orderedi
		for j0 = string.byte('A')-1, string.byte('Z') do
			local j = string.char(j0)
			if j0 == string.byte('A')-1 then j =  end
			if entry'color' .. j then noColors = false end
			if entry'network' .. j then setNetwork = true end
			if entry'start' .. j then allReleased = false end
			if entry'end' .. j then noEndDates = false end
		end
	end
	
	if title.namespace == 0 and not args.multiseries and not allReleased and noEndDates then
		categories = categories .. 'หมวดหมู่:บทความที่ใช้แม่แบบ:Series overview ด้วยรูปแบบพารามิเตอร์เริ่มต้นที่ไม่ใช้แล้ว'
	end
	
	-- Top info cell
	-- @ = string.char(64), A = string.char(65)
	local topInfoCell = numInfoCells > 0 and string.char(numInfoCells + (string.byte('A') - 1)) or '@'
	
	-- Networks are included if the very first entry sets the first network
	local networkTransclude = args.network_transclude
	if (networkTransclude == 'onlyinclude' and title.fullText == initialArticle) or (networkTransclude == 'noinclude' and title.fullText ~= initialArticle) then
		setNetwork = false
	end
	-- Headers
	do
		if args.multiseries or not args.series then
			local headerRow = root:tag('tr')
			headerRow
				:css('text-align', 'center')
			
			-- Base series/season content on the format of the first date; Series = D M Y, Season = M D, Y
			local matchDMY = false
			local thisStart = firstRow.start or firstRow.startA or firstRow.released or firstRow.releasedA
			if thisStart then
				if string.match(thisStart:gsub(" "," "), '(%d+)%s(%a+)%s(%d+)') then
					matchDMY = true
				end
			end
			
			-- Multiple series header
			if args.multiseries then
				headerRow:tag('th')
					:attr('scope', 'col')
					:css('padding', cellPadding)
					:attr('rowspan', allReleased and 1 or 2)
					:wikitext('ชื่อ')
			end
			
			-- Season header
			headerRow:tag('th')
				:attr('scope', 'col')
				:attr('rowspan', allReleased and 1 or 2)
				:css('min-width', '50px')
				:css('padding', cellPadding)
				:wikitext(args.seriesT or args.seasonT or (matchDMY and 'ชื่อ') or 'ปี')
			
			for _a = 1, 3 do
				if _a == 1 or _a == 3 then
					-- Aux headers
					local auxtype = (_a == 3 and 'post' or ) .. 'aux'
					for i = string.byte('A'), string.byte('Z') do
						local param = auxtype .. string.char(i)
						if argsparam then
							numAuxCells = numAuxCells + 1
							headerRow:tag('th')
								:attr('scope', 'col')
								:css('padding', cellPadding)
								:attr('rowspan', allReleased and 1 or 2)
								:wikitext(argsparam)
						end
					end
				end
				
				if _a == 2 then
					-- Episodes header
					headerRow:tag('th')
						:attr('scope', 'col')
						:attr('rowspan', allReleased and 1 or 2)
						:attr('colspan', 2)
						:css('padding', cellPadding)
						:wikitext(args.episodesT or 'จำนวนตอน')
				end
			end
			-- Originally aired header
			local OriginallyColspan = (not allReleased and setNetwork) and 3 or 2
			local countryBlurb = 
			if args.country then
				countryBlurb = ' (' .. args.country .. ')'
			end
			headerRow:tag('th')
				:attr('scope', (setNetwork and allReleased) and 'col' or 'colgroup')
				:attr('colspan', OriginallyColspan)
				:wikitext('วันที่ออกอากาศครั้งแรก' .. countryBlurb)
			
			-- Network subheader for released series
			if setNetwork and allReleased then
				headerRow:tag('th')
					:attr('scope', 'col')
					:attr('rowspan', allReleased and 1 or 2)
					:css('padding', cellPadding)
					:wikitext('เครือข่าย')
			end
			
			-- Information headers
			if topInfoCell ~= '@' then
				for i = string.byte('A'), string.byte(topInfoCell) do
					local param = 'info' .. string.char(i)
					local infoTransclude = argsparam .. '_transclude'
					if (infoTransclude == 'onlyinclude' and title.fullText == initialArticle) or (infoTransclude == 'noinclude' and title.fullText ~= initialArticle) then else
						headerRow:tag('th')
							:attr('scope', 'col')
							:attr('rowspan', allReleased and 1 or 2)
							:css('padding', cellPadding)
							:wikitext(argsparam)
					end
				end
			end
			
			-- Subheader row
			local subheaderRow = mw.html.create('tr')
			if not allReleased then
				-- First aired subheader
				subheaderRow:tag('th')
					:attr('scope', 'col')
					:wikitext('วันแรก')
				-- Last aired subheader
				subheaderRow:tag('th')
					:attr('scope', 'col')
					:wikitext('วันสุดท้าย')
				
				-- Network subheader for aired series
				if setNetwork then
					subheaderRow:tag('th')
						:attr('scope', 'col')
						:css('padding', cellPadding)
						:wikitext('เครือข่าย')
				end
			end
		
			-- Check for scenarios with an empty subheaderRow
			if not allReleased or numInfoCells > 0 then
				root:node(subheaderRow)
			end
		end
	end
	-- Season rows
	do
		if args.multiseries then
			-- Multi series individual entries
			if args.multiseries ~= "y" then
				root:node(args.multiseries)
			end
		else
			-- One row entries, only categorized in the mainspace
			if title.namespace == 0 and #SeasonEntries == 1 then
				categories = categories .. 'หมวดหมู่:บทความที่ใช้แม่แบบ:Series overview ที่มีเพียงแถวเดียว'
			end
		
			-- Determine number of rows in the whole overview
			local SeasonEntriesRows = 0
			for X = 1, #SeasonEntries_ordered do
				local season, entry = SeasonEntries_orderedX, SeasonEntriesSeasonEntries_orderedX
				local splits = 0
				for i = string.byte('A'), string.byte('Z') do
					local paramS = 'start' .. string.char(i)
					local paramR = 'released' .. string.char(i)
					if entryparamS or entryparamR then splits = splits + 1 end
				end
				if splits == 0 then splits = 1 end
				SeasonEntriesRows = SeasonEntriesRows + splits
			end
			
			for X = 1, #SeasonEntries_ordered do
				local season, entry = SeasonEntries_orderedX, SeasonEntriesSeasonEntries_orderedX
				
				-- Determine number of splits in a season
				local splits = 0
				for i = string.byte('A'), string.byte('Z') do
					local paramS = 'start' .. string.char(i)
					local paramR = 'released' .. string.char(i)
					if entryparamS or entryparamR then splits = splits + 1 end
				end
				local splitSeason = (splits > 1)
				
				-- Season rows for each season
				for k0 = string.byte('A')-1, string.byte('Z') do
					local k = string.char(k0)
					if k0 == string.byte('A')-1 then k =  end
					
					-- Part header
					if entry.part and k ==  then
						root:node(entry.part)
					end
					
					-- New season row
					local seasonRow = (entry'start' .. k or entry'released' .. k) and root:tag('tr') or mw.html.create('tr')
					seasonRow:css('height', '100%')
					
					local borderBottom = '2px solid #8D939A'
					
					-- Series name for group overviews
					if X == 1 and (k ==  or k == 'A') and args.series then
						seasonRow:tag('th')
							:attr('scope', 'row')
							:attr('rowspan', SeasonEntriesRows)
							:wikitext(args.series)
							:css('border-bottom', borderBottom)
					end
					if X == #SeasonEntries_ordered and args.series then
						seasonRow:css('border-bottom', borderBottom)
					end
					
					-- Season number link, included only in the first row
					local cellColor
					if not noColors then
						if entry'color' .. k ~= nil and HTMLcolorentry'color' .. k == nil then 
							entry'color' .. k = '#'..(mw.ustring.match(entry'color' .. k, '^%s#*(a-fA-F0-9*)%s*$') or )
						end
						if splitSeason then
							if entry.color then
								cellColor = entry.color
							else
								cellColor = "linear-gradient(to bottom"
								for i = 0, splits-1 do
									local _color = 'color' .. string.upper(string.char(i+97))
									cellColor = cellColor .. ", " .. (entry_color or 'rgba(0,0,0,0)') .. " " .. (100/splits *i) .. "%"
														  .. ", " .. (entry_color or 'rgba(0,0,0,0)') .. " " .. (100/splits *(i+1)) .. "%"
								end
								cellColor = cellColor .. ")"
							end
						else
							cellColor = entry'color' .. k
						end
					end
					
					if k ==  or k == 'A' then
						local colorWidth = '14px'
						
						-- Overall table cell
						local cellRow = mw.html.create(args.series and 'td' or 'th')
							:attr('scope', splitSeason and 'rowgroup' or 'row')
							:attr('rowspan', splitSeason and splits or nil)
							:attr('colspan', entry.special and not entry.episodes and 3+numAuxCells or 1)
							:css('height', 'inherit')
							:css('padding', '0')
							
						-- Overall inner span
						local spanRow = mw.html.create('span')
						spanRow
							:css('width: 100%')
							:css('text-align', 'center')
							:css('float', 'left')
							:css('width', '100%')
							:css('height', '100%')
						
						-- Coloured nested span
						local spanRow2 = mw.html.create('span')
						spanRow2
							:css('width', colorWidth)
							:css('background', cellColor)
							:css('color', '#202122')
							:css('height', '100%')
							:css('float', 'left')
							:css('box-shadow', 'inset -1px 0 #A2A9B1')
						
						-- Link nested span
						local spanRow3 = mw.html.create('span')
						spanRow3
							:css('height', '100%')
							:css('width', not noColors and 'calc(100% - ' .. colorWidth .. ' - 8px)' or '100%')
							:css('display', 'flex')
							:css('vertical-align', 'middle')
							:css('align-items', 'center')
							:css('justify-content', 'center')
							:css('padding', not noColors and '0 4px' or )
							
						local spanRow4 = mw.html.create('span')
						spanRow4
							:addClass('nowrap')
						
						-- Coloured span first into the overall span
						if not noColors then
							spanRow:node(spanRow2)
						end
						-- Link into the blank span
						spanRow4:wikitext((entry.link and  .. (entry.linkT or season) ..  or (entry.linkT or season)) .. (entry.linkR or ))
						-- Blank span into the Link nested span
						spanRow3:node(spanRow4)
						-- Link span second into the overall span
						spanRow:node(spanRow3)
						-- Overall span into the actual cell
						cellRow:node(spanRow)
						-- The actual cell into the season row
						seasonRow:node(cellRow)
					end
					
					for _a = 1, 3 do
						if _a == 1 or _a == 3 then
							-- Aux headers
							local auxtype = (_a == 3 and 'post' or ) .. 'aux'
							-- Aux cells
							for i = string.byte('A'), string.byte('Z') do
								local param = auxtype .. string.char(i)
								if entryparam .. k then
									local thisCell = SeriesOverview.season_cell(entryparam .. k, frame)
										:attr('scope', 'col')
										:attr('rowspan', SeriesOverview.cellspan(SeasonEntries, SeasonEntries_ordered, param, X, (args.series and true or false), entryparam .. k .. 'span' or nil))
										:css('padding', cellPadding)
									seasonRow:node(thisCell)
								end
							end
						end
						
						if _a == 2 then
							-- Episodes counts
							if ((splitSeason and k == 'A' and entry.episodes ~= 'hide') or not splitSeason) then
								if entry.episodes then
									local thisCell = SeriesOverview.season_cell(entry.episodes, frame)
										:attr('colspan', (splitSeason and entry.episodesA ~= 'hide') and 1 or 2)
										:attr('rowspan', splitSeason and splits or nil)
									seasonRow:node(thisCell)
								elseif not entry.special then
									local infoCell = SeriesOverview.series_attributes( frame:expandTemplate{title='N/A',args={'TBA'}} )
									infoCell
										:attr('colspan', (splitSeason and entry.episodesA ~= 'hide') and 1 or 2)
										:attr('rowspan', splitSeason and splits or nil)
									seasonRow:node(infoCell)
								end
							end
							if splitSeason and entry.episodesA ~= 'hide' then
								if entry'episodes' .. k then
									local thisCell = SeriesOverview.season_cell(entry'episodes' .. k, frame)
										:attr('colspan', (entry.episodes ~= 'hide') and 1 or 2)
									seasonRow:node(thisCell)
								else
									local infoCell = SeriesOverview.series_attributes( frame:expandTemplate{title='N/A',args={'TBA'}} )
										:attr('colspan', (entry.episodes ~= 'hide') and 1 or 2)
									seasonRow:node(infoCell)
								end
							end
						end
					end
					
					-- Start date
					if entry'start' .. k or entry'released' .. k then
						local thisCell = oldDateSupport and (
						SeriesOverview.season_cell(entry'start' .. k or entry'released' .. k, frame)
						) or (
						SeriesOverview.season_cell((not allReleased or entry'end' .. k) and entry'start' .. k or entry'released' .. k, frame)
						)
						thisCell:attr('colspan', oldDateSupport and (
							((not entry.special and entry'released' .. k) or (entry.special and not entry'end' .. k) or allReleased) and 2 or 1
							) or (
							(entry'released' .. k or allReleased) and 2 or 1
							))
							:css('padding',basePadding)
						seasonRow:node(thisCell)
					else
						local infoCell = SeriesOverview.series_attributes( frame:expandTemplate{title='N/A',args={'TBA'}} )
						infoCell:css('padding',basePadding)
						seasonRow:node(infoCell)
					end
					
					-- End date
					local canIncludeEnd = oldDateSupport and (
					(not allReleased and not entry'released' .. k and ((entry.special and entry'end' .. k) or not entry.special)) and 'yes' or 'no'
					) or (
					(not allReleased and not entry'released' .. k and ((entry.special and entry'end' .. k) or not entry.special)) and 'yes' or 'no'
					)
					if canIncludeEnd == 'yes' then
						if entry'end' .. k then
							local thisCell = SeriesOverview.season_cell(entry'end' .. k, frame)
								:css('padding',cellPadding)
							seasonRow:node(thisCell)
						else
							local infoCell = SeriesOverview.series_attributes( frame:expandTemplate{title='N/A',args={'TBA'}} )
							infoCell:css('padding',cellPadding)
							seasonRow:node(infoCell)
						end
					end
					
					-- Network
					if entry'network' .. k and setNetwork then
						local thisCell = SeriesOverview.season_cell(entry'network' .. k, frame)
							:attr('rowspan', SeriesOverview.cellspan(SeasonEntries, SeasonEntries_ordered, 'network', X, (args.series and true or false), entry'network' .. k .. 'span' or nil))
						seasonRow:node(thisCell)
					end
					
					-- Information
					for i = string.byte('A'), string.byte(topInfoCell) do
						local param0 = 'info' .. string.char(i)
						local param = 'info' .. string.char(i) .. k
						
						local infoTransclude = argsparam .. '_transclude'
						if (infoTransclude == 'onlyinclude' and title.fullText == initialArticle) or (infoTransclude == 'noinclude' and title.fullText ~= initialArticle) then else
							local infoParam = entryparam
							
							if infoParam and splitSeason and k ==  and not entryparam .. 'A' then
								entryparam .. 'A' = entryparam
								entryparam .. 'spanning' = 'y'
							end
							
							local rowspan = (entryparam0 .. 'spanning' and splits) or
											(args.series and SeriesOverview.cellspan(SeasonEntries, SeasonEntries_ordered, param0, X, (args.series and true or false), entryparam0 .. 'span' or nil))
											or nil
							
							if k == 'A' or (k ~= 'A' and not entryparam0 .. 'spanning') then
								-- Cells with ... already expanded
								if infoParam then
									if string.sub(infoParam,1,5) == 'style' then
										local infoCell = SeriesOverview.series_attributes(infoParam)
										infoCell:attr('rowspan', rowspan)
										seasonRow:node(infoCell)
									else
										-- Unstyled content info cell
										local thisCell = SeriesOverview.season_cell(infoParam, frame)
											:attr('rowspan', rowspan)
										seasonRow:node(thisCell)
									end
								else
									if not args.series then
										local infoCell = SeriesOverview.series_attributes( frame:expandTemplate{title='N/A',args={'TBA'}} )
										infoCell:attr('rowspan', rowspan)
										seasonRow:node(infoCell)
									end
								end
							elseif not entryparam0 .. 'spanning' then
								if not args.series then
									local infoCell = SeriesOverview.series_attributes( frame:expandTemplate{title='N/A',args={'TBA'}} )
									infoCell:attr('rowspan', rowspan)
									seasonRow:node(infoCell)
								end
							end
						end
					end
				
				end -- End k0 string.byte
			end -- End 'for' SeasonEntries_ordered
		end -- End 'if' multiseries
	end -- End 'do' season rows
	
	local rootdiv
	if args.multiseries or not args.series then
		rootdiv = mw.html.create('div')
		rootdiv
			:css('display', 'block')
			:css('overflow-x', 'auto')
		rootdiv:node(root)
		rootdiv = tostring(rootdiv)
	else
		rootdiv = tostring(root)
	end
	
	if args.dontclose then 
		rootdiv = mw.ustring.gsub(rootdiv, "", "")
		rootdiv = mw.ustring.gsub(rootdiv, "", "")
	end
	return rootdiv .. categories
end
--------------------------------------------------------------------------------
-- Exports
--------------------------------------------------------------------------------
local p = {}
function p.main(frame)
	local args = require('มอดูล:Arguments').getArgs(frame, {
		wrappers = 'แม่แบบ:Series overview'
	})
	return SeriesOverview.new(frame, args)
end
function p._end(frame)
	return SeriesOverview.endtable()
end
return p