模块:Lua banner

local yesno = require('Module:Yesno')
local mTableTools = require('Module:TableTools')
local mMessageBox = require('Module:Message box')

local p = {}

function p.main3(frame)
	local args = (require('Module:Arguments').getArgs(frame))
	local currentTitle = mw.title.getCurrentTitle()
	local items = {}
	for k, v in mTableTools.sparseIpairs(args) do
		local title = mw.title.new(v, 828)
		table.insert(items, string.format('[[%s|%s]]', tostring(title), v))
	end
	if #items == 0 then
		local subpageBlacklist = {doc=true,sandbox=true,sandbox2=true,testcases=true}
		local fallback = subpageBlacklist[currentTitle.subpageText] and currentTitle.baseText or currentTitle.text
		local title = mw.title.new(fallback, 828)
		table.insert(items, string.format('[[%s|%s]]', tostring(title), fallback))
	end
	return table.concat(items, '、') .. p.renderTrackingCategories(args, args, currentTitle)
end

function p.main(frame)
	local origArgs = frame:getParent().args
	local args = {}
	for k, v in pairs(origArgs) do
		v = v:match('^%s*(.-)%s*$')
		if v ~= '' then args[k] = v end
	end
	return p._main(frame, args)
end

function p._main(frame, args)
	local modules = mTableTools.compressSparseArray(args)
	return p.renderBox(modules) .. p.renderTrackingCategories(args, modules)
end

function p.main2(frame)
	local origArgs = frame:getParent().args
	local args = {}
	for k, v in pairs(origArgs) do
		v = v:match('^%s*(.-)%s*$')
		if v ~= '' then args[k] = v end
	end
	return p.renderBox(mTableTools.compressSparseArray(args))
end

function p.renderBox(modules)
	local boxArgs = {}
	if #modules < 1 then
		boxArgs.text = '<strong class="error">错误：未指定模块</strong>'
	else
		local links = {}
		for i, m in ipairs(modules) do
			local link = '[[:Module:' .. mw.title.new(m).text .. ']]'
			local sb = mw.title.new(m .. '/sandbox')
			if sb and sb.exists then
				link = link .. string.format('（[[:%s|沙盒]]）', sb.fullText)
			end
			links[i] = link
		end
		local t = mw.title.getCurrentTitle()
		if t.subpageText == 'doc' then t = t.basePageTitle end
		local pfx
		if t:inNamespaces(828,829) then pfx = '此模块'
		elseif t.namespace == 10 then pfx = '此模板'
		else pfx = '此页面' end
		boxArgs.text = pfx .. '基于以下[[Lua]]模块编写：' .. table.concat(links, '、')
	end
	boxArgs.type = 'notice'
	boxArgs.small = true
	boxArgs.image = '[[File:Lua-Logo.svg|30px|alt=|link=]]'
	return mMessageBox.main('mbox', boxArgs)
end

function p.renderTrackingCategories(args, modules, titleObj)
	if yesno(args.nocat) then return '' end
	local cats = {}
	if #modules < 1 then cats[#cats + 1] = '有错误的Lua模板' end
	titleObj = titleObj or mw.title.getCurrentTitle()
	local bl = {doc=true,sandbox=true,sandbox2=true,testcases=true}
	if not bl[titleObj.subpageText] and titleObj.namespace == 10 then
		local cat = args.category
		if not cat then
			local cs = {['Module:String'] = '使用Module:String的模板'}
			cat = modules[1] and cs[modules[1]]
		end
		if cat then cats[#cats + 1] = cat end
		cats[#cats + 1] = 'Lua模板'
		local pcn = '使用受保护Lua模块的模板'
		if not args.noprotcat then
			local pl = {autoconfirmed=1,extendedconfirmed=2,templateeditor=3,sysop=4}
			local cp = titleObj.protectionLevels["edit"][1]
			cp = cp and pl[cp] or 0
			for i, m in ipairs(modules) do
				local mt = mw.title.new(m)
				if mt and mt.protectionLevels["edit"] then
					local mp = mt.protectionLevels["edit"][1]
					mp = mp and pl[mp] or 0
					if mp < cp then cats[#cats + 1] = pcn; break end
				end
			end
		end
	end
	for i, c in ipairs(cats) do cats[i] = string.format('[[Category:%s]]', c) end
	return table.concat(cats)
end

return p