111 lines
3.2 KiB
Go
111 lines
3.2 KiB
Go
// Package api 定义API接口层
|
||
// 包含所有对外暴露的HTTP接口定义
|
||
package api
|
||
|
||
import (
|
||
"errors"
|
||
"net/http"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/model"
|
||
"github.com/LoveLosita/smartflow/backend/respond"
|
||
"github.com/LoveLosita/smartflow/backend/service"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
type UserHandler struct {
|
||
// 伸出手：准备接住 Service
|
||
svc *service.UserService
|
||
}
|
||
|
||
// NewUserHandler：组装 Handler 的“工厂”
|
||
func NewUserHandler(svc *service.UserService) *UserHandler {
|
||
return &UserHandler{
|
||
svc: svc, // 把传进来的 Service 揣进口袋里
|
||
}
|
||
}
|
||
|
||
// UserRegister 用户注册API
|
||
// 处理用户注册请求
|
||
func (api *UserHandler) UserRegister(c *gin.Context) {
|
||
var user model.UserRegisterRequest
|
||
err := c.ShouldBindJSON(&user)
|
||
if err != nil {
|
||
c.JSON(http.StatusBadRequest, respond.WrongParamType)
|
||
return
|
||
}
|
||
|
||
retUser, err := api.svc.UserRegister(user)
|
||
if err != nil {
|
||
switch {
|
||
case errors.Is(err, respond.InvalidName), errors.Is(err, respond.MissingParam),
|
||
errors.Is(err, respond.ParamTooLong): //如果是无效ID或者缺少参数的错误
|
||
c.JSON(http.StatusBadRequest, err)
|
||
return
|
||
default:
|
||
c.JSON(http.StatusInternalServerError, respond.InternalError(err))
|
||
return
|
||
}
|
||
}
|
||
|
||
c.JSON(http.StatusOK, respond.RespWithData(respond.Ok, retUser))
|
||
}
|
||
|
||
func (api *UserHandler) UserLogin(c *gin.Context) {
|
||
var req model.UserLoginRequest
|
||
err := c.ShouldBindJSON(&req)
|
||
if err != nil {
|
||
c.JSON(http.StatusOK, respond.WrongParamType)
|
||
return
|
||
}
|
||
tokens, err := api.svc.UserLogin(&req)
|
||
if err != nil {
|
||
switch {
|
||
case errors.Is(err, respond.WrongName), errors.Is(err, respond.WrongPwd): //如果是无效ID或者缺少参数的错误
|
||
c.JSON(http.StatusBadRequest, err)
|
||
return
|
||
default:
|
||
c.JSON(http.StatusInternalServerError, respond.InternalError(err))
|
||
return
|
||
}
|
||
}
|
||
c.JSON(http.StatusOK, respond.RespWithData(respond.Ok, tokens))
|
||
}
|
||
|
||
func (api *UserHandler) RefreshTokenHandler(c *gin.Context) {
|
||
var requestBody struct {
|
||
RefreshToken string `json:"old_refresh_token"`
|
||
}
|
||
if err := c.ShouldBindJSON(&requestBody); err != nil {
|
||
c.JSON(http.StatusBadRequest, respond.WrongParamType)
|
||
return
|
||
}
|
||
if requestBody.RefreshToken == "" {
|
||
c.JSON(http.StatusBadRequest, respond.MissingParam)
|
||
}
|
||
tokens, err := api.svc.RefreshTokenHandler(requestBody.RefreshToken)
|
||
if err != nil {
|
||
switch {
|
||
case errors.Is(err, respond.InvalidRefreshToken), errors.Is(err, respond.InvalidClaims),
|
||
errors.Is(err, respond.InvalidTokenSingingMethod), errors.Is(err, respond.UserLoggedOut): //如果是无效刷新令牌或者无效claims或者无效签名方法
|
||
c.JSON(http.StatusBadRequest, err)
|
||
return
|
||
default:
|
||
c.JSON(http.StatusInternalServerError, respond.InternalError(err))
|
||
}
|
||
}
|
||
c.JSON(http.StatusOK, respond.RespWithData(respond.Ok, tokens))
|
||
}
|
||
|
||
func (api *UserHandler) UserLogout(c *gin.Context) {
|
||
//1.从上下文中获取 jti 和 expireTime
|
||
claims, _ := c.Get("claims")
|
||
cl := claims.(*model.MyCustomClaims)
|
||
//2.调用 Service 层的 UserLogout 方法
|
||
err := api.svc.UserLogout(cl.Jti, cl.ExpiresAt.Time)
|
||
if err != nil {
|
||
c.JSON(http.StatusInternalServerError, respond.InternalError(err))
|
||
return
|
||
}
|
||
c.JSON(http.StatusOK, respond.Ok)
|
||
}
|