cas_tt_cloud_backend/app/enterprise/controller/account/register.go

package account

import (
	model2 "SciencesServer/app/common/model"
	"SciencesServer/app/enterprise/model"
	"SciencesServer/app/handle"
	"SciencesServer/serve/orm"
	"SciencesServer/utils"
	"errors"
	"gorm.io/gorm"
)

type Register struct{}

type RegisterHandle func() *Register

type (
	RegisterParams struct {
		Name, Mobile, Captcha, Password, RepeatPass string
		Identity                                    int
	}
)

func (c *RegisterParams) checkPassword() bool {
	return c.Password == c.RepeatPass
}

func (c *RegisterParams) checkUserExist(mUserInstance *model2.UserInstance) (bool, error) {
	var count int64

	if err := model2.Count(mUserInstance, &count, model2.NewWhere("mobile", c.Mobile)); err != nil {
		return false, err
	}
	return count > 0, nil
}

func (c *RegisterParams) checkCaptcha() (bool, error) {
	return handle.NewCaptcha().Validate(&handle.CaptchaSms{Captcha: c.Captcha, Mobile: c.Mobile})
}

// Launch 发起注册
func (c *Register) Launch(params *RegisterParams) (*InstanceLoginReturn, error) {
	if params.checkPassword() {
		return nil, errors.New("两次密码不一致")
	}
	mUserInstance := model.NewUserInstance()

	pass, err := params.checkUserExist(mUserInstance.UserInstance)

	if err != nil {
		return nil, err
	} else if pass {
		return nil, errors.New("当前手机号码已注册")
	}
	if pass, err = params.checkCaptcha(); err != nil {
		return nil, err
	} else if !pass {
		return nil, errors.New("验证码错误或已过期")
	}
	mUserInstance.Password = utils.GetRandomString(12)
	mUserInstance.Mobile = params.Mobile
	mUserInstance.Password = params.Password
	mUserInstance.Identity = params.Identity

	mUserTenant := model.NewUserTenant()

	if err = orm.GetDB().Transaction(func(tx *gorm.DB) error {
		if err = model2.Create(mUserInstance.UserInstance, tx); err != nil {
			return err
		}
		mUserTenant.UID = mUserInstance.UUID
		mUserTenant.Name = params.Name
		mUserTenant.Identity = params.Identity
		mUserTenant.Selected = model2.UserTenantSelectedForYes
		return model2.Create(mUserTenant.UserTenant, tx)
	}); err != nil {
		return nil, err
	}
	return NewInstance()().Login()(&InstanceLoginParams{
		UID: mUserTenant.UUID, Name: mUserTenant.Name, Mobile: mUserInstance.Mobile,
		Identity: mUserInstance.Identity, SelectIdentity: params.Identity,
	}), err
}

func NewRegister() RegisterHandle {
	return func() *Register {
		return &Register{}
	}
}