پودمان:Protect

از ویکی تراث

local function processResult(options, success, ...)
	if not success then
		local message = tostring(... or '(not message)')
		if options.removeLocation then
			message = mw.ustring.gsub(message, '^Module:[^:]+:%d+: ', '', 1)
		end
		return mw.ustring.format(options.errFormat, message)
	end
	return ...
end

local function protect(func, errFormat, options)
	if type(errFormat) == 'table' then
		options = options or errFormat
		errFormat = nil
	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(options, pcall(func, ...))
	end
end

return protect