Editing Module:Sandbox/Jay D. Easy/tud

-- This module implements the {{Tracks and uses Wikidata}} template.
local yesno = require('Module:Yesno')
local mList = require('Module:List')
local mTableTools = require('Module:TableTools')
local mMessageBox = require('Module:Message box')

local p = {}

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(args)
end

function p._main(args)
	local tuWikidata = mTableTools.compressSparseArray(args)
	local box = p.renderBox(tuWikidata)
	local trackingCategories = p.renderTrackingCategories(args, tuWikidata)
	return box .. trackingCategories
end

function p.renderBox(tuWikidata)
	local boxArgs = {}
	if #tuWikidata < 1 then
		boxArgs.text = '<strong class="error">Error: no properties specified</strong>'
	else
		local tuWikidataLinks = {}
		for i, ts in ipairs(tuWikidata) do
			local sandboxLink = nil
			local tsTitle = mw.title.new(ts)
			if tsTitle then
				local tsSandboxTitle = mw.title.new(string.format('%s:%s/sandbox/%s', tsTitle.nsText, tsTitle.baseText, tsTitle.subpageText))
				if tsSandboxTitle and tsSandboxTitle.exists then
					sandboxLink = string.format(' ([[:%s|sandbox]])', tsSandboxTitle.prefixedText)
				end
			end
			tuWikidataLinks[i] = string.format('[[:d:Property:%s]]', ts, sandboxLink or '')
		end
		local tuWikidataList = mList.makeList('bulleted', tuWikidataLinks)
		boxArgs.text = 'This ' .. 
			(mw.title.getCurrentTitle():inNamespaces(828,829) and 'module' or 'template') ..
			' tracks and uses [[:d:Wikidata:List of properties|Wikidata property]]:\n' .. tuWikidataList
	end
	boxArgs.type = 'notice'
	boxArgs.small = true
	boxArgs.image = '[[File:Wikidata-logo-without-paddings.svg|30px|class=noviewer|alt=|link=]]'
	return mMessageBox.main('mbox', boxArgs)
end

function p.renderTrackingCategories(args, tuWikidata, titleObj)
	if yesno(args.nocat) then
		return ''
	end
	
	local cats = {}
	
	-- Error category
	if #tuWikidata < 1 then
		cats[#cats + 1] = 'Uses Tracks and uses Wikidata templates with errors'
	end
	
	-- Tracks and uses Wikidata category
	titleObj = titleObj or mw.title.getCurrentTitle()
	local subpageBlacklist = {
		doc = true,
		sandbox = true,
		sandbox2 = true,
		testcases = true
	}
	if (titleObj.namespace == 10 or titleObj.namespace == 828)
		and not subpageBlacklist[titleObj.subpageText]
	then
		local category = args.category
		if not category then
			category = category or 'Templates using Tracks and uses Wikidata'
		end
		cats[#cats + 1] = category
		if not yesno(args.noprotcat) then
			local currentProt = titleObj.protectionLevels["edit"] and titleObj.protectionLevels["edit"][1] or nil
			local addedLevelCat = false
			local addedPadlockCat = false
			for i, ts in ipairs(tuWikidata) do
				local tsTitleObj = mw.title.new(ts)
				local tsProt = tsTitleObj.protectionLevels["edit"] and tsTitleObj.protectionLevels["edit"][1] or nil
				if tsProt ~= currentProt and not addedLevelCat then
					cats[#cats + 1] = "Templates using Tracks and uses Wikidata with a different protection level"
				end
				if tsProt and not addedPadlockCat then
					local content = tsTitleObj:getContent()
					if not content:find("{{pp-") then
						cats[#cats + 1] = "Templates using Tracks and uses Wikidata without padlocks"
						addedPadlockCat = true
					end
				end
			end
		end
	end
	for i, cat in ipairs(cats) do
		cats[i] = string.format('[[Category:%s]]', cat)
	end
	return table.concat(cats)
end

return p