hongze_cygx/controllers/wechat.go

package controllers

import (
	"encoding/json"
	"github.com/medivhzhan/weapp/v2"
	"hongze/hongze_cygx/models"
	"hongze/hongze_cygx/services"
	"hongze/hongze_cygx/utils"
	"strconv"
	"time"
)

type WechatController struct {
	BaseAuthController
}

type WechatCommonController struct {
	BaseCommonController
}

//
//// @Title 微信登录接口
//// @Description 微信登录接口
//// @Param   Code   query   string  true       "微信唯一编码code"
//// @Success 200 {object} models.WxLoginResp
//// @router /login [get]
//func (this *WechatCommonController) WechatLogin() {
//	br := new(models.BaseResponse).Init()
//	defer func() {
//		this.Data["json"] = br
//		this.ServeJSON()
//	}()
//	code := this.GetString("Code")
//	if code == "" {
//		br.Msg = "参数错误"
//		br.ErrMsg = "Code 为空"
//		return
//	}
//	wxInfo, err := weapp.Login(utils.WxAppId, utils.WxAppSecret, code)
//	if err != nil {
//		br.Msg = "获取用户信息失败"
//		br.ErrMsg = "获取用户信息失败,Err:" + err.Error()
//		return
//	}
//	if err = wxInfo.GetResponseError(); err != nil {
//		br.Msg = "获取用户信息失败"
//		br.ErrMsg = "获取用户信息失败,code:" + strconv.Itoa(wxInfo.ErrCode) + ",msg:" + wxInfo.ErrMSG
//		return
//	}
//	firstLogin := 1
//	userId := 0
//	wxUser, err := models.GetWxUserItemByUnionid(wxInfo.UnionID)
//	if err != nil && err.Error() != utils.ErrNoRow() {
//		br.Msg = "获取用户信息失败"
//		br.ErrMsg = "根据openid获取用户信息失败，Eerr:" + err.Error()
//		return
//	}
//	if wxUser == nil || (err != nil && err.Error() == utils.ErrNoRow()) {
//		user := new(models.WxUser)
//		user.OpenId = wxInfo.OpenID
//		user.CompanyId = 1
//		user.CreatedTime = time.Now()
//		user.UnionId = wxInfo.UnionID
//		user.Unionid = wxInfo.UnionID
//		user.FirstLogin = 1
//		user.Enabled = 1
//		user.RegisterPlatform = 1
//		user.RegisterTime = time.Now()
//		user.SessionKey = wxInfo.SessionKey
//		newId, err := models.AddWxUser(user)
//		if err != nil {
//			br.Msg = "登录失败"
//			br.ErrMsg = "登录失败，新增联系人失败，Eerr:" + err.Error()
//			return
//		}
//		userId = int(newId)
//	} else {
//		firstLogin = wxUser.FirstLogin
//		userId = wxUser.UserId
//		models.ModifyWxUserSessionKey(wxInfo.SessionKey, userId)
//	}
//
//	var token string
//	tokenItem, err := models.GetTokenByUid(userId)
//	if err != nil && err.Error() != utils.ErrNoRow() {
//		br.Msg = "登录失败"
//		br.ErrMsg = "登录失败，获取token失败:" + err.Error()
//		return
//	}
//
//	if tokenItem == nil || (err != nil && err.Error() == utils.ErrNoRow()) {
//		timeUnix := time.Now().Unix()
//		timeUnixStr := strconv.FormatInt(timeUnix, 10)
//		token = utils.MD5(strconv.Itoa(userId)) + utils.MD5(timeUnixStr)
//		//新增session
//		{
//			session := new(models.CygxSession)
//			session.OpenId = wxInfo.OpenID
//			session.UnionId = wxInfo.UnionID
//			session.UserId = userId
//			session.CreatedTime = time.Now()
//			session.LastUpdatedTime = time.Now()
//			session.ExpireTime = time.Now().AddDate(0, 3, 0)
//			session.AccessToken = token
//			err = models.AddSession(session)
//			if err != nil {
//				br.Msg = "登录失败"
//				br.ErrMsg = "登录失败，新增用户session信息失败:" + err.Error()
//				return
//			}
//		}
//	} else {
//		token = tokenItem.AccessToken
//	}
//	{
//		newItem, _ := models.GetWxUserItemByUserId(userId)
//		if newItem.Mobile == "" && newItem.Email == "" {
//			firstLogin = 1
//		}
//	}
//
//	//新增登录日志
//	{
//		loginLog := new(models.WxUserLog)
//		loginLog.UserId = userId
//		loginLog.OpenId = wxInfo.OpenID
//		loginLog.UnionId = wxInfo.UnionID
//		loginLog.CreateTime = time.Now()
//		loginLog.Handle = "wechat_login_cygx"
//		loginLog.Remark = token
//		go models.AddWxUserLog(loginLog)
//	}
//
//	{
//		codeLog := new(models.WxUserCode)
//		codeLog.WxCode = code
//		codeLog.UserId = userId
//		codeLog.Code = 0
//		codeLog.FirstLogin = firstLogin
//		codeLog.Authorization = token
//		codeLog.UserPermission = 1
//		codeLog.CreateTime = time.Now()
//		models.AddWxUserCode(codeLog)
//	}
//
//	resp := new(models.WxLoginResp)
//	resp.UserId = userId
//	resp.FirstLogin = firstLogin
//	resp.Authorization = token
//	br.Ret = 200
//	br.Success = true
//	br.Msg = "登录成功"
//	br.Data = resp
//}

// @Title 微信登录接口
// @Description 微信登录接口
// @Param   Code   query   string  true       "微信唯一编码code"
// @Success 200 {object} models.WxLoginResp
// @router /login [get]
func (this *WechatCommonController) WechatLogin() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	code := this.GetString("Code")
	if code == "" {
		br.Msg = "参数错误"
		br.ErrMsg = "Code 为空"
		return
	}
	wxInfo, err := weapp.Login(utils.WxAppId, utils.WxAppSecret, code)
	if err != nil {
		br.Msg = "获取用户信息失败"
		br.ErrMsg = "获取用户信息失败,Err:" + err.Error()
		return
	}
	if err = wxInfo.GetResponseError(); err != nil {
		br.Msg = "获取用户信息失败"
		br.ErrMsg = "获取用户信息失败,code:" + strconv.Itoa(wxInfo.ErrCode) + ",msg:" + wxInfo.ErrMSG
		return
	}

	wxUserInfo := new(services.WxUserInfo)
	wxUserInfo.Unionid = wxInfo.UnionID
	wxUserInfo.Openid = wxInfo.OpenID
	wxUserInfo.Unionid = wxInfo.UnionID
	wxUserInfo.Errcode = wxInfo.ErrCode
	wxUserInfo.Errmsg = wxInfo.ErrMSG
	wxUserInfo.SessionKey = wxInfo.SessionKey
	token, userId, firstLogin, _, err := services.WxLogin(code, wxInfo.OpenID, wxInfo.UnionID, wxUserInfo)
	if err != nil {
		br.Msg = "微信登录失败"
		br.ErrMsg = "微信登录失败，err:" + err.Error()
		return
	}

	//新增登录日志
	{
		loginLog := new(models.WxUserLog)
		loginLog.UserId = userId
		loginLog.OpenId = wxInfo.OpenID
		loginLog.UnionId = wxInfo.UnionID
		loginLog.CreateTime = time.Now()
		loginLog.Handle = "wechat_login_cygx"
		loginLog.Remark = token
		go models.AddWxUserLog(loginLog)
	}

	{
		codeLog := new(models.WxUserCode)
		codeLog.WxCode = code
		codeLog.UserId = userId
		codeLog.Code = 0
		codeLog.FirstLogin = firstLogin
		codeLog.Authorization = token
		codeLog.UserPermission = 1
		codeLog.CreateTime = time.Now()
		models.AddWxUserCode(codeLog)
	}

	resp := new(models.WxLoginResp)
	resp.UserId = userId
	resp.FirstLogin = firstLogin
	resp.Authorization = token
	br.Ret = 200
	br.Success = true
	br.Msg = "登录成功"
	br.Data = resp
}
//
//// @Title 小程序获取用户信息
//// @Description 小程序获取用户信息接口(需要登录)
//// @Param	request	body models.WxGetUserInfoReq true "type json string"
//// @Success 200 {object} models.WxGetUserInfoResp
//// @router /getUserInfo [post]
//func (this *WechatController) GetUserInfo() {
//	br := new(models.BaseResponse).Init()
//	defer func() {
//		this.Data["json"] = br
//		this.ServeJSON()
//	}()
//	var req models.WxGetUserInfoReq
//	err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
//	if err != nil {
//		br.Msg = "参数解析异常！"
//		br.ErrMsg = "参数解析失败，Err:" + err.Error()
//		return
//	}
//	if req.RawData == "" || req.EncryptedData == "" || req.Signature == "" || req.Iv == "" {
//		br.Msg = "参数错误"
//		return
//	}
//	user := this.User
//	if user == nil {
//		br.Msg = "请登陆"
//		br.Ret = 408
//		return
//	}
//	sessionKey := user.SessionKey
//	userInfo, err := weapp.DecryptUserInfo(sessionKey, req.RawData, req.EncryptedData, req.Signature, req.Iv)
//	if err != nil {
//		br.Msg = "解析用户信息失败"
//		br.ErrMsg = "解析用户信息失败，DecryptUserInfo Err:" + err.Error()
//		return
//	}
//	userId := 0
//	//修改用户微信信息
//	oldUser, err := models.GetWxUserItemByUnionid(userInfo.UnionID)
//	if err != nil && err.Error() != utils.ErrNoRow() {
//		br.Msg = "获取用户信息失败"
//		br.ErrMsg = "获取用户信息失败 Err:" + err.Error()
//		return
//	}
//	if oldUser == nil || (err != nil && err.Error() == utils.ErrNoRow()) {
//		err = models.ModifyWxUserInfo(userInfo.UnionID, userInfo.Nickname, userInfo.Province, userInfo.City, userInfo.Country, userInfo.Avatar, userInfo.Gender, user.UserId)
//		if err != nil && err.Error() != utils.ErrNoRow() {
//			br.Msg = "获取用户信息失败"
//			br.ErrMsg = "修改用户信息失败 Err:" + err.Error()
//			return
//		}
//		userId = user.UserId
//	} else {
//		if user.UserId == oldUser.UserId {
//			err = models.ModifyWxUserInfo(userInfo.UnionID, userInfo.Nickname, userInfo.Province, userInfo.City, userInfo.Country, userInfo.Avatar, userInfo.Gender, user.UserId)
//			if err != nil && err.Error() != utils.ErrNoRow() {
//				br.Msg = "获取用户信息失败"
//				br.ErrMsg = "修改用户信息失败 Err:" + err.Error()
//				return
//			}
//			userId = user.UserId
//		} else {
//			userId = oldUser.UserId
//			err = models.ModifyWxUserInfo(userInfo.UnionID, userInfo.Nickname, userInfo.Province, userInfo.City, userInfo.Country, userInfo.Avatar, userInfo.Gender, oldUser.UserId)
//			if err != nil && err.Error() != utils.ErrNoRow() {
//				br.Msg = "获取用户信息失败"
//				br.ErrMsg = "修改用户信息失败 Err:" + err.Error()
//				return
//			}
//			err = models.DeleteWxUserByUserId(user.UserId)
//			if err != nil {
//				br.Msg = "删除用户信息失败"
//				br.ErrMsg = "删除用户信息失败 Err:" + err.Error()
//				return
//			}
//		}
//	}
//	var token string
//	tokenItem, err := models.GetTokenByUid(userId)
//	if err != nil && err.Error() != utils.ErrNoRow() {
//		br.Msg = "登录失败"
//		br.ErrMsg = "登录失败，获取token失败:" + err.Error()
//		return
//	}
//
//	if tokenItem == nil || (err != nil && err.Error() == utils.ErrNoRow()) {
//		timeUnix := time.Now().Unix()
//		timeUnixStr := strconv.FormatInt(timeUnix, 10)
//		token = utils.MD5(strconv.Itoa(userId)) + utils.MD5(timeUnixStr)
//		//新增session
//		{
//			session := new(models.CygxSession)
//			session.OpenId = userInfo.OpenID
//			session.UnionId = userInfo.UnionID
//			session.UserId = userId
//			session.CreatedTime = time.Now()
//			session.LastUpdatedTime = time.Now()
//			session.ExpireTime = time.Now().AddDate(0, 3, 0)
//			session.AccessToken = token
//			err = models.AddSession(session)
//			if err != nil {
//				br.Msg = "登录失败"
//				br.ErrMsg = "登录失败，新增用户session信息失败:" + err.Error()
//				return
//			}
//		}
//	} else {
//		token = tokenItem.AccessToken
//	}
//	resp := new(models.WxGetUserInfoResp)
//	resp.Authorization = token
//	br.Msg = "获取成功！"
//	br.Ret = 200
//	br.Success = true
//}


// @Title 小程序获取用户信息
// @Description 小程序获取用户信息接口(需要登录)
// @Param	request	body models.WxGetUserInfoReq true "type json string"
// @Success 200 {object} models.WxGetUserInfoResp
// @router /getUserInfo [post]
func (this *WechatController) GetUserInfo() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	var req models.WxGetUserInfoReq
	err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
	if err != nil {
		br.Msg = "参数解析异常！"
		br.ErrMsg = "参数解析失败，Err:" + err.Error()
		return
	}
	if req.RawData == "" || req.EncryptedData == "" || req.Signature == "" || req.Iv == "" {
		br.Msg = "参数错误"
		return
	}
	user := this.User
	if user == nil {
		br.Msg = "请登陆"
		br.Ret = 408
		return
	}
	userId := user.UserId
	sessionKey := user.SessionKey
	userInfo, err := weapp.DecryptUserInfo(sessionKey, req.RawData, req.EncryptedData, req.Signature, req.Iv)
	if err != nil {
		br.Msg = "解析用户信息失败"
		br.ErrMsg = "解析用户信息失败，DecryptUserInfo Err:" + err.Error()
		return
	}
	//修改用户微信信息
	err = models.ModifyUserRecordInfo(userInfo.UnionID, userInfo.Nickname, userInfo.Avatar, userInfo.City, userInfo.Province, userInfo.Country, userInfo.Gender, userId)
	if err != nil {
		br.Msg = "授权失败"
		br.ErrMsg = "授权失败，修改用户信息失败:" + err.Error()
		return
	}
	var token string
	tokenItem, err := models.GetTokenByOpenId(userInfo.OpenID)
	if err != nil && err.Error() != utils.ErrNoRow() {
		br.Msg = "授权失败"
		br.ErrMsg = "授权失败，获取token失败:" + err.Error()
		return
	}

	if tokenItem == nil || (err != nil && err.Error() == utils.ErrNoRow()) {
		timeUnix := time.Now().Unix()
		timeUnixStr := strconv.FormatInt(timeUnix, 10)
		token = utils.MD5(userInfo.OpenID) + utils.MD5(timeUnixStr)
		//新增session
		{
			session := new(models.CygxSession)
			session.OpenId = userInfo.OpenID
			session.UnionId = userInfo.UnionID
			session.UserId = userId
			session.CreatedTime = time.Now()
			session.LastUpdatedTime = time.Now()
			session.ExpireTime = time.Now().AddDate(0, 3, 0)
			session.AccessToken = token
			err = models.AddSession(session)
			if err != nil {
				br.Msg = "授权失败"
				br.ErrMsg = "授权失败，新增用户session信息失败:" + err.Error()
				return
			}
		}
	} else {
		token = tokenItem.AccessToken
	}
	resp := new(models.WxGetUserInfoResp)
	resp.Authorization = token
	br.Msg = "获取成功！"
	br.Ret = 200
	br.Success = true
}

// @Title 小程序获取用户绑定手机号
// @Description 小程序获取用户绑定手机号接口(需要登录)
// @Param	request	body models.WxGetPhoneNumberReq true "type json string"
// @Success 200 {object} models.WxGetPhoneNumberResp
// @router /getPhoneNumber [post]
func (this *WechatController) GetPhoneNumber() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	var req models.WxGetPhoneNumberReq
	err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
	if err != nil {
		br.Msg = "参数解析异常！"
		br.ErrMsg = "参数解析失败，Err:" + err.Error()
		return
	}
	if req.EncryptedData == "" || req.Iv == "" {
		br.Msg = "参数错误"
		return
	}
	user := this.User
	if user == nil {
		br.Msg = "请登陆"
		br.Ret = 408
		return
	}
	sessionKey := user.SessionKey
	wxMobile, err := weapp.DecryptMobile(sessionKey, req.EncryptedData, req.Iv)
	if err != nil {
		br.Msg = "解析用户手机号信息失败"
		br.ErrMsg = "解析用户手机号信息失败，Err:" + err.Error()
		return
	}
	err = models.ModifyUsersMobile(user.UserId, wxMobile.PurePhoneNumber)
	if err != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "获取失败,Err:" + err.Error()
		return
	}

	resp := new(models.WxGetPhoneNumberResp)
	resp.PhoneNumber = wxMobile.PhoneNumber
	resp.PurePhoneNumber = wxMobile.PurePhoneNumber
	resp.CountryCode = wxMobile.CountryCode
	br.Msg = "获取成功！"
	br.Ret = 200
	br.Success = true
	br.Data = resp
}

// @Title 获取短信验证码
// @Description 获取短信验证码接口
// @Param   Mobile   query   string  true       "手机号码"
// @Param   AreaNum   query   string  true       "地区编码"
// @Success Ret=200 获取成功
// @router /getSmsCode [get]
func (this *WechatController) GetSmsCode() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	mobile := this.GetString("Mobile")
	if mobile == "" {
		br.Msg = "请输入手机号"
		return
	}
	areaNum := this.GetString("AreaNum")
	msgCode := utils.GetRandDigit(4)
	var result bool
	if areaNum == "86" || areaNum == "" || areaNum == "0" {
		result = services.SendSmsCode(mobile, msgCode)
	} else {
		result = services.SendSmsCodeGj(mobile, msgCode, areaNum)
	}
	//发送成功
	if result {
		item := new(models.MsgCode)
		item.OpenId = ""
		item.Code = msgCode
		item.Mobile = mobile
		item.ExpiredIn = time.Now().Add(15 * time.Minute).Unix()
		item.Enabled = 1
		item.CreatedTime = time.Now()
		err := models.AddMsgCode(item)
		if err != nil {
			br.Msg = "发送失败"
			br.ErrMsg = "发送失败，Err:" + err.Error()
			return
		}
		br.Msg = "发送成功"
	} else {
		br.Msg = "发送失败"
	}
	br.Ret = 200
	br.Success = true
}

// @Title 获取邮件验证码
// @Description 获取邮件验证码接口
// @Param   Email   query   string  true       "邮箱"
// @Success Ret=200 获取成功
// @router /getEmailCode [get]
func (this *WechatController) GetEmailCode() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	email := this.GetString("Email")
	if email == "" {
		br.Msg = "请输入邮箱地址"
		return
	}
	if !utils.ValidateEmailFormatat(email) {
		br.Msg = "邮箱格式错误，请重新输入"
		return
	}
	msgCode := utils.GetRandDigit(4)
	content := "尊敬的用户：</br>您好，感谢您使用弘则研究，您正在进行邮箱验证，本次请求的验证码为：" + msgCode + "（为了保障您账号的安全性，请在15分钟内完成验证。）</br>弘则研究团队 </br>2019年05月11日"
	title := "弘则研究登陆验证"
	//发送邮件
	result, err := utils.SendEmailByHz(title, content, email)
	if err != nil {
		br.Msg = "发送失败"
		br.ErrMsg = "发送失败,Err:" + err.Error()
		return
	}
	if result {
		item := new(models.MsgCode)
		item.OpenId = ""
		item.Code = msgCode
		item.Mobile = email
		item.ExpiredIn = time.Now().Add(15 * time.Minute).Unix()
		item.Enabled = 1
		item.CreatedTime = time.Now()
		err := models.AddMsgCode(item)
		if err != nil {
			br.Msg = "发送失败"
			br.ErrMsg = "发送失败，Err:" + err.Error()
			return
		}
		br.Msg = "发送成功"
	} else {
		br.Msg = "发送失败"
	}
	br.Ret = 200
	br.Success = true
}

//
//func init() {
//	fmt.Println("start")
//	sessionKey:=`CBDTrqAe0z8uvC8gFd3lJw==`
//	rawData:=`{"nickName":"freedom","gender":0,"language":"zh_CN","city":"","province":"","country":"","avatarUrl":"https://thirdwx.qlogo.cn/mmopen/vi_32/RsZCoxzBXkoeKlKjCkWPh8DR88rtD9PlOYLB7NaFsOmwQr2DOic3m9nh2azEnkZD2mzHZ7lf0oJ53Fh8ARQr9vA/132"}`
//	encryptedData:=`OuyYSyhocFgQSj68B4MVvJYJdLCJkSdSu0TuCU+nhcRB6Vzji8e7v/5nDJHsRLopEzUvi/fcUdJKoUtDap08kIxfFsvxKTRXn2OeaCams8Wf6dE4N27HaSpq91Ykfq7Zm33WJJi/PK4kn86t/2a1bBe2FQRrcLH40cLveglPn88MDC0N4cJ119Ab8V4OhTzIbeUNWtlD883GQL7saSsPBsiGhd4a8qoHVmQiZGVjjRJ0vH9X2K404MNKi0kmtcmTGTiecTptY/nf2ZOwohnVYFXVg6C2qFwpZw2ozRNRs7mBzBK818SBM8MHvS6gRuTwkc27/Yhn/7SH4Gq/DTDBoseEa/RENZJdYi1OqRSuS9q+qb5kfvf8GB8I6g+rA35oxvwdqhA3iyotzh91znHr1LMyqPg/ol59TL8T1GYh9ppcg5rrkgywmlF9N+YWaSqpruB9VUxoieKjZNtre1aq1GZAnxY7bwoe7a7u1Uu6vVpnP2kScJC2zK4IgmBExV4/`
//
//	signature:=`a1a8a81ae75388bcbb66d92a6682ef4d16d4f634`
//	iv:=`ibri5rJxiWOT0Ar/6AfUXw==`
//	userInfo, err := weapp.DecryptUserInfo(sessionKey, rawData, encryptedData,signature, iv)
//	fmt.Println(userInfo)
//	fmt.Println(err)
//	fmt.Println("end")
//}