Jump to content

Module:ArrayList

From Wikipedia, the free encyclopedia
This is an old revision of this page, as edited by Bungle (talk | contribs) at 22:05, 1 February 2025 (refine min/max code). The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.
local p = {}

-- Helper function to trim whitespace
local function trim(s)
    return s:match("^%s*(.-)%s*$")
end

-- Escape special characters in the delimiter for pattern matching
local function escapePattern(str)
    return str:gsub("([%^%$%(%)%%%.%[%]%*%+%?%-])", "%%%1")
end

-- Main function to call other functions dynamically
function p.main(frame)
    local func = frame.args[1]
    if p[func] then
        return p[func](frame)
    else
        return "void:notfound " .. tostring(func)
    end
end

-- Count occurrences of delimiter in a string
function p.count(frame)
    local str = frame.args[2] or ""
    local delimiter = frame.args[3] or ","
    local nostrip = frame.args["nostrip"]
        
    delimiter = escapePattern(delimiter)
    
    -- If nostrip is not set to "true", strip leading/trailing delimiters
    if nostrip ~= "true" then
        -- Remove leading and trailing delimiters (along with any surrounding whitespace)
        str = str:gsub("^%s*" .. delimiter .. "%s*", ""):gsub("%s*" .. delimiter .. "%s*$", "")

        -- Normalize internal consecutive delimiters to a single delimiter (replace ",," with ",")
        str = str:gsub("%s*" .. delimiter .. "%s*" .. delimiter .. "%s*", delimiter)
    end
    
    local count = select(2, str:gsub(delimiter, ""))
    return count + 1
end

-- Get the Nth item in a delimited string, supporting negative indices
function p.get(frame)
    local str = frame.args[2] or ""
    local delimiter = frame.args[3] or ","
    local index = frame.args[4]
    delimiter = escapePattern(delimiter)
    str = str:gsub("^%s*" .. delimiter .. "%s*(.-)%s*" .. delimiter .. "$", "%1")
    local items = {}
    for item in string.gmatch(str, "([^" .. delimiter .. "]+)") do
        table.insert(items, trim(item))
    end
    if index == "last" then
        index = #items
    elseif index and tonumber(index) then
        index = tonumber(index)
        if index < 0 then
            index = #items + index + 1
        end
    else
        return "void:invalid"
    end
    return items[index] or "void:outrange"
end

-- Find the position of the Nth occurrence of a matching item in a delimited string
function p.pos(frame)
    local str = frame.args[2] or ""
    local delimiter = frame.args[3] or ","
    local item = frame.args[4] or ""
    local occurrence = tonumber(frame.args[5])
    delimiter = escapePattern(delimiter)
    str = str:gsub("^%s*" .. delimiter .. "%s*(.-)%s*" .. delimiter .. "$", "%1")
    local positions = {}
    local index = 1
    for subitem in string.gmatch(str, "([^" .. delimiter .. "]+)") do
        subitem = trim(subitem)
        if subitem == item then
            table.insert(positions, index)
        end
        index = index + 1
    end
    if not occurrence then
        return #positions > 0 and table.concat(positions, ",") or "void:nomatch"
    else
        return positions[occurrence] or -1
    end
end

-- Perform mathematical operations on numeric array items
function p.math(frame)
    local str = frame.args[2] or ""
    local delimiter = frame.args[3] or ","
    local operation = frame.args[4]
    local nth = tonumber(frame.args[5]) -- Optional Nth occurrence argument
    delimiter = escapePattern(delimiter)
    str = str:gsub("^%s*" .. delimiter .. "%s*(.-)%s*" .. delimiter .. "$", "%1")
    local items = {}
    for item in string.gmatch(str, "([^" .. delimiter .. "]+)") do
        local number = tonumber(trim(item))
        if number then
            table.insert(items, number)
        else
            return "void:isalpha"
        end
    end
    if #items == 0 then
        return "void:nonumeric"
    elseif operation == "sum" then
        local total = 0
        for _, num in ipairs(items) do
            total = total + num
        end
        return total
    elseif operation == "min" or operation == "max" then
		local extreme = items[1]
		local comparison = operation == "min" and function(a, b) return a < b end or function(a, b) return a > b end
		for _, num in ipairs(items) do
		    if comparison(num, extreme) then
		        extreme = num
		    end
		end
		return extreme
    else
        return "void:unsupported"
    end
end

return p