Module:Class

-- This module implements [[Template:Class]], [[Template:Class/icon]] and
-- [[Template:Class/colour]].

local mArguments, mIcon -- modules to lazily load
local data = mw.loadData('Module:Class/data')

local p = {}

-- Initialize helper function variables
local getRawArgs, makeInvokeFunction, fetchTrimmedArg, fetchNormalizedArg
local trim, validateCode

--------------------------------------------------------------------------------
-- Argument helper functions
--------------------------------------------------------------------------------

function getRawArgs(frame, wrapper)
	mArguments = mArguments or require('Module:Arguments')
	return mArguments.getArgs(frame, {
		wrappers = wrapper,
		trim = false,
		removeBlanks = false
	})
end

function makeInvokeFunction(func, wrapper)
	return function (frame)
		local args = getRawArgs(frame, wrapper)
		return func(args)
	end
end

function fetchTrimmedArg(args, key)
	local val = args[key]
	if type(val) == 'string' then
		val = trim(val)
		return val
	else
		return val
	end
end

function fetchNormalizedArg(args, key)
	local val = fetchTrimmedArg(args, key)
	if val ~= '' then
		return val
	end
end

--------------------------------------------------------------------------------
-- String helper functions
--------------------------------------------------------------------------------

function trim(s)
	return s:match('^%s*(.-)%s*$')
end

function validateCode(code)
	if code then
		return trim(code):lower()
	else
		return ''
	end
end

--------------------------------------------------------------------------------
-- Color functions
--------------------------------------------------------------------------------

function p._color(code)
	code = validateCode(code)
	return data.colors[code] or data.colors._DEFAULT
end

function p.color(frame)
	local args = getRawArgs(frame, 'Template:Class/colour')
	local color = p._color(args[1])
	-- We need nowiki tags as template output beginning with "#" triggers
	-- bug 14974.
	return frame:extensionTag('nowiki', color)
end

--------------------------------------------------------------------------------
-- Icon functions
--------------------------------------------------------------------------------

function p._icon(args)
	local code = validateCode(args.class or args[1])
	local iconData = data.icons[code]
	if iconData then
		local span = mw.html.create('span')
		span
			:cssText(args.style)
			:attr('title', iconData.title)
			:wikitext(string.format(
				'[[File:%s|%s|16px%s]]',
				iconData.image,
				iconData.tooltip,
				iconData.suppressLink and '|link=|alt=' or ''
			))
		return tostring(span)
	else
		mIcon = require('Module:Icon')
		return mIcon._main{args[1]}
	end
end

p.icon = makeInvokeFunction(p._icon, 'Template:Class/icon')

--------------------------------------------------------------------------------
-- Class functions
--------------------------------------------------------------------------------

function p._class(args)
	local cell = fetchArg(args, 'heading') and 'th' or 'td'
	cell = mw.html.create(cell)
	cell
		:addClass('assess')
		:addClass('assess' .. mw.ustring.lower(args[1] or ''))
		:addClass(fetchArg(args, 'class'))
		:css('text-align', 'center')
		:css('white-space', 'nowrap')
		:css('font-weight', args.bold ~= 'no' and 'bold' or nil)
		:css('background', p._color())
end

p.class = makeInvokeFunction(p._class, 'Template:Class')

return p
[1] This is assumed to be a light-mode colour; it might be desirable to split this into light- and dark-mode properties.
[underscores-2] Underscores are probably better for file names because they're more easily used in URLs, and don't hurt anything being used in wikitext.
[lowerTitle-3] Lowercase is better for full or generic words, e.g. "featured", but title case is better for constructed values, e.g. "Start-Class".
[spaces-4] Spaces are probably better for referencing wiki pages because they're easily used as strings or in wikitext; use in URLs would have to do other transformations anyway.
[model 1]
[model 2]
[model 3]
[model 4]
Module:Class

Usage

Documentation for Module:Class/definition.json

Object model notes
