90 lines
2.3 KiB
Go
90 lines
2.3 KiB
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{}
|
|
}
|
|
}
|