Module:Shortcut: Difference between revisions

    From Nonbinary Wiki
    (add a nocat option to suppress category output)
    (use the number of shortcuts to generate the plural for the shortcut heading, rather than the number of list items; also, don't output the heading if we don't have any shortcuts to display)
    Line 34: Line 34:
    end
    end
    table.insert(listItems, options.msg)
    table.insert(listItems, options.msg)
    local nListItems = #listItems


    -- Exit if we have nothing to display
    -- Exit if we have nothing to display
    if nListItems < 1 then
    if #listItems < 1 then
    return nil
    return nil
    end
    end
    Line 56: Line 55:


    -- Shortcut heading
    -- Shortcut heading
    local shortcutHeading = mw.message.newRawMessage(
    local shortcutHeading
    cfg['shortcut-heading'],
    do
    nListItems
    local nShortcuts = #shortcuts
    ):plain()
    if nShortcuts > 0 then
    shortcutHeading = frame:preprocess(shortcutHeading)
    shortcutHeading = mw.message.newRawMessage(
    cfg['shortcut-heading'],
    nShortcuts
    ):plain()
    shortcutHeading = frame:preprocess(shortcutHeading)
    shortcutHeading = shortcutHeading .. '\n'
    end
    end


    -- Shortcut box
    -- Shortcut box
    Line 79: Line 85:
    :tag('small')
    :tag('small')
    :wikitext(shortcutHeading)
    :wikitext(shortcutHeading)
    :newline()
    :tag('ul')
    :tag('ul')
    for i, item in ipairs(listItems) do
    for i, item in ipairs(listItems) do

    Revision as of 08:32, 17 December 2014

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

    -- This module implements {{shortcut}}.
    
    -- Set constants
    local CONFIG_MODULE = 'Module:Shortcut/config'
    
    -- Load required modules
    local yesno = require('Module:Yesno')
    local checkType = require('libraryUtil').checkType
    
    local p = {}
    
    function p._main(shortcuts, options, frame, cfg)
    	checkType('_main', 1, shortcuts, 'table')
    	checkType('_main', 2, options, 'table', true)
    	options = options or {}
    	frame = frame or mw.getCurrentFrame()
    	cfg = cfg or mw.loadData(CONFIG_MODULE)
    
    	-- Validate shortcuts
    	for i, shortcut in ipairs(shortcuts) do
    		if type(shortcut) ~= 'string' or #shortcut < 1 then
    			error(string.format(
    				'shortcut #%d was invalid (shortcuts must be strings of ' ..
    				'at least one character in length)'
    			), 2)
    		end
    	end
    
    	-- Make the list items. These are the shortcuts plus any extra lines such
    	-- as options.msg.
    	local listItems = {}
    	for i, shortcut in ipairs(shortcuts) do
    		listItems[i] = string.format('[[%s]]', shortcut)
    	end
    	table.insert(listItems, options.msg)
    
    	-- Exit if we have nothing to display
    	if #listItems < 1 then
    		return nil
    	end
    
    	local root = mw.html.create()
    
    	-- Anchors
    	local anchorDiv = root
    		:tag('div')
    			:css('position', 'relative')
    			:css('top', '-3em')
    	for i, shortcut in ipairs(shortcuts) do
    		local anchor = mw.uri.anchorEncode(shortcut)
    		anchorDiv:tag('span'):attr('id', anchor)
    	end
    
    	root:newline() -- To match the old [[Template:Shortcut]]
    
    	-- Shortcut heading
    	local shortcutHeading
    	do
    		local nShortcuts = #shortcuts
    		if nShortcuts > 0 then
    			shortcutHeading = mw.message.newRawMessage(
    				cfg['shortcut-heading'],
    				nShortcuts
    			):plain()
    			shortcutHeading = frame:preprocess(shortcutHeading)
    			shortcutHeading = shortcutHeading .. '\n'
    		end
    	end
    
    	-- Shortcut box
    	local shortcutList = root
    		:tag('table')
    			:addClass('shortcutbox noprint')
    			:css('float', 'right')
    			:css('border', '1px solid #aaa')
    			:css('background', '#fff')
    			:css('margin', '.3em .3em .3em 1em')
    			:css('padding', '3px')
    			:css('text-align', 'center')
    				:tag('tr')
    					:tag('th')
    						:addClass('plainlist')
    						:css('border', 'none')
    						:css('background', 'transparent')
    						:tag('small')
    							:wikitext(shortcutHeading)
    							:tag('ul')
    	for i, item in ipairs(listItems) do
    		shortcutList:tag('li'):wikitext(item)
    	end
    
    	-- Error category
    	if shortcuts[1] and not yesno(options.nocat) then
    		local title = mw.title.new(shortcuts[1])
    		if not title or not title.exists then
    			root:wikitext(string.format(
    				'[[%s:%s]]',
    				mw.site.namespaces[14].name,
    				cfg['first-parameter-error-category']
    			))
    		end
    	end
    
    	return tostring(root)
    end
    
    function p.main(frame)
    	local args = require('Module:Arguments').getArgs(frame, {
    		wrappers = 'Template:Shortcut'
    	})
    
    	-- Separate shortcuts from options
    	local shortcuts, options = {}, {}
    	for k, v in pairs(args) do
    		if type(k) == 'number' then
    			shortcuts[k] = v
    		else
    			options[k] = v
    		end
    	end
    
    	-- Compress the shortcut array, which may contain nils.
    	local function compressArray(t)
    		local nums, ret = {}, {}
    		for k in pairs(t) do
    			nums[#nums + 1] = k
    		end
    		table.sort(nums)
    		for i, num in ipairs(nums) do
    			ret[i] = t[num]
    		end
    		return ret
    	end
    	shortcuts = compressArray(shortcuts)
    
    	return p._main(shortcuts, options, frame)
    end
    
    return p