Jump to content

Module:Random portal component/sandbox: Difference between revisions

From Wikipedia, the free encyclopedia
Content deleted Content added
split the "selected page" check out into an isSelectedPage function
rename isSelectedPage() to isSelectedHeader(), as we are not dealing with actual page titles; and make a new isRootPortalTitle() function to use in all category checks (this excludes subpages from Category:Portals needing attention)
Line 10: Line 10:
}
}


local function isSelectedPage(header)
local function isSelectedHeader(header)
-- Whether a page is selected (a selected article, selected list, etc.)
-- Whether a page is selected (a selected article, selected list, etc.).
-- The "'*" part of the pattern checks for italic or bold text, like
-- ''Selected'' or '''Selected'''.
return string.find(header, "^'*[sS]elected") ~= nil
return string.find(header, "^'*[sS]elected") ~= nil
end
end
Line 27: Line 29:
and not portalNamesContainingSlashes[title.text]
and not portalNamesContainingSlashes[title.text]
)
)
end

local function isRootPortalTitle(title)
-- Whether a title object is for a root portal page.
return isInPortalNamespace(title) and not isPortalSubpage(title)
end
end


Line 34: Line 41:
local i;
local i;


-- don't track DYK etc, only selected articles etc
-- Only track root portal pages that have "Selected x" sections.
if not isSelectedPage(header) then
if not isSelectedHeader(header) or not isRootPortalTitle(title) then
return ""
end

-- no tracking unless we are in Portal namespace
if not isInPortalNamespace(currentTitle) then
return ""
end

-- no tracking if this is a subpage
if isPortalSubpage(currentTitle) then
return ""
return ""
end
end
Line 126: Line 123:
title
title
)
)
if isInPortalNamespace(currentTitle) then
if isRootPortalTitle(currentTitle) then
msg = msg .. '[[Category:Portals needing attention]]'
msg = msg .. '[[Category:Portals needing attention]]'
end
end

Revision as of 05:50, 2 May 2019

-- This module implements [[Template:Random portal component]]

local p = {}

local mRandom = require('Module:Random')
local currentTitle = mw.title.getCurrentTitle()

local portalNamesContainingSlashes = {
	["AC/DC"] = true,
}

local function isSelectedHeader(header)
	-- Whether a page is selected (a selected article, selected list, etc.).
	-- The "'*" part of the pattern checks for italic or bold text, like
	-- ''Selected'' or '''Selected'''.
	return string.find(header, "^'*[sS]elected") ~= nil
end

local function isInPortalNamespace(title)
	-- Whether a title object is in the portal namespace.
	return title.namespace == 100
end

local function isPortalSubpage(title)
	-- Whether a title object is a portal subpage
	return (
		isInPortalNamespace(title)
		and title.isSubpage
		and not portalNamesContainingSlashes[title.text]
	)
end

local function isRootPortalTitle(title)
	-- Whether a title object is for a root portal page.
	return isInPortalNamespace(title) and not isPortalSubpage(title)
end

local function subPageTrackingCategories(pages, max, header)
	local retval = "";
	local availableSubPageCount = 0;
	local i;

	-- Only track root portal pages that have "Selected x" sections.
	if not isSelectedHeader(header) or not isRootPortalTitle(title) then
		return ""
	end

	-- limit checking to prevent Lua overload
	local myMaxCheck = 60
	if tonumber(max) < myMaxCheck then
		myMaxCheck = tonumber(max)
	end
	for i=1,myMaxCheck do 
		local aSubPage = mw.title.new(pages.subpage .. '/' .. i)
		if (aSubPage.exists) then
			availableSubPageCount = availableSubPageCount + 1;
		end
	end
	if myMaxCheck >= tonumber(max) then
		if (availableSubPageCount < tonumber(max)) then
			retval = retval .. "[[Category:Random portal component with fewer available subpages than specified max]]"
		elseif (availableSubPageCount > tonumber(max)) then
			retval = retval .. "[[Category:Random portal component with more available subpages than specified max]]"
		end
	end
	if (availableSubPageCount < 2) then
		retval = retval .. "[[Category:Random portal component with less than 2 available  subpages]]"
	elseif (availableSubPageCount <= 5) then
		retval = retval .. "[[Category:Random portal component with 2–5 available  subpages]]"
	elseif (availableSubPageCount <= 10) then
		retval = retval .. "[[Category:Random portal component with 6–10 available  subpages]]"
	elseif (availableSubPageCount <= 15) then
		retval = retval .. "[[Category:Random portal component with 11–15 available  subpages]]"
	elseif (availableSubPageCount <= 20) then
		retval = retval .. "[[Category:Random portal component with 16–20 available subpages]]"
	elseif (availableSubPageCount <= 25) then
		retval = retval .. "[[Category:Random portal component with 21–25 available subpages]]"
	elseif (availableSubPageCount <= 30) then
		retval = retval .. "[[Category:Random portal component with 26–30 available subpages]]"
	elseif (availableSubPageCount <= 40) then
		retval = retval .. "[[Category:Random portal component with 31–40 available subpages]]"
	elseif (availableSubPageCount <= 50) then
		retval = retval .. "[[Category:Random portal component with 41–50 available subpages]]"
	else
		retval = retval .. "[[Category:Random portal component with over 50 available subpages]]"
	end
	return retval;
end

local function getRandomNumber(max)
	-- gets a random integer between 1 and max; max defaults to 1
	return mRandom.number{max or 1}
end

local function expandArg(args, key)
	-- Emulate how unspecified template parameters appear in wikitext. If the
	-- specified argument exists, its value is returned, and if not the argument
	-- name is returned inside triple curly braces.
	local val = args[key]
	if val then
		return val
	else
		return string.format('{{{%s}}}', key)
	end
end

local function getPages(args)
	local pages = {}
	pages.root = args.rootpage or currentTitle.prefixedText
	pages.subpage = pages.root .. '/' .. expandArg(args, 'subpage')
	pages.random = pages.subpage .. '/' .. getRandomNumber(args.max)
	pages.footer = 'Template:Box-footer'
	return pages
end

local function tryExpandTemplate(frame, title, args)
	local success, result = pcall(frame.expandTemplate, frame, {title = title, args = args})
	if success then
		return result
	else
		local msg = string.format(
			'<strong class="error">The page "[[%s]]" does not exist.</strong>',
			title
		)
		if isRootPortalTitle(currentTitle) then
			msg = msg .. '[[Category:Portals needing attention]]'
		end
		return msg
	end
end

local function getHeader(frame, pages, header, template)
	return tryExpandTemplate(
		frame,
		template or pages.root .. '/box-header',
		{header, pages.random}
	)
end

local function getRandomSubpageContent(frame, pages)
	return tryExpandTemplate(
		frame,
		pages.random
	)
end

local function getFooter(frame, pages, link)
	return tryExpandTemplate(
		frame,
		pages.footer,
		{link}
	)
end

function p._main(args, frame)
	frame = frame or mw.getCurrentFrame()
	local pages = getPages(args)

	local ret = {}
	ret[#ret + 1] = getHeader(frame, pages, args.header or 'subpage', args.headertemplate)
	ret[#ret + 1] = getRandomSubpageContent(frame, pages)
	if not args.footer or not args.footer:find('%S') then
		ret[#ret + 1] = '<div style="clear:both;"></div></div>'
	else
		ret[#ret + 1] = getFooter(frame, pages, string.format(
			'[[%s|%s]]',
			pages.subpage,
			expandArg(args, 'footer')
		))
	end

	return table.concat(ret, '\n') .. subPageTrackingCategories(pages, args.max, args.header)
end

function p._nominate(args, frame)
	frame = frame or mw.getCurrentFrame()
	local pages = getPages(args)

	local ret = {}
	ret[#ret + 1] = getHeader(frame, pages, expandArg(args, 'header'), args.headertemplate)
	ret[#ret + 1] = getRandomSubpageContent(frame, pages)
	ret[#ret + 1] = getFooter(frame, pages, string.format(
		'[[/Nominate/%s|Suggest]] • [[%s|%s]] ',
		expandArg(args, 'subpage'),
		pages.subpage,
		args.footer or 'Archive'
	))

	return table.concat(ret, '\n') .. subPageTrackingCategories(pages, args.max, args.header)
end

local function makeInvokeFunction(func)
	return function (frame)
		local args = require('Module:Arguments').getArgs(frame, {
			trim = false,
			removeBlanks = false,
			wrappers = {
				'Template:Random portal component',
				'Template:Random portal component/BHG-test',
				'Template:Random portal component with nominate'
			}
		})
		return func(args, frame)
	end
end

p.main = makeInvokeFunction(p._main)
p.nominate = makeInvokeFunction(p._nominate)

return p