更新

go/controllers/platform_reminder.go

@@ -11,6 +11,7 @@ import (
 	"time"
 
 	"server/models"
+	"server/pkg/identity"
 	"server/pkg/jwtutil"
 	"server/services"
 
@@ -22,22 +23,7 @@ type PlatformReminderController struct {
 }
 
 func (c *PlatformReminderController) platformClaims() (*jwtutil.Claims, error) {
-	auth := c.Ctx.Request.Header.Get("Authorization")
-	if auth == "" {
-		return nil, fmt.Errorf("未登录")
-	}
-	parts := strings.SplitN(auth, " ", 2)
-	if len(parts) != 2 || parts[0] != "Bearer" {
-		return nil, fmt.Errorf("认证信息格式错误")
-	}
-	claims, err := jwtutil.ParseToken(parts[1])
-	if err != nil {
-		return nil, fmt.Errorf("无效的token")
-	}
-	if claims.UserType != "platform" {
-		return nil, fmt.Errorf("无权访问")
-	}
-	return claims, nil
+	return identity.Claims(c.Ctx.Request.Header.Get("Authorization"), identity.EndPlatform)
 }
 
 func (c *PlatformReminderController) jsonErr(httpStatus, bizCode int, msg string) {
@@ -75,7 +61,8 @@ type reminderFormPayload struct {
 // GetReminderList GET /platform/reminder/list
 func (c *PlatformReminderController) GetReminderList() {
 	if _, err := c.platformClaims(); err != nil {
-		c.jsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.jsonErr(code, code, err.Error())
 		return
 	}
 
@@ -151,7 +138,8 @@ func (c *PlatformReminderController) GetReminderList() {
 // GetReminderDetail GET /platform/reminder/:id
 func (c *PlatformReminderController) GetReminderDetail() {
 	if _, err := c.platformClaims(); err != nil {
-		c.jsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.jsonErr(code, code, err.Error())
 		return
 	}
 
@@ -222,7 +210,8 @@ func (c *PlatformReminderController) GetReminderDetail() {
 func (c *PlatformReminderController) CreateReminder() {
 	claims, err := c.platformClaims()
 	if err != nil {
-		c.jsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.jsonErr(code, code, err.Error())
 		return
 	}
 
@@ -316,7 +305,8 @@ func (c *PlatformReminderController) CreateReminder() {
 // UpdateReminder PUT /platform/reminder/:id
 func (c *PlatformReminderController) UpdateReminder() {
 	if _, err := c.platformClaims(); err != nil {
-		c.jsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.jsonErr(code, code, err.Error())
 		return
 	}
 
@@ -443,7 +433,8 @@ func (c *PlatformReminderController) UpdateReminder() {
 // DeleteReminder DELETE /platform/reminder/:id
 func (c *PlatformReminderController) DeleteReminder() {
 	if _, err := c.platformClaims(); err != nil {
-		c.jsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.jsonErr(code, code, err.Error())
 		return
 	}
 
@@ -499,7 +490,8 @@ type reminderBatchDeletePayload struct {
 // BatchDeleteReminder POST /platform/reminder/batchDelete
 func (c *PlatformReminderController) BatchDeleteReminder() {
 	if _, err := c.platformClaims(); err != nil {
-		c.jsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.jsonErr(code, code, err.Error())
 		return
 	}
 
@@ -559,7 +551,8 @@ func (c *PlatformReminderController) BatchDeleteReminder() {
 // 切换日程提醒的结束状态：若尚有提醒未结束则全部置为已结束；若已全部结束则恢复为待提醒。
 func (c *PlatformReminderController) FinishReminder() {
 	if _, err := c.platformClaims(); err != nil {
-		c.jsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.jsonErr(code, code, err.Error())
 		return
 	}
 
@@ -618,7 +611,8 @@ type reminderTestPayload struct {
 func (c *PlatformReminderController) TestReminder() {
 	claims, err := c.platformClaims()
 	if err != nil {
-		c.jsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.jsonErr(code, code, err.Error())
 		return
 	}