fix: fix CasbinJsGetPermissionForUser() (#251)

Zxilly · web-flow · commit 98c11f1451e7 · 2021-04-05T17:12:34.000+08:00
* fix: fix CasbinJsGetPermissionForUser()

it has unexpected side affect, will change policies when it read from
that

Signed-off-by: Zxilly &lt;zhouxinyu1001@gmail.com&gt;

* test: enhance CasbinJsGetPermissionForUser() tests

Signed-off-by: Zxilly &lt;zhouxinyu1001@gmail.com&gt;

* feat: add deepCopy function

Signed-off-by: Zxilly &lt;zhouxinyu1001@gmail.com&gt;

* style: remove comment

Signed-off-by: Zxilly &lt;zhouxinyu1001@gmail.com&gt;

* feat: add missing import

Signed-off-by: Zxilly &lt;zhouxinyu1001@gmail.com&gt;
diff --git a/src/frontend.ts b/src/frontend.ts
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 import { Enforcer } from './enforcer';
+import { deepCopy } from './util';
 
 /**
  * Experiment!
@@ -41,7 +42,7 @@ export async function casbinJsGetPermissionForUser(e: Enforcer, user: string): P
   s += '[matchers]\n';
   s += `m = ${m.get('m')?.get('m')?.value.replace(/_/g, '.')}`;
   obj['m'] = s;
-  obj['p'] = await e.getPolicy();
+  obj['p'] = deepCopy(await e.getPolicy());
   for (const arr of obj['p']) {
     arr.splice(0, 0, 'p');
   }
diff --git a/src/util/util.ts b/src/util/util.ts
@@ -159,6 +159,17 @@ async function generatorRunAsync(iterator: Generator<any>): Promise<any> {
   }
 }
 
+function deepCopy(obj: Array<any> | any): any {
+  if (typeof obj !== 'object') return;
+  const newObj: any = obj instanceof Array ? [] : {};
+  for (const key in obj) {
+    if (obj.hasOwnProperty(key)) {
+      newObj[key] = typeof obj[key] === 'object' ? deepCopy(obj[key]) : obj[key];
+    }
+  }
+  return newObj;
+}
+
 export {
   escapeAssertion,
   removeComments,
@@ -175,4 +186,5 @@ export {
   getEvalValue,
   generatorRunSync,
   generatorRunAsync,
+  deepCopy,
 };
diff --git a/test/frontend.test.ts b/test/frontend.test.ts
@@ -18,6 +18,11 @@ import { casbinJsGetPermissionForUser } from '../src/frontend';
 
 test('TestCasbinJsGetPermissionForUser', async () => {
   const e = await newEnforcer('examples/rbac_model.conf', 'examples/rbac_with_hierarchy_policy.csv');
+  const a = await casbinJsGetPermissionForUser(e, 'alice');
+  const b = await casbinJsGetPermissionForUser(e, 'alice');
+  if (a !== b) {
+    throw new Error('Unexpected side affect.');
+  }
   const received = JSON.parse(await casbinJsGetPermissionForUser(e, 'alice'));
   const expectedModelStr = readFileSync('examples/rbac_model.conf').toString();
   expect(received['m']).toBe(expectedModelStr.replace(/\n\n/g, '\n'));
