Module:Class mask

-- This module implements [[Template:Class mask]].

local lang -- Lazily load a language object

local p = {}

local function getDocWarning(title)
	if title.namespace == 10 and title.subpageText == 'class' then
		return mw.getCurrentFrame():expandTemplate{
			title = 'Class mask/doc warning'
		}
	end
end

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

local function ucfirst(s)
	-- Returns the given string with the first character in upper case.
	-- Should not be used with non-ascii strings.
	return s:sub(1, 1):upper() .. s:sub(2, -1)
end

local function isTruthyBParam(s)
	s = s and s:lower()
	return not s or s == 'yes' or s == 'y' or s == '1' or s == 'pass' or s == 'na' or s == 'n/a' or s == '¬' or s == 'unused'
end

local function resolveFQSgrade(grade, args)
	if (args[grade] or args.FQS) == 'yes' then
		return ucfirst(grade)
	else
		return 'NA'
	end
end

local function resolveExtraGrade(grade, args)
	if args[grade] == 'yes' then
		return ucfirst(grade)
	else
		return 'NA'
	end
end

local function getGrade(args, title)
	local grade = args[1]
	grade = grade and trim(grade):lower()

	local ret

	-- Undefined
	if not grade or grade == '¬' then
		ret = '¬'

	-- Ucfirst
	elseif (grade == 'start' or grade == 'stub' or grade == 'list')
		and args[grade] ~= 'no'
	then
		ret = ucfirst(grade)

	-- B
	elseif grade == 'b' then
		local bParams = {'b1', 'b2', 'b3', 'b4', 'b5', 'b6'}
		local isExtended = false
		for _, param in ipairs(bParams) do
			if args[param] then
				isExtended = true
				break
			end
		end
		if isExtended then
			local isB = true
			for _, param in ipairs(bParams) do
				if not isTruthyBParam(args[param]) then
					isB = false
					break
				end
			end
			ret = isB and 'B' or 'C'
		elseif args.b ~= 'no' then
			ret = 'B'
		end

	-- Upper-case
	elseif (grade == 'fa' or grade == 'fl' or grade == 'a' or grade == 'ga' or grade == 'c')
		and args[grade] ~= 'no'
	then
		ret = grade:upper()

	-- File
	elseif grade == 'file' or grade == 'image' or grade == 'img' then
		ret = resolveFQSgrade('file', args)

	-- Category
	elseif grade == 'category' or grade == 'cat' or grade == 'categ' then
		ret = resolveFQSgrade('category', args)

	-- Disambguation
	elseif grade == 'dab' or grade == 'disambig' or grade == 'disambiguation' or grade == 'disamb' then
		ret = resolveFQSgrade('disambig', args)

	-- Redirect
	elseif grade == 'redirect' or grade == 'red' or grade == 'redir' then
		ret = resolveExtraGrade('redirect', args)

	-- Portal, Project and Draft
	elseif grade == 'portal' or grade == 'project' or grade == 'draft' then
		ret = resolveFQSgrade(grade, args)

	-- Template
	elseif grade == 'template' or grade == 'temp' or grade == 'tpl' or grade == 'templ' then
		ret = resolveFQSgrade('template', args)

	-- Book
	elseif grade == 'book' then
		ret = resolveExtraGrade('book', args)

	-- FM
	elseif grade == 'fm' then
		if args.fm == 'yes' then
			ret = 'FM'
		else
			ret = resolveFQSgrade('file', args)
		end

	end

	return ret
end

function p._main(args, title)
	title = title or mw.title.getCurrentTitle()
	local docWarning = getDocWarning(title) or ''
	local grade = getGrade(args, title) or ''
	return docWarning .. grade
end

function p.main(frame)
	return p._main(frame:getParent().args)
end

return p
Quality function
