模块:Navplate SMW row display all

来自决策链云智库
Zeroclanzhang讨论 | 贡献2024年1月27日 (六) 01:33的版本

可在模块: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:[[nodename::' .. pageName .. ']]|?nodeicon' ..
            '|format=list' ..
            '|link=none' ..  -- 不生成链接,仅获取属性值
            '|headers=hide' ..
            '|searchlabel=' ..
            '|class=smwlist' ..
            '|sep=,}}' -- 使用逗号作为分隔符
    local result = frame:preprocess(ask)
    return trim(result)
end

-- 新增函数用于从nodeicon值中提取src属性
local function extractSrcFromNodeicon(nodeicon)
    local srcPattern = 'src="([^"]+)"'
    local src = nodeicon:match(srcPattern)
    return src
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=none' ..  -- 不生成链接,仅列出页面名称
            '|headers=hide' ..
            '|searchlabel=' ..
            '|class=smwlist' ..
            '|sep=,}}' -- 使用逗号作为分隔符

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

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

    -- 解析查询结果并处理图标src
    local parsedResults = {}
    for pageName in string.gmatch(results, '%[%[(.-)%]%]') do
        local trimmedPageName = trim(pageName)
        local nodeicon = getnodeicon(trimmedPageName, frame) -- 获取页面的nodeicon属性
        local imageSrc = ''
        if nodeicon and nodeicon ~= '' then
            -- 提取图标src
            imageSrc = extractSrcFromNodeicon(nodeicon)
        end
        -- 构造带有图标src的隐藏链接
        local formattedPageName = string.format('<span style="display:none;"><a href="/wiki/%s">%s</a></span>', mw.text.encode(trimmedPageName), imageSrc)
        table.insert(parsedResults, formattedPageName)
    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=]]&nbsp;&nbsp;&nbsp;' .. 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())
    finalHtml = trim(finalHtml)

    return finalHtml
end

return p