Jump to content

Module:Rail

From Wikipedia, the free encyclopedia
This is an old revision of this page, as edited by Ythlev (talk | contribs) at 19:19, 28 January 2019 (Created page with 'local p, data, defaultData, lineData, typeData = {}, {} local function dig(...) -- Digs through a table with sub-tables using arguments as keys, returning the v...'). The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.
(diff) ← Previous revision | Latest revision (diff) | Newer revision → (diff)

local p, data, defaultData, lineData, typeData = {}, {}
local function dig(...)
	-- Digs through a table with sub-tables using arguments as keys, returning the value of the last key argument
	-- Analogous to returning a file given a file path with sub-folders
	-- Returns nil if any given sub-table does not exist
	local arg = {...}
	local a, i, n = arg[1], 1, select('#', ...)
	while a and i < n do
		a = a[arg[i + 1]]
		i = i + 1
	end
	return a
end
local function makeInvokeFunc(funcName)
	return function (frame)
		local args = require('Module:Arguments').getArgs(frame)
		if funcName == '_adjacent' then
			local tableTools = require('Module:TableTools')
			args = tableTools.numData(args)
			if args.other then
				args[1] = args[1] or {}
				for k, _ in pairs(args.other) do args[1][k] = args[1][k] or args['other'][k] end
			end
			return p[funcName](tableTools.compressSparseArray(args))
		else
			args.system = args.system or funcName == '_infoboxStation' and args[2] or args[1]
			local s, t = 'Module:Rail/' .. args.system, 'Module:Adjacent stations/' .. args.system
			data = mw.title.new(s).exists and mw.loadData(s) or mw.title.new(t).exists and mw.loadData(t)
			if funcName ~= '_station' and funcName ~= '_infoboxStation' then
				args.line = args.line or args[2]
				args['type'] = args['type'] or args[3]
				defaultData = dig(data, 'lines', '_default')
				if args.line then
					lineData = 
						dig(data, 'lines', args.line) or
						dig(data, 'lines', dig(data, 'aliases', string.lower(args.line)))
				end
				typeData = dig(lineData, 'types', args['type'])
			end
			return p[funcName](args, not data and frame)
		end
	end
end
local function getStationTitle(station, n)
	if station then
		local stationFormat = n and data[n]['station format'] or data['station format']
		local link = stationFormat[station] or stationFormat[1]
		return
			string.match(link, '%[%[.+%]%]') and (string.gsub(link, '%%1', station)) or
			table.concat({'[[', (string.gsub(link, '%%1', station)), '|', station, ']]'})
	end
end
p.main = makeInvokeFunc('_adjacent')
p.adjacent = makeInvokeFunc('_adjacent')
function p._adjacent(args)
	local yesNo = require('Module:Yesno')
	local i18n = require('Module:Adjacent stations/i18n')
	local root, lang = mw.html.create('table'), 'en-GB'
	root:addClass('wikitable adjacent-stations')
	local function renderHeader(stopNoun, systemIcon, systemTitle)
		root
			:tag('tr')
				:tag('th')
					:addClass('hcA')
					:wikitext(i18n[lang]['preceding'](stopNoun))
					:done()
				:tag('th')
					:attr('colspan', 3)
					:css('vertical-align', 'middle')
					:wikitext(systemIcon and systemIcon .. '&nbsp;' .. systemTitle or systemTitle)
					:done()
				:tag('th')
					:addClass('hcA')
					:wikitext(i18n[lang]['following'](stopNoun))
	end
	local function renderSubHeader(subHeader)
		root
			:tag('tr')
				:tag('th')
					:attr('colspan', 5)
					:addClass('hmA')
					:wikitext(subHeader)
	end
	local function renderSideCell(row, rowSpan, adjacent, terminus, oneWay, circular, through, Reverse, note)
		local mainText, subText = mw.html.create('div')
		if adjacent then
			mainText:wikitext(adjacent)
			subText = mw.html.create('div')
			subText:addClass('isA')
			if adjacent == terminus then
				subText:wikitext('Terminus')
			else
				subText:wikitext(oneWay and 'one-way operation' or circular and terminus or i18n[lang]['towards'](terminus))
			end
		else
			mainText:css('font-style', 'italic')
			mainText:wikitext(Reverse and 'Reverses direction' or through and i18n[lang]['through'](through) or 'Terminus')
		end
		row
			:tag('td')
				:attr('rowspan', rowSpan)
				:addClass('bcA')
				:node(mainText)
					:tag('div')
						:css('font-size', 'smaller')
						:wikitext(note)
						:done()
				:node(subText)
	end
	local function renderMidCells(row, rowSpan, colour, backgroundColour, lineTitle, typeTitle, note, transfer)
		if colour then
			colour = string.match(colour, '#') and colour or '#' .. colour
		end
		row
			:tag('td')
				:attr('rowspan', rowSpan)
				:addClass('bbA')
				:css('background-color', colour)
				:done()
			:tag('td')
				:attr('rowspan', rowSpan)
				:addClass('bcA')
				:css('background-color', backgroundColour)
				:wikitext(lineTitle)
					:tag('div')
						:wikitext(typeTitle)
						:done()
					:tag('div')
						:css('font-size', 'smaller')
						:wikitext(note)
						:done()
					:tag('div')
						:addClass('isA')
						:wikitext(i18n[lang]['transfer'](transfer))
						:done()
				:done()
			:tag('td')
				:attr('rowspan', rowSpan)
				:addClass('bbA')
				:css('background-color', colour)
	end
	local function renderNonStopRow(title, colour, isFormer)
		if colour then
			colour = string.match(colour, '#') and colour or '#' .. colour
		end
		root
			:tag('tr')
				:tag('td')
					:attr('colspan', 5)
					:addClass('bcA')
						:tag('div')
							:tag('span')
								:css('border', '1px solid black')
								:css('background-color', colour)
								:wikitext(string.rep('&nbsp;', 4))
								:done()
							:wikitext('&nbsp;', isFormer == true and i18n[lang]['nonstop_past'](title) or i18n[lang]['nonstop_present'](title))
	end
	local function renderNoteRow(note)
		root
			:tag('tr')
				:tag('td')
					:attr('colspan', 5)
					:addClass('bcA')
					:wikitext(note)
	end
	local function getTerminus(termini, n, to)
		if type(termini) == 'string' then
			return getStationTitle(termini, n)
		elseif type(termini) == 'table' then
			local i, t = 1, {}
			while termini[i] and to ~= termini[i] do
				t[i] = getStationTitle(termini[i], n)
				i = i + 1
			end
			return getStationTitle(termini[i], n) or mw.text.listToText(t, nil, ' or ')
		end
	end
	local j, k, l = 2, 2, 2
	for i, v in ipairs(args) do
		args[i]['system'] = args[i]['system'] or args[i - 1]['system']
		data[i] = mw.loadData('Module:Adjacent stations/' .. args[i]['system'])
		lang = data[i]['lang'] or 'en-GB'
		defaultData = function (n) return dig(data[n], 'lines', '_default') end
		if args[i]['line'] then
			args[i]['line'] =
				dig(data[i], 'lines', args[i]['line']) and args[i]['line'] or
				dig(data[i], 'aliases', string.lower(args[i]['line'])) or
				error(i18n[lang]['error_unknown'](args[i]['line']))
		else
			args[i]['line'] = i == 1 and '_default' or args[i - 1]['line']
		end
		lineData = function (n, line) return
			dig(data[n], 'lines', line or args[n]['line']) or
			dig(data[n], 'lines', dig(data[n], 'aliases', string.lower(line or args[n]['line'])))
		end
		typeData = function (n) return dig(lineData(n), 'types', args[n]['type']) end
		local function fallback(parameter, n)
			return dig(typeData(n or i), parameter) or dig(lineData(n or i), parameter) or dig(defaultData(n or i), parameter)
		end
		if i == 1 or args[i]['system'] ~= args[i - 1]['system'] then
			renderHeader(
				data[i]['header stop noun'] or i18n[lang]['stop_noun'],
				data[i]['system icon'],
				data[i]['system title'] or '[[' .. args[i]['system'] .. ']]'
			)
		end
		if v.header then renderSubHeader(v.header) end
		if v.nonstop then
			renderNonStopRow(fallback('title'), fallback('color'), v.nonstop == 'former')
		else
			local row = root:tag('tr')
			if i > j - 2  then
				while args[j] and
					args[j]['left'] == args[i]['left'] and
					args[j]['to-left'] == args[i]['to-left'] and
					args[j]['oneway-left'] == args[i]['oneway-left'] and
					args[j]['note-left'] == args[i]['note-left'] and
					(args[j]['through-left'] or args[j]['through']) == (args[i]['through-left'] or args[i]['through']) and
					(args[j]['reverse-left'] or args[j]['reverse']) == (args[i]['reverse-left'] or args[i]['reverse']) and
					fallback('oneway-left', j) == fallback('oneway-left') and
					fallback('circular', j) == fallback('circular') do
					j = j + 1
				end
				renderSideCell(
					row,
					j - i,
					getStationTitle(v.left, i),
					yesNo(fallback('circular')) and fallback('left terminus') or getTerminus(fallback('left terminus'), i, v['to-left'] or v.to),
					yesNo(v['oneway-left'] or fallback('oneway-left')),
					yesNo(fallback('circular')),
					(v['through-left'] or v['through']) and dig(lineData(i, v['through-left'] or v['through']), 'title'),
					yesNo(v['reverse-left'] or v['reverse']),
					v['note-left']
				)
				j = j + 1
			end
			if i > k - 2 then
				while args[k] and
					args[k]['line'] == args[i]['line'] and
					args[k]['type'] == args[i]['type'] and
					args[k]['note-mid'] == args[i]['note-mid']
					do
					k = k + 1
				end
				renderMidCells(
					row,
					k - i,
					fallback('color'),
					fallback('background color'),
					lineData(i)['title'] or (string.gsub(dig(defaultData(i), 'title'), '%%1', v.line)),
					typeData(i) and typeData(i)['title'],
					v['note-mid'] or lineData(i)['note-mid'],
					getStationTitle(v.transfer, i)
				)
				k = k + 1
			end
			if i > l - 2  then
				while args[l] and
					args[l]['right'] == args[i]['right'] and
					args[l]['to-right'] == args[i]['to-right'] and
					args[l]['oneway-right'] == args[i]['oneway-right'] and
					args[l]['note-right'] == args[i]['note-right'] and
					(args[l]['through-right'] or args[l]['through']) == (args[i]['through-right'] or args[i]['through']) and
					(args[l]['reverse-right'] or args[l]['reverse']) == (args[i]['reverse-right'] or args[i]['reverse']) and
					fallback('oneway-right', l) == fallback('oneway-right') and
					fallback('circular', l) == fallback('circular') do
					l = l + 1
				end
				renderSideCell(
					row,
					l - i,
					getStationTitle(v.right, i),
					yesNo(fallback('circular')) and fallback('right terminus') or getTerminus(fallback('right terminus'), i, v['to-right'] or v.to),
					yesNo(v['oneway-right'] or fallback('oneway-right')),
					yesNo(fallback('circular')),
					(v['through-right'] or v['through']) and dig(lineData(i, v['through-right'] or v['through']), 'title'),
					yesNo(v['reverse-right'] or v['reverse']),
					v['note-right']
				)
				l = l + 1
			end
		end
		if v['note-row'] then renderNoteRow(v['note-row']) end
	end
	return root
end
p.station = makeInvokeFunc('_station')
function p._station(args, frame)
	if data then
		return getStationTitle(args.station or args[2])
	else
		return frame:expandTemplate{
			title = args.system .. ' stations',
			args = {
				['station'] = args.station or args[2],
				['line'] = args.line or args[3],
				['branch'] = args['type'] or args[4]
			}
		}
	end
end
p.line = makeInvokeFunc('_line')
function p._line(args, frame)
	if data then
		return
			typeData and typeData['title'] and table.concat({lineData['title'], ' (', typeData['title'], ')'}) or
			lineData and lineData['title'] or
			defaultData and (string.gsub(defaultData['title'], '%%1', args.line or '_default'))
	else
		return frame:expandTemplate{
			title = args.system .. ' stations',
			args = {
				args.line,
				['branch'] = args['type']
			}
		}
	end
end
p.color = makeInvokeFunc('_color')
function p._color(args, frame)
	if data then
		return
			typeData and typeData['color'] or
			lineData and lineData['color'] or
			defaultData and defaultData['color']
	else
		return frame:expandTemplate{
			title = args.system .. ' color',
			args = {
				args.line,
				['branch'] = args['type']
			}
		}
	end
end
p.icon = makeInvokeFunc('_icon')
function p._icon(args, frame)
	if data then
		local s =
			typedata and typeData['icon'] or
			lineData and lineData['icon'] or
			data and data['system icon']
		return args.link and (string.gsub(s, '%[%[(.+)|.+%]%]', args.link)) or s
	else
		return frame:expandTemplate{
			title = 'Rail-interchange',
			args = {
				args.system,
				args.line,
				['size'] = args.size
			}
		}
	end
end
p.box = makeInvokeFunc('_box')
function p._box(args, frame)
	local root = mw.html.create('div')
	local colour, lineTitle = p._color(args, frame), p._line(args, frame)
	local style = args.style or args.inline
	if colour then
		colour = string.match(colour, '#') and colour or '#' .. colour
	end
	local linkedBox =
		style == 'link' or
		style == 'ldot' or
		style == 'lsquare' or
		style and string.match(style, 'route')
	if linkedBox then
		root:wikitext(string.match(lineTitle, '%[%[.+|'))
	end
	local div, box = mw.html.create('div'), mw.html.create('span')
	root:node(div)
	div:node(box)
	if style == nil then
		div
			:addClass('legend')
			:css('-webkit-column-break-inside', 'avoid')
			:css('page-break-inside', 'avoid')
			:css('break-inside', 'avoid-column')
		box
			:addClass('legend-color')
			:css('display', 'inline-block')
			:css('width', '1.5em')
			:css('height', '1.5em')
			:css('margin', '1px')
	end
	if
		style == 'dot' or
		style == 'ldot' or
		style == 'square' or
		style == 'lsquare' then
		box:css('line-height', 'initial')
	end
	if
		style == 'dot' or
		style == 'ldot' or
		style == 'square' or
		style == 'lsquare' or
		style == 'xroute' then
		box:css('color', colour)
	else
		box:css('background-color', colour)
	end
	if
		style == nil or
		style == 'link' or
		style == 'inline' or style == 'yes' or
		style == 'box' then
		box:css('border', '1px solid black')
	elseif style and string.match(style, 'route') then
		box
			:css('border', '.075em solid ' .. (
				typeData and typeData['border color'] or
				lineData and lineData['border color'] or
				colour
				)
			)
			:css('padding', '0 .3em')
		if style ~= 'route' then
			box:css('border-radius', '.5em')
		end
	end
	if style == 'dot' or style == 'ldot' then
		box:wikitext('●')
	elseif style == 'square' or style == 'lsquare' then
		box:wikitext('■')
	elseif style and string.match(style, 'route') then
		box
			:css('color',
				typeData and typeData['text color'] or
				lineData and lineData['text color'] or
				style == 'xroute' and colour or
				'black'
			)
			:css('font-weight', args.bold == 'no' or 'bold')
			:css('font-size', 'inherit')
			:css('white-space', 'nowrap')
			:wikitext(args.line)
	else
		if style == nil or style == 'small' then
			box:wikitext(string.rep('&nbsp;',1))
		else
			box:wikitext(string.rep('&nbsp;',4))
		end
	end	
	if linkedBox then root:wikitext(']]') end
	if
		style == nil or
		style == 'inline' or style == 'yes' or
		style == 'small' or
		style == 'dot' or
		style == 'square' then
		div:wikitext('&nbsp;', lineTitle)
	end
	return root
end
p.style = makeInvokeFunc('_infoboxStation')
p.infoboxStation = makeInvokeFunc('_infoboxStation')
function p._infoboxStation(args, frame)
	if data then
		local root = mw.html.create('div')
		root
			:css('color', '#222222')
			:css('background-color', '#EFEFEF')
			:css(dig(data, 'infobox station', args[1]))
		return string.match(tostring(root), '<div style="(.*)"></div>')
	else
		return frame:expandTemplate{
			title = args.system .. ' style',
			args = {
				args[1] == 'header' and 'name_format' or 'thbgcolor',
			}
		}
	end
end
return p