package cn.teammodel.security.service; import java.util.Set; import cn.teammodel.model.entity.User; import cn.teammodel.security.utils.SecurityUtils; import org.apache.commons.lang3.StringUtils; import org.springframework.security.core.Authentication; import org.springframework.stereotype.Service; import org.springframework.util.CollectionUtils; /** * 自定义权限实现,ss取自SpringSecurity首字母
*

* 1. IES 类似角色判断: hasAuth
* 2. role 身份判断(authToken 解析出来的用户身份): hasRole
* 3. permission 权限判断(authToken 解析出来的用户权限): hasPermi
*

* @author ruoyi */ @Service("ss") public class PermissionService { /** 所有权限标识 */ private static final String ALL_PERMISSION = "*:*:*"; /** 管理员角色权限标识 */ private static final String SUPER_ADMIN = "admin"; private static final String ROLE_DELIMETER = ","; private static final String PERMISSION_DELIMETER = ","; /** * access_token 是否拥有 auth 角色 * @param auth: * @return: boolean * @author: winter * @date: 2023/11/10 10:05 * @description: */ public boolean hasAuth(String auth) { if (StringUtils.isEmpty(auth)) { return false; } Authentication authentication = SecurityUtils.getAuthentication(); if (authentication == null || CollectionUtils.isEmpty(authentication.getAuthorities())) { return false; } return authentication.getAuthorities().contains(auth); } /** * 验证用户是否具备某权限 * * @param permission 权限字符串 * @return 用户是否具备某权限 */ public boolean hasPermi(String permission) { if (StringUtils.isEmpty(permission)) { return false; } User loginUser = SecurityUtils.getLoginUser(); if (loginUser == null || CollectionUtils.isEmpty(loginUser.getPermissions())) { return false; } return loginUser.getPermissions().contains(permission); } /** * 验证用户是否不具备某权限,与 hasPermi逻辑相反 * * @param permission 权限字符串 * @return 用户是否不具备某权限 */ public boolean lacksPermi(String permission) { return !hasPermi(permission); } /** * 验证用户是否具有以下任意一个权限 * * @param permissions 以 PERMISSION_DELIMETER 为分隔符的权限列表 * @return 用户是否具有以下任意一个权限 */ public boolean hasAnyPermi(String permissions) { if (StringUtils.isEmpty(permissions)) { return false; } User loginUser = SecurityUtils.getLoginUser(); if (loginUser == null || CollectionUtils.isEmpty(loginUser.getPermissions())) { return false; } Set authorities = loginUser.getPermissions(); for (String permission : permissions.split(PERMISSION_DELIMETER)) { if (permission != null && authorities.contains(permission)) { return true; } } return false; } /** * 判断用户是否拥有某个角色 * * @param role 角色字符串 * @return 用户是否具备某角色 */ public boolean hasRole(String role) { if (StringUtils.isEmpty(role)) { return false; } User loginUser = SecurityUtils.getLoginUser(); if (loginUser == null || CollectionUtils.isEmpty(loginUser.getRoles())) { return false; } return loginUser.getRoles().contains(role); } /** * 验证用户是否不具备某角色,与 isRole逻辑相反。 * * @param role 角色名称 * @return 用户是否不具备某角色 */ public boolean lacksRole(String role) { return !hasRole(role); } /** * 验证用户是否具有以下任意一个角色 * * @param roles 以 ROLE_NAMES_DELIMETER 为分隔符的角色列表 * @return 用户是否具有以下任意一个角色 */ public boolean hasAnyRoles(String roles) { if (StringUtils.isEmpty(roles)) { return false; } User loginUser = SecurityUtils.getLoginUser(); if (loginUser == null || CollectionUtils.isEmpty(loginUser.getRoles())) { return false; } for (String role : roles.split(ROLE_DELIMETER)) { if (hasRole(role)) { return true; } } return false; } }