소스 검색

fix: 内部员工图表权限

hsun 3 년 전
부모
커밋
76fdcaa7e5
2개의 변경된 파일16개의 추가작업 그리고 6개의 파일을 삭제
  1. 1 1
      logic/user/user.go
  2. 15 5
      middleware/check_chart_auth.go

+ 1 - 1
logic/user/user.go

@@ -201,7 +201,7 @@ func GetUserInfo(userInfo user.UserInfo) (userDetail Detail, err error, errMsg s
 		errMsg = "系统异常"
 		return
 	}
-	if ok == true {
+	if ok {
 		isInner = 1
 	}
 

+ 15 - 5
middleware/check_chart_auth.go

@@ -14,16 +14,26 @@ func CheckChartAuth() gin.HandlerFunc {
 
 		userInfo := user.GetInfoByClaims(c)
 
-		ok, permissionCheckInfo, err := company.CheckUserChartPermission(userInfo.CompanyID, int(userInfo.UserID))
+		// 是否为内部员工
+		ok, _, err := user.GetAdminByUserInfo(userInfo)
 		if err != nil {
-			response.FailMsg("图库权限验证失败", "图库权限验证失败" + err.Error(), c)
+			response.FailMsg("图库权限验证失败", "获取系统用户信息失败" + err.Error(), c)
 			c.Abort()
 			return
 		}
+
 		if !ok {
-			response.AuthError(permissionCheckInfo, "暂无权限", c)
-			c.Abort()
-			return
+			checkOk, permissionCheckInfo, tempErr := company.CheckUserChartPermission(userInfo.CompanyID, int(userInfo.UserID))
+			if tempErr != nil {
+				response.FailMsg("图库权限验证失败", "图库权限验证失败" + tempErr.Error(), c)
+				c.Abort()
+				return
+			}
+			if !checkOk {
+				response.AuthError(permissionCheckInfo, "暂无权限", c)
+				c.Abort()
+				return
+			}
 		}
 
 		c.Next()