Module:Traits

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

require('Module:CommonFunctions')
local getArgs = require('Module:Arguments').getArgs
local p = {}

-- Main process
function p.main(frame)
    local args = getArgs(frame);

    -- Argument init
    local traits = split(args[3])
    local skill = args[2]

    -- Descriptions
    local effects = {
        [1] = {},
        [2] = {}
    }
    if args.effects1 ~= nil then
        effects[1] = split(args.effects1)
    end
    if args.effects2 ~= nil then
        effects[2] = split(args.effects2)
    end

    local desc = {
        [1] = {},
        [2] = {}
    }

    -- Attribute Effect
    local counts = {}
    local header_data = {
        [1] = {},
        [2] = {}
    }
    for i=1,2,1 do
        local desc_arg = args['desc' .. i];

        if desc_arg ~= nil then
            local headers = split(desc_arg, ';')[1];
            local data = split(desc_arg, ';')[2];
            headers = split(headers);

            for k,v in pairs(headers) do
                table.insert(header_data[i], v)
            end

            data = split(data);
            counts[i] = 1 + #headers;

            local index = 1
            while index <= #headers do
                desc[i][headers[index]] = data[index]
                index = index + 1
            end
        end

    end

    -- Main block
    local trait_table = mw.html.create('table'):attr({
        ['cellpadding'] = '5',
        ['border'] = '1'
    }):css({
        ['border-collapse'] = 'collapse',
        ['text-align'] = 'center'
    })

    -- Get table head color
    local function color(char)
        char = char or args[1]
        return frame:expandTemplate{title = 'ColorSel', args = {'CharLight', char}}:gsub("&#35;", "#")
    end

    Tr = {}
    function Tr:new()
        return trait_table:tag('tr'):css('background-color', color())
    end

    -- Spawn headers
    local thead = Tr:new()
    thead:tag('th'):attr('colspan', counts[1]):wikitext(traits[1] .. ' ' .. skill)
    thead:tag('th'):attr('colspan', counts[2]):wikitext(traits[2] .. ' ' .. skill)

    if (args.effects1 ~= nil or args.effects2 ~= nil) then
        local tr_2 = Tr:new();
        for i=1,2,1 do
            tr_2:tag('th'):wikitext('Attribute Effect')
            if (args['effects' .. i] ~= nil) then
                for k,v in pairs(header_data[i]) do
                    v = frame:preprocess(v);
                    tr_2:tag('th'):wikitext(v)
                end
            end
        end
    end

    -- Spawn information
    local tr = trait_table:tag('tr')
    for i=1,2,1 do
        tr:tag('td'):wikitext(
            frame:expandTemplate{
                title = 'SkillText', args = {traits[i], effects[i][1], effects[i][2]}
            }
        )
        for k,v in pairs(desc[i]) do
            tr:tag('td'):wikitext(
                frame:preprocess(v)
            )
        end
    end
    

    -- Output point
    return tostring(trait_table);

end

return p