Module:Sandbox/SkillTree

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

local p = {}

local GameData = require('Module:GameData')

function p.getSkillTreeNodes(checkFunc)
    local nodes = {}
    
    for skillName, skillData in pairs(GameData.skillData) do
        local skillTrees = skillData.skillTrees
        if skillTrees then
            for _, skillTree in ipairs(skillTrees) do
                for _, node in ipairs(GameData.getEntities(skillTree.nodes, checkFunc)) do
                    local nodeCopy = {}
                    for k, v in pairs(node) do
                        nodeCopy[k] = v
                    end

                    nodeCopy.skillName = skillName

                    table.insert(nodes, nodeCopy)
                end
            end
        end
    end
    return nodes
end

function p.getSkillTreeNodesFromSkillName(skillName)
	mw.log(GameData.skillData)
	
    local skillData = GameData.skillData[skillName]
    if not skillData then return nil end

    local nodes = {}
    for _, node in ipairs(skillData.skillTrees[1].nodes) do
        table.insert(nodes, node)
    end
    return nodes
end

function p.generateSkillTree(frame)
    local args = frame.args ~= nil and frame.args or frame
    local skillName = args[1]
    
    if not skillName then
        return "Invalid skillName"
    end
    
    local skillNodes = p.getSkillTreeNodesFromSkillName(skillName)

    if not skillNodes then
        return "No skill tree found for: " .. skillName
    end

    local html = mw.html.create('div'):addClass('skill-tree-container'):css({
        ['position'] = 'relative',
        ['width'] = '100%',
        ['height'] = 'auto'
    })

    for _, node in ipairs(skillNodes) do
        local nodeDiv = html:tag('div'):css({
            ['position'] = 'absolute',
            ['width'] = '225px',
            ['height'] = '150px',
            ['top'] = node.top .. 'px',
            ['left'] = node.left .. 'px',
        })

        nodeDiv:tag('span'):wikitext(node.name):css({
            ['position'] = 'absolute',
            ['top'] = '6px',
            ['left'] = '6px',
            ['z-index'] = 1,
            ['font-size'] = 'smaller',
            ['font-weight'] = '700',
        }):done()

        nodeDiv:tag('img'):attr('src', node.imageSrc):css({
            ['border'] = '2px solid ' .. (node.locked and 'red' or 'green')
        }):done()

        nodeDiv:tag('div'):wikitext(node.description):css({
            ['text-align'] = 'center',
            ['overflow-y'] = 'auto',
            ['padding'] = '10px'
        }):done()

        if node.connections then
            for _, connection in ipairs(node.connections) do
                html:tag('svg')
                    :attr('width', '100%')
                    :attr('height', '100%')
                    :attr('preserveAspectRatio', 'none')
                    :tag('path'):attr('d', connection.path)
                    :css({
                        ['stroke'] = connection.strokeColor,
                        ['stroke-width'] = '3px',
                        ['fill'] = 'none'
                    }):done()
            end
        end

        html:done()
    end

    return tostring(html)
end

return p