Module:Jctrdt and Module:Jctrdt/sandbox: Difference between pages

local statenameModuleName = "Module:Jct/statename" .. moduleSuffix -- TODO transition

-- Shields

local defaultShieldSize = 17

local function addContextBanner(route, name, suffix, bannerSpec)

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 = defaultShieldSize

-- Add banner plate.

insert(bannerSpec, {shield, shieldSize})

end

end

local function bannerSpec(banner, bannerSize, bannerSuffix, route)

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)

end

if route.to then

addContextBanner(route, 'to', bannerSuffix, banners)

end

return banners

end

local function shieldSpec(route, mainShield)

local shieldSpec = {}

local shield

if mainShield then shield = parser(route, "shieldmain") end

if not shield then shield = parser(route, 'shield') or '' end

if shield == '' then return shieldSpec end

local orientation = parser(route, 'orientation')

local shieldsize = defaultShieldSize

local banner = parser(route, 'banner') or {}

local bannersize = defaultShieldSize

local bannersuffix = parser(route, 'bannersuffix')

local bannerIsNotTable = type(banner) ~= "table"

local bannersizeIsNotTable = type(bannersize) ~= "table"

local bannersuffixIsNotTable = type(bannersuffix) ~= "table"

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)

})

end

elseif shield ~= '' then

if shieldsize == "" then shieldsize = nil end

insert(shieldSpec, {

shield = {shield, shieldsize},

banners = bannerSpec(banner, bannersize, bannersuffix, route)

})

end

return shieldSpec

end

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

local function render(shieldEntry, scale, showLink)

local shield = shieldEntry.shield

local banners = shieldEntry.banners

local exists, landscape = shieldExists(shield[1])

if not exists then return '' end

local size

if shield[2] then

local width, height = mw.ustring.match(shield[2], "(%d*)x?(%d*)")

width = tonumber(width)

height = tonumber(height)

local sizeparts = {}

if width then

insert(sizeparts, format("%d", width * scale))

end

if height then

insert(sizeparts, format("x%d", height * scale))

end

size = concat(sizeparts)

else

size = format("%s%d", landscape, defaultShieldSize * scale)

end

local link = showLink and "" or "|link="

local shieldCode = format("[[File:%s|x%spx%s|%s]]", shield[1], size, link, shield[1])

if not banners[1] then return shieldCode end

for _,banner in ipairs(banners) do

shieldCode = format("[[File:%s|%spx|link=|alt=]]<br>%s",

banner[1],

defaultShieldSize,

shieldCode)

end

return '<span style="display: inline-block; vertical-align: baseline; line-height: 0; text-align: center;">' .. shieldCode .. '</span>'

end

function p.shield(route, scale, showLink, mainShield)

missingShields = {}

scale = 1

local rendered = {}

for _,entry in ipairs(shieldSpec(route, mainShield)) do

insert(rendered, render(entry, scale, showLink))

end

return concat(rendered)

end

function p.link(route)

local abbr, errMsg = parser(route, 'abbr')

if not abbr then 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

local function stateName(args)

-- TODO transition

local data = mw.loadData(statenameModuleName)

local abbr = args.state or args.province

local countryData = data[args.country]

return countryData and countryData[abbr]

end

link = p.link(route)

local shield = p.shield(route)