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
			local found = {}
			if sfind(restrictionsRow[2],'{i}',1,true) then
				found[1] = true
			end
			if sfind(restrictionsRow[2],'{j}',1,true) then
				found[2] = true
			end
			if sfind(restrictionsRow[2],'{k}',1,true) then
				found[3] = true
			end
			for i = 0, 100 do
				local entry = ''
				if found[1] then
					entry = sreplace(restrictionsRow[2],'{i}',i)
				else
					entry = restrictionsRow[2] .. '/' .. i
				end
				if found[2] then
					for j = 0, 100 do
						entry = sreplace(entry,'{j}',j)
						if found[3] then
							for k = 0, 100 do
								entry = sreplace(entry,'{k}',k)
								if not dataRow[entry] or dataRow[entry] == '' then
									break
								elseif dataRow[entry] == restrictionsRow[3] then
									inList = true
								end
							end
						else
							if not dataRow[entry] or dataRow[entry] == '' then
								break
							elseif dataRow[entry] == restrictionsRow[3] then
								inList = true
							end
						end
					end
				else
					if not dataRow[entry] or dataRow[entry] == '' then
						break
					elseif dataRow[entry] == restrictionsRow[3] then
						inList = true
					end
				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