Module:RestrictionChecker

Documentation for this module may be created at Module:RestrictionChecker/doc

-- Helper Module for checking restrictions
local p = {}


local CommaSplitModule = require('Module:CommaSplit')
local StringReplaceModule = require('Module:StringReplace')
local sreplace = StringReplaceModule.replace
local slen = string.len
local sfind = string.find
local ssub = string.sub

---@param restrictionsTable is a table of restrictions
---@dataRow is the row from the data table to check the restrictions against.
---@return true if all restrictions pass
function p.AllPass(restrictionsTable, dataRow)
	local passesAllRestrictions = true
	for _, restrictionsRowText in ipairs(restrictionsTable) do -- Consider each restrictionsTable row one by one:
		local restrictionsRow = CommaSplitModule.CommaSplit(restrictionsRowText)
		restrictionsRow[3] = mw.text.trim(restrictionsRow[3])
		if restrictionsRow[1] == 'Match' then
			if dataRow[restrictionsRow[2]] ~= restrictionsRow[3] then
				passesAllRestrictions = false
			end
		elseif restrictionsRow[1] == 'Match Any' then
			local foundMatch = false
			for i = 3, 100 do
				if dataRow[restrictionsRow[2]] == restrictionsRow[i] then
					foundMatch = true
				end
			end
			if not foundMatch then
				passesAllRestrictions = false
			end
		elseif restrictionsRow[1] == 'Not Match' then
			if dataRow[restrictionsRow[2]] == restrictionsRow[3] then
				passesAllRestrictions = false
			end
		elseif restrictionsRow[1] == 'Contains' then
			if not sfind(dataRow[restrictionsRow[2]], restrictionsRow[3], 1, true) then
				passesAllRestrictions = false
			end
		elseif restrictionsRow[1] == 'In List' then
			local inList = false
			for i = 0, 100 do
				local entry = ''
				if sfind(restrictionsRow[2],'{i}',1,true) then
					entry = sreplace(restrictionsRow[2],'{i}',i)
				else
					entry = restrictionsRow[2] .. '/' .. i
				end
				if not dataRow[entry] or dataRow[entry] == '' then
					break
				elseif dataRow[entry] == restrictionsRow[3] then
					inList = true
				end
			end
			if not inList then
				passesAllRestrictions = false
			end
		elseif restrictionsRow[1] == 'Not In List' then
			local inList = false
			for i = 0, 100 do
				local entry = ''
				if sfind(restrictionsRow[2],'{i}',1,true) then
					entry = sreplace(restrictionsRow[2],'{i}',i)
				else
					entry = restrictionsRow[2] .. '/' .. i
				end
				if not dataRow[entry] or dataRow[entry] == '' then
					break
				elseif dataRow[entry] == restrictionsRow[3] then
					inList = true
				end
			end
			if inList then
				passesAllRestrictions = false
			end
		end
	end
	return passesAllRestrictions
end

return p