更新

go/controllers/backend_dict.go

@@ -10,6 +10,7 @@ import (
 	"time"
 
 	"server/models"
+	"server/pkg/identity"
 	"server/pkg/jwtutil"
 
 	"github.com/beego/beego/v2/client/orm"
@@ -42,22 +43,7 @@ var backendDictTagTypes = map[string]bool{
 }
 
 func (c *BackendDictController) dictClaims() (*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 != "backend" {
-		return nil, fmt.Errorf("无权访问")
-	}
-	return claims, nil
+	return identity.Claims(c.Ctx.Request.Header.Get("Authorization"), identity.EndBackend)
 }
 
 func (c *BackendDictController) dictErr(httpStatus, bizCode int, msg string) {
@@ -139,7 +125,8 @@ func isGlobalDictRow(tenantID string) bool { return tenantID == globalDictTenant
 func (c *BackendDictController) ListTypes() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	page, pageSize, offset := dictPageSize(&c.Controller)
@@ -188,7 +175,8 @@ func (c *BackendDictController) ListTypes() {
 func (c *BackendDictController) CreateType() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	body, _ := io.ReadAll(c.Ctx.Request.Body)
@@ -249,7 +237,8 @@ func (c *BackendDictController) CreateType() {
 func (c *BackendDictController) UpdateType() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	id := dictPathID(&c.Controller)
@@ -303,7 +292,8 @@ func (c *BackendDictController) UpdateType() {
 func (c *BackendDictController) DeleteType() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	id := dictPathID(&c.Controller)
@@ -349,7 +339,8 @@ func (c *BackendDictController) DeleteType() {
 func (c *BackendDictController) ListItems() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	dictID, _ := c.GetInt64("dict_id")
@@ -411,7 +402,8 @@ func (c *BackendDictController) ListItems() {
 func (c *BackendDictController) CreateItem() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	body, _ := io.ReadAll(c.Ctx.Request.Body)
@@ -501,7 +493,8 @@ func (c *BackendDictController) CreateItem() {
 func (c *BackendDictController) UpdateItem() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	id := dictPathID(&c.Controller)
@@ -559,7 +552,8 @@ func (c *BackendDictController) UpdateItem() {
 func (c *BackendDictController) ToggleItemStatus() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	id := dictPathID(&c.Controller)
@@ -604,7 +598,8 @@ func (c *BackendDictController) ToggleItemStatus() {
 func (c *BackendDictController) DeleteItem() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	id := dictPathID(&c.Controller)
@@ -639,7 +634,8 @@ func (c *BackendDictController) DeleteItem() {
 func (c *BackendDictController) OptionsAll() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	tenantID := strconv.Itoa(claims.TenantId)
@@ -697,7 +693,8 @@ func (c *BackendDictController) OptionsAll() {
 func (c *BackendDictController) OptionsBatch() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	body, _ := io.ReadAll(c.Ctx.Request.Body)
@@ -780,7 +777,8 @@ func (c *BackendDictController) OptionsBatch() {
 func (c *BackendDictController) Options() {
 	claims, err := c.dictClaims()
 	if err != nil {
-		c.dictErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.dictErr(code, code, err.Error())
 		return
 	}
 	code := strings.TrimSpace(c.Ctx.Input.Param(":code"))