From 1eacef3a2dc53030e291eef9e3782e53e8ef41e3 Mon Sep 17 00:00:00 2001
From: wucongxing8150 <815046773@qq.com>
Date: Wed, 16 Oct 2024 13:47:58 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9D=83=E9=99=90=E7=8A=B6=E6=80=81=E7=A0=81?=
 =?UTF-8?q?=E6=9B=B4=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/middlewares/auth.go | 66 ++++++++++++++++++++++++++++++++++-------
 1 file changed, 55 insertions(+), 11 deletions(-)

diff --git a/api/middlewares/auth.go b/api/middlewares/auth.go
index 8fe0fac..ddcecfe 100644
--- a/api/middlewares/auth.go
+++ b/api/middlewares/auth.go
@@ -33,31 +33,51 @@ func Auth() gin.HandlerFunc {
 		adminUserDao := dao.AdminUserDao{}
 		adminUser, err := adminUserDao.GetAdminUserFirstById(userId)
 		if err != nil || adminUser == nil {
-			responses.FailWithMessage("用户数据错误", c)
+			c.JSON(http.StatusUnauthorized, gin.H{
+				"message": "用户数据错误",
+				"code":    consts.UserStatusError,
+				"data":    "",
+			})
 			c.Abort()
 			return
 		}
 
 		if adminUser.Status == 2 {
-			responses.FailWithMessage("用户审核中", c)
+			c.JSON(http.StatusUnauthorized, gin.H{
+				"message": "用户审核中",
+				"code":    consts.UserStatusError,
+				"data":    "",
+			})
 			c.Abort()
 			return
 		}
 
 		if adminUser.Status == 3 {
-			responses.FailWithMessage("用户已删除或禁用", c)
+			c.JSON(http.StatusUnauthorized, gin.H{
+				"message": "用户已删除或禁用",
+				"code":    consts.UserStatusError,
+				"data":    "",
+			})
 			c.Abort()
 			return
 		}
 
 		if adminUser.RoleID == 0 {
-			responses.FailWithMessage("用户角色错误", c)
+			c.JSON(http.StatusUnauthorized, gin.H{
+				"message": "用户角色错误",
+				"code":    consts.UserStatusError,
+				"data":    "",
+			})
 			c.Abort()
 			return
 		}
 
 		if adminUser.IsDisabled == 1 {
-			responses.FailWithMessage("用户已禁用", c)
+			c.JSON(http.StatusUnauthorized, gin.H{
+				"message": "用户已禁用",
+				"code":    consts.UserStatusError,
+				"data":    "",
+			})
 			c.Abort()
 			return
 		}
@@ -72,14 +92,22 @@ func Auth() gin.HandlerFunc {
 		adminRoleDao := dao.AdminRoleDao{}
 		adminRole, err := adminRoleDao.GetAdminRoleFirstById(roleId)
 		if err != nil || adminRole == nil {
-			responses.FailWithMessage("角色错误", c)
+			c.JSON(http.StatusUnauthorized, gin.H{
+				"message": "角色错误",
+				"code":    consts.UserStatusError,
+				"data":    "",
+			})
 			c.Abort()
 			return
 		}
 
 		// 检测角色是否已被禁用
 		if adminRole.RoleStatus == 2 {
-			responses.FailWithMessage("角色已被禁用", c)
+			c.JSON(http.StatusUnauthorized, gin.H{
+				"message": "角色已被禁用",
+				"code":    consts.UserStatusError,
+				"data":    "",
+			})
 			c.Abort()
 			return
 		}
@@ -95,13 +123,21 @@ func Auth() gin.HandlerFunc {
 			adminDeptDao := dao.AdminDeptDao{}
 			adminDept, err := adminDeptDao.GetAdminDeptById(adminUser.DeptID)
 			if err != nil || adminDept == nil {
-				responses.FailWithMessage("用户部门数据错误", c)
+				c.JSON(http.StatusUnauthorized, gin.H{
+					"message": "用户部门数据错误",
+					"code":    consts.UserStatusError,
+					"data":    "",
+				})
 				c.Abort()
 				return
 			}
 
 			if adminDept.DeptStatus == 2 {
-				responses.FailWithMessage("您所在的部门已被删除，请联系管理员修改", c)
+				c.JSON(http.StatusUnauthorized, gin.H{
+					"message": "您所在的部门已被删除，请联系管理员修改",
+					"code":    consts.UserStatusError,
+					"data":    "",
+				})
 				c.Abort()
 				return
 			}
@@ -112,13 +148,21 @@ func Auth() gin.HandlerFunc {
 			adminPostDao := dao.AdminPostDao{}
 			adminPost, err := adminPostDao.GetAdminPostById(adminUser.PostID)
 			if err != nil || adminPost == nil {
-				responses.FailWithMessage("用户岗位数据错误", c)
+				c.JSON(http.StatusUnauthorized, gin.H{
+					"message": "用户岗位数据错误",
+					"code":    consts.UserStatusError,
+					"data":    "",
+				})
 				c.Abort()
 				return
 			}
 
 			if adminPost.PostStatus == 2 {
-				responses.FailWithMessage("您所在的岗位已被删除，请联系管理员修改", c)
+				c.JSON(http.StatusUnauthorized, gin.H{
+					"message": "您所在的岗位已被删除，请联系管理员修改",
+					"code":    consts.UserStatusError,
+					"data":    "",
+				})
 				c.Abort()
 				return
 			}