Refactored RoleMgtServiceImpl.

components/device-mgt/org.wso2.carbon.device.mgt.api/src/main/java/org/wso2/carbon/device/mgt/jaxrs/service/impl/RoleManagementServiceImpl.java

@@ -143,12 +143,7 @@ public class RoleManagementServiceImpl implements RoleManagementService {
         }
     }
 
-    private UIPermissionNode getAllRolePermissions(String roleName) {
-        try {
-            final UserRealm userRealm = DeviceMgtAPIUtils.getUserRealm();
-            if (!userRealm.getUserStoreManager().isExistingRole(roleName)) {
-                throw new IllegalArgumentException("No role exists with the name '" + roleName + "'");
-            }
+    private UIPermissionNode getAllRolePermissions(String roleName, UserRealm userRealm) throws UserAdminException {
         org.wso2.carbon.user.core.UserRealm userRealmCore = null;
         if (userRealm instanceof org.wso2.carbon.user.core.UserRealm) {
             userRealmCore = (org.wso2.carbon.user.core.UserRealm) userRealm;
@@ -157,12 +152,6 @@ public class RoleManagementServiceImpl implements RoleManagementService {
         final UIPermissionNode rolePermissions =
                 userRealmProxy.getRolePermissions(roleName, MultitenantConstants.SUPER_TENANT_ID);
         return rolePermissions;
-        } catch (UserAdminException e) {
-            log.error("Error occurred while retrieving the permissions of user role : '" + roleName + "'", e);
-        } catch (UserStoreException e) {
-            log.error("Error occurred while retrieving the permissions of user role : '" + roleName + "'", e);
-        }
-        return null;
     }
 
     private UIPermissionNode getUIPermissionNode(String roleName, UserRealm userRealm)
@@ -413,15 +402,15 @@ public class RoleManagementServiceImpl implements RoleManagementService {
 
             if (roleInfo.getPermissions() != null) {
                 // Get all role permissions
-                final UIPermissionNode rolePermissions = this.getAllRolePermissions(roleName);
+                final UIPermissionNode rolePermissions = this.getAllRolePermissions(roleName, userRealm);
                 List<String> permissions = new ArrayList<String>();
                 this.getAuthorizedPermissions(rolePermissions, permissions);
                 for (String permission : roleInfo.getPermissions()) {
                     permissions.add(permission);
                 }
-                String [] allapplicablePerms = new String[permissions.size()];
-                allapplicablePerms = permissions.toArray(allapplicablePerms);
-                roleInfo.setPermissions(allapplicablePerms);
+                String [] allApplicablePerms = new String[permissions.size()];
+                allApplicablePerms = permissions.toArray(allApplicablePerms);
+                roleInfo.setPermissions(allApplicablePerms);
 
                 // Delete all authorizations for the current role before authorizing the permission tree
                 authorizationManager.clearRoleAuthorization(roleName);
@@ -440,6 +429,11 @@ public class RoleManagementServiceImpl implements RoleManagementService {
             log.error(msg, e);
             return Response.serverError().entity(
                     new ErrorResponse.ErrorResponseBuilder().setMessage(msg).build()).build();
+        } catch (UserAdminException e) {
+            String msg = "Error occurred while updating permissions of the role '" + roleName + "'";
+            log.error(msg, e);
+            return Response.serverError().entity(
+                    new ErrorResponse.ErrorResponseBuilder().setMessage(msg).build()).build();
         }
     }