Module:ProcessCastes

From Ant Colony Wiki
Jump to navigation Jump to search

Documentation for this module may be created at Module:ProcessCastes/doc

local p = {}

function p.process(frame)
    local castes = frame.args[1] or ""
    local result = ""
    for caste in string.gmatch(castes, '([^,]+)') do
        result = result .. frame:expandTemplate{title = "Caste", args = {caste:match("^%s*(.-)%s*$")}}
    end
    return result
end

function p.countCastes(frame)
    local castes = frame.args[1] or ""
    local count = 0

    for _ in string.gmatch(castes, '([^,]+)') do
        count = count + 1
    end

    return count
end

function p.categorize(frame)
    local castes = frame.args.castes or ""
    local result = ""

    for caste in string.gmatch(castes, '([^,]+)') do
        result = result .. "[[Category:Ant Caste " .. caste .. "]]\n"
    end

    return result
end

return p