package com.jcdm.framework.web.service; import java.util.Set; import org.springframework.stereotype.Service; import org.springframework.util.CollectionUtils; import com.jcdm.common.constant.Constants; import com.jcdm.common.core.domain.entity.SysRole; import com.jcdm.common.core.domain.model.LoginUser; import com.jcdm.common.utils.SecurityUtils; import com.jcdm.common.utils.StringUtils; import com.jcdm.framework.security.context.PermissionContextHolder; /** * 自定义æƒé™å®žçŽ°ï¼Œsså–自SpringSecurity首嗿¯ * * @author jc */ @Service("ss") public class PermissionService { /** * 验è¯ç”¨æˆ·æ˜¯å¦å…·å¤‡æŸæƒé™ * * @param permission æƒé™å—符串 * @return 用户是å¦å…·å¤‡æŸæƒé™ */ public boolean hasPermi(String permission) { if (StringUtils.isEmpty(permission)) { return false; } LoginUser loginUser = SecurityUtils.getLoginUser(); if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getPermissions())) { return false; } PermissionContextHolder.setContext(permission); return hasPermissions(loginUser.getPermissions(), permission); } /** * 验è¯ç”¨æˆ·æ˜¯å¦ä¸å…·å¤‡æŸæƒé™ï¼Œä¸Ž hasPermi逻辑相å * * @param permission æƒé™å—符串 * @return 用户是å¦ä¸å…·å¤‡æŸæƒé™ */ public boolean lacksPermi(String permission) { return hasPermi(permission) != true; } /** * 验è¯ç”¨æˆ·æ˜¯å¦å…·æœ‰ä»¥ä¸‹ä»»æ„一个æƒé™ * * @param permissions 以 PERMISSION_DELIMETER 为分隔符的æƒé™åˆ—表 * @return 用户是å¦å…·æœ‰ä»¥ä¸‹ä»»æ„一个æƒé™ */ public boolean hasAnyPermi(String permissions) { if (StringUtils.isEmpty(permissions)) { return false; } LoginUser loginUser = SecurityUtils.getLoginUser(); if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getPermissions())) { return false; } PermissionContextHolder.setContext(permissions); Set<String> authorities = loginUser.getPermissions(); for (String permission : permissions.split(Constants.PERMISSION_DELIMETER)) { if (permission != null && hasPermissions(authorities, permission)) { return true; } } return false; } /** * 判æ–ç”¨æˆ·æ˜¯å¦æ‹¥æœ‰æŸä¸ªè§’色 * * @param role 角色å—符串 * @return 用户是å¦å…·å¤‡æŸè§’色 */ public boolean hasRole(String role) { if (StringUtils.isEmpty(role)) { return false; } LoginUser loginUser = SecurityUtils.getLoginUser(); if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getUser().getRoles())) { return false; } for (SysRole sysRole : loginUser.getUser().getRoles()) { String roleKey = sysRole.getRoleKey(); if (Constants.SUPER_ADMIN.equals(roleKey) || roleKey.equals(StringUtils.trim(role))) { return true; } } return false; } /** * 验è¯ç”¨æˆ·æ˜¯å¦ä¸å…·å¤‡æŸè§’色,与 isRole逻辑相å。 * * @param role 角色åç§° * @return 用户是å¦ä¸å…·å¤‡æŸè§’色 */ public boolean lacksRole(String role) { return hasRole(role) != true; } /** * 验è¯ç”¨æˆ·æ˜¯å¦å…·æœ‰ä»¥ä¸‹ä»»æ„一个角色 * * @param roles 以 ROLE_NAMES_DELIMETER 为分隔符的角色列表 * @return 用户是å¦å…·æœ‰ä»¥ä¸‹ä»»æ„一个角色 */ public boolean hasAnyRoles(String roles) { if (StringUtils.isEmpty(roles)) { return false; } LoginUser loginUser = SecurityUtils.getLoginUser(); if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getUser().getRoles())) { return false; } for (String role : roles.split(Constants.ROLE_DELIMETER)) { if (hasRole(role)) { return true; } } return false; } /** * åˆ¤æ–æ˜¯å¦åŒ…嫿ƒé™ * * @param permissions æƒé™åˆ—表 * @param permission æƒé™å—符串 * @return 用户是å¦å…·å¤‡æŸæƒé™ */ private boolean hasPermissions(Set<String> permissions, String permission) { return permissions.contains(Constants.ALL_PERMISSION) || permissions.contains(StringUtils.trim(permission)); } }