Module:Portal bar: Difference between revisions

-- This module implements {{portal bar}}.

local p = {}

local htmlBuilder = require( 'Module:HtmlBuilder' )
local buildNavbox = require( 'Module:Navbox' )._navbox
local getImageName = require( 'Module:Portal' ).image

-- Builds the portal bar used by {{portal bar}}.
function p._main( portals )
    if #portals < 1 then return '' end -- Don't display a blank navbox if no portals were specified.
    local root = htmlBuilder.create( 'ul' )
    for i, portal in ipairs( portals ) do
        root
            .tag( 'li' )
                .css( 'display', 'inline' )
                .css( 'white-space', 'nowrap' )
                .tag( 'span' )
                    .css( 'margin', 'auto 0.5em' )
                    .wikitext( mw.ustring.format( '[[File:%s|24x21px]]', getImageName{ portal } ) )
                    .done()
                .tag( 'span' )
                    .css( 'font-weight', 'bold' )
                    .wikitext( mw.ustring.format( '[[Portal:%s|%s portal]]', portal, portal ) )
    end
    return buildNavbox{
        name = 'Portal bar',
        bodyclass = 'noprint',
        list1 = tostring( root )
    }
end

-- Processes external arguments and sends them to the other functions.
function p.main( frame )
    -- If called via #invoke, use the args passed into the invoking
    -- template, or the args passed to #invoke if any exist. Otherwise
    -- assume args are being passed directly in from the debug console
    -- or from another Lua module.
    local args
    if frame == mw.getCurrentFrame() then
        args = frame:getParent().args
        for k, v in pairs( frame.args ) do
            args = frame.args
            break
        end
    else
        args = frame
    end
    -- Process the args to make an array of portal names that can be used with ipairs. We need to use ipairs because we want to list
    -- all the portals in the order they were passed to the template, but we also want to be able to deal with positional arguments
    -- passed explicitly, for example {{portal|2=Politics}}. The behaviour of ipairs is undefined if nil values are present, so we
    -- need to make sure they are all removed.
    local portals = {}
    for k, v in pairs( args ) do
        if type( k ) == 'number' and type( v ) == 'string' then -- Make sure we have no non-string portal names.
            if mw.ustring.find( v, '%S' ) then -- Remove blank values.
                table.insert( portals, k )
            end
        end
    end
    table.sort( portals )
    for i, v in ipairs( portals ) do
        portals[ i ] = mw.text.trim( args[ v ] ) -- Swap keys with values, trimming whitespace.
    end
    return p._main( portals )
end

return p