Module:TemplateBox

Module documentation[ create · purge ]

This documentation is transcluded from Module:TemplateBox/doc.

local p = {}

function tobool(st)
    if type( st ) == 'string' then
        return st == 'true'
    else
        return not not st
    end
end

-- InterfaceText:
-- A free-form string (no wikitext) in the content-language of the wiki, or, an object containing those strings keyed by language code.
local paraminfoTemplate = {
    description = {
        required = true,
        default = '',
        t = 'InterfaceText',
        alias = 'desc'
    }
}
local paraminfoTLParams = {
    label = {
        required = false,
        default = '',
        t = 'InterfaceText'
    },
    required = {
        required = true,
        default = false,
        extract = function(pargs, number, paramVal)
            local req = (pargs[number .. 'stat'] == 'required')
            return tobool( paramVal or req )
        end
    },
    description = {
        required = false,
        default = '',
        t = 'InterfaceText',
        alias = 'd'
    },
    deprecated = {
        required = true,
        default = false,
        extract = function(pargs, number, paramVal)
            local depr = (pargs[number .. 'stat'] == 'deprecated')
            return tobool( paramVal or depr )
        end
    },
    aliases = {
        required = false,
        default = '',
        extract = function(pargs, number, paramVal)
            local key = number .. 'aliases'
            local tdkey = key .. '-td'
            local aliases = pargs[tdkey] or pargs[key]
            if aliases then
                aliases = mw.text.split( aliases, '/', true )
            end
            return aliases
        end
    },
    default = {
        required = false,
        default = '',
        t = 'InterfaceText',
        alias = 'def'
    },
    type = {
        required = false,
        default = 'unknown',
        t = 'selection',
        selection = 'unknown|number|string|string/wiki-user-name|string/wiki-page-name'
    },
    inherits = {
        required = false,
        default = nil,
        t = 'string'
    }
    -- sets will be treated differently because we can only have a plain structure in wikitext
}

-- Initialize param info
function init( which )
    local setDefault = function(v)
        if v.t == nil and v.default ~= nil then
            v.t = type( v.default )
        end
        if v.selection then
            v.selection = '|' .. v.selection .. '|'
        end
    end
    for a, v in pairs( which ) do
        setDefault(v)
    end
end
init( paraminfoTemplate )
init( paraminfoTLParams )

function p.templatedata(frame)
    -- A real parser would look differently but it would likely be much more complex
    local JSON = require('Module:JSON')
    local pargs = mw.clone( ( frame:getParent() or {} ).args or {} )
    local templateArgs = {}
    local templateArgsCount = 0
    local i18nTemplateArgs = {}
    JSON.strictTypes = true
    local tdata = JSON:decode('{"description":"","params":{},"sets":[]}')
    local extractData = function( pi, number )
        local prefix = number or ''
        local key, key2, tdkey, tdkey2, paramVal
        local paramKey, paramTable
        if number then
            paramKey = mw.text.trim( pargs[number] )
            if '' == paramKey then
                paramKey = tostring( number )
            end
            tdata.params[paramKey] = {}
            paramTable = tdata.params[paramKey]
        end
        for p, info in pairs( pi ) do
            key = prefix .. (info.alias or p)
            key2 = prefix .. p
            tdkey = key .. '-td'
            tdkey2 = key2 .. '-td'
            paramVal = pargs[tdkey] or pargs[tdkey2] or pargs[key] or pargs[key2] or info.default
            -- Do we have a multilingual item?
            if i18nTemplateArgs[tdkey2] then
                paramVal = i18nTemplateArgs[tdkey2]
            end
            if i18nTemplateArgs[tdkey] then
                paramVal = i18nTemplateArgs[tdkey]
            end
            if 'function' == type( info.extract ) then
                if 'string' == type( paramVal ) then
                    paramVal = mw.text.trim( paramVal )
                    if '' == paramVal then
                        paramVal = nil
                    end
                end
                paramVal = info.extract( pargs, number, paramVal )
            end
            
            local insertValue = function()
                if number then
                    paramTable[p] = paramVal
                else
                    tdata[p] = paramVal
                end
            end
            
            if info.selection then
                if info.selection:find( paramVal, 1, true ) then
                    insertValue()
                end
            elseif 'InterfaceText' == info.t then
                if ({ table=1, string=1 })[type( paramVal )] then
                    insertValue()
                end
            else
                local paramType = type( paramVal )
                if 'string' == info.t and 'string' == paramType then
                    paramVal = mw.text.trim( paramVal )
                    if '' ~= paramVal then
                        insertValue()
                    end
                elseif 'boolean' == info.t then
                    paramVal = tobool(paramVal)
                    insertValue()
                elseif 'number' == info.t then
                    paramVal = tonumber(paramVal)
                    insertValue()
                elseif paramType == info.t then
                    insertValue()
                elseif paramType == 'nil' then
                    -- Do nothing
                else
                    error( p .. ':' .. tostring( paramVal ) .. ' is of type ' ..  paramType .. ' but should be of type ' .. (info.t or 'unknown'), 1 )
                end
            end
        end
        -- Now, treat sets
        key = prefix .. 'set'
        tdkey = prefix .. 'set-td'
        paramVal = pargs[tdkey] or pargs[key]
        if paramVal then
            local found = false
            for i, s in ipairs( tdata.sets ) do
                if s.label == paramVal then
                    table.insert( s.params, p )
                    found = true
                end
            end
            if not found then
                table.insert( tdata.sets, {
                    label = paramVal, 
                    params = { p }
                } )
            end
        end
    end
    
    if not pargs.useTemplateData then
        local warning = "Warning: Module:TemplateBox - templatedata invoked but not requested by user (setting useTemplateData=1)."
        mw.log(warning)
        tdata.description = warning
        return JSON:encode(tdata);
    end
    
    -- First, analyse the structure of the provided arguments
    for a, v in pairs( pargs ) do
        if type( a ) == 'number' then
            table.insert( templateArgs, a )
            templateArgsCount = templateArgsCount + 1
        else
            local argSplit = mw.text.split( a, '-', true )
            local argUnitl = {}
            local argAfter = {}
            local isTDArg = false
            for i, part in ipairs( argSplit ) do
                if isTDArg then
                    -- This is likely a language version
                    table.insert( argAfter, part )
                else
                    table.insert( argUnitl, part )
                end
                if part == 'td' then
                    isTDArg = true
                end
            end
            argUnitl = table.concat( argUnitl, '-' )
            argAfter = table.concat( argAfter, '-' )
            i18nTemplateArgs[argUnitl] = i18nTemplateArgs[argUnitl] or {}
            i18nTemplateArgs[argUnitl][argAfter] = v
        end
    end
    -- Then, start building the actual template
    extractData( paraminfoTemplate )
    for i, number in pairs( templateArgs ) do
        extractData( paraminfoTLParams, number )
    end
    
    -- And finally return the result
    return JSON:encode(tdata);
end

return p