ark-member/internal/logic/generate_refresh_code_logic.go

package logic

import (
	"context"
	"crypto/rand"
	"fmt"
	"math/big"
	"member/gen_result/pb/member"
	"member/internal/domain"
	ers "member/internal/lib/error"
	"member/internal/svc"
	"strconv"

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

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

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

type generateRefreshCodeReq struct {
	Account string `json:"account" validate:"account"`
	// CodeType 1 email 2 phone
	CodeType int32 `json:"code_type" validate:"required,oneof=1 2 3"`
}

var codeMap = map[int32]string{
	1: "email",
	2: "phone",
}

func getCodeNameByCode(code int32) (string, bool) {
	res, ok := codeMap[code]
	if !ok {
		return "", false
	}

	return res, true
}

func generateVerifyCode(digits int) (string, error) {
	if digits <= 0 {
		// 預設為六位數
		digits = 6
	}

	// 計算最大值 (10^digits - 1)
	exp := new(big.Int).Exp(big.NewInt(10), big.NewInt(int64(digits)), nil)
	// 生成隨機數
	randomNumber, err := rand.Int(rand.Reader, exp)
	if err != nil {
		return "", err
	}

	// 將隨機數轉換為 string
	verifyCode := strconv.Itoa(int(randomNumber.Int64()))
	// 如果隨機數的位數少於指定的位數，則補 0
	if len(verifyCode) < digits {
		verifyCode = fmt.Sprintf("%0*d", digits, randomNumber)
	}

	return verifyCode, nil
}

// GenerateRefreshCode 這個帳號驗證碼（十分鐘），通用的
func (l *GenerateRefreshCodeLogic) GenerateRefreshCode(in *member.GenerateRefreshCodeReq) (*member.GenerateRefreshCodeResp, error) {
	// 驗證資料
	if err := l.svcCtx.Validate.ValidateAll(&generateRefreshCodeReq{
		Account:  in.GetAccount(),
		CodeType: in.GetCodeType(),
	}); err != nil {
		return nil, ers.InvalidFormat(err.Error())
	}

	checkType, status := getCodeNameByCode(in.GetCodeType())
	if !status {
		return nil, ers.InvalidFormat(fmt.Errorf("failed to get correct code type").Error())
	}
	rk := fmt.Sprintf("verify:%s:%s", checkType, in.GetAccount())
	// 拿過就不要再拿了
	get, err := l.svcCtx.Redis.Get(rk)
	if err != nil {
		return nil, ers.DBError("failed to connect to redis", err.Error())
	}
	if get != "" {
		return &member.GenerateRefreshCodeResp{
			Status: &member.BaseResp{
				Code:    domain.CodeOk.ToString(),
				Message: "success",
				Error:   "",
			},
			Data: &member.VerifyCode{
				VerifyCode: get,
			},
		}, nil
	}

	code, err := generateVerifyCode(6)
	if !status {
		return nil, ers.ArkInternal(err.Error())
	}

	err = l.svcCtx.Redis.Setex(rk, code, 600)
	if err != nil {
		return nil, ers.ArkInternal(err.Error())
	}

	return &member.GenerateRefreshCodeResp{
		Status: &member.BaseResp{
			Code:    domain.CodeOk.ToString(),
			Message: "success",
			Error:   "",
		},
		Data: &member.VerifyCode{
			VerifyCode: code,
		},
	}, nil
}