Module:For loop/sandbox

Module documentation[view] [edit] [history] [purge]

This module implements Template:For loop (edit | talk | history | links | watch | logs). Please see the template page for documentation.

The above documentation is transcluded from Module:For loop/doc. (edit | history)
Editors can experiment in this module's sandbox (edit | diff) and testcases (edit | run) pages.
Add categories to the /doc subpage. Subpages of this module.

p = {}
 
local args
 
local function callTemplate(template, targs)
    return mw.getCurrentFrame():expandTemplate{title = template, args = targs}
end
 
local function getArgNums(prefix, suffix)
    -- Returns a table containing the numbers of the arguments that exist
    -- for the specified prefix and suffix.
    local nums = {}
    local pattern = '^' .. prefix .. '([1-9]%d*)' .. suffix .. '$'
    for k, v in pairs(args) do
        local num = tostring(k):match(pattern)
        if num then
        	nums[#nums + 1] = tonumber(num)
        end
    end
    table.sort(nums)
    return nums
end
 
local function getConstants()
    local constantArgNums = getArgNums('pc', 'n')
    local constantArgs = {}
    for _, num in ipairs(constantArgNums) do
        local keyArg = 'pc' .. tostring(num) .. 'n'
        local valArg = 'pc' .. tostring(num) .. 'v'
        local key = args[keyArg]
        key = tonumber(key) or key
        local value = args[valArg]
        constantArgs[key] = value
    end
    return constantArgs
end
 
local function getVariableVals()
    local variableVals = {}
    for i, v in ipairs(args) do
        if i ~= 1 then
            variableVals[i - 1] = v
        end
    end
    return variableVals
end
 
local function _main()
    local template = args['call'] or 'void'
    local variableParam = args.pv
    variableParam = tonumber(variableParam) or variableParam or 1 -- fix for positional parameters
    local variableValPrefix = args.prefix or ''
    local variableValPostfix = args.postfix or ''
    local sep = args[1] or ''
    local constantArgs = getConstants()
    local variableVals = getVariableVals()
	
    local result = ''
    for i, v in ipairs(variableVals) do
        v = mw.text.trim(v) -- trim whitespace
        local targs = constantArgs
        targs[variableParam] = variableValPrefix .. v .. variableValPostfix
        result = result .. tostring(callTemplate(template, targs))
        if variableVals[i + 1] then
            result = result .. sep
        end
    end
    return result
end
 
function p.main(frame)
    -- If called via #invoke, use the args passed into the invoking template.
    -- Otherwise, for testing purposes, assume args are being passed directly in.
    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
    return _main()
end
 
return p