Module:Documentation: Difference between revisions

@@ Line 31: / Line 31: @@
 cfg.sandboxNoticeTemplate = 'template sandbox notice' -- The name of the template to display at the top of sandbox pages.
 cfg.sandboxNoticeLivepageParam = 1 -- The parameter of the sandbox notice template to send the cfg.livepageArg to.
+cfg.protectionTemplate = 'pp-template' -- The name of the template that displays the protection icon (a padlock on enwiki).
+cfg.protectionTemplateArgs = {docusage = 'yes'} -- Any arguments to send to the protection template.
 -- Display settings
@@ Line 148: / Line 150: @@
 function p.protectionTemplate()
+	local protectionTemplate = cfg.protectionTemplate
-	if currentTitle.namespace == 10 then -- We are in the template namespace.
+	if not (protectionTemplate and currentTitle.namespace == 10) then
-		local frame = mw.getCurrentFrame()
+		-- Don't display the protection template if we are not in the template namespace.
+		return nil
-		local function getProtectionLevel(protectionType)
+	end
-			-- Gets the protection level for the current page.
+	local frame = mw.getCurrentFrame()
-			local level = frame:callParserFunction('PROTECTIONLEVEL', protectionType)
+	local function getProtectionLevel(protectionType)
-			if level ~= '' then
+		-- Gets the protection level for the current page.
-				return level
+		local level = frame:callParserFunction('PROTECTIONLEVEL', protectionType)
-			else
+		if level ~= '' then
-				return nil -- The parser function returns the blank string if there is no match.
+			return level
-			end
-		end
+		else
+			return nil -- The parser function returns the blank string if there is no match.
-		if getProtectionLevel('move') == 'sysop' or getProtectionLevel('edit') then
-			-- The page is full-move protected, or full, template, or semi-protected.
-			return frame:expandTemplate{title = 'pp-template', args = {docusage = 'yes'}}
 		end
+	end
+	if getProtectionLevel('move') == 'sysop' or getProtectionLevel('edit') then
+		-- The page is full-move protected, or full, template, or semi-protected.
+		return frame:expandTemplate{title = protectionTemplate, args = cfg.protectionTemplateArgs}
 	end
 	return nil