require('strict')
local p = {}
local getArgs = require('Module:Arguments').getArgs
local yesno = require('Module:Yesno')
local data = mw.loadData('Module:Jcon/data')
local placeTypes = { -- Place types to remove from lowercase road type
'municipality',
'municipal',
'city',
'township',
'district',
'county',
'counties',
'united counties',
'region',
'regional',
'regional municipality',
'road'
}
-- Get the road info table for the given type
local function getRoadInfo(roadType)
roadType = (roadType or ''):lower() -- Make the road type lowercase
-- local route = args[2] or '' -- Get the route number
for index, placeType in ipairs(placeTypes) do
roadType = roadType -- Remove the place types from the road type
:gsub('^' .. placeType .. ' of ', '')
:gsub(' ' .. placeType .. '$', '')
:gsub(' ' .. placeType .. ' road$', '')
end
roadType = data.aliases[roadType] or roadType -- Transform alias into proper name
-- if data.signs[roadType] or data.signs[route:lower()] then
-- return data.signs[roadType] or data.signs[route:lower()] -- Return signs symbols like airport and hospital
-- end
return data.types[roadType] -- Get road type info from the data module
end
-- Generate the wikitext for the shield
local function shieldWikitext(route, roadInfo, args)
local size = args.size or '20px' -- Image size
local fileName
local titleObj
if roadInfo.prefix == 'Ontario' and (tonumber(({route:gsub('%D', '')})[1]) or 0) >= 500 then -- Exception for secondary and tertiary highways
fileName = 'Ontario Highway ' .. route .. '.svg'
elseif roadInfo.prefix == 'Ontario' and route:gsub('%s+', ''):upper() == '407ETR' then -- Exception for 407 ETR
fileName = yesno(args.shield) and '407 Express Toll Route Traffic Sign.svg' or 'Highway407crest.svg'
elseif roadInfo.shield and yesno(args.shield) then -- Shield format (used as reassurance marker)
fileName = roadInfo.shield:format(route)
elseif roadInfo.guide then -- Guide format (used on guide signs)
fileName = roadInfo.guide:format(route)
else
return '' -- Return nothing if no file format was found
end
titleObj = mw.title.new('File:' .. fileName)
if not titleObj or not titleObj.file.exists then
return '' -- Return nothing if no existing file was found
end
return '[[File:' .. fileName .. '|alt=|link=|' .. size .. ']]' -- Return the file wikitext
end
-- Generate the text part of the output
local function getText(route, roadInfo, args)
local link = ''
local display = ''
if roadInfo.prefix == 'Ontario' and route:gsub('%s+', ''):upper() == '407ETR' then -- Exception for the 407 ETR
link = 'Ontario Highway 407'
display = '407 ETR'
elseif roadInfo.prefix == 'Ontario' and route:upper() == 'QEW' then -- Exception for the QEW
link = 'Queen Elizabeth Way'
display = 'Queen Elizabeth Way'
elseif roadInfo.prefix == 'Toronto' and route:upper() == 'DVP' then -- Exception for the DVP
link = 'Don Valley Parkway'
display = 'Don Valley Parkway'
elseif roadInfo.prefix == 'Toronto' and route == 'Gardiner' then -- Exception for the Gardiner Expressway
link = 'Gardiner Expressway'
display = 'Gardiner Expressway'
else
link = roadInfo.prefix .. ' ' .. roadInfo.type .. ' ' .. route
display = roadInfo.type .. ' ' .. route
end
local titleObj = mw.title.new(link)
if (((titleObj and titleObj.exists) or yesno(args.showred)) and not yesno(args.nolink)) then -- Check if the link show be shown
return '[[' .. link .. '|' .. display .. ']]' -- Return the link
else
return display -- Fallback to returning the display text
end
end
-- Gets the wikitext for a place
local function getPlace(place, args)
local placeArticle = place .. ', Ontario'
local titleObj = mw.title.new(placeArticle)
if (titleObj and titleObj.exists) or yesno(args.showred) then
return '[[' .. placeArticle .. '|' .. place .. ']]'
else
return place
end
end
-- Entry function
function p.jcon(frame)
local args = getArgs(frame)
if yesno(args.ot) then -- Set correct arguments if output should be only text
args.nosh = 'yes'
args.nolink = 'yes'
end
local roadInfo = getRoadInfo(args[1])
if not roadInfo or not args[2] then
return '​' -- Return ZWSP if road type is not supported or the route is not specified
end
local shield = ''
local text = ''
local conNumber = 0
local paramName = 2
local typeParamName = nil
local dirParamName = 'dir'
local showShield = not yesno(args.nosh)
local showText = not yesno(args.notext)
repeat
local routeRoadInfo = roadInfo
if typeParamName and args[typeParamName] then
routeRoadInfo = getRoadInfo(args[typeParamName])
end
if showShield then
if shield ~= '' then shield = shield .. ' ' end
shield = shield .. shieldWikitext(args[paramName], routeRoadInfo, args)
end
if showText then
if text ~= '' then text = text .. ' / ' end
text = text .. getText(args[paramName], routeRoadInfo, args) -- Add text of the second concurrency
end
if args[dirParamName] then
text = text .. ' ' .. args[dirParamName] -- Add the direction
end
conNumber = conNumber + 1
local paramPrefix = conNumber == 1 and '' or conNumber
paramName = 'con' .. paramPrefix
typeParamName = 'contype' .. paramPrefix
dirParamName = 'condir' .. paramPrefix
until not args[paramName]
if yesno(args.tch) then
if showShield then shield = shield .. '[[File:TCH-blank.svg|x20px]]' end -- Add the TCH shield
if showText then text = text .. ' / [[Trans-Canada Highway|TCH]]' end -- Add the TCH text
end
local output = ''
if not yesno(args['pic aft']) then
output = output .. shield -- Add the shield if it goes first
end
if text ~= '' then
if output ~= '' then output = output .. ' ' end -- Add a NBSP after the shield if it exists
output = output .. text
end
if args[3] then
output = output .. ' (' .. args[3] .. ')' -- Add the name to the output
end
if args.city or args.town then
output = output .. ' – ' .. getPlace(args.city or args.town, args) -- Add the first city
local extraCityNum = 2
while args['city' .. extraCityNum] or args['town' .. extraCityNum] do
local val = args['city' .. extraCityNum] or args['town' .. extraCityNum]
output = output .. ', ' .. getPlace(val, args) -- Add extra cities
extraCityNum = extraCityNum + 1
end
end
if yesno(args['pic aft']) and shield then
if output ~= '' then output = output .. ' ' end -- Add a space if output already has text
output = output .. shield -- Add the shield if it goes last
end
return output
end
-- Generates a list of supported regions
function p.supported(frame)
local reverseAliases = {}
local entries = {}
local list = mw.html.create('ul') -- Create output list element
for alias, name in pairs(data.aliases) do -- Reverse the alias table to allow lookup by name
if not reverseAliases[name] then
reverseAliases[name] = {}
end
table.insert(reverseAliases[name], alias)
end
for name, info in pairs(data.types) do -- Create tables for each region
if reverseAliases[name] then
for _, alias in ipairs(reverseAliases[name]) do
name = name .. ' / ' .. alias -- Add alias to list item
end
end
list:tag('li'):wikitext(name, ' ', '(' .. info.prefix .. ' ' .. info.type .. ')')
end
for sign, wikitext in pairs(data.signs) do
if reverseAliases[sign] then
for _, alias in ipairs(reverseAliases[sign]) do
sign = sign .. ' / ' .. alias -- Add alias to list item
end
end
list:tag('li'):wikitext(sign, ' ', wikitext)
end
return tostring(list)
end
return p
