add modify role and permission method

lidianhao123 · Nov 13, 2017 · 4b9c978 · 4b9c978
1 parent 8512ed8
commit 4b9c978
Showing 1 changed file with 27 additions and 4 deletions.
diff --git a/lib/mongoose/index.js b/lib/mongoose/index.js
@@ -26,10 +26,6 @@ module.exports = exports = class mongooseStorage {
       });
   }
 
-  addPermission(_id, permissionIds) {
-    return this.Role.updateOne({ _id }, { $addToSet: { grants: { $each: permissionIds } } });
-  }
-
   newPermission({ name, alias }) {
     return this.Permission.findOne({ name })
       .then(oldPermission => {
@@ -43,6 +39,33 @@ module.exports = exports = class mongooseStorage {
       });
   }
 
+  modifyRoleAlias(_id, alias) {
+    return this.Role.updateOne({ _id }, { $set: { alias } });
+  }
+
+  modifyPermissionAlias(_id, alias) {
+    return this.Permission.updateOne({ _id }, { $set: { alias } });
+  }
+
+  removeRole(_id) {
+    return this.Role.remove({ _id });
+  }
+
+  removePermission(_id) {
+    return Promise.all([
+      this.Permission.remove({ _id }),
+      this.Role.update({}, { $pull: { grants: _id } }),
+    ]);
+  }
+
+  addPermission(_id, permissionIds) {
+    return this.Role.updateOne({ _id }, { $addToSet: { grants: { $each: permissionIds } } });
+  }
+
+  removePermissions(_id, permissionIds) {
+    return this.Role.updateOne({ _id }, { $pull: { grants: { $in: permissionIds } } });
+  }
+
   insertManyPermission(permissions) {
     return this.Permission.insertMany(permissions)
       .then(result => {