fusenapi/server/home-user-auth/internal/logic/userloginlogic.go

package logic

import (
	"context"
	"errors"
	"fmt"
	"net/http"
	"time"

	"fusenapi/server/home-user-auth/internal/svc"
	"fusenapi/server/home-user-auth/internal/types"
	"fusenapi/utils/auth"
	"fusenapi/utils/basic"

	"github.com/zeromicro/go-zero/core/logx"
	"gorm.io/gorm"
)

type UserLoginLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext

	token string
}

func NewUserLoginLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UserLoginLogic {
	return &UserLoginLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *UserLoginLogic) AfterLogic(w http.ResponseWriter, r *http.Request) {
	if l.token != "" {
		w.Header().Add("Authorization", fmt.Sprintf("Bearer %s", l.token))
	}
}

func (l *UserLoginLogic) UserLogin(req *types.RequestUserLogin, userinfo *auth.UserInfo) (resp *basic.Response) {
	// 创建一个 FsUserModel 对象 m 并实例化之，该对象用于操作 MySQL 数据库中的用户数据表。
	m := l.svcCtx.AllModels.FsUser

	// 在用户数据表中根据登录名(email)查找用户记录，并返回 UserModel 类型的结构体对象 userModel。
	user, err := m.FindUserByEmail(l.ctx, req.Email)
	if errors.Is(err, gorm.ErrRecordNotFound) {
		return resp.SetStatus(basic.CodeEmailNotFoundErr)
	}

	// 如果在用户数据表中找到了登录名匹配的用户记录，则判断密码是否匹配。
	if *user.PasswordHash != req.Password {
		logx.Info("密码错误")
		return resp.SetStatus(basic.CodePasswordErr)
	}

	// 如果密码匹配，则生成 JWT Token。
	nowSec := time.Now().Unix()
	jwtToken, err := auth.GenerateJwtToken(&l.svcCtx.Config.Auth.AccessSecret, l.svcCtx.Config.Auth.AccessExpire, nowSec, user.Id, 0)

	// 如果生成 JWT Token 失败，则抛出错误并返回未认证的状态码。
	if err != nil {
		logx.Error(err)
		return resp.SetStatus(basic.CodeUnAuth)
	}

	// 如果更新 VerificationToken 字段失败，则返回未认证的状态码。
	if err != nil {
		return resp.SetStatus(basic.CodeUnAuth)
	}

	// 构造 DataUserLogin 类型的数据对象 data 并设置其属性值为生成的 JWT Token。
	data := &types.DataUserLogin{
		Token: jwtToken,
	}

	l.token = jwtToken

	// 返回认证成功的状态码以及数据对象 data 和 JWT Token。
	return resp.SetStatus(basic.CodeOK, data)
}