更新

go/controllers/backend_oa_schedule.go

@@ -12,6 +12,7 @@ import (
 
 	"server/models"
 	"server/pkg/holiday"
+	"server/pkg/identity"
 	"server/pkg/jwtutil"
 
 	"github.com/beego/beego/v2/client/orm"
@@ -25,22 +26,7 @@ type BackendOaScheduleController struct {
 }
 
 func (c *BackendOaScheduleController) oaScheduleClaims() (*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 *BackendOaScheduleController) oasJsonErr(httpStatus, bizCode int, msg string) {
@@ -147,7 +133,8 @@ func oaScheduleQS(tid int, userID int, f oaScheduleFilter) orm.QuerySeter {
 func (c *BackendOaScheduleController) List() {
 	claims, err := c.oaScheduleClaims()
 	if err != nil {
-		c.oasJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.oasJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -268,7 +255,8 @@ func oaScheduleListData(tid, userID, page, pageSize int,
 // 响应里的 builtin_years 字段告知前端内置数据覆盖到哪一年。
 func (c *BackendOaScheduleController) Holidays() {
 	if _, err := c.oaScheduleClaims(); err != nil {
-		c.oasJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.oasJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -416,7 +404,8 @@ func mergeOaHolidays(list []holiday.Day) []holiday.Day {
 func (c *BackendOaScheduleController) Stats() {
 	claims, err := c.oaScheduleClaims()
 	if err != nil {
-		c.oasJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.oasJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -687,7 +676,8 @@ func parseOaSchedulePayloadBody(raw []byte) (oaSchedulePayload, string) {
 func (c *BackendOaScheduleController) Create() {
 	claims, err := c.oaScheduleClaims()
 	if err != nil {
-		c.oasJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.oasJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -758,7 +748,8 @@ func oaScheduleCreateCore(claims *jwtutil.Claims, payload oaSchedulePayload) (*m
 func (c *BackendOaScheduleController) Update() {
 	claims, err := c.oaScheduleClaims()
 	if err != nil {
-		c.oasJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.oasJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -845,7 +836,8 @@ func oaScheduleUpdateCore(claims *jwtutil.Claims, id uint64, payload oaScheduleP
 func (c *BackendOaScheduleController) SetImages() {
 	claims, err := c.oaScheduleClaims()
 	if err != nil {
-		c.oasJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.oasJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -907,7 +899,8 @@ func oaScheduleSetImagesCore(claims *jwtutil.Claims, id uint64, images []oaSched
 func (c *BackendOaScheduleController) UpdateContent() {
 	claims, err := c.oaScheduleClaims()
 	if err != nil {
-		c.oasJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.oasJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -953,7 +946,8 @@ func (c *BackendOaScheduleController) UpdateContent() {
 func (c *BackendOaScheduleController) Delete() {
 	claims, err := c.oaScheduleClaims()
 	if err != nil {
-		c.oasJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.oasJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -1008,7 +1002,8 @@ func oaScheduleDeleteCore(claims *jwtutil.Claims, id uint64) (int, string) {
 func (c *BackendOaScheduleController) FinishSchedule() {
 	claims, err := c.oaScheduleClaims()
 	if err != nil {
-		c.oasJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.oasJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -1148,7 +1143,8 @@ func carryOaSchedule(item *models.OaSchedule, targetDate string, claims *jwtutil
 func (c *BackendOaScheduleController) CarrySchedule() {
 	claims, err := c.oaScheduleClaims()
 	if err != nil {
-		c.oasJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.oasJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -1204,7 +1200,8 @@ func (c *BackendOaScheduleController) CarrySchedule() {
 func (c *BackendOaScheduleController) CarryPending() {
 	claims, err := c.oaScheduleClaims()
 	if err != nil {
-		c.oasJsonErr(401, 401, "未登录或无权限")
+		code := identity.Status(err)
+		c.oasJsonErr(code, code, err.Error())
 		return
 	}