goadmin/server/api/v1/app/user.go

package app

import (
	"ln/nft/global"
	"ln/nft/model/app"
	appReq "ln/nft/model/app/request"
	appRes "ln/nft/model/app/response"
	"ln/nft/model/common/response"
	"ln/nft/model/system"
	systemReq "ln/nft/model/system/request"
	"ln/nft/utils"

	"github.com/gin-gonic/gin"
	"github.com/go-redis/redis/v8"
	"go.uber.org/zap"
)

// @Tags User
// @Summary 用户注册账号
// @Produce  application/json
// @Param data body systemReq.Register true "电子邮箱, 验证码, 密码"
// @Success 200 {object} response.Response{data=systemRes.UserResponse,msg=string} "用户注册账号,返回包括用户信息"
// @Router /user/admin_register [post]
func (b *BaseApi) Register(c *gin.Context) {
	var r appReq.Register
	_ = c.ShouldBindJSON(&r)
	var verify map[string][]string
	if r.Email != "" { // 邮箱注册
		verify = utils.RegisterVerifyEmail
	} else { // 手机号注册
		verify = utils.RegisterVerifyPhone
	}
	if err := utils.Verify(r, verify); err != nil {
		response.FailWithMessage(err.Error(), c)
		return
	}
	var authorities []app.Authority
	for _, v := range r.AuthorityIds {
		authorities = append(authorities, app.Authority{
			AuthorityId: v,
		})
	}
	user := &app.User{Email: r.Email, Password: r.Password, AuthorityId: r.AuthorityId, Authorities: authorities}
	userReturn, err := userService.Register(*user)
	if err != nil {
		global.GVA_LOG.Error("注册失败", zap.Error(err))
		response.FailWithDetailed(appRes.UserResponse{User: userReturn}, err.Error(), c)
	} else {
		response.OkWithDetailed(appRes.UserResponse{User: userReturn}, "注册成功", c)
	}

}

// @Tags Base
// @Summary 用户登录
// @Produce  application/json
// @Param data body systemReq.Login true "电子邮箱/手机号, 验证码/密码"
// @Success 200 {object} response.Response{data=systemRes.LoginResponse,msg=string} "返回包括用户信息,token,过期时间"
// @Router /base/login [post]
func (b *BaseApi) Login(c *gin.Context) {
	var l appReq.Login
	_ = c.ShouldBindJSON(&l)
	if err := utils.Verify(l, utils.LoginVerify); err != nil {
		response.FailWithMessage(err.Error(), c)
		return
	}

	u := &app.User{Email: l.Email, Phone: l.Phone, Password: l.Password}
	if user, err := userService.Login(u); err != nil {
		global.GVA_LOG.Error("登陆失败! 账号不存在或者密码错误!", zap.Error(err))
		response.FailWithMessage("账号不存在或者密码错误", c)
	} else {
		if user.Enable != 1 {
			global.GVA_LOG.Error("登陆失败! 用户被禁止登录!")
			response.FailWithMessage("用户被禁止登录", c)
			return
		}
		b.TokenNext(c, *user)
	}
}

// 登录以后签发jwt
func (b *BaseApi) TokenNext(c *gin.Context, user app.User) {
	j := &utils.JWT{SigningKey: []byte(global.GVA_CONFIG.JWT.SigningKey)} // 唯一签名
	claims := j.CreateClaims(systemReq.BaseClaims{
		UUID:        user.UUID,
		ID:          user.ID,
		NickName:    user.NickName,
		Email:       user.Email,
		Phone:       user.Phone,
		AuthorityId: user.AuthorityId,
	})
	token, err := j.CreateToken(claims)
	if err != nil {
		global.GVA_LOG.Error("获取token失败!", zap.Error(err))
		response.FailWithMessage("获取token失败", c)
		return
	}
	if !global.GVA_CONFIG.System.UseMultipoint {
		response.OkWithDetailed(appRes.LoginResponse{
			User:      user,
			Token:     token,
			ExpiresAt: claims.StandardClaims.ExpiresAt * 1000,
		}, "登录成功", c)
		return
	}

	if jwtStr, err := jwtService.GetRedisJWT(user.Email + user.Phone); err == redis.Nil {
		if err := jwtService.SetRedisJWT(token, user.Email+user.Phone); err != nil {
			global.GVA_LOG.Error("设置登录状态失败!", zap.Error(err))
			response.FailWithMessage("设置登录状态失败", c)
			return
		}
		response.OkWithDetailed(appRes.LoginResponse{
			User:      user,
			Token:     token,
			ExpiresAt: claims.StandardClaims.ExpiresAt * 1000,
		}, "登录成功", c)
	} else if err != nil {
		global.GVA_LOG.Error("设置登录状态失败!", zap.Error(err))
		response.FailWithMessage("设置登录状态失败", c)
	} else {
		var blackJWT system.JwtBlacklist
		blackJWT.Jwt = jwtStr
		if err := jwtService.JsonInBlacklist(blackJWT); err != nil {
			response.FailWithMessage("jwt作废失败", c)
			return
		}
		if err := jwtService.SetRedisJWT(token, user.Email+user.Phone); err != nil {
			response.FailWithMessage("设置登录状态失败", c)
			return
		}
		response.OkWithDetailed(appRes.LoginResponse{
			User:      user,
			Token:     token,
			ExpiresAt: claims.StandardClaims.ExpiresAt * 1000,
		}, "登录成功", c)
	}
}

// @Tags User
// @Summary 用户修改密码
// @Security ApiKeyAuth
// @Produce  application/json
// @Param data body appReq.ChangePasswordStruct true "原密码, 新密码"
// @Success 200 {object} response.Response{msg=string} "用户修改密码"
// @Router /user/changePassword [post]
func (b *BaseApi) ChangePassword(c *gin.Context) {
	var user appReq.ChangePasswordStruct
	_ = c.ShouldBindJSON(&user)
	if err := utils.Verify(user, utils.AppChangePasswordVerify); err != nil {
		response.FailWithMessage(err.Error(), c)
		return
	}
	claims := c.MustGet("claims").(*systemReq.CustomClaims)
	u := &app.User{Password: user.Password, UUID: claims.UUID}
	if _, err := userService.ChangePassword(u, user.NewPassword); err != nil {
		global.GVA_LOG.Error("修改失败!", zap.Error(err))
		response.FailWithMessage("修改失败，原密码与当前账户不符", c)
	} else {
		response.OkWithMessage("修改成功", c)
	}
}