135 lines
3.0 KiB
Go
135 lines
3.0 KiB
Go
package handler
|
|
|
|
import (
|
|
"errors"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"kra/internal/biz/system"
|
|
"kra/internal/service/dto"
|
|
service "kra/internal/service/liveclient"
|
|
)
|
|
|
|
type LiveClient struct{ service *service.Service }
|
|
|
|
func NewLiveClient(service *service.Service) *LiveClient { return &LiveClient{service: service} }
|
|
|
|
func (h *LiveClient) Register(c *gin.Context) {
|
|
in, ok := bindJSON[dto.LiveClientRegisterRequest](c)
|
|
if !ok {
|
|
return
|
|
}
|
|
result, err := h.service.Register(c.Request.Context(), in, c.ClientIP(), c.Request.UserAgent())
|
|
if err != nil {
|
|
clientAuthFail(c, err)
|
|
return
|
|
}
|
|
setClientToken(c, result)
|
|
Write(c, CodeSuccess, result, "注册成功")
|
|
}
|
|
|
|
func (h *LiveClient) Login(c *gin.Context) {
|
|
in, ok := bindJSON[dto.LiveClientLoginRequest](c)
|
|
if !ok {
|
|
return
|
|
}
|
|
result, err := h.service.Login(c.Request.Context(), in, c.ClientIP(), c.Request.UserAgent())
|
|
if err != nil {
|
|
clientAuthFail(c, err)
|
|
return
|
|
}
|
|
setClientToken(c, result)
|
|
Write(c, CodeSuccess, result, "登录成功")
|
|
}
|
|
|
|
func (h *LiveClient) Me(c *gin.Context) {
|
|
claims := Claims(c)
|
|
if claims == nil {
|
|
NoAuth(c, "请登录")
|
|
return
|
|
}
|
|
result, err := h.service.Me(c.Request.Context(), claims.ID)
|
|
if err != nil {
|
|
businessFail(c, err)
|
|
return
|
|
}
|
|
OKWithData(c, result)
|
|
}
|
|
|
|
func (h *LiveClient) Balance(c *gin.Context) {
|
|
claims := Claims(c)
|
|
if claims == nil {
|
|
NoAuth(c, "请登录")
|
|
return
|
|
}
|
|
result, err := h.service.Balance(c.Request.Context(), claims.ID)
|
|
if err != nil {
|
|
businessFail(c, err)
|
|
return
|
|
}
|
|
OKWithData(c, result)
|
|
}
|
|
|
|
func (h *LiveClient) UpdateProfile(c *gin.Context) {
|
|
claims := Claims(c)
|
|
if claims == nil {
|
|
NoAuth(c, "请登录")
|
|
return
|
|
}
|
|
in, ok := bindJSON[dto.LiveClientProfileRequest](c)
|
|
if !ok {
|
|
return
|
|
}
|
|
result, err := h.service.UpdateProfile(c.Request.Context(), claims.ID, in)
|
|
if err != nil {
|
|
businessFail(c, err)
|
|
return
|
|
}
|
|
OKWithData(c, result)
|
|
}
|
|
|
|
func (h *LiveClient) ChangePassword(c *gin.Context) {
|
|
claims := Claims(c)
|
|
if claims == nil {
|
|
NoAuth(c, "请登录")
|
|
return
|
|
}
|
|
in, ok := bindJSON[dto.LiveClientPasswordRequest](c)
|
|
if !ok {
|
|
return
|
|
}
|
|
result, err := h.service.ChangePassword(c.Request.Context(), claims.ID, in, c.ClientIP(), c.Request.UserAgent())
|
|
if err != nil {
|
|
clientAuthFail(c, err)
|
|
return
|
|
}
|
|
setClientToken(c, result)
|
|
Write(c, CodeSuccess, result, "密码修改成功")
|
|
}
|
|
|
|
func setClientToken(c *gin.Context, result *dto.LiveClientAuthResponse) {
|
|
maxAge := int(time.UnixMilli(result.ExpiresAt).Unix() - time.Now().Unix())
|
|
SetTokenCookie(c, result.Token, maxAge)
|
|
}
|
|
|
|
func clientAuthFail(c *gin.Context, err error) {
|
|
switch {
|
|
case errors.Is(err, system.ErrAccountLocked):
|
|
var locked *system.AccountLockedError
|
|
errors.As(err, &locked)
|
|
minutes := 0
|
|
if locked != nil {
|
|
minutes = locked.Minutes
|
|
}
|
|
Fail(c, "账号已锁定，请 "+strconv.Itoa(minutes)+" 分钟后再试")
|
|
case errors.Is(err, system.ErrInvalidCredentials):
|
|
Fail(c, "用户名不存在或者密码错误")
|
|
case errors.Is(err, system.ErrUserDisabled):
|
|
Fail(c, "用户被禁止登录")
|
|
default:
|
|
businessFail(c, err)
|
|
}
|
|
}
|