youngee_mc_api/app/service/youngee_talent_service/LoginFromSms.go

package youngee_talent_service

import (
	"encoding/json"
	"fmt"
	"github.com/gogf/gf/crypto/gmd5"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/net/ghttp"
	"github.com/gogf/gf/os/gtime"
	"net/http"
	"youngmini_server/app/dao"
	"youngmini_server/app/model"
	"youngmini_server/app/model/youngee_talent_model"
	"youngmini_server/app/utils"
)

type LoginReq struct {
	Code  string `json:"code"` //来自uni.login返回的code
	Phone string `json:"phone"`
	Vcode string `json:"vcode"`
}

// 通过短信登录
func LoginFromSms(r *ghttp.Request) *TalentHttpResult {
	l := LoginReq{}
	err := r.ParseForm(&l)
	if err != nil {
		fmt.Printf("前端数据解析错误")
	}
	//通过手机号和验证码构造出key。
	VcodeKey := fmt.Sprintf("%s%s", "c_user:", l.Phone)
	//viewCount, err := g.Redis().DoVar("GET", projectViewKey)
	Vcode, err := g.Redis().DoVar("GET", VcodeKey)
	//验证码正确
	if Vcode.String() == l.Vcode {
		//进行微信登录
		appId := g.Config().GetString("miniapp.appid")
		secret := g.Config().GetString("miniapp.appsecret")
		url := fmt.Sprintf(urlformat, appId, secret, l.Code)
		//通过前端uni.login获得了code用于拼接url
		resp, err := http.Get(url)
		if err != nil {
			return &TalentHttpResult{Code: -2, Msg: err.Error()}
		}
		//http.Get(url)之后要关闭
		defer resp.Body.Close()

		// 解码微信服务端传来的信息
		wxResp := WxLoginResult{}
		decoder := json.NewDecoder(resp.Body)
		if err = decoder.Decode(&wxResp); err != nil {
			return &TalentHttpResult{Code: -3, Msg: "decode json from wx fail"}
		}

		if wxResp.ErrCode != 0 {
			fmt.Printf("错误码：%d， 错误信息：%s", wxResp.ErrCode, wxResp.ErrMsg)
			return &TalentHttpResult{Code: -4, Msg: fmt.Sprintf("errCode:%d, errmsg:%s", wxResp.ErrCode, wxResp.ErrMsg)}
		}

		// 根据电话号码，查询达人信息。达人与电话号码一一对应
		rec, err := g.DB().Model("youngee_talent_info").One("talent_phone_number", l.Phone)
		if err != nil {
			return &TalentHttpResult{Code: -5, Msg: "get talent info failed"}
		}

		// 如果达人被拉黑，则返回
		if rec != nil && rec[dao.YoungeeTalentInfo.Columns.InBlacklist].Int() > 0 {
			return &TalentHttpResult{Code: -6, Msg: "in black list"}
		}

		//手机号登录使用默认名称和头像
		//获得登录后的返回结果对象
		res := youngee_talent_model.LoginResultData{}
		var newTalentId string
		var defaultAvatar = "https://horastar.obs.cn-east-3.myhuaweicloud.com/talent/logoxin01.png"
		if rec == nil {
			// 如果数据库中不存在此达人，则插入新的达人信息
			// 首先生成达人唯一id
			newTalentId = utils.GetUuid.GetTalentId()
			talentInfo := model.YoungeeTalentInfo{
				Id:                newTalentId,
				TalentWxOpenid:    wxResp.OpenId,
				Avatar:            defaultAvatar,        //使用默认头像
				TalentWxNickname:  "样叽用户" + newTalentId, //使用默认名称
				TalentPhoneNumber: l.Phone,              //达人电话
				Canwithdraw:       0,
				Income:            0,
				Withdrawing:       0,
				Withdrawed:        0,
				LastLoginDate:     gtime.Now(),
				CreateDate:        gtime.Now(),
			}

			res.Avatar = defaultAvatar
			res.Nickname = "样叽用户" + newTalentId

			_, err = g.DB().Model(dao.YoungeeTalentInfo.Table).Data(talentInfo).Insert()
			if err != nil {
				return &TalentHttpResult{Code: -7, Msg: "get talentId failed"}
			}
		} else { //已存在达人，则读取达人id
			newTalentId = rec["id"].String()
			res.Avatar = rec["avatar"].String()
			res.Nickname = rec["talent_wx_nickname"].String()
			// 更新达人最近登录时间
			_, err = g.DB().Model(dao.YoungeeTalentInfo.Table).Data(
				g.Map{
					"last_login_date": gtime.Now(),
				}).Where("id", newTalentId).Update()
			if err != nil {
				return &TalentHttpResult{Code: -8, Msg: "update talent last login date failed"}
			}
		}
		// 用微信的openid和SessionKey的md5做为token。会一直生效
		token, err := gmd5.EncryptString(wxResp.OpenId + wxResp.SessionKey)
		fmt.Println("token is " + token)
		if err != nil {
			return &TalentHttpResult{Code: -9, Msg: "generate key failed"}
		}
		res.Token = token

		// 以token为键保存session 存储在服务器中
		//自动生成一个 session ID 并通过 Cookie 将其发送给客户端，以便在后续请求中识别该客户端的会话
		//自动生成一个 session ID，并通过 `Set-Cookie` 头将其发送给客户端，前端可以通过r.header['Set-Cookie']获取cookie
		err = r.Session.Set(token, g.Map{
			"talentId":     newTalentId,
			"wxOpenId":     wxResp.OpenId,
			"wxSessionKey": wxResp.SessionKey,
		})

		if err != nil {
			return &TalentHttpResult{Code: -10, Msg: "set session failed"}
		}

		return &TalentHttpResult{Code: 0, Msg: "SmsLoginSuccess", Data: res}

	} else { //验证码不正确
		return &TalentHttpResult{Code: -1, Msg: "验证码错误"}
	}
}