feat: Add keymatch5 for ignoring params in url

dacongda · Jun 3, 2022 · c29c95f · c29c95f
1 parent 601a6c0
commit c29c95f
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 0 deletions.
diff --git a/Casbin.UnitTest/UtilTests/BuiltInFunctionTest.cs b/Casbin.UnitTest/UtilTests/BuiltInFunctionTest.cs
@@ -172,6 +172,24 @@ public void TestKeyMatch4(string key1, string key2, bool expectedResult)
                 BuiltInFunctions.KeyMatch4(key1, key2));
         }
 
+        public static IEnumerable<object[]> KeyMatch5TestData = new[]
+        {
+            new object[] { "/parent/child?status=1&type=2", "/parent/child", true},
+            new object[] { "/parent?status=1&type=2", "/parent/child", false},
+
+            new object[] { "/parent/child/?status=1&type=2", "/parent/child/", true},
+            new object[] { "/parent/child/?status=1&type=2", "/parent/child", false},
+            new object[] { "/parent/child?status=1&type=2", "/parent/child/", false}
+        };
+
+        [Theory]
+        [MemberData(nameof(KeyMatch5TestData))]
+        public void TestKeyMatch5(string key1, string key2, bool expectedResult)
+        {
+            Assert.Equal(expectedResult,
+                BuiltInFunctions.KeyMatch5(key1, key2));
+        }
+
         public static IEnumerable<object[]> GlobMatchTestData = new[]
         {
             new object[] {"/foo", "/foo", true},

diff --git a/Casbin/Model/FunctionMap.cs b/Casbin/Model/FunctionMap.cs
@@ -24,6 +24,7 @@ internal static FunctionMap LoadFunctionMap()
             map.AddFunction("keyMatch2", BuiltInFunctions.KeyMatch2);
             map.AddFunction("keyMatch3", BuiltInFunctions.KeyMatch3);
             map.AddFunction("keyMatch4", BuiltInFunctions.KeyMatch4);
+            map.AddFunction("keyMatch5", BuiltInFunctions.KeyMatch5);
             map.AddFunction("regexMatch", BuiltInFunctions.RegexMatch);
             map.AddFunction("ipMatch", BuiltInFunctions.IPMatch);
             map.AddFunction("globMatch", BuiltInFunctions.GlobMatch);

diff --git a/Casbin/Util/BuiltInFunctions.cs b/Casbin/Util/BuiltInFunctions.cs
@@ -128,6 +128,22 @@ public static bool KeyMatch4(string key1, string key2)
 
             return true;
         }
+        /// <summary>
+        /// KeyMatch determines whether key1 matches the pattern of key2 and ignores the parameters in key2.
+        /// For example, "/foo/bar?status=1&amp;type=2" matches "/foo/bar"
+        /// </summary>
+        /// <param name="key1"> The first argument. </param>
+        /// <param name="key2"> The second argument. </param>
+        /// <returns></returns>
+        public static bool KeyMatch5(string key1, string key2)
+        {
+            var key1Span = key1.AsSpan();
+            var key2Span = key2.AsSpan();
+            int index = key1Span.IndexOf('?');
+            return index is -1
+                ? key1Span.Equals(key2Span, StringComparison.Ordinal)
+                : key1Span.Slice(0, index).Equals(key2Span, StringComparison.Ordinal);
+        }
 
         /// <summary>
         /// Determines whether IP address ip1 matches the pattern of IP address ip2,