Module:Sandbox/Ythlev/Adjacent stations

local p, data, defaultData, lineData, typeData, funcIsAdjacent = {} 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, n = {...}, select('#', ...) local a, i = arg[1], 1 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 funcIsAdjacent = true 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 args[1] if args.system then local s = 'Module:Rail/' .. args.system local t = '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) end if funcName ~= '_infoboxStation' then args.line = args.line or args[funcName == '_station' and 3 or 2] args['type'] = args['type'] or args[funcName == '_station' and 4 or 3] if funcName ~= '_station' then 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']) if funcName == '_icon' or funcName == '_box' then args.style = args.style or args.inline or							dig(typeData, 'icon format') or							dig(lineData, 'icon format') or							dig(data, 'system icon format') args.style = args.style == 'image' and nil or args.style funcName = args.style and '_box' end end end return p[funcName](args, not data and frame) end end end p.style = makeInvokeFunc('_infoboxStation') p.infoboxStation = makeInvokeFunc('_infoboxStation') function p._infoboxStation(args, frame) local root = mw.html.create('div') if args[3] == '_subheader' then root:css('background-color', '#EFEFEF') end if args.system then if args[3] == '_subheader' then root:css('color', 'white') end if data then root:css(				dig(data, 'infobox station', args[2], args[3]) or				dig(data, 'infobox station', args[3])			) else if args[3] == '_header' then root:cssText(frame:expandTemplate{					title = args.system .. ' style',					args = {'name_format'}				}) else local thcolor = frame:expandTemplate{ title = args.system .. ' style', args = {'thcolor'} }				root:css('color', thcolor and '#' .. thcolor) local thbgcolor = frame:expandTemplate{ title = args.system .. ' style', args = {'thbgcolor'} }				root:css('background-color', thbgcolor and '#' .. thbgcolor) end end end return string.match(tostring(root), ' ') end local function stationTitle(station, line, typ, n)	if station and data then local data, link = funcIsAdjacent and data[n] or data if type(data['station format']) == 'table' then local defaultFormat = data['station format'] if type(defaultFormat[station]) == 'table' then local stationFormat = defaultFormat[station] if line then line = stationFormat[line] and line or						dig(data, 'aliases', string.lower(line)) end if type(stationFormat[line]) == 'table' then local lineFormat = stationFormat[line] link = lineFormat[typ] or lineFormat[1] else link = stationFormat[line] or stationFormat[1] end else link = defaultFormat[station] or defaultFormat[1] end else link = data['station format'] end link = (string.gsub(link, '%%1', station)) link = line and (string.gsub(link, '%%2', line)) or link link = typ and (string.gsub(link, '%%3', typ)) or link return string.match(link, '%[%[.*%]%]') and link or			table.concat({, station, }) end end p.station = makeInvokeFunc('_station') function p._station(args, frame) args.station = args.station or args[2] if data then return stationTitle(args.station, args.line, args['type']) else return frame:expandTemplate{ title = args.system .. ' stations', args = { ['station'] = args.station, ['line'] = args.line, ['branch'] = args['type'] }		}	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 .. ' lines', args = {args.line, ['branch'] = args['type']} }	end end p.color = makeInvokeFunc('_color') function p._color(args, frame) if data then return mw.text.nowiki(			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 = {string.lower(args.system), string.lower(args.line),	['size'] = args.size} }	end end p.box = makeInvokeFunc('_box') function p._box(args, frame) local root, style = mw.html.create('div'), args.style local colour, lineTitle = p._color(args, frame), p._line(args, frame) if colour then colour = string.match(colour, '#') and colour or '#' .. colour end if args.line then args.line = dig(data, 'lines', args.line) and args.line or			dig(data, 'aliases', string.lower(args.line)) end local box = mw.html.create('span') if style == nil then root :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 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				'white'			) :css('font-weight', args.bold == 'no' or 'bold') :css('font-size', 'inherit') :css('white-space', 'nowrap') end local boxText = { ['inline'] = string.rep(' ',4), ['yes'] = string.rep(' ',4), ['small'] = string.rep(' ',1), ['link'] = string.rep(' ',4), ['box'] = string.rep(' ', 4), ['dot'] = '●', ['ldot'] = '●', ['square'] = '■', ['lsquare'] = '■', ['route'] = args.line, ['croute'] = args.line, ['xroute'] = args.line }	box:wikitext(boxText[style] or string.rep(' ',1)) if style == 'link' or		style == 'ldot' or		style == 'lsquare' or		style and string.match(style, 'route') then if string.match(lineTitle, '|') then root:wikitext((string.gsub(lineTitle, '%[%[.*|(.*)%]%]', tostring(box)))) else root:wikitext((string.gsub(lineTitle, '%]%]', '|' .. tostring(box) .. ']]'))) end elseif style == 'box' then root:wikitext(tostring(box)) else root:wikitext(tostring(box), ' ', lineTitle) end return root 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 .. ' ' .. 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(' ', 4)) :done :wikitext(' ', 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 makeTerminusFunc(n, fallback) return function (side) local termini = fallback(side .. ' terminus') if type(termini) == 'string' then return stationTitle(termini, args[n]['line'], args[n]['type'], n)			elseif type(termini) == 'table' then local i, t, to = 1, {}, args[n]['to-' .. side] or args[n]['to'] while termini[i] and to ~= termini[i] do					t[i] = stationTitle(termini[i], args[n]['line'], args[n]['type'], n)					i = i + 1 end return stationTitle(termini[i], args[n]['line'], args[n]['type'], n) or mw.text.listToText(t, nil, ' or ') end end end data = {} 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 v.line) or			dig(data[n], 'lines', dig(data[n], 'aliases', string.lower(line or v.line))) end typeData = function (n) return dig(lineData(n), 'types', v['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 local terminus = makeTerminusFunc(i, fallback) 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') and not args[j]['nonstop'] and not args[j]['note-row'] do					j = j + 1 end renderSideCell(					row,					j - i,					stationTitle(v.left, v.line, v['type'], i),					yesNo(fallback('circular')) and fallback('left terminus') or terminus('left'),					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'] and not args[k]['nonstop'] and not args[k]['note-row'] 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'],					stationTitle(v.transfer, v.line, v['type'], 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') and not args[l]['nonstop'] and not args[l]['note-row'] do					l = l + 1 end renderSideCell(					row,					l - i,					stationTitle(v.right, v.line, v['type'], i),					yesNo(fallback('circular')) and fallback('right terminus') or terminus('right'),					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 function p.convert(frame) local args = frame.args local code = mw.text.split(mw.ustring.gsub(args[1], '^%s*%s*$', '%1'), '%s*}}%s*{{%s*') local system local group = 0 local delete = { ['s-rail'] = true, ['s-rail-next'] = true, ['s-rail-national'] = true, ['s-start'] = true, ['s-rail-start'] = true, ['start'] = true, ['s-end'] = true, ['end'] = true }	local order = { 'line', 'left', 'right', 'to-left', 'to-right', 'oneway-left', 'oneway-right', 'through-left', 'through-right', 'reverse', 'reverse-left', 'reverse-right', 'note-left', 'note-mid', 'note-right', 'transfer' -- circular: use module subpage -- state: not implemented }	local replace = { ['previous'] = 'left', ['next'] = 'right', ['type'] = 'to-left', ['type2'] = 'to-right', ['branch'] = 'type', ['note'] = 'note-left', ['notemid'] = 'note-mid', ['note2'] = 'note-right', ['oneway1'] = 'oneway-left', ['oneway2'] = 'oneway-right', ['through1'] = 'through-left', ['through2'] = 'through-right' }	local remove_rows = {} local data = {} for i, v in ipairs(code) do		code[i] = mw.ustring.gsub(code[i], '\n', ' ') local template = mw.ustring.lower(mw.text.trim(mw.ustring.match(code[i], '^[^|]+'))) code[i] = mw.ustring.match(code[i], '(|.+)$') if template == 's-line' then data[i] = {} local this_system = mw.text.trim(mw.ustring.match(code[i], '|%s*system%s*=([^|]+)')) code[i] = mw.text.split(code[i], '%s*|%s*') for m, n in ipairs(code[i]) do				local tmp = mw.text.split(n, '%s*=%s*') if tmp[3] then tmp[2] = mw.ustring.gsub(n, '^.-%s*=', '') end tmp[1] = replace[tmp[1]] or tmp[1] if tmp[2] then -- checks for matching brackets local curly = select(2, mw.ustring.gsub(tmp[2], "{", ""))-select(2, mw.ustring.gsub(tmp[2], "}", "")) local square = select(2, mw.ustring.gsub(tmp[2], "%[", ""))-select(2, mw.ustring.gsub(tmp[2], "%]", "")) if not (curly+square==0) then local count = mw.clone(m)+1 while not (curly+square==0) do							tmp[2] = tmp[2]..'|'..code[i][count] curly = curly+select(2, mw.ustring.gsub(code[i][count], "{", ""))-select(2, mw.ustring.gsub(code[i][count], "}", "")) square = square+select(2, mw.ustring.gsub(code[i][count], "%[", ""))-select(2, mw.ustring.gsub(code[i][count], "%]", "")) code[i][count] = '' count = count+1 end end data[i][tmp[1]] = tmp[2] end end if (this_system ~= system) or (not system) then system = this_system data[i]['system'] = system else data[i]['system'] = nil end local last = data[i-1] or data[i-2] or data[i-3] if last then for r, s in pairs({					['hide1'] = {'left', 'to-left', 'note-left', 'oneway-left'},					['hide2'] = {'right', 'to-right', 'note-right', 'oneway-right'},					['hidemid'] = {'type', 'note-mid'}					}) do					if data[i][r] then for m, n in ipairs(s) do							if not data[i][n] then data[i][n] = last[n] end end end end end code[i] = {} local X = '|' local Y = (i+group)..'=' if data[i]['system'] then table.insert(code[i], '|system') table.insert(code[i], Y)				table.insert(code[i], data[i]['system']) table.insert(code[i], '\n') end for m, n in ipairs(order) do				if data[i][n] then table.insert(code[i], X)					table.insert(code[i], n)					table.insert(code[i], Y)					table.insert(code[i], data[i][n]) end end code[i] = table.concat(code[i]) elseif template == 's-note' then code[i] = mw.ustring.gsub(code[i], '|%s*text%s*=', '|header'..i+group..'=') code[i] = mw.ustring.gsub(code[i], '|%s*wide%s*=[^|]*', '') elseif template == 's-text' then code[i] = mw.ustring.gsub(code[i], '|%s*text%s*=', '|note-row'..i+group..'=') elseif delete[template] then code[i] = '' table.insert(remove_rows, 1, i) -- at the start, so that the rows are deleted in reverse order group = group-1 end end for i, v in ipairs(remove_rows) do		table.remove(code, v)	end code = table.concat(code, '\n') local t = {'{{Adjacent stations', '\n}}'} system = mw.ustring.match(code, '|system(%d*)=') code = mw.ustring.gsub(code, '\n\n+', '\n') if tonumber(system) > 1 then -- If s-line isn't the first template then the system will have to be moved to the top system = mw.ustring.match(code, '|system%d*=([^|]*[^|\n])') code = mw.ustring.gsub(code, '|system%d*=[^|]*', '') code = '\n|system1='..system..code elseif not mw.ustring.match(code, '^[^{%[]*|[^=|]+2=') then -- If there's only one parameter group then there's no need to have line breaks code = mw.ustring.gsub(code, '\n', '') code = mw.ustring.gsub(code, '(|[^=|]+)1=', '%1=') t[2] = '}}' if not mw.ustring.match(code, '[%[{]') then code = mw.ustring.gsub(code, '|[^=|]*=$', '') code = mw.ustring.gsub(code, '|[^=|]*$', '') end end if not mw.ustring.match(code, '[%[{]') then code = mw.ustring.gsub(code, '|[^=|]*=|', '|') code = mw.ustring.gsub(code, '|[^=|]*|', '|') code = mw.ustring.gsub(code, '|[^=|]*=\n', '\n') code = mw.ustring.gsub(code, '|[^=|]*\n', '\n') end return t[1]..code..t[2] end return p