更新

go/controllers/backend_oa_notice.go

@@ -9,6 +9,7 @@ import (
 	"unicode/utf8"
 
 	"server/models"
+	"server/pkg/identity"
 	"server/pkg/jwtutil"
 
 	"github.com/beego/beego/v2/client/orm"
@@ -22,22 +23,7 @@ type BackendOaNoticeController struct {
 }
 
 func (c *BackendOaNoticeController) oaNoticeClaims() (*jwtutil.Claims, error) {
-	auth := c.Ctx.Request.Header.Get("Authorization")
-	if auth == "" {
-		return nil, orm.ErrNoRows
-	}
-	parts := strings.SplitN(auth, " ", 2)
-	if len(parts) != 2 || parts[0] != "Bearer" {
-		return nil, orm.ErrNoRows
-	}
-	claims, err := jwtutil.ParseToken(parts[1])
-	if err != nil {
-		return nil, err
-	}
-	if claims.UserType != "backend" {
-		return nil, orm.ErrNoRows
-	}
-	return claims, nil
+	return identity.Claims(c.Ctx.Request.Header.Get("Authorization"), identity.EndBackend)
 }
 
 func (c *BackendOaNoticeController) ontJsonErr(httpStatus, bizCode int, msg string) {
@@ -104,7 +90,8 @@ func noticeBase(tid int) orm.QuerySeter {
 func (c *BackendOaNoticeController) List() {
 	claims, err := c.oaNoticeClaims()
 	if err != nil {
-		c.ontJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.ontJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -156,7 +143,8 @@ func (c *BackendOaNoticeController) List() {
 func (c *BackendOaNoticeController) Portal() {
 	claims, err := c.oaNoticeClaims()
 	if err != nil {
-		c.ontJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.ontJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -187,7 +175,8 @@ func (c *BackendOaNoticeController) Portal() {
 func (c *BackendOaNoticeController) Detail() {
 	claims, err := c.oaNoticeClaims()
 	if err != nil {
-		c.ontJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.ontJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -216,7 +205,8 @@ func (c *BackendOaNoticeController) Detail() {
 func (c *BackendOaNoticeController) Create() {
 	claims, err := c.oaNoticeClaims()
 	if err != nil {
-		c.ontJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.ontJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -258,7 +248,8 @@ func (c *BackendOaNoticeController) Create() {
 func (c *BackendOaNoticeController) Update() {
 	claims, err := c.oaNoticeClaims()
 	if err != nil {
-		c.ontJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.ontJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -308,7 +299,8 @@ func (c *BackendOaNoticeController) Update() {
 func (c *BackendOaNoticeController) Delete() {
 	claims, err := c.oaNoticeClaims()
 	if err != nil {
-		c.ontJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.ontJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -338,7 +330,8 @@ func (c *BackendOaNoticeController) Delete() {
 func (c *BackendOaNoticeController) Publish() {
 	claims, err := c.oaNoticeClaims()
 	if err != nil {
-		c.ontJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.ontJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -377,7 +370,8 @@ func (c *BackendOaNoticeController) Publish() {
 func (c *BackendOaNoticeController) Top() {
 	claims, err := c.oaNoticeClaims()
 	if err != nil {
-		c.ontJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.ontJsonErr(code, code, err.Error())
 		return
 	}