package logic

import (
	"backend/usercenter/orm"
	"backend/usercenter/rpc/internal/svc"
	"backend/usercenter/rpc/pb/usercenter"
	"backend/utils"
	"context"
	"errors"
	"regexp"
	"strconv"

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

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 (l *RegisterLogic) Register(in *usercenter.RegisterRequest) (*usercenter.RegisterResponse, error) {
	// 1. 检查用户是否存在
	user := new(orm.User)
	res := l.svcCtx.UsercenterDB.First(user, "name = ?", in.Username)
	// 有错误返回错误，找到数据返回找到
	if res.Error != nil && res.Error != gorm.ErrRecordNotFound {
		return nil, errors.New(res.Error.Error())
	}
	if res.RowsAffected > 0 {
		return nil, errors.New("用户已存在")
	}

	// 2. 检查手机号是否存在
	res = l.svcCtx.UsercenterDB.First(user, "phone = ?", in.Mobile)
	if res.Error != nil && res.Error != gorm.ErrRecordNotFound {
		return nil, errors.New(res.Error.Error())
	}
	if res.RowsAffected > 0 {
		return nil, errors.New("手机号已存在")
	}
	// 3. 检查用户名是否符合要求
	if len(in.Username) < 4 || len(in.Username) > 16 {
		return nil, errors.New("用户名长度必须在4-16之间")
	}
	// 4. 检查密码是否符合要求
	if len(in.Password) < 6 || len(in.Password) > 16 {
		return nil, errors.New("密码长度必须在6-16之间")
	}
	// 5. 检查手机号是否符合要求
	if len(in.Mobile) != 11 {
		return nil, errors.New("手机号长度必须为11位")
	}
	// 6. 正则检查手机号
	if !regexp.MustCompile(`^1[3-9]\d{9}$`).MatchString(in.Mobile) {
		return nil, errors.New("手机号格式不正确")
	}
	// 检查完成，创建用户
	// 密码加盐, utils.bcrypt.HashPassword
	salt, err := utils.HashPassword(in.Password)
	if err != nil {
		return nil, errors.New(err.Error())
	}
	user.Name = in.Username
	user.Phone = in.Mobile
	user.Password = salt

	res = l.svcCtx.UsercenterDB.Create(user)
	if res.Error != nil {
		return nil, errors.New(res.Error.Error())
	}
	// 创建成功，返回成功
	return &usercenter.RegisterResponse{
		UserId:  strconv.FormatUint(uint64(user.ID), 10),
		Message: "注册成功",
	}, nil
}