更新

go/controllers/backend_erp_supplier.go

@@ -9,6 +9,7 @@ import (
 	"time"
 
 	"server/models"
+	"server/pkg/identity"
 	"server/pkg/jwtutil"
 
 	beego "github.com/beego/beego/v2/server/web"
@@ -20,22 +21,7 @@ type BackendErpSupplierController struct {
 }
 
 func (c *BackendErpSupplierController) supplierClaims() (*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 *BackendErpSupplierController) supplierJsonErr(httpStatus, bizCode int, msg string) {
@@ -76,13 +62,15 @@ type supplierFormPayload struct {
 	BankAccount       string `json:"bank_account"`
 	RegisteredAddress string `json:"registered_address"`
 	RegisteredPhone   string `json:"registered_phone"`
+	PaymentDays       int    `json:"payment_days"`
 }
 
 // List GET /backend/erp/supplier/list
 func (c *BackendErpSupplierController) List() {
 	claims, err := c.supplierClaims()
 	if err != nil {
-		c.supplierJsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.supplierJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -156,7 +144,8 @@ func (c *BackendErpSupplierController) List() {
 func (c *BackendErpSupplierController) Detail() {
 	claims, err := c.supplierClaims()
 	if err != nil {
-		c.supplierJsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.supplierJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -187,7 +176,8 @@ func (c *BackendErpSupplierController) Detail() {
 func (c *BackendErpSupplierController) ConvertToCustomer() {
 	claims, err := c.supplierClaims()
 	if err != nil {
-		c.supplierJsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.supplierJsonErr(code, code, err.Error())
 		return
 	}
 	id, _ := strconv.ParseUint(c.Ctx.Input.Param(":id"), 10, 64)
@@ -226,7 +216,8 @@ func (c *BackendErpSupplierController) ConvertToCustomer() {
 func (c *BackendErpSupplierController) Create() {
 	claims, err := c.supplierClaims()
 	if err != nil {
-		c.supplierJsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.supplierJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -276,6 +267,7 @@ func (c *BackendErpSupplierController) Create() {
 		BankAccount:       strings.TrimSpace(p.BankAccount),
 		RegisteredAddress: strings.TrimSpace(p.RegisteredAddress),
 		RegisteredPhone:   strings.TrimSpace(p.RegisteredPhone),
+		PaymentDays:       p.PaymentDays,
 		CreateUserID:      fmt.Sprintf("%d", claims.UserID),
 		CreateTime:        time.Now(),
 		UpdateTime:        time.Now(),
@@ -316,7 +308,8 @@ func (c *BackendErpSupplierController) Create() {
 func (c *BackendErpSupplierController) Update() {
 	claims, err := c.supplierClaims()
 	if err != nil {
-		c.supplierJsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.supplierJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -371,6 +364,7 @@ func (c *BackendErpSupplierController) Update() {
 	supplier.BankAccount = strings.TrimSpace(p.BankAccount)
 	supplier.RegisteredAddress = strings.TrimSpace(p.RegisteredAddress)
 	supplier.RegisteredPhone = strings.TrimSpace(p.RegisteredPhone)
+	supplier.PaymentDays = p.PaymentDays
 	supplier.UpdateTime = time.Now()
 
 	if p.RegisterTime != "" {
@@ -408,7 +402,8 @@ func (c *BackendErpSupplierController) Update() {
 func (c *BackendErpSupplierController) Delete() {
 	claims, err := c.supplierClaims()
 	if err != nil {
-		c.supplierJsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.supplierJsonErr(code, code, err.Error())
 		return
 	}
 
@@ -455,7 +450,8 @@ func (c *BackendErpSupplierController) Delete() {
 func (c *BackendErpSupplierController) ToggleStatus() {
 	claims, err := c.supplierClaims()
 	if err != nil {
-		c.supplierJsonErr(401, 401, err.Error())
+		code := identity.Status(err)
+		c.supplierJsonErr(code, code, err.Error())
 		return
 	}