Module:Protect: Difference between revisions

Module:Protect (edit)

262 bytes added , 4 years ago

m

5 revisions imported from wikipedia:Module:Protect

540

edits

@@ Line 1: / Line 1: @@
-local function wrapMessage(errMessage)
+local function processResult(options, success, ...)
-	return '<span class="error">' .. errMessage .. '</span>'
-end
-local defaultFormat = wrapMessage('Error: %s')
-local function processResult(errFormat, success, ...)
 	if not success then
-		return string.format(errFormat, tostring(... or '(no message)'))
+		local message = tostring(... or '(no message)')
+		if options.removeLocation then
+			message = string.gsub(message, '^Module:[^:]+:%d+: ', '', 1)
+		end
+		return string.format(options.errFormat, message)
 	end
 	return ...
 end
-local function protect(func, errFormat, raw)
+local function protect(func, errFormat, options)
-	if not errFormat then
+	if type(errFormat) == 'table' then
-		errFormat = defaultFormat
+		options = options or errFormat
-	elseif not raw then
+		errFormat = nil
-		errFormat = wrapMessage(errFormat)
+	end
+	options = mw.clone(options) or {}
+	options.errFormat = errFormat or options.errFormat or 'Error: %s'
+	if not options.raw then
+		options.errFormat = '<strong class="error">' .. options.errFormat .. '</strong>'
 	end
+	options.removeLocation = options.removeLocation == nil or options.removeLocation
 	return function (...)
-		return processResult(errFormat, pcall(func, ...))
+		return processResult(options, pcall(func, ...))
 	end
 end
 return protect