add: user auth accomplished

app/users/rpc/internal/logic/loginLogic.go

@@ -2,9 +2,11 @@ package logic
 
 import (
 	"context"
-
+	"errors"
 	"juwan-backend/app/users/rpc/internal/svc"
+	utils2 "juwan-backend/app/users/rpc/internal/utils"
 	"juwan-backend/app/users/rpc/pb"
+	"juwan-backend/common/utils"
 
 	"github.com/zeromicro/go-zero/core/logx"
 )
@@ -24,7 +26,29 @@ func NewLoginLogic(ctx context.Context, svcCtx *svc.ServiceContext) *LoginLogic
 }
 
 func (l *LoginLogic) Login(in *pb.LoginReq) (*pb.LoginResp, error) {
-	// todo: add your logic here and delete this line
+	user, err := l.svcCtx.UsersModelRO.FindOneByUsername(l.ctx, in.Username)
+	if err != nil {
+		logx.WithContext(l.ctx).Errorf("LoginLogic.Login error:%v", err)
+		return nil, err
+	}
+	if !utils.VerifyPassword(user.Passwd, in.Passwd) {
+		logx.WithContext(l.ctx).Errorf("User %s Login failed", user.Username)
+		return nil, errors.New("incorrect password")
+	}
 
-	return &pb.LoginResp{}, nil
+	token, err := l.svcCtx.JwtManager.New(l.ctx, &utils2.TokenPayload{
+		UserId:  user.UserId,
+		IsAdmin: false,
+	})
+	if err != nil {
+		logx.Errorf("LoginLogic.Login gen jwt for user %v error:%v", user.UserId, err)
+		return nil, err
+	}
+
+	return &pb.LoginResp{
+		Token:    token,
+		Username: user.Username,
+		Email:    user.Email,
+		Id:       user.UserId,
+	}, nil
 }

app/users/rpc/internal/logic/logoutLogic.go

@@ -0,0 +1,34 @@
+package logic
+
+import (
+	"context"
+
+	"juwan-backend/app/users/rpc/internal/svc"
+	"juwan-backend/app/users/rpc/pb"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type LogoutLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewLogoutLogic(ctx context.Context, svcCtx *svc.ServiceContext) *LogoutLogic {
+	return &LogoutLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *LogoutLogic) Logout(in *pb.LogoutReq) (*pb.LogoutResp, error) {
+	// todo: add your logic here and delete this line
+	err := l.svcCtx.JwtManager.Logout(l.ctx, in.UserId)
+	if err != nil {
+		logx.WithContext(l.ctx).Errorf("Logout failed: %s", err.Error())
+		return nil, err
+	}
+	return &pb.LogoutResp{}, nil
+}

app/users/rpc/internal/logic/registerLogic.go

@@ -0,0 +1,92 @@
+package logic
+
+import (
+	"context"
+	"crypto/rand"
+	"encoding/hex"
+	"errors"
+	"fmt"
+	"strconv"
+	"strings"
+
+	"juwan-backend/app/snowflake/rpc/snowflake"
+	"juwan-backend/app/users/rpc/internal/models"
+	"juwan-backend/app/users/rpc/internal/svc"
+	"juwan-backend/app/users/rpc/pb"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type RegisterLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewRegisterLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RegisterLogic {
+	return &RegisterLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func mustNewRandomNickname() string {
+	bytes := make([]byte, 5)
+	_, err := rand.Read(bytes)
+	if err != nil {
+		return "NewUser"
+	}
+	nickname := strings.Builder{}
+	nickname.WriteString("user_")
+	nickname.WriteString(hex.EncodeToString(bytes))
+	return nickname.String()
+}
+
+func (l *RegisterLogic) Register(in *pb.RegisterReq) (*pb.RegisterResp, error) {
+	// todo: add your logic here and delete this line
+	if in.Phone == "" || in.Username == "" || in.Passwd == "" {
+		logx.Error("invalid input")
+		return nil, errors.New("invalid input")
+	}
+
+	redisKey := fmt.Sprintf("vcode:%s:%s:%s", in.RequestId, "register", in.Email)
+	vcode, err := l.svcCtx.RedisCluster.Get(l.ctx, redisKey).Result()
+	logx.Infof("vcode:%s, err:%v", vcode, err)
+	if err != nil {
+		logx.Error("invalid verification code")
+		return nil, errors.New("invalid verification code")
+	}
+
+	code, err := strconv.ParseInt(vcode, 10, 32)
+	if err != nil || int32(code) != in.Vcode {
+		logx.Error("invalid verification code")
+		return nil, errors.New("invalid verification code")
+	}
+
+	resp, err := l.svcCtx.Snowflake.NextId(l.ctx, &snowflake.NextIdReq{})
+	if err != nil {
+		return nil, errors.New("generate user ID failed")
+	}
+
+	user := models.Users{
+		UserId:     resp.Id,
+		Username:   in.Username,
+		Nickname:   mustNewRandomNickname(),
+		Passwd:     in.Passwd,
+		Phone:      in.Phone,
+		Email:      in.Email,
+		RoleType:   0,
+		IsVerified: false,
+	}
+
+	_, err = l.svcCtx.UsersModelRW.Insert(l.ctx, &user)
+	if err != nil {
+		logx.Error("failed to create user: ", err)
+		return nil, errors.New("failed to create user")
+	}
+
+	return &pb.RegisterResp{
+		Res: "user registered successfully",
+	}, nil
+}

app/users/rpc/internal/logic/validateTokenLogic.go

@@ -2,6 +2,7 @@ package logic
 
 import (
 	"context"
+	"fmt"
 
 	"juwan-backend/app/users/rpc/internal/svc"
 	"juwan-backend/app/users/rpc/pb"
@@ -9,6 +10,8 @@ import (
 	"github.com/zeromicro/go-zero/core/logx"
 )
 
+var USER_TOKEN_TEMP = "jwt:%v"
+
 type ValidateTokenLogic struct {
 	ctx    context.Context
 	svcCtx *svc.ServiceContext
@@ -24,7 +27,20 @@ func NewValidateTokenLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Val
 }
 
 func (l *ValidateTokenLogic) ValidateToken(in *pb.ValidateTokenReq) (*pb.ValidateTokenResp, error) {
-	// todo: add your logic here and delete this line
+	redisKey := fmt.Sprintf(USER_TOKEN_TEMP, in.UserId)
+	_, err := l.svcCtx.JwtManager.Valid(l.ctx, redisKey)
+	if err != nil {
+		return nil, err
+	}
+	users, err := l.svcCtx.UsersModelRO.FindOne(l.ctx, in.UserId)
+	if err != nil {
+		return nil, err
+	}
 
-	return &pb.ValidateTokenResp{}, nil
+	return &pb.ValidateTokenResp{
+		Valid:    true,
+		Message:  "OK",
+		UserId:   in.UserId,
+		RoleType: users.RoleType,
+	}, nil
 }