From c93b6661406b61ba9bca170273e74ae769bc31a6 Mon Sep 17 00:00:00 2001
From: 盘古给你一斧 <56600535@qq.com>
Date: 星期四, 09 十一月 2023 17:37:56 +0800
Subject: [PATCH] !444 @Sensitive脱敏增加角色和权限校验 * update 优化 脱敏增加角色及权限校验
---
ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysSensitiveServiceImpl.java | 25 +++++++++++++++++++++++--
1 files changed, 23 insertions(+), 2 deletions(-)
diff --git a/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysSensitiveServiceImpl.java b/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysSensitiveServiceImpl.java
index 296574b..2f2493a 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysSensitiveServiceImpl.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysSensitiveServiceImpl.java
@@ -1,7 +1,10 @@
package org.dromara.system.service.impl;
+import cn.dev33.satoken.stp.StpUtil;
+import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.satoken.utils.LoginHelper;
import org.dromara.common.sensitive.core.SensitiveService;
+import org.dromara.common.tenant.helper.TenantHelper;
import org.springframework.stereotype.Service;
/**
@@ -19,8 +22,26 @@
* 鏄惁鑴辨晱
*/
@Override
- public boolean isSensitive() {
- return !LoginHelper.isSuperAdmin() || !LoginHelper.isTenantAdmin();
+ public boolean isSensitive(String roleKey, String perms) {
+ if (!StpUtil.isLogin()) {
+ return true;
+ }
+ boolean roleExist = StringUtils.isNotEmpty(roleKey);
+ boolean permsExist = StringUtils.isNotEmpty(perms);
+ if (roleExist && permsExist) {
+ if (StpUtil.hasRole(roleKey) && StpUtil.hasPermission(perms)) {
+ return false;
+ }
+ } else if (roleExist && StpUtil.hasRole(roleKey)) {
+ return false;
+ } else if (permsExist && StpUtil.hasPermission(perms)) {
+ return false;
+ }
+
+ if (TenantHelper.isEnable()) {
+ return !LoginHelper.isSuperAdmin() && !LoginHelper.isTenantAdmin();
+ }
+ return !LoginHelper.isSuperAdmin();
}
}
--
Gitblit v1.9.3