feat: 封装中间件

app/zeus/api/lark/internal/user/user.go

@@ -1,6 +1,7 @@
 package user
 
 import (
+	"git.icechen.cn/monorepo/backend/app/zeus/service/lark/pkg/admin"
 	"git.icechen.cn/monorepo/backend/pkg/api"
 	"git.icechen.cn/monorepo/backend/pkg/proto/zeus/lark"
 	"git.icechen.cn/monorepo/backend/pkg/rpc"
@@ -27,22 +28,8 @@ func LoginHandle(c *fiber.Ctx) error {
 	return c.JSON(response)
 }
 
-// UserInfoHandle 管理员信息
-func UserInfoHandle(c *fiber.Ctx) error {
-	var p = struct {
-		Token string `json:"token"`
-	}{}
-	err := api.Bind(c, &p)
-	if err != nil {
-		return err
-	}
-
-	response, err := lark.NewUserClient(rpc.GetServiceConn("lark")).Info(c.Context(), &lark.InfoRequest{
-		Token: p.Token,
-	})
-	if err != nil {
-		return err
-	}
-
-	return c.JSON(response)
+// InfoHandle 管理员信息
+func InfoHandle(c *fiber.Ctx) error {
+	userInfo := admin.GetUserInfo(c)
+	return c.JSON(userInfo)
 }

app/zeus/api/lark/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"git.icechen.cn/monorepo/backend/app/zeus/api/lark/internal/user"
+	"git.icechen.cn/monorepo/backend/app/zeus/service/lark/pkg/admin"
 	"git.icechen.cn/monorepo/backend/pkg/api"
 	"github.com/gofiber/fiber/v2"
 	ctxLogger "github.com/luizsuper/ctxLoggers"
@@ -17,7 +18,7 @@ func main() {
 
 	userRouter := router.Group("/user")
 	userRouter.Post("/login", user.LoginHandle)
-	userRouter.Post("/info", user.UserInfoHandle)
+	userRouter.Post("/info", admin.MiddlewareLarkAdmin, user.InfoHandle)
 
 	err := router.Listen(":8080")
 	if err != nil {