feat: add user interactions in casbinJsGetPermissionForUser method

casbin · Jun 15, 2024 · 844cc65 · 844cc65
1 parent 536d194
commit 844cc65
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/main/java/org/casbin/jcasbin/main/Frontend.java b/src/main/java/org/casbin/jcasbin/main/Frontend.java
@@ -37,6 +37,16 @@ public static String casbinJsGetPermissionForUser(Enforcer e, String user) {
       }
     }
     m.put("p", policies);
+    policies = new ArrayList<>();
+    for (String ptype : model.model.get("g").keySet()) {
+        List<List<String>> policy = model.getPolicy("g", ptype);
+        for (List<String> p : policy) {
+          List<String> tmp = new ArrayList<>(p);
+          tmp.add(0, ptype);
+          policies.add(tmp);
+        }
+    }
+    m.put("g", policies);
     return new Gson().toJson(m);
   }
 }
diff --git a/src/test/java/org/casbin/jcasbin/main/FrontendUnitTest.java b/src/test/java/org/casbin/jcasbin/main/FrontendUnitTest.java
@@ -22,6 +22,7 @@
 import java.nio.file.Paths;
 import java.util.HashMap;
 import java.util.List;
+import java.util.regex.Pattern;
 
 import static org.junit.Assert.assertEquals;
 
@@ -35,6 +36,7 @@ public void testCasbinJsGetPermissionForUser() throws IOException {
     assertEquals(received.get("m"), expectedModelStr);
 
     String expectedPolicyStr = new String(Files.readAllBytes(Paths.get("examples/rbac_with_hierarchy_policy.csv")));
+    expectedPolicyStr = Pattern.compile("\n+").matcher(expectedPolicyStr).replaceAll("\n");
     String[] expectedPolicyItem = expectedPolicyStr.split(",|\n");
     int i = 0;
     for (List<String> sArr : (List<List<String>>) received.get("p")) {
@@ -43,5 +45,11 @@ public void testCasbinJsGetPermissionForUser() throws IOException {
         i++;
       }
     }
+    for (List<String> sArr : (List<List<String>>) received.get("g")) {
+      for (String s : sArr) {
+        assertEquals(expectedPolicyItem[i].trim(), s.trim());
+        i++;
+      }
+    }
   }
 }