KnownUser.V3.Lua/SDK/ValidatorHelpers.lua at master · queueit/KnownUser.V3.Lua · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
local utils = require("Utils")
local comparisonOperatorHelper = require("ComparisonOperatorHelper")

local model = {
	UrlValidatorHelper = {
		evaluate = function(triggerPart, url)
			-- Private functions
			local function getUrlPart(_urlPart, _url)
				-- Private functions
				local function getHostNameFromUrl(__url)
					return utils.toString(__url:match('^%w+://([^/]+)'))
				end

				local function getPathFromUrl(__url)
					local pathAndQuery = utils.toString(__url:match('^%w+://[^/]+(.*)'))
					local query = utils.toString(__url:match('%?+.*'))
					local path = pathAndQuery:gsub(utils.escapeMagicChars(query), "")

					return path
				end
				-- END Private functions

				if (_urlPart == "PagePath") then
					return getPathFromUrl(_url)
				end
				if (_urlPart == "PageUrl") then
					return _url
				end
				if (_urlPart == "HostName") then
					return getHostNameFromUrl(_url)
				end

				return ""
			end
			-- END Private functions

			if (triggerPart == nil or
				triggerPart["Operator"] == nil or
				triggerPart["IsNegative"] == nil or
				triggerPart["IsIgnoreCase"] == nil or
				triggerPart["UrlPart"] == nil) then
				return false
			end

			return comparisonOperatorHelper.evaluate(
				triggerPart["Operator"],
				triggerPart["IsNegative"],
				triggerPart["IsIgnoreCase"],
				getUrlPart(triggerPart["UrlPart"], url),
				triggerPart["ValueToCompare"],
				triggerPart["ValuesToCompare"])
		end
	},
	CookieValidatorHelper = {
		evaluate = function(triggerPart, request)
			if (triggerPart == nil or
				triggerPart["Operator"] == nil or
				triggerPart["IsNegative"] == nil or
				triggerPart["IsIgnoreCase"] == nil or
				triggerPart["CookieName"] == nil) then
				return false
			end

			local cookieValue = ""
			local cookieName = triggerPart["CookieName"]
			if (cookieName ~= nil and request.getUnescapedCookieValue(cookieName) ~= nil) then
				cookieValue = request.getUnescapedCookieValue(cookieName)
			end

			return comparisonOperatorHelper.evaluate(
				triggerPart["Operator"],
				triggerPart["IsNegative"],
				triggerPart["IsIgnoreCase"],
				cookieValue,
				triggerPart["ValueToCompare"],
				triggerPart["ValuesToCompare"])
		end
	},
	UserAgentValidatorHelper  = {
		evaluate = function(triggerPart, request)
			if (triggerPart == nil or
				triggerPart["Operator"] == nil or
				triggerPart["IsNegative"] == nil or
				triggerPart["IsIgnoreCase"] == nil) then
				return false
			end

			local headerValue = request.getHeader("user-agent")
			if (headerValue == nil) then
				headerValue = ""
			end

			return comparisonOperatorHelper.evaluate(
				triggerPart["Operator"],
				triggerPart["IsNegative"],
				triggerPart["IsIgnoreCase"],
				headerValue,
				triggerPart["ValueToCompare"],
				triggerPart["ValuesToCompare"])
		end
	},
	HttpHeaderValidatorHelper = {
		evaluate = function(triggerPart, request)
			if (triggerPart == nil or
				triggerPart["Operator"] == nil or
				triggerPart["IsNegative"] == nil or
				triggerPart["IsIgnoreCase"] == nil or
				triggerPart["HttpHeaderName"] == nil) then
				return false
			end

			local headerValue = ""
			local headerName = triggerPart["HttpHeaderName"]
			if (headerName ~= nil and request.getHeader(string.lower(headerName)) ~= nil) then
				headerValue = request.getHeader(string.lower(headerName))
			end

			return comparisonOperatorHelper.evaluate(
				triggerPart["Operator"],
				triggerPart["IsNegative"],
				triggerPart["IsIgnoreCase"],
				headerValue,
				triggerPart["ValueToCompare"],
				triggerPart["ValuesToCompare"])
		end
	},
	RequestBodyValidatorHelper = {
		evaluate = function(triggerPart, request)
			if (triggerPart == nil or
				triggerPart["Operator"] == nil or
				triggerPart["IsNegative"] == nil or
				triggerPart["IsIgnoreCase"] == nil) then
				return false
			end

			local requestBody = request.getBody()

			return comparisonOperatorHelper.evaluate(
				triggerPart["Operator"],
				triggerPart["IsNegative"],
				triggerPart["IsIgnoreCase"],
				requestBody,
				triggerPart["ValueToCompare"],
				triggerPart["ValuesToCompare"])
		end
	}
}

return model