feat: Add GetPolicyTypes APIs of IReadOnlyPolicyStore

casbin · Oct 4, 2022 · a6c464e · a6c464e
1 parent 5c007fd
commit a6c464e
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/Casbin/Abstractions/Model/IReadOnlyPolicyStore.cs b/Casbin/Abstractions/Model/IReadOnlyPolicyStore.cs
@@ -12,7 +12,11 @@ public interface IReadOnlyPolicyStore
 
  public IEnumerable<IPolicyValues> GetPolicy(string section, string policyType);
 
- public IDictionary<string, IEnumerable<IPolicyValues>> GetPolicyAllType(string section);
+ public IEnumerable<string> GetPolicyTypes(string section);
+
+ public IDictionary<string, IEnumerable<string>> GetPolicyTypesAllSections();
+
+ public IDictionary<string, IEnumerable<IPolicyValues>> GetPolicyAllType(string section); 
 
  public IEnumerable<IPolicyValues> GetFilteredPolicy(string section, string policyType, int fieldIndex,
  IPolicyValues fieldValues);

diff --git a/Casbin/Model/DefaultPolicyStore.cs b/Casbin/Model/DefaultPolicyStore.cs
@@ -40,6 +40,19 @@ public bool AddNode(string section, string type, PolicyAssertion policyAssertion
  public IEnumerable<IPolicyValues> GetPolicy(string section, string policyType)
  => GetNode(section, policyType).GetPolicy();
 
+ public IEnumerable<string> GetPolicyTypes(string section)
+ => GetNodes(section).Select(x => x.Key);
+
+ public IDictionary<string, IEnumerable<string>> GetPolicyTypesAllSections()
+ {
+ Dictionary<string, IEnumerable<string>> res = new Dictionary<string, IEnumerable<string>>();
+ foreach (var keyValuePair in _nodesMap)
+ {
+ res.Add(keyValuePair.Key, keyValuePair.Value.Select(x => x.Key));
+ }
+ return res;
+ }
+
  public IDictionary<string, IEnumerable<IPolicyValues>> GetPolicyAllType(string section)
  => GetNodes(section).ToDictionary(kv =>
  kv.Key, x => GetPolicy(section, x.Key));