cas_tt_cloud_backend/app/api/enterprise/controller/account/login.go

package account

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

type Login struct{ tenantID uint64 }

type (
	LoginHandle func(tenantID uint64) *Login
)

type (
	LoginParams struct {
		Captcha struct {
			Mobile, Captcha string
		}
		Password struct {
			Account, Password string
		}
		Platform struct {
			OpenID string
		}
	}
)

// LoginMode 登陆模式
type LoginMode int

const (
	LoginModeForSmsCaptcha LoginMode = iota + 1e2 + 1 // 短信验证码登陆
	LoginModeForPassword                              // 密码登陆
	LoginModeForWechat                                // 微信登陆
	LoginModeForQQ                                    // QQ登陆
)

var loginHandle = map[LoginMode]func(*LoginParams, uint64) (*InstanceLoginParams, error){
	LoginModeForSmsCaptcha: loginForSmsCaptcha, LoginModeForPassword: loginForPassword,
}

// loginForSmsCaptcha 短信验证码登陆
func loginForSmsCaptcha(params *LoginParams, tenantID uint64) (*InstanceLoginParams, error) {
	if !utils.ValidateMobile(params.Captcha.Mobile) {
		return nil, errors.New("操作错误，手机号码格式异常")
	}
	pass, err := handle.NewCaptcha().Validate(&handle.CaptchaSms{
		Mobile: params.Captcha.Mobile, Captcha: params.Captcha.Captcha,
	})
	if err != nil {
		return nil, err
	} else if !pass {
		return nil, errors.New("操作错误，验证码错误或已过期")
	}
	var isExist bool
	// 查询账号信息
	mUserInstance := model.NewUserInstance()

	if isExist, err = model2.FirstField(mUserInstance.UserInstance, []string{"id", "uuid", "avatar", "name", "mobile",
		"is_vip", "vip_deadline", "status"},
		model2.NewWhere("mobile", params.Captcha.Mobile)); err != nil {
		return nil, err
	}
	mUserIdentity := model.NewUserIdentity()
	// 用户资产信息
	mUserAssets := new(model.UserAssets)

	if isExist {
		// 查询该区域下最后一次选中的信息
		if err = mUserIdentity.LastChooseInfo(mUserInstance.UUID); err != nil {
			return nil, err
		}

		if mUserAssets, err = model.NewUserAssets().Assets(mUserInstance.UUID); err != nil {
			return nil, err
		}
		goto RETURNS
	}
	if err = orm.GetDB().Transaction(func(tx *gorm.DB) error {
		mUserInstance.Name = params.Captcha.Mobile
		mUserInstance.Mobile = params.Captcha.Mobile
		mUserInstance.Password = utils.GetRandomString(12)

		if err = model2.Create(mUserInstance.UserInstance, tx); err != nil {
			return err
		}
		mUserAssets.UID = mUserInstance.UUID

		return model2.Create(mUserAssets.UserAssets, tx)
	}); err != nil {
		return nil, err
	}
RETURNS:
	return &InstanceLoginParams{
		UID:    mUserInstance.UUID,
		Avatar: mUserInstance.Avatar, Name: mUserInstance.Name, Mobile: mUserInstance.Mobile,
		Vip: mUserInstance.Vip, VipStatus: mUserInstance.VipStatus(), VipDeadline: mUserInstance.VipDeadline,
		Currency: mUserAssets.Currency, Identity: mUserInstance.Identity, SelectIdentity: mUserIdentity.Identity,
		Status: mUserInstance.Status,
	}, nil
}

// loginForPassword 密码登陆
func loginForPassword(params *LoginParams, tenantID uint64) (*InstanceLoginParams, error) {
	if !utils.ValidateMobile(params.Password.Account) {
		return nil, errors.New("操作错误，手机号码格式异常")
	}
	mUserInstance := model.NewUserInstance()

	isExist, err := model2.FirstField(mUserInstance.UserInstance, []string{"id", "uuid", "name", "avatar", "mobile",
		"is_vip", "vip_deadline", "password", "salt", "status"},
		model2.NewWhere("mobile", params.Password.Account))

	if err != nil {
		return nil, err
	} else if !isExist {
		return nil, errors.New("操作错误，当前手机号码未注册")
	}
	if !mUserInstance.ValidatePassword(params.Password.Password) {
		return nil, errors.New("操作错误，账户或密码错误")
	}
	// 用户资产信息
	mUserAssets := new(model.UserAssets)

	if mUserAssets, err = model.NewUserAssets().Assets(mUserInstance.UUID); err != nil {
		return nil, err
	}
	// 最后一次选中的身份信息
	mUserIdentity := model.NewUserIdentity()

	if err = mUserIdentity.LastChooseInfo(mUserInstance.UUID); err != nil {
		return nil, err
	}
	return &InstanceLoginParams{
		UID:    mUserInstance.UUID,
		Avatar: mUserInstance.Avatar, Name: mUserInstance.Name, Mobile: mUserInstance.Mobile,
		Vip: mUserInstance.Vip, VipStatus: mUserInstance.VipStatus(), VipDeadline: mUserInstance.VipDeadline,
		Currency: mUserAssets.Currency, Identity: mUserInstance.Identity, SelectIdentity: mUserIdentity.Identity,
		Status: mUserInstance.Status,
	}, nil
}

func loginForPlatform(params *LoginParams) error {
	return nil
}

func (c *Login) Launch(mode LoginMode, params *LoginParams) (*InstanceLoginReturn, error) {
	_handle, has := loginHandle[mode]

	if !has {
		return nil, errors.New("操作错误，未知的登陆模式")
	}
	ret, err := _handle(params, c.tenantID)

	if err != nil {
		return nil, err
	}
	if ret.Status != model2.AccountStatusForEnable {
		return nil, errors.New("操作错误，该账号已禁止登陆，请联系管理员")
	}
	return NewInstance()().Login()(ret), err
}

func NewLogin() LoginHandle {
	return func(tenantID uint64) *Login {
		return &Login{tenantID: tenantID}
	}
}