基本完成重置密码和忘记密码

server/auth/internal/handler/userresetpasswordhandler.go

@@ -14,7 +14,7 @@ import (
 func UserResetPasswordHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
 
-		var req types.RequestUserLogin
+		var req types.RequestUserResetPassword
 		userinfo, err := basic.RequestParse(w, r, svcCtx, &req)
 		if err != nil {
 			return

server/auth/internal/logic/userresetpasswordlogic.go

@@ -1,6 +1,8 @@
 package logic
 
 import (
+	"fmt"
+	"fusenapi/model/gmodel"
 	"fusenapi/utils/auth"
 	"fusenapi/utils/basic"
 
@@ -10,6 +12,7 @@ import (
 	"fusenapi/server/auth/internal/types"
 
 	"github.com/zeromicro/go-zero/core/logx"
+	"gorm.io/gorm"
 )
 
 type UserResetPasswordLogic struct {
@@ -30,10 +33,41 @@ func NewUserResetPasswordLogic(ctx context.Context, svcCtx *svc.ServiceContext)
 // func (l *UserResetPasswordLogic) BeforeLogic(w http.ResponseWriter, r *http.Request) {
 // }
 
-func (l *UserResetPasswordLogic) UserResetPassword(req *types.RequestUserLogin, userinfo *auth.UserInfo) (resp *basic.Response) {
+func (l *UserResetPasswordLogic) UserResetPassword(req *types.RequestUserResetPassword, userinfo *auth.UserInfo) (resp *basic.Response) {
 	// 返回值必须调用Set重新返回, resp可以空指针调用 resp.SetStatus(basic.CodeOK, data)
 	// userinfo 传入值时, 一定不为null
 
+	if !userinfo.IsUser() {
+		return resp.SetStatus(basic.CodeUnAuth)
+	}
+
+	rt, err := l.svcCtx.ResetTokenManger.Decrypt(req.ResetToken)
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatus(basic.CodeOAuthResetTokenDecryptErr, err.Error())
+	}
+
+	// TODO: 存储
+	if rt.OperateType != auth.OpTypeResetToken {
+		return resp.SetStatus(basic.CodeOAuthTypeErr, "error OperateType: rt.OperateType != auth.OpTypeResetToken")
+	}
+
+	err = l.svcCtx.AllModels.FsUser.Transaction(l.ctx, func(tx *gorm.DB) error {
+		user := &gmodel.FsUser{Id: int64(rt.UserId)}
+		err := tx.Take(user).Error
+		if err != nil {
+			return err
+		}
+		if *user.PasswordHash != rt.OldPassword {
+			return fmt.Errorf("password had beed updated")
+		}
+		return tx.Update("PasswordHash", req.Password).Error
+	})
+
+	if err != nil {
+		return resp.SetStatus(basic.CodeDbSqlErr, err.Error())
+	}
+
 	return resp.SetStatus(basic.CodeOK)
 }
 

server/auth/internal/logic/userresettokenlogic.go

@@ -50,13 +50,21 @@ func (l *UserResetTokenLogic) UserResetToken(req *types.RequestUserResetToken, u
 		UserId:      uint64(userinfo.UserId),
 		Wid:         req.Wid,
 		Email:       *user.Email,
-		Password:    *user.PasswordHash,
+		OldPassword: *user.PasswordHash,
 		CreateAt:    time.Now(),
 	}
 
-	l.svcCtx.ResetTokenManger.Encrypt(token)
+	rtoken, err := l.svcCtx.ResetTokenManger.Encrypt(token)
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatus(basic.CodeOAuthResetTokenEncryptErr, err.Error())
+	}
 
-	return resp.SetStatus(basic.CodeOK)
+	data := types.DataResetToken{
+		ResetToken: rtoken,
+	}
+
+	return resp.SetStatus(basic.CodeOK, data)
 }
 
 // 处理逻辑后 w,r 如:重定向, resp 必须重新处理