Module:Section link: Difference between revisions

    From Nonbinary Wiki
    m>Mr. Stradivarius
    (fix error placement and tweak default conjunction)
    m>Mr. Stradivarius
    (actually, just make the sections optional)
    Line 15: Line 15:
    checkType('_main', 1, page, 'string', true)
    checkType('_main', 1, page, 'string', true)
    checkType('_main', 3, options, 'table', true)
    checkType('_main', 3, options, 'table', true)
    if type(sections) == 'string' then
    if sections == nil then
    sections = {}
    elseif type(sections) == 'string' then
    sections = {sections}
    sections = {sections}
    elseif type(sections) ~= 'table' then
    elseif type(sections) ~= 'table' then
    error(string.format(
    error(string.format(
    "type error in argument #2 to '_main' (string or table expected, got %s)",
    "type error in argument #2 to '_main' " ..
    "(string, table or nil expected, got %s)",
    type(sections)
    type(sections)
    ), 2)
    ), 2)
    Line 28: Line 31:
    -- Make the link(s).
    -- Make the link(s).
    local isShowingPage = not options.nopage
    local isShowingPage = not options.nopage
    if #sections == 1 then
    if #sections <= 1 then
    local linkPage = page or ''
    local linkPage = page or ''
    local section = sections[1] or 'Notes'
    local section = sections[1] or 'Notes'
    Line 38: Line 41:
    return makeSectionLink(linkPage, section, display)
    return makeSectionLink(linkPage, section, display)
    else
    else
    -- Multiple sections. First, validate the sections table.
    -- Multiple sections. First, make a list of the links to display.
    if #sections < 1 then
    error('at least one section must be specified', 2)
    end
     
    -- Make the list of links.
    local ret = {}
    local ret = {}
    for i, section in ipairs(sections) do
    for i, section in ipairs(sections) do

    Revision as of 10:09, 9 December 2014

    Documentation for this module may be created at Module:Section link/doc

    -- This module implements {{section link}}.
    
    local checkType = require('libraryUtil').checkType
    
    local p = {}
    
    local function makeSectionLink(page, section, display)
    	display = display or section
    	page = page or ''
    	return string.format('[[%s#%s|%s]]', page, section, display)
    end
    
    function p._main(page, sections, options, title)
    	-- Validate input.
    	checkType('_main', 1, page, 'string', true)
    	checkType('_main', 3, options, 'table', true)
    	if sections == nil then
    		sections = {}
    	elseif type(sections) == 'string' then
    		sections = {sections}
    	elseif type(sections) ~= 'table' then
    		error(string.format(
    			"type error in argument #2 to '_main' " ..
    			"(string, table or nil expected, got %s)",
    			type(sections)
    		), 2)
    	end
    	options = options or {}
    	title = title or mw.title.getCurrentTitle()
    
    	-- Make the link(s).
    	local isShowingPage = not options.nopage
    	if #sections <= 1 then
    		local linkPage = page or ''
    		local section = sections[1] or 'Notes'
    		local display = '§&thinsp;' .. section
    		if isShowingPage then
    			page = page or title.prefixedText
    			display = page .. ' ' .. display
    		end
    		return makeSectionLink(linkPage, section, display)
    	else
    		-- Multiple sections. First, make a list of the links to display.
    		local ret = {}
    		for i, section in ipairs(sections) do
    			ret[i] = makeSectionLink(page, section)
    		end
    
    		-- Assemble the list of links into a string with mw.text.listToText.
    		-- We use the default separator for mw.text.listToText, but a custom
    		-- conjunction. There is also a special case conjunction if we only
    		-- have two links.
    		local conjunction
    		if #sections == 2 then
    			conjunction = '&#8203; and '
    		else
    			conjunction = ', and '
    		end
    		ret = mw.text.listToText(ret, nil, conjunction)
    
    		-- Add the intro text.
    		local intro = '§§&thinsp;'
    		if isShowingPage then
    			intro = (page or title.prefixedText) .. ' ' .. intro
    		end
    		ret = intro .. ret
    
    		return ret
    	end
    end
    
    function p.main(frame)
    	local args = require('Module:Arguments').getArgs(frame, {
    		wrappers = 'Template:Section link'
    	})
    
    	-- Sort the arguments.
    	local page
    	local sections, options = {}, {}
    	for k, v in pairs(args) do
    		if k == 1 then
    			-- Doing this in the loop because of a bug in [[Module:Arguments]]
    			-- when using pairs with deleted arguments.
    			page = v
    		elseif type(k) == 'number' then
    			sections[k] = v
    		else
    			options[k] = v
    		end
    	end
    
    	-- Compress the sections array.
    	local function compressArray(t)
    		local nums, ret = {}, {}
    		for num in pairs(t) do
    			nums[#nums + 1] = num
    		end
    		table.sort(nums)
    		for i, num in ipairs(nums) do
    			ret[i] = t[num]
    		end
    		return ret
    	end
    	sections = compressArray(sections)
    
    	return p._main(page, sections, options)
    end
    
    return p