| | |
| | | import cn.dev33.satoken.session.SaSession; |
| | | import cn.dev33.satoken.stp.SaLoginModel; |
| | | import cn.dev33.satoken.stp.StpUtil; |
| | | import cn.hutool.core.collection.CollUtil; |
| | | import cn.hutool.core.convert.Convert; |
| | | import cn.hutool.core.util.ObjectUtil; |
| | | import lombok.AccessLevel; |
| | | import lombok.NoArgsConstructor; |
| | | import org.dromara.common.core.constant.SystemConstants; |
| | | import org.dromara.common.core.constant.TenantConstants; |
| | | import org.dromara.common.core.constant.UserConstants; |
| | | import org.dromara.common.core.domain.model.LoginUser; |
| | | import org.dromara.common.core.enums.UserType; |
| | | |
| | | import java.util.Set; |
| | | |
| | | |
| | | /** |
| | | * 登录鉴权助手 |
| | |
| | | public static final String USER_NAME_KEY = "userName"; |
| | | public static final String DEPT_KEY = "deptId"; |
| | | public static final String DEPT_NAME_KEY = "deptName"; |
| | | public static final String DEPT_CATEGORY_KEY = "deptCategory"; |
| | | public static final String CLIENT_KEY = "clientid"; |
| | | |
| | | /** |
| | |
| | | StpUtil.login(loginUser.getLoginId(), |
| | | model.setExtra(TENANT_KEY, loginUser.getTenantId()) |
| | | .setExtra(USER_KEY, loginUser.getUserId()) |
| | | .setExtra(DEPT_KEY, loginUser.getDeptId()) |
| | | .setExtra(TENANT_KEY, loginUser.getTenantId()) |
| | | .setExtra(DEPT_NAME_KEY, loginUser.getDeptName()) |
| | | .setExtra(USER_NAME_KEY, loginUser.getUsername()) |
| | | .setExtra(DEPT_KEY, loginUser.getDeptId()) |
| | | .setExtra(DEPT_NAME_KEY, loginUser.getDeptName()) |
| | | .setExtra(DEPT_CATEGORY_KEY, loginUser.getDeptCategory()) |
| | | ); |
| | | SaSession tokenSession = StpUtil.getTokenSession(); |
| | | tokenSession.updateTimeout(model.getTimeout()); |
| | | tokenSession.set(LOGIN_USER_KEY, loginUser); |
| | | StpUtil.getTokenSession().set(LOGIN_USER_KEY, loginUser); |
| | | } |
| | | |
| | | /** |
| | |
| | | } |
| | | |
| | | /** |
| | | * 获取用户id |
| | | */ |
| | | public static String getUserIdStr() { |
| | | return Convert.toStr(getExtra(USER_KEY)); |
| | | } |
| | | |
| | | /** |
| | | * 获取用户账户 |
| | | */ |
| | | public static String getUsername() { |
| | | return Convert.toStr(getExtra(USER_NAME_KEY)); |
| | | } |
| | | |
| | | /** |
| | | * 获取租户ID |
| | | */ |
| | | public static String getTenantId() { |
| | |
| | | return Convert.toLong(getExtra(DEPT_KEY)); |
| | | } |
| | | |
| | | /** |
| | | * 获取部门名 |
| | | */ |
| | | public static String getDeptName() { |
| | | return Convert.toStr(getExtra(DEPT_NAME_KEY)); |
| | | } |
| | | |
| | | /** |
| | | * 获取部门类别编码 |
| | | */ |
| | | public static String getDeptCategory() { |
| | | return Convert.toStr(getExtra(DEPT_CATEGORY_KEY)); |
| | | } |
| | | |
| | | /** |
| | | * 获取当前 Token 的扩展信息 |
| | | * |
| | | * @param key 键值 |
| | | * @return 对应的扩展数据 |
| | | */ |
| | | private static Object getExtra(String key) { |
| | | try { |
| | | return StpUtil.getExtra(key); |
| | | } catch (Exception e) { |
| | | return null; |
| | | } |
| | | |
| | | } |
| | | |
| | | /** |
| | | * 获取用户账户 |
| | | */ |
| | | public static String getUsername() { |
| | | return getLoginUser().getUsername(); |
| | | } |
| | | |
| | | /** |
| | |
| | | * @return 结果 |
| | | */ |
| | | public static boolean isSuperAdmin(Long userId) { |
| | | return UserConstants.SUPER_ADMIN_ID.equals(userId); |
| | | } |
| | | |
| | | public static boolean isSuperAdmin() { |
| | | return isSuperAdmin(getUserId()); |
| | | return SystemConstants.SUPER_ADMIN_ID.equals(userId); |
| | | } |
| | | |
| | | /** |
| | | * 是否为超级管理员 |
| | | * |
| | | * @return 结果 |
| | | */ |
| | | public static boolean isSuperAdmin() { |
| | | return isSuperAdmin(getUserId()); |
| | | } |
| | | |
| | | /** |
| | | * 是否为租户管理员 |
| | | * |
| | | * @param rolePermission 角色权限标识组 |
| | | * @return 结果 |
| | | */ |
| | | public static boolean isTenantAdmin(Set<String> rolePermission) { |
| | | if (CollUtil.isEmpty(rolePermission)) { |
| | | return false; |
| | | } |
| | | return rolePermission.contains(TenantConstants.TENANT_ADMIN_ROLE_KEY); |
| | | } |
| | | |
| | | /** |
| | | * 是否为租户管理员 |
| | | * |
| | | * @return 结果 |
| | | */ |
| | | public static boolean isTenantAdmin() { |
| | | return Convert.toBool(isTenantAdmin(getLoginUser().getRolePermission())); |
| | | } |
| | | |
| | | /** |
| | | * 检查当前用户是否已登录 |
| | | * |
| | | * @return 结果 |
| | | */ |
| | | public static boolean isLogin() { |
| | | return getLoginUser() != null; |
| | | try { |
| | | return getLoginUser() != null; |
| | | } catch (Exception e) { |
| | | return false; |
| | | } |
| | | } |
| | | |
| | | } |