hongze_yb/services/user/user_bind.go

package user

import (
	"context"
	"errors"
	"fmt"
	"hongze/hongze_yb/global"
	"hongze/hongze_yb/models/tables/rddp/msg_code"
	"hongze/hongze_yb/models/tables/rddp/session"
	"hongze/hongze_yb/models/tables/user_record"
	"hongze/hongze_yb/models/tables/wx_user"
	"hongze/hongze_yb/models/tables/wx_user_log"
	"hongze/hongze_yb/utils"
	"strconv"
	"time"
)

// BindWxUser 用户注册/绑定
func BindWxUser(openid, mobile, email, code string, bindType, areaNum, registerPlatform int) (token string, userInfo UserInfo, err error, errMsg string) {
	switch bindType {
	case 1: //手机号
		_, tmpErr := msg_code.GetMsgCode(mobile, code)
		if tmpErr != nil {
			if err == utils.ErrNoRow {
				err = errors.New("校验验证码失败，Err:" + tmpErr.Error())
				errMsg = "校验验证码失败"
				return
			} else {
				err = errors.New("校验验证码失败，Err:" + tmpErr.Error())
				errMsg = "验证码错误,请重新输入"
				return
			}
		}
	case 2: //邮箱
		_, tmpErr := msg_code.GetMsgCode(email, code)
		if tmpErr != nil {
			if err == utils.ErrNoRow {
				err = errors.New("校验验证码失败，Err:" + tmpErr.Error())
				errMsg = "校验验证码失败"
				return
			} else {
				err = errors.New("校验验证码失败，Err:" + tmpErr.Error())
				errMsg = "验证码错误,请重新输入"
				return
			}
		}
	default:
		err = errors.New("无效的绑定方式,bindType:" + strconv.Itoa(bindType))
		errMsg = "无效的绑定方式"
		return
	}

	userInfo, err = bindWxUser(openid, mobile, email, areaNum, registerPlatform)

	if err != nil {
		errMsg = "绑定失败：" + err.Error()
		return
	}

	sessionItem, err := session.GetTokenByOpenId(userInfo.OpenID)
	if err != nil && err != utils.ErrNoRow {
		errMsg = "登录失败"
		err = errors.New("登录失败，获取token失败:" + err.Error())
		return
	}

	if sessionItem == nil || (err != nil && err == utils.ErrNoRow) {
		timeUnix := time.Now().Unix()
		timeUnixStr := strconv.FormatInt(timeUnix, 10)
		token = utils.MD5(userInfo.OpenID) + utils.MD5(timeUnixStr)
		//新增session
		{
			sessionInfo := &session.Session{
				OpenID:          userInfo.OpenID,
				UserID:          int64(userInfo.UserID),
				CreatedTime:     time.Now(),
				LastUpdatedTime: time.Now(),
				ExpireTime:      time.Now().AddDate(0, 1, 0),
				AccessToken:     token,
			}
			tmpErr := sessionInfo.Create()
			if tmpErr != nil {
				errMsg = "登录失败"
				err = errors.New("登录失败，新增用户session信息失败:" + tmpErr.Error())
				return
			}
		}
	} else {
		token = sessionItem.AccessToken
		_ = sessionItem.UpdateSession(int64(userInfo.UserID), time.Now().AddDate(0, 1, 0))
	}

	tmpErr := wx_user.ModifyFirstLogin(userInfo.UserID)
	if err != nil {
		fmt.Println("变更联系人是否第一次登录失败，ERR:", tmpErr)
		return
	}

	//新增登录日志
	//新增登录日志
	{
		loginLog := &wx_user_log.WxUserLog{
			UserID:     int(userInfo.UserID),
			OpenID:     userInfo.OpenID,
			UnionID:    userInfo.UnionID,
			Email:      userInfo.Email,
			Mobile:     userInfo.Mobile,
			CreateTime: time.Now(),
			Handle:     "yb_login",
			Remark:     token,
		}
		go loginLog.Create()
	}
	return
}

// bindWxUser 用户注册/绑定
func bindWxUser(openid, mobile, email string, areaNum, registerPlatform int) (userInfo UserInfo, err error) {
	var source int8
	source = 6 //绑定来源，1:微信端,2:pc网页端，3:查研观向小程序,4:每日咨询
	if mobile == "" && email == "" {
		err = errors.New("手机号或邮箱必填一个")
		return
	}
	var bindAccount string
	var wxUser *wx_user.WxUser
	needCreateWxUser := true

	//根据手机号获取用户信息
	if mobile != "" {
		tmpWxUser, wxUserErr := wx_user.GetByMobile(mobile)
		if wxUserErr != nil {
			if wxUserErr != utils.ErrNoRow {
				err = wxUserErr
				return
			}
		} else {
			needCreateWxUser = false
			wxUser = tmpWxUser
			bindAccount = mobile
		}
	}
	//根据邮箱获取用户信息
	if needCreateWxUser && email != "" {
		tmpWxUser, wxUserErr := wx_user.GetByEmail(email)
		if wxUserErr != nil {
			if wxUserErr != utils.ErrNoRow {
				err = wxUserErr
				return
			}
		} else {
			needCreateWxUser = false
			wxUser = tmpWxUser
			bindAccount = email
		}
	}

	//查询openid的第三方（微信）信息
	userRecord, err := user_record.GetByOpenID(openid)
	if err != nil {
		return
	}

	var userId int
	//如果查询出来的结果是空，那么需要新增用户
	if needCreateWxUser {
		key := "bind_wx_user:mobile:" + mobile + ":email:" + email
		isHas, _ := global.Redis.Exists(context.TODO(), key).Result()
		if isHas > 0 {
			err = errors.New("多次提交，请关闭页面重新进入")
			return
		}
		global.Redis.SetNX(context.TODO(), key, "ok", time.Second*300)
		addwxUser := &wx_user.WxUser{
			CompanyID:           1,
			CreatedTime:         time.Now(),
			FirstLogin:          1,
			Enabled:             1,
			RegisterPlatform:    int8(registerPlatform), //账号注册来源，注册平台,1:微信端,2:PC网页端
			RegisterTime:        time.Now(),
			Mobile:              mobile,
			Email:               email,
			IsRegister:          1,
			Source:              source,
			CountryCode:         strconv.Itoa(areaNum),
			OutboundMobile:      mobile,
			OutboundCountryCode: strconv.Itoa(areaNum),
		}

		addUserErr := addwxUser.Create()
		//添加完成，清除缓存
		_ = global.Redis.Del(context.TODO(), key)
		if addUserErr != nil {
			err = addUserErr
			return
		}
		userId = int(addwxUser.UserID)
		tmpWxUser, _ := wx_user.GetByUserId(userId)
		wxUser = tmpWxUser
	} else {
		userId = int(wxUser.UserID)
	}
	//如果存在该手机号/邮箱，那么需要校验
	if userRecord.UserID > 0 && userRecord.UserID != userId {
		err = errors.New(fmt.Sprint("用户已绑定其他账户,已绑定的用户编号：", userRecord.UserID, "，不允许重复绑定"))
		return
	}
	if userRecord.UserID == 0 {
		userRecord.BindAccount = bindAccount
		userRecord.UserID = userId
		var updateCols = []string{"UserID", "BindAccount"}
		tmpErr := userRecord.Update(updateCols)
		if tmpErr != nil {
			err = tmpErr
			return
		}
	}

	//如果当前该第三方用户信息的昵称为空串的话，那么需要去查询该用户的第一个绑定信息的数据作为来源做数据修复
	if userRecord.NickName == "" {
		oldUserRecord, err := user_record.GetUserThirdRecordByUserId(userId)
		if err == nil && oldUserRecord != nil {
			//如果该用户绑定的第一条数据的头像信息不为空串，那么就去做新数据的修复
			if oldUserRecord.NickName != "" {
				_ = userRecord.ModifyUserRecordInfo(oldUserRecord.NickName, oldUserRecord.Headimgurl, oldUserRecord.City, oldUserRecord.Province, oldUserRecord.Country, oldUserRecord.Sex)
			}
		}
	}

	//如果该用户 绑定注册状态 字段处于 未注册 的情况下，那么去修改该数据
	if wxUser.IsRegister == 0 {
		err = wxUser.ModifyWxUserRegisterStatus(1, source, time.Now())
		if err != nil {
			return
		}
	}

	//格式化用户数据
	userInfo = formatWxUserAndUserRecord(wxUser, userRecord)
	return
}