更新

go/controllers/app_notebook.go

@@ -5,6 +5,7 @@ import (
 	"io"
 	"server/middleware"
 	"server/models"
+	"server/pkg/identity"
 	"server/pkg/jwtutil"
 	"strconv"
 	"strings"
@@ -18,23 +19,9 @@ type AppNotebookController struct {
 	beego.Controller
 }
 
+// 只放行移动端与租户端令牌：/api/** 仅被 uniapp 调用，平台令牌不该能打移动端接口。
 func (c *AppNotebookController) appNotebookClaims() (*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" && claims.UserType != "app" && claims.UserType != "platform" {
-		return nil, orm.ErrNoRows
-	}
-	return claims, nil
+	return identity.Claims(c.Ctx.Request.Header.Get("Authorization"), identity.EndApp, identity.EndBackend)
 }
 
 func (c *AppNotebookController) nbJsonErr(httpStatus, bizCode int, msg string) {
@@ -52,7 +39,7 @@ func (c *AppNotebookController) nbOk(data interface{}) {
 func (c *AppNotebookController) GetList() {
 	claims, err := c.appNotebookClaims()
 	if err != nil {
-		c.nbJsonErr(401, 401, "未登录或无权限")
+		c.nbJsonErr(identity.Status(err), identity.Status(err), err.Error())
 		return
 	}
 
@@ -112,7 +99,7 @@ func (c *AppNotebookController) GetList() {
 func (c *AppNotebookController) GetDetail() {
 	claims, err := c.appNotebookClaims()
 	if err != nil {
-		c.nbJsonErr(401, 401, "未登录或无权限")
+		c.nbJsonErr(identity.Status(err), identity.Status(err), err.Error())
 		return
 	}
 
@@ -153,7 +140,7 @@ func (c *AppNotebookController) GetDetail() {
 func (c *AppNotebookController) Create() {
 	claims, err := c.appNotebookClaims()
 	if err != nil {
-		c.nbJsonErr(401, 401, "未登录或无权限")
+		c.nbJsonErr(identity.Status(err), identity.Status(err), err.Error())
 		return
 	}
 
@@ -212,7 +199,7 @@ func (c *AppNotebookController) Create() {
 func (c *AppNotebookController) Update() {
 	claims, err := c.appNotebookClaims()
 	if err != nil {
-		c.nbJsonErr(401, 401, "未登录或无权限")
+		c.nbJsonErr(identity.Status(err), identity.Status(err), err.Error())
 		return
 	}
 
@@ -289,7 +276,7 @@ func (c *AppNotebookController) Update() {
 func (c *AppNotebookController) Delete() {
 	claims, err := c.appNotebookClaims()
 	if err != nil {
-		c.nbJsonErr(401, 401, "未登录或无权限")
+		c.nbJsonErr(identity.Status(err), identity.Status(err), err.Error())
 		return
 	}
 
@@ -330,7 +317,7 @@ func (c *AppNotebookController) Delete() {
 func (c *AppNotebookController) TogglePin() {
 	claims, err := c.appNotebookClaims()
 	if err != nil {
-		c.nbJsonErr(401, 401, "未登录或无权限")
+		c.nbJsonErr(identity.Status(err), identity.Status(err), err.Error())
 		return
 	}