|
local p = {}
local roadDataModule = require('Module:Road data/sandbox') -- REMOVE SANDBOX
local concat = table.concat
local insert = table.insert
local format = mw.ustring.format
local trim = mw.text.trim
function p._shieldmain(args)
local parserModule = require("Module:Road data/parser")
local parsershields = parserModule.parser{}
local routeCount = 1
local util = require("Module:Road data/util")
local sizeModule = require("Module:Road data/size/sandbox") -- REMOVE SANDBOX
-- Shields
local defaultShieldSize = 24
local function addContextBanner(route, name, suffix, bannerSpec, sizeStyle)
local bannerModule = 'Module:Road data/banners/' .. string.upper(route.country)
local shieldfield = name .. 'shield'
local shield = parser(route, shieldfield)
if shield == nil then
-- This route type does not define shield.
-- Find shield in the default banner table.
shield = parser(route, 'shield', name, bannerModule)
if shield and shield ~= '' then
if suffix == nil then
suffix = parser(route, 'shield', 'suffix', bannerModule)
end
if suffix and suffix ~= '' then
shield = shield .. " " .. suffix
end
shield = shield .. ".svg"
end
end
if shield and shield ~= '' then
local shieldSize = sizeModule._size({
style = sizeStyle,
type = 'banner'
})
-- Add banner plate.
insert(bannerSpec, {shield, shieldSize})
end
end
local function bannerSpec(banner, bannerSize, bannerSuffix, route, sizeStyle)
local banners = {}
if type(banner) == "table" then
local bannerSizeIsNotTable = type(bannerSize) ~= "table"
for i,filename in ipairs(banner) do
local bannersize = bannerSizeIsNotTable and bannerSize or bannerSize[i] or defaultShieldSize
insert(banners, {filename, bannersize})
end
elseif banner ~= '' then
insert(banners, {banner, bannerSize})
end
if route.dir then
addContextBanner(route, 'dir', bannerSuffix, banners, sizeStyle)
end
if route.to then
addContextBanner(route, 'to', bannerSuffix, banners, sizeStyle)
end
return banners
end
local function shieldSpec(route, shieldStyle, sizeStyle)
local shieldSpec = {}
local shield
if route.to then shield = parser(route, 'shieldto') end
if not shield and shieldStyle == 'main' then shield = parser(route, 'shieldmain') end
if not shield and shieldStyle == 'list' then shield = parser(route, 'shieldlist') end
if not shield then shield = parser(route, 'shield') or '' end
if shield == '' then return shieldSpec end
local orientation = parser(route, 'orientation')
local shieldsize = sizeModule._size({
style = sizeStyle,
type = orientation
})
while args[routeCount * 2 - 1] do
local banner = parser(route, 'banner') or {}
local bannersizeroute = sizeModule._size({
country = args['country' .. routeCount] or args.country,
style = sizeStyle,
state = args['state' .. routeCount] or args['province' .. routeCount] or args.state or args.province,
type = 'banner'
type = args[routeCount * 2 - 1],
})
route = args[routeCount * 2]
local bannersuffix = parser(route, 'bannersuffix')
}
local bannerIsNotTable = type(banner) ~= "table"
if not route.country then
local bannersizeIsNotTable = type(bannersize) ~= "table"
local countryModule = mw.loadData('Module:Road data/countrymask')
local bannersuffixIsNotTable = type(bannersuffix) ~= "table"
route.country = countryModule[route.state] or 'UNK'
if type(shield) == "table" then
for i,filename in ipairs(shield) do
local size = shieldsize or shieldsize[i]
if size == "" then size = nil end
-- banner.all describes banners that apply to all multiple shields.
local shieldBanner = bannerIsNotTable and banner or (banner[i] or banner.all or {})
-- Banner size is default if the corresponding entry
-- in bannerSize table is not set.
local shieldBannerSize =
bannersizeIsNotTable and bannersize
or (bannersize[i] or bannersize.all or defaultShieldSize)
local shieldBannerSuffix = bannersuffix and (bannersuffixIsNotTable and bannersuffix or bannersuffix[i])
insert(shieldSpec, {
shield = {filename, size},
banners = bannerSpec(shieldBanner, shieldBannerSize, shieldBannerSuffix, route, sizeStyle),
route = route
})
end
elseif shield ~= '' then
if shieldsize == "" then shieldsize = nil end
insert(shieldSpec, {
shield = {shield, shieldsize},
banners = bannerSpec(banner, bannersize, bannersuffix, route, sizeStyle),
route = route
})
end
local shield = roadDataModule.shield(route, 'main', 'infobox', true)
return shieldSpec
table.insert(shields, shield)
end
routeCount = routeCount + 1
local missingShields
local shieldExistsCache = {}
-- Return up to two booleans.
-- The first boolean is false if `shield` does not exist, and true otherwise.
-- If the first boolean is true, the second boolean is true if the shield is
-- landscape (width >= height), and false otherwise.
local function shieldExists(shield)
local result = shieldExistsCache[shield]
if result == nil then
local file = mw.title.new(shield, 'Media').file
-- Cache result.
local exists = file.exists
result = {exists}
if exists then result[2] = file.width >= file.height end
shieldExistsCache[shield] = result
end
if result[1] then return true, result[2] end
insert(missingShields, shield)
return false
end
return table.concat(shields, ' ')
local function render(shieldEntry, addAlt)
local shield = shieldEntry.shield
local banners = shieldEntry.banners
local alt = ''
if addAlt then
alt = (parser(shieldEntry.route, 'abbr') or '') .. ' marker'
end
local shieldCode = format("[[File:%s|%s|link=|alt=%s]]", shield[1], shield[2], alt)
if not banners[1] then return shieldCode end
for _,banner in ipairs(banners) do
shieldCode = format("[[File:%s|%s|link=|alt=%s]]<br>%s",
banner[1],
banner[2],
alt,
shieldCode)
end
return '<span style="display: inline-block; vertical-align: baseline; line-height: 0; text-align: center;">' .. shieldCode .. '</span>'
end
function p.shieldshieldmain(route, shieldStyle, sizeStyle, addAltframe)
local args = require('Module:Arguments').getArgs(frame)
missingShields = {}
return p._shieldmain(args)
local rendered = {}
for _,entry in ipairs(shieldSpec(route, shieldStyle, sizeStyle)) do
insert(rendered, render(entry, addAlt))
end
return concat(rendered), missingShields
end
function p.link(route)
local abbr, errMsg = parser(route, 'abbr')
if not abbr then
route.typeerror = true
return util.err(errMsg or format("Invalid type: %s", route.type or "(nil)"))
end
if route.nolink then return abbr, abbr end
local link = parser(route, 'link') or ''
if link == '' then return abbr, abbr end
return format("[[%s|%s]]", link, abbr), abbr
end
| 