Difference between revisions of "Module:Main/sandbox"

From blackwiki
Jump to navigation Jump to search
test>Nihiltres
(Prototype improvement: offload common stuff onto Module:Labelled list hatnote)
test>Frietjes
Line 10: Line 10:
 
local mHatnote = require('Module:Hatnote')
 
local mHatnote = require('Module:Hatnote')
 
local mHatlist = require('Module:Hatnote list')
 
local mHatlist = require('Module:Hatnote list')
local mLabelledList = require('Module:Labelled list hatnote')
 
 
local mArguments -- lazily initialise
 
local mArguments -- lazily initialise
 
local p = {}
 
local p = {}
Line 17: Line 16:
 
mArguments = require('Module:Arguments')
 
mArguments = require('Module:Arguments')
 
local args = mArguments.getArgs(frame, {parentOnly = true})
 
local args = mArguments.getArgs(frame, {parentOnly = true})
local pages = mLabelledList.preprocessDisplays(args)
+
local pages = {}
 +
for k, v in pairs(args) do
 +
if type(k) == 'number' then
 +
local display = args['label ' .. k] or args['l' .. k]
 +
local page = display and
 +
string.format('%s|%s', string.gsub(v, '|.*$', ''), display) or v
 +
pages[#pages + 1] = page
 +
end
 +
end
 +
if #pages == 0 and mw.title.getCurrentTitle().namespace == 0 then
 +
return mHatnote.makeWikitextError(
 +
'no page names specified',
 +
'Template:Main#Errors',
 +
args.category
 +
)
 +
end
 
local options = {
 
local options = {
selfref = args.selfref,
+
selfref = args.selfref
category = args.category
 
 
}
 
}
 
return p._main(pages, options)
 
return p._main(pages, options)
 
end
 
end
  
function p._main(pages, options)
+
function p._main(args, options)
 
-- Get the list of pages. If no first page was specified we use the current
 
-- Get the list of pages. If no first page was specified we use the current
 
-- page name.
 
-- page name.
 
local currentTitle = mw.title.getCurrentTitle()
 
local currentTitle = mw.title.getCurrentTitle()
if #pages == 0 then
+
if #args == 0 then args = {currentTitle.text} end
if mw.title.getCurrentTitle().namespace == 0 then
+
local firstPage = string.gsub(args[1], '|.*$', '')
return mHatnote.makeWikitextError(
 
'no page names specified',
 
'Template:Main#Errors',
 
options.category
 
)
 
else
 
pages = {currentTitle.text}
 
end
 
end
 
local firstPage = string.gsub(pages[1], '|.*$', '')
 
 
-- Find the pagetype.
 
-- Find the pagetype.
local pageType = mHatnote.findNamespaceId(firstPage) == 0 and 'article' or 'page'
+
local pageType = (string.match(firstPage, '#') and (#args < 2) and 'section') or (mHatnote.findNamespaceId(firstPage) == 0 and 'article' or 'page')
 +
-- Make the formatted link text
 +
list = mHatlist.andList(args, true)
 +
-- Build the text.
 +
local isPlural = #args > 1
 +
local mainForm
 
local curNs = currentTitle.namespace
 
local curNs = currentTitle.namespace
 
if (curNs == 14) or (curNs == 15) then --category/talk namespaces
 
if (curNs == 14) or (curNs == 15) then --category/talk namespaces
local mainForm = #pages > 1 and
+
mainForm = isPlural and
'The main %ss for this [[Wikipedia:Categorization|category]] are %s.'
+
'The main %ss for this [[Wikipedia:Categorization|category]] are %s'
 
or
 
or
'The main %s for this [[Wikipedia:Categorization|category]] is %s.'
+
'The main %s for this [[Wikipedia:Categorization|category]] is %s'
return mHatnote._hatnote(
 
string.format(mainForm, pageType, mHatlist.andList(pages, true)),
 
{selfref = options.selfref}
 
)
 
 
else
 
else
return mLabelledList._labelledList(
+
mainForm = isPlural and 'Main %ss: %s' or 'Main %s: %s'
pages,
 
{
 
string.format('Main %s', pageType),
 
string.format('Main %ss', pageType)
 
},
 
{
 
category = options.category,
 
selfref = options.selfref,
 
template = 'Template:Main'
 
}
 
)
 
 
end
 
end
 +
local text = string.format(mainForm, pageType, list)
 +
-- Process the options and pass the text to the _rellink function in
 +
-- [[Module:Hatnote]].
 +
options = options or {}
 +
local hnOptions = {
 +
selfref = options.selfref
 +
}
 +
return mHatnote._hatnote(text, hnOptions)
 
end
 
end
  
 
return p
 
return p

Revision as of 00:33, 31 January 2017

Documentation for this module may be created at Module:Main/sandbox/doc

--[[
-- This module produces a link to a main article or articles. It implements the
-- template {{main}}.
-- 
-- If the module is used in category or category talk space, it produces "The
-- main article for this category is xxx". Otherwise, it produces
-- "Main article: xxx".
--]]

local mHatnote = require('Module:Hatnote')
local mHatlist = require('Module:Hatnote list')
local mArguments -- lazily initialise
local p = {}

function p.main(frame)
	mArguments = require('Module:Arguments')
	local args = mArguments.getArgs(frame, {parentOnly = true})
	local pages = {}
	for k, v in pairs(args) do
		if type(k) == 'number' then
			local display = args['label ' .. k] or args['l' .. k]
			local page = display and
				string.format('%s|%s', string.gsub(v, '|.*$', ''), display) or v
			pages[#pages + 1] = page
		end
	end
	if #pages == 0 and mw.title.getCurrentTitle().namespace == 0 then
		return mHatnote.makeWikitextError(
			'no page names specified',
			'Template:Main#Errors',
			args.category
		)
	end
	local options = {
		selfref = args.selfref
	}
	return p._main(pages, options)
end

function p._main(args, options)
	-- Get the list of pages. If no first page was specified we use the current
	-- page name.
	local currentTitle = mw.title.getCurrentTitle()
	if #args == 0 then args = {currentTitle.text} end
	local firstPage = string.gsub(args[1], '|.*$', '')
	-- Find the pagetype.
	local pageType = (string.match(firstPage, '#') and (#args < 2) and 'section') or (mHatnote.findNamespaceId(firstPage) == 0 and 'article' or 'page')
	-- Make the formatted link text
	list = mHatlist.andList(args, true)
	-- Build the text.
	local isPlural = #args > 1
	local mainForm
	local curNs = currentTitle.namespace
	if (curNs == 14) or (curNs == 15) then --category/talk namespaces
		mainForm = isPlural and
			'The main %ss for this [[Wikipedia:Categorization|category]] are %s'
			or
			'The main %s for this [[Wikipedia:Categorization|category]] is %s'
	else
		mainForm = isPlural and 'Main %ss: %s' or 'Main %s: %s'
	end
	local text = string.format(mainForm, pageType, list)
	-- Process the options and pass the text to the _rellink function in
	-- [[Module:Hatnote]].
	options = options or {}
	local hnOptions = {
		selfref = options.selfref
	}
	return mHatnote._hatnote(text, hnOptions)
end

return p