Module:BuildVersionTree

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

-- Module:VersionTree
local p = {}

-- Helper function to dump a table for debugging
local function dumpTable(tbl, indent)
    indent = indent or ""
    local result = {}
    
    for k, v in pairs(tbl) do
        if type(v) == "table" then
            table.insert(result, indent .. tostring(k) .. ":")
            table.insert(result, dumpTable(v, indent .. "  "))
        else
            table.insert(result, indent .. tostring(k) .. ": " .. tostring(v))
        end
    end
    
    return table.concat(result, "\n")
end

-- Helper function to build a tree structure from flat data
local function buildTree(versions, parentField)
    local tree = {}
    local nodes = {}
    
    -- First pass: Create all nodes
    for _, version in pairs(versions) do
        local title = version[1]
        nodes[title] = {
            title = title,
            children = {},
            processed = false
        }
    end
    
    -- Second pass: Build relationships
    for _, version in pairs(versions) do
        local title = version[1]
        local parent = version[parentField]
        
        if parent and parent[1] and nodes[parent[1]] then
            -- If it has a valid parent, add it as a child
            table.insert(nodes[parent[1]].children, nodes[title])
        else
            -- If no parent or invalid parent, it's a root node
            table.insert(tree, nodes[title])
        end
    end
    
    return tree
end

-- Helper function to generate unique IDs for nodes
local function generateId(text)
    return 'node-' .. mw.uri.encode(text):gsub('%%', '-')
end

-- Helper function to render a tree node and its children
local function renderNode(node, level)
    local result = {}
    
    -- Generate unique ID for this node
    local nodeId = generateId(node.title)
    
    -- Create node container
    table.insert(result, string.format('<div class="tree-node level-%d" id="%s">', level, nodeId))
    
    -- Add node content
    table.insert(result, string.format('<div class="node-content"><div class="node-box">[[%s]]</div></div>', node.title))
    
    if #node.children > 0 then
        -- Add container for children
        table.insert(result, '<div class="node-children">')
        
        -- Sort children by title
        table.sort(node.children, function(a, b)
            return a.title < b.title
        end)
        
        -- Render children
        for _, child in ipairs(node.children) do
            table.insert(result, renderNode(child, level + 1))
        end
        
        table.insert(result, '</div>')
    end
    
    table.insert(result, '</div>')
    
    return table.concat(result, '\n')
end

-- Function to generate CSS styles
local function getStyles()
    return mw.html.create('style')
        :wikitext([[
.version-tree {
    text-align: center;
    padding: 20px;
    font-family: sans-serif;
}

.tree-node {
    display: inline-block;
    vertical-align: top;
    text-align: center;
    padding: 0 10px;
    position: relative;
}

.node-content {
    position: relative;
    padding: 10px 0;
}

.node-box {
    display: inline-block;
    border: 1px solid #ccc;
    border-radius: 5px;
    padding: 5px 15px;
    background: #f8f9fa;
    position: relative;
    z-index: 1;
}

.node-children {
    position: relative;
    margin-top: 20px;
}

/* Lines connecting nodes */
.node-children:before {
    content: '';
    position: absolute;
    top: -20px;
    left: 50%;
    border-left: 1px solid #ccc;
    height: 20px;
}

.tree-node .tree-node:before {
    content: '';
    position: absolute;
    top: 0;
    left: 50%;
    border-left: 1px solid #ccc;
    height: 10px;
}

.node-children .tree-node:first-child:after,
.node-children .tree-node:last-child:after {
    content: '';
    position: absolute;
    top: 0;
    width: 50%;
    height: 10px;
    border-top: 1px solid #ccc;
}

.node-children .tree-node:first-child:after {
    right: 50%;
}

.node-children .tree-node:last-child:after {
    left: 50%;
}

.node-children .tree-node:only-child:after {
    display: none;
}

/* Horizontal line connecting siblings */
.node-children {
    position: relative;
}

.node-children:after {
    content: '';
    position: absolute;
    top: 0;
    left: 0;
    right: 0;
    border-top: 1px solid #ccc;
}

.node-children .tree-node:only-child:before {
    height: 10px;
}
]])
        :done()
end

-- Main function to process the versions and generate tree view
function p.getTree(frame)
    local versions = mw.smw.ask({
        "[[Category:Version||Secondary Version]]",
        "?Has Parent Version"
    })
    
    if not versions then
        return "SMW query returned no results."
    end
    
    -- Build tree
    local tree = buildTree(versions, "Has Parent Version")
    
    -- Sort root nodes
    table.sort(tree, function(a, b)
        return a.title < b.title
    end)
    
    -- Create HTML builder
    local html = mw.html.create()
    
    -- Add styles
    html:node(getStyles())
    
    -- Create container
    local container = html:tag('div')
        :addClass('version-tree')
    
    -- Render each root node
    for _, node in ipairs(tree) do
        container:wikitext(renderNode(node, 0))
    end
    
    return html
end

return p