use of org.wso2.carbon.user.mgt.ui.UserAdminClient in project carbon-identity-framework by wso2.
the class UserAndRoleManagementServlet method doPost.
@Override
protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
String category = request.getParameter(CATEGORY);
HttpSession session = request.getSession();
if (USERS.equals(category)) {
boolean error = false;
boolean newFilter = false;
boolean doUserList = true;
boolean showFilterMessage = false;
boolean multipleUserStores = false;
String forwardTo = "user-mgt.jsp";
FlaggedName[] datas = null;
FlaggedName exceededDomains = null;
String[] claimUris = null;
FlaggedName[] users = null;
String[] domainNames = null;
int pageNumber = 0;
int cachePages = 3;
int noOfPageLinksToDisplay = 5;
int numberOfPages = 0;
Map<Integer, PaginatedNamesBean> flaggedNameMap = null;
String BUNDLE = "org.wso2.carbon.userstore.ui.i18n.Resources";
ResourceBundle resourceBundle = ResourceBundle.getBundle(BUNDLE, request.getLocale());
// remove session data
session.removeAttribute("userBean");
session.removeAttribute(UserAdminUIConstants.USER_DISPLAY_NAME);
session.removeAttribute(UserAdminUIConstants.USER_LIST_UNASSIGNED_ROLE_CACHE);
session.removeAttribute(UserAdminUIConstants.USER_LIST_UNASSIGNED_ROLE_CACHE_EXCEEDED);
session.removeAttribute(UserAdminUIConstants.USER_LIST_ASSIGNED_ROLE_CACHE);
session.removeAttribute(UserAdminUIConstants.USER_LIST_ASSIGNED_ROLE_CACHE_EXCEEDED);
session.removeAttribute(UserAdminUIConstants.USER_LIST_ADD_USER_ROLE_CACHE);
session.removeAttribute(UserAdminUIConstants.USER_LIST_ADD_USER_ROLE_CACHE_EXCEEDED);
session.removeAttribute(UserAdminUIConstants.USER_LIST_ASSIGN_ROLE_FILTER);
session.removeAttribute(UserAdminUIConstants.USER_LIST_UNASSIGNED_ROLE_FILTER);
session.removeAttribute(UserAdminUIConstants.USER_LIST_VIEW_ROLE_FILTER);
session.removeAttribute(UserAdminUIConstants.USER_LIST_CACHE);
// retrieve session attributes
UserRealmInfo userRealmInfo = (UserRealmInfo) session.getAttribute(UserAdminUIConstants.USER_STORE_INFO);
if (userRealmInfo != null) {
multipleUserStores = userRealmInfo.getMultipleUserStore();
}
java.lang.String errorAttribute = (java.lang.String) session.getAttribute(UserAdminUIConstants.DO_USER_LIST);
String claimUri = request.getParameter("claimUri");
if (claimUri == null || claimUri.length() == 0) {
claimUri = (java.lang.String) session.getAttribute(UserAdminUIConstants.USER_CLAIM_FILTER);
}
session.setAttribute(UserAdminUIConstants.USER_CLAIM_FILTER, claimUri);
exceededDomains = (FlaggedName) session.getAttribute(UserAdminUIConstants.USER_LIST_CACHE_EXCEEDED);
// search filter
String selectedDomain = request.getParameter(DOMAIN);
if (selectedDomain == null || selectedDomain.trim().length() == 0) {
selectedDomain = (String) session.getAttribute(UserAdminUIConstants.USER_LIST_DOMAIN_FILTER);
if (selectedDomain == null || selectedDomain.trim().length() == 0) {
selectedDomain = UserAdminUIConstants.ALL_DOMAINS;
}
} else {
newFilter = true;
}
session.setAttribute(UserAdminUIConstants.USER_LIST_DOMAIN_FILTER, selectedDomain.trim());
String filter = request.getParameter(UserAdminUIConstants.USER_LIST_FILTER);
if (filter == null || filter.trim().length() == 0) {
/*filter = (java.lang.String) session.getAttribute(UserAdminUIConstants.USER_LIST_FILTER);
if (filter == null || filter.trim().length() == 0) {
filter = "*";
}*/
filter = "*";
} else {
if (filter.contains(UserAdminUIConstants.DOMAIN_SEPARATOR)) {
selectedDomain = UserAdminUIConstants.ALL_DOMAINS;
session.removeAttribute(UserAdminUIConstants.USER_LIST_DOMAIN_FILTER);
}
newFilter = true;
}
String userDomainSelector;
String modifiedFilter = filter.trim();
if (!UserAdminUIConstants.ALL_DOMAINS.equalsIgnoreCase(selectedDomain)) {
modifiedFilter = selectedDomain + UserAdminUIConstants.DOMAIN_SEPARATOR + filter;
modifiedFilter = modifiedFilter.trim();
userDomainSelector = selectedDomain + UserAdminUIConstants.DOMAIN_SEPARATOR + "*";
} else {
userDomainSelector = "*";
}
session.setAttribute(UserAdminUIConstants.USER_LIST_FILTER, filter.trim());
// check page number
String pageNumberStr = request.getParameter("pageNumber");
if (pageNumberStr == null) {
pageNumberStr = "0";
}
if (userRealmInfo != null) {
claimUris = userRealmInfo.getRequiredUserClaims();
}
try {
pageNumber = Integer.parseInt(pageNumberStr);
} catch (NumberFormatException ignored) {
// page number format exception
}
flaggedNameMap = (Map<Integer, PaginatedNamesBean>) session.getAttribute(UserAdminUIConstants.USER_LIST_CACHE);
if (flaggedNameMap != null) {
PaginatedNamesBean bean = flaggedNameMap.get(pageNumber);
if (bean != null) {
users = bean.getNames();
if (users != null && users.length > 0) {
numberOfPages = bean.getNumberOfPages();
doUserList = false;
}
}
}
if (errorAttribute != null) {
error = true;
session.removeAttribute(UserAdminUIConstants.DO_USER_LIST);
}
if ((doUserList || newFilter) && !error) {
// don't call the back end if some kind of message is showing
try {
java.lang.String cookie = (java.lang.String) session.getAttribute(ServerConstants.ADMIN_SERVICE_COOKIE);
java.lang.String backendServerURL = CarbonUIUtil.getServerURL(getServletConfig().getServletContext(), session);
ConfigurationContext configContext = (ConfigurationContext) getServletConfig().getServletContext().getAttribute(CarbonConstants.CONFIGURATION_CONTEXT);
UserAdminClient client = new UserAdminClient(cookie, backendServerURL, configContext);
if (userRealmInfo == null) {
userRealmInfo = client.getUserRealmInfo();
session.setAttribute(UserAdminUIConstants.USER_STORE_INFO, userRealmInfo);
}
if (userRealmInfo != null) {
claimUris = userRealmInfo.getRequiredUserClaims();
}
if (filter.length() > 0) {
if (claimUri != null && !"select".equalsIgnoreCase(claimUri)) {
ClaimValue claimValue = new ClaimValue();
claimValue.setClaimURI(claimUri);
claimValue.setValue(modifiedFilter);
datas = client.listUserByClaimWithPermission(claimValue, userDomainSelector, PERMISSION_VIEWTASKS, -1);
} else {
datas = client.listAllUsersWithPermission(modifiedFilter, PERMISSION_VIEWTASKS, -1);
}
List<FlaggedName> dataList = new ArrayList<>(Arrays.asList(datas));
exceededDomains = dataList.remove(dataList.size() - 1);
session.setAttribute(UserAdminUIConstants.USER_LIST_CACHE_EXCEEDED, exceededDomains);
if (dataList.size() == 0) {
session.removeAttribute(UserAdminUIConstants.USER_LIST_FILTER);
showFilterMessage = true;
}
flaggedNameMap = new HashMap<>();
int max = pageNumber + cachePages;
for (int i = (pageNumber - cachePages); i < max; i++) {
if (i < 0) {
max++;
continue;
}
PaginatedNamesBean bean = Util.retrievePaginatedFlaggedName(i, dataList);
flaggedNameMap.put(i, bean);
if (bean.getNumberOfPages() == i + 1) {
break;
}
}
users = flaggedNameMap.get(pageNumber).getNames();
numberOfPages = flaggedNameMap.get(pageNumber).getNumberOfPages();
session.setAttribute(UserAdminUIConstants.USER_LIST_CACHE, flaggedNameMap);
}
} catch (Exception e) {
String message = MessageFormat.format(resourceBundle.getString("error.while.user.filtered"), e.getMessage());
}
String resp = "";
if (users != null) {
UserBean[] userList = new UserBean[users.length];
for (int i = 0; i < users.length; i++) {
FlaggedName flaggedName = users[i];
UserBean user = new UserBean();
user.setUsername(flaggedName.getItemName());
userList[i] = user;
}
UserSearchResult userSearchResult = new UserSearchResult();
userSearchResult.setPageNumber(pageNumber);
userSearchResult.setNumberOfPages(numberOfPages);
userSearchResult.setNoOfPageLinksToDisplay(noOfPageLinksToDisplay);
userSearchResult.setUserBeans(userList);
Gson gson = new Gson();
resp = gson.toJson(userSearchResult);
}
response.setContentType("application/json");
response.getWriter().write(resp);
}
} else if (ROLES.equals(category)) {
boolean error = false;
boolean newFilter = false;
boolean doRoleList = true;
boolean showFilterMessage = false;
boolean multipleUserStores = false;
List<FlaggedName> datasList = null;
FlaggedName[] roles = null;
FlaggedName exceededDomains = null;
String[] domainNames = null;
int pageNumber = 0;
int cachePages = 3;
int noOfPageLinksToDisplay = 5;
int numberOfPages = 0;
Map<Integer, PaginatedNamesBean> flaggedNameMap = null;
UserRealmInfo userRealmInfo = null;
// clear session data
session.removeAttribute("roleBean");
session.removeAttribute(UserAdminUIConstants.ROLE_READ_ONLY);
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_UNASSIGNED_USER_CACHE);
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_UNASSIGNED_USER_CACHE_EXCEEDED);
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_ASSIGNED_USER_CACHE);
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_ASSIGNED_USER_CACHE_EXCEEDED);
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_ADD_ROLE_USER_CACHE);
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_ADD_ROLE_USER_CACHE_EXCEEDED);
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_ASSIGN_USER_FILTER);
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_UNASSIGNED_USER_FILTER);
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_VIEW_USER_FILTER);
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_CACHE);
session.removeAttribute(PREVIOUS_ROLE);
// search filter
String selectedDomain = request.getParameter(DOMAIN);
if (StringUtils.isBlank(selectedDomain)) {
selectedDomain = (String) session.getAttribute(UserAdminUIConstants.ROLE_LIST_DOMAIN_FILTER);
if (selectedDomain == null || selectedDomain.trim().length() == 0) {
selectedDomain = UserAdminUIConstants.ALL_DOMAINS;
}
} else {
newFilter = true;
}
session.setAttribute(UserAdminUIConstants.ROLE_LIST_DOMAIN_FILTER, selectedDomain.trim());
String filter = request.getParameter(UserAdminUIConstants.ROLE_LIST_FILTER);
if (filter == null || filter.trim().length() == 0) {
filter = (String) session.getAttribute(UserAdminUIConstants.ROLE_LIST_FILTER);
if (filter == null || filter.trim().length() == 0) {
filter = "*";
}
} else {
if (filter.contains(UserAdminUIConstants.DOMAIN_SEPARATOR)) {
selectedDomain = UserAdminUIConstants.ALL_DOMAINS;
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_DOMAIN_FILTER);
}
newFilter = true;
}
String modifiedFilter = filter.trim();
if (!UserAdminUIConstants.ALL_DOMAINS.equalsIgnoreCase(selectedDomain)) {
modifiedFilter = selectedDomain + UserAdminUIConstants.DOMAIN_SEPARATOR + filter;
modifiedFilter = modifiedFilter.trim();
}
session.setAttribute(UserAdminUIConstants.ROLE_LIST_FILTER, filter.trim());
userRealmInfo = (UserRealmInfo) session.getAttribute(UserAdminUIConstants.USER_STORE_INFO);
if (userRealmInfo != null) {
multipleUserStores = userRealmInfo.getMultipleUserStore();
}
String errorAttribute = (String) session.getAttribute(UserAdminUIConstants.DO_ROLE_LIST);
exceededDomains = (FlaggedName) session.getAttribute(UserAdminUIConstants.ROLE_LIST_CACHE_EXCEEDED);
// check page number
String pageNumberStr = request.getParameter("pageNumber");
if (pageNumberStr == null) {
pageNumberStr = "0";
}
try {
pageNumber = Integer.parseInt(pageNumberStr);
} catch (NumberFormatException ignored) {
// page number format exception
}
flaggedNameMap = (Map<Integer, PaginatedNamesBean>) session.getAttribute(UserAdminUIConstants.ROLE_LIST_CACHE);
if (flaggedNameMap != null) {
PaginatedNamesBean bean = flaggedNameMap.get(pageNumber);
if (bean != null) {
roles = bean.getNames();
if (roles != null && roles.length > 0) {
numberOfPages = bean.getNumberOfPages();
doRoleList = false;
}
}
}
if (errorAttribute != null) {
error = true;
session.removeAttribute(UserAdminUIConstants.DO_ROLE_LIST);
}
if ((doRoleList || newFilter) && !error) {
try {
String cookie = (String) session.getAttribute(ServerConstants.ADMIN_SERVICE_COOKIE);
String backendServerURL = CarbonUIUtil.getServerURL(getServletConfig().getServletContext(), session);
ConfigurationContext configContext = (ConfigurationContext) getServletConfig().getServletContext().getAttribute(CarbonConstants.CONFIGURATION_CONTEXT);
UserAdminClient client = new UserAdminClient(cookie, backendServerURL, configContext);
boolean sharedRoleEnabled = client.isSharedRolesEnabled();
session.setAttribute(UserAdminUIConstants.SHARED_ROLE_ENABLED, sharedRoleEnabled);
if (filter.length() > 0) {
FlaggedName[] datas = client.getAllPermittedRoleNames(modifiedFilter, PERMISSION_VIEWTASKS, -1);
datasList = new ArrayList<FlaggedName>(Arrays.asList(datas));
exceededDomains = datasList.remove(datasList.size() - 1);
session.setAttribute(UserAdminUIConstants.ROLE_LIST_CACHE_EXCEEDED, exceededDomains);
datas = datasList.toArray(new FlaggedName[datasList.size()]);
if (datas == null || datas.length == 0) {
session.removeAttribute(UserAdminUIConstants.ROLE_LIST_FILTER);
showFilterMessage = true;
}
}
if (userRealmInfo == null) {
userRealmInfo = client.getUserRealmInfo();
session.setAttribute(UserAdminUIConstants.USER_STORE_INFO, userRealmInfo);
}
if (datasList != null) {
flaggedNameMap = new HashMap<Integer, PaginatedNamesBean>();
int max = pageNumber + cachePages;
for (int i = (pageNumber - cachePages); i < max; i++) {
if (i < 0) {
max++;
continue;
}
PaginatedNamesBean bean = Util.retrievePaginatedFlaggedName(i, datasList);
flaggedNameMap.put(i, bean);
if (bean.getNumberOfPages() == i + 1) {
break;
}
}
roles = flaggedNameMap.get(pageNumber).getNames();
numberOfPages = flaggedNameMap.get(pageNumber).getNumberOfPages();
session.setAttribute(UserAdminUIConstants.ROLE_LIST_CACHE, flaggedNameMap);
}
} catch (Exception e) {
}
String resp = "";
if (roles != null) {
RoleBean[] roleList = new RoleBean[roles.length];
for (int i = 0; i < roles.length; i++) {
FlaggedName flaggedName = roles[i];
RoleBean role = new RoleBean();
role.setRoleName(flaggedName.getItemName());
roleList[i] = role;
}
RoleSearchResult roleSearchResult = new RoleSearchResult();
roleSearchResult.setPageNumber(pageNumber);
roleSearchResult.setNumberOfPages(numberOfPages);
roleSearchResult.setNoOfPageLinksToDisplay(noOfPageLinksToDisplay);
roleSearchResult.setRoleBeans(roleList);
Gson gson = new Gson();
resp = gson.toJson(roleSearchResult);
}
response.setContentType("application/json");
response.getWriter().write(resp);
}
}
}
Aggregations