查看“模块:Inherit”的源代码

local SMWUtils = require("Module:SMWUtils")
local ListUtils = require("Module:ListUtils")
local p = {}

function _merge_keys(a, b)
    local res = {}
    for k, _ in pairs(a) do
        res[k] = true
    end
    for k, _ in pairs(b) do
        res[k] = true
    end
    return res
end


function p.query_inherited_property(query_property, parent_list)
    local results = {}
    for idx, parent in ipairs(parent_list) do
        local anchor_query = "[["..parent.."]]"
        local property_query = "[[" .. query_property .. "::+]]"
        local query = anchor_query .. property_query
        local res = SMWUtils.query_page_properties(query, {query_property})
        for j, r in ipairs(res) do
            results[r.properties[query_property]] = true
        end
    end
    return results
end

function p.query_by_reference(name, query_property)
    local query = "[["..query_property.."::"..name.."]]"
    return ListUtils.list_to_set(SMWUtils.query_pages(query))
end


function _create_list(root, title, list)
    local node = root:tag("div")
    node:wikitext([[ ''' ]] .. title .. [[''':]])
    local ul = node:tag("ul")
    for k, v in ipairs(list) do
        ul:tag("li"):wikitext("[["..v.."]]")
    end
end


function _split_or_empty(value)
    if value and #value > 0 then
        return mw.text.split(value, ",")
    else
        return {}
    end
end

function p.inherited_lists(frame)
    local name = frame.args.name
    local parents = _split_or_empty(frame.args.parents)
    local parent_property = frame.args.parent_property
    local parent_title = frame.args.parent_title
    local properties = mw.text.split(frame.args.properties, ",")
    local children_title = frame.args.children_title

    local root = mw.html.create("div")

    _create_list(root, parent_title, parents)
    for i, parent in ipairs(parents) do
        mw.smw.set({[parent_property]=parent})
    end

    for i, prop in ipairs(properties) do
        local title = frame.args[prop.."_title"]
        local values = _split_or_empty(frame.args[prop.."_values"])
        local reversed = frame.args[prop.."_reversed"]
        local values_set = ListUtils.list_to_set(values)
        local inherited_values = p.query_inherited_property(prop, parents)
        local final_values = _merge_keys(values_set, inherited_values)
        if reversed then
            local reversed_values = p.query_by_reference(name, reversed)
            final_values = _merge_keys(final_values, reversed_values)
        end

        local final_list = ListUtils.set_to_list(final_values)
        _create_list(root, title, final_list)
        for i, k in ipairs(final_list) do
            mw.smw.set({[prop]=k})
        end
    end

    if children_title then
        local children = p.query_by_reference(name, parent_property)
        local children_list = ListUtils.set_to_list(children)
        _create_list(root, children_title, children_list)
    end

    return root:done()
end

return p