fix: getImplicitPermissionsForUser works with rmMap (#272)

Zxilly · web-flow · commit 0d5923998fa7 · 2021-05-19T12:29:54.000+08:00
Signed-off-by: Zxilly &lt;zhouxinyu1001@gmail.com&gt;
diff --git a/src/coreEnforcer.ts b/src/coreEnforcer.ts
@@ -129,6 +129,13 @@ export class CoreEnforcer {
     return <RoleManager>this.rmMap.get('g');
   }
 
+  /**
+   * getNamedRoleManager gets role manager by name.
+   */
+  public getNamedRoleManager(name: string): RoleManager | undefined {
+    return this.rmMap.get(name);
+  }
+
   /**
    * setEffector sets the current effector.
    *
diff --git a/src/enforcer.ts b/src/enforcer.ts
@@ -292,13 +292,15 @@ export class Enforcer extends ManagementEnforcer {
     const q = [name];
     let n: string | undefined;
     while ((n = q.shift()) !== undefined) {
-      const role = await this.getRoleManager().getRoles(n, ...domain);
-      role.forEach((r) => {
-        if (!res.has(r)) {
-          res.add(r);
-          q.push(r);
-        }
-      });
+      for (const rm of this.rmMap.values()) {
+        const role = await rm.getRoles(n, ...domain);
+        role.forEach((r) => {
+          if (!res.has(r)) {
+            res.add(r);
+            q.push(r);
+          }
+        });
+      }
     }
 
     return Array.from(res);

Original file line number	Diff line number	Diff line change
`@@ -129,6 +129,13 @@ export class CoreEnforcer {`
`129`	`129`	`return <RoleManager>this.rmMap.get('g');`
`130`	`130`	`}`
`131`	`131`
	`132`	`+ /**`
	`133`	`+ * getNamedRoleManager gets role manager by name.`
	`134`	`+ */`
	`135`	`+ public getNamedRoleManager(name: string): RoleManager \| undefined {`
	`136`	`+ return this.rmMap.get(name);`
	`137`	`+ }`
	`138`	`+`
`132`	`139`	`/**`
`133`	`140`	`* setEffector sets the current effector.`
`134`	`141`	`*`