Jump to content

Module:Category main article/sandbox

From Wikipedia, the free encyclopedia
This is an old revision of this page, as edited by Mclay1 (talk | contribs) at 14:58, 5 November 2025. The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.
-- This module implements {{Category main article}}.

local mHatnote = require('Module:Hatnote')
local mFormatLink = require('Module:Format link')
local yesno = require('Module:Yesno')
local mTableTools -- lazily initialise
local mArguments -- lazily initialise

local p = {}

-- Helper: true if title is Template:Category main article or subpage
local function isTemplateOrSubpage(title)
	local t = type(title) == 'string' and title or (title and title.prefixedText or '')
	return t == 'Template:Category main article'
		or t:match('^Template:Category main article/.+')
end

function p.catMain(frame)
	mTableTools = require('Module:TableTools')
	mArguments = require('Module:Arguments')

	local args = mArguments.getArgs(frame)
	local pages = mTableTools.compressSparseArray(args)
	if #pages == 0 and args[1] then
		pages = { args[1] }
	end

	local options = {
		article = args.article,
		selfref = args.selfref,
		_rawPages = pages,
	}

	local thisTitle = mw.title.getCurrentTitle()
	local titleText = thisTitle and thisTitle.prefixedText or ''
	local ns = thisTitle and thisTitle.namespace or nil

	-- Skip tracking categories on template itself
	local isTemplate = isTemplateOrSubpage(titleText)

	-- Generate hatnote text and get first output link
	local text, firstLink = p._catMain(options, unpack(pages))

	if isTemplate then
		return text
	end

	-- Add tracking categories
	local cats = {}

	-- Wrong namespace detection (Article, Draft)
	if ns == 0 or ns == 118 then
		table.insert(cats, '[[Category:Articles using category hatnotes]]')
	end

	-- Title mismatch detection
	if #pages == 1 and firstLink then
		local compareStr = firstLink:gsub('^[^:]*:', '')  -- remove namespace
		local catName = (thisTitle and thisTitle.text or ''):gsub('^[^:]*:', '')
		if compareStr ~= catName then
			table.insert(cats, '[[Category:Category main article does not match category title]]')
		end
	end

	if #cats > 0 then
		text = text .. '\n' .. table.concat(cats, '\n')
	end

	return text
end

function p._catMain(options, ...)
	options = options or {}
	local pages = {...}
	local thisTitle = mw.title.getCurrentTitle()
	local fullTitle = thisTitle and thisTitle.prefixedText or ''
	local thisText = thisTitle and thisTitle.text or (pages[1] or '')

	local isTemplate = isTemplateOrSubpage(fullTitle)

	-- Red link detection
	local rawLinks = mFormatLink.formatPages({
		categorizeMissing = not isTemplate and 'Categories with hatnote templates targeting a non-existent page' or nil
	}, pages)

	local links = {}
	for i, link in ipairs(rawLinks) do
		links[i] = tostring(link)
	end

	local firstOutputLink

	if not links[1] or links[1] == '' then
		-- auto-fill
		local title = mw.title.new(thisText)
		if title.isRedirect then
			title = title.redirectTarget
		end

		-- Use categorizeMissing only if not template
		links[1] = tostring(mFormatLink._formatLink{
			link = title.text,
			categorizeMissing = not isTemplate and 'Categories with hatnote templates targeting a non-existent page' or nil
		})

		pages = { title.text }
		firstOutputLink = title.text  -- resolved redirect
	else
		-- Use the resolved link text (strip formatting)
		firstOutputLink = links[1]:gsub("'''(.-)'''", "%1"):gsub("#.*$", "")
	end

	-- Bold links
	for i, link in ipairs(links) do
		links[i] = string.format("'''%s'''", link)
	end

	-- Determine pagetype
	local pagetype
	if options.article ~= nil then
		pagetype = yesno(options.article) ~= false and 'article' or 'page'
	elseif pages and pages[1] then
		local page = pages[1]:gsub("|.*","")
		pagetype = mw.title.new(page).namespace == 0 and "article" or "page"
	else
		pagetype = "article"
	end

	-- Work out whether we need to be singular or plural
	local stringToFormat
	if #links > 1 then
		stringToFormat = 'The main %ss for this [[Help:Categories|category]] are %s.'
	else
		stringToFormat = 'The main %s for this [[Help:Categories|category]] is %s.'
	end

	-- Build text
	local text = string.format(
		stringToFormat,
		pagetype,
		mw.text.listToText(links)
	)

	-- Pass through Module:Hatnote
	local hnOptions = { selfref = options.selfref }
	text = mHatnote._hatnote(text, hnOptions)

	return text, firstOutputLink
end

return p