模块:Navplate SMW row display all

来自决策链云智库
Zeroclanzhang讨论 | 贡献2024年1月26日 (五) 22:31的版本

可在模块:Navplate SMW row display all/doc创建此模块的帮助文档

-- 自定义的trim函数
local function trim(s)
    return (s:gsub("^%s*(.-)%s*$", "%1"))
end

local p = {}
local mw = require('mw')

p.cache = {} -- 初始化缓存

-- 新增函数用于获取页面的nodeicon属性
local function getNodeIcon(pageName, frame)
    local ask = '{{#ask:[[' .. pageName .. ']]|?Nodeicon|mainlabel=-}}'
    local result = frame:preprocess(ask)
    return trim(result)
end

function p.navplateSmwRow(frame)
    local args = frame:getParent().args
    local categoryName = args[1]
    local property = args[2]
    local text = args[3]
    local icon = args[4]

    -- 使用缓存结果,减少查询
    local cacheKey = categoryName .. (property or '') .. (text or '')
    if p.cache[cacheKey] then
        return p.cache[cacheKey]
    end

    -- 构建查询
    local query = '[[Category:' .. mw.text.encode(categoryName) .. ']]'
    if property and property ~= '' then
        -- 确保属性查询被正确添加
        query = query .. '[[Has property::' .. mw.text.encode(property) .. ']]'
    end

    local ask = '{{#ask:' .. query ..
            '|format=list' ..
            '|link=all' ..
            '|headers=hide' ..
            '|searchlabel=' ..
            '|class=smwlist' ..
            '|sep=, ' ..
            '}}'

    -- 执行查询
    local results = frame:preprocess(ask)

    -- 缓存查询结果
    p.cache[cacheKey] = results

    -- 解析查询结果并添加图标
    local parsedResults = {}
    for page in string.gmatch(results, "[^,]+") do
        local trimmedPage = trim(page)
        local nodeIcon = getNodeIcon(trimmedPage, frame) -- 获取页面的nodeicon属性
        local iconMarkup = ''
        if nodeIcon and nodeIcon ~= '' then
            iconMarkup = '[[File:' .. mw.text.encode(nodeIcon) .. '|20px|link=]] '
        end
        table.insert(parsedResults, iconMarkup .. trimmedPage)
    end

    -- 格式化输出
    local itemDiv = mw.html.create('div'):addClass('template-navplate-item')
    local metadataAndTextDiv = mw.html.create('div'):addClass('template-navplate-item__label')
    local metadata = mw.html.create('span'):addClass('metadata'):wikitext('[[File:' .. mw.text.encode(icon) .. '|20px|link=]]   ' .. mw.text.encode(text))

    metadataAndTextDiv:node(metadata)

    local listDiv = mw.html.create('div'):addClass('template-navplate-item__list'):wikitext(table.concat(parsedResults, ', '))

    itemDiv
        :node(metadataAndTextDiv)
        :node(listDiv)

    -- tostring之后进行修剪
    local finalHtml = tostring(itemDiv:allDone())
    -- 使用自定义的trim函数去除首尾空格
    finalHtml = trim(finalHtml)

    return finalHtml
end

return p