wechat.go 8.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "github.com/medivhzhan/weapp/v2"
  6. adminResp "hongze/hongze_mobile_admin/models/response/admin"
  7. "hongze/hongze_mobile_admin/models/response/wechat"
  8. "hongze/hongze_mobile_admin/services"
  9. "hongze/hongze_mobile_admin/utils"
  10. "strconv"
  11. )
  12. type WeChatCommon struct {
  13. BaseNotAuth
  14. }
  15. // WeChatLogin
  16. // @Title 微信登录接口
  17. // @Description 微信登录接口
  18. // @Param Code query string true "微信唯一编码code"
  19. // @Success 200 {object} wechat.WxLoginResp
  20. // @router /login [get]
  21. func (c *WeChatCommon) WeChatLogin() {
  22. code := c.GetString("Code")
  23. fmt.Println("code:", code)
  24. utils.FileLog.Info("WechatLogin code:%s", code)
  25. item, err := services.WxGetUserOpenIdByCode(code)
  26. if err != nil {
  27. c.FailWithMessage(fmt.Sprintf("%v", err), fmt.Sprintf("%v", err))
  28. return
  29. }
  30. if item.Errcode != 0 {
  31. c.FailWithMessage("获取用户信息失败", "获取access_token 失败 errCode:"+strconv.Itoa(item.Errcode)+" ;errMsg:"+item.Errmsg)
  32. }
  33. openId := item.Openid
  34. if openId == "" {
  35. c.FailWithMessage("获取用户信息失败", "获取openid失败,openid:"+item.Openid)
  36. }
  37. accessToken, err := services.WxGetAccessToken()
  38. if err != nil {
  39. c.FailWithMessage("获取用户信息失败", "获取access_token失败,err:"+err.Error())
  40. }
  41. //获取用户信息
  42. wxUserInfo, err := services.WxGetUserInfo(openId, accessToken)
  43. if err != nil {
  44. c.FailWithMessage("获取用户信息失败", "获取微信用户信息失败,Err:"+err.Error())
  45. }
  46. if wxUserInfo.Errcode != 0 {
  47. userInfoJson, _ := json.Marshal(wxUserInfo)
  48. c.FailWithMessage("登录失败", "获取用户信息失败,err:"+string(userInfoJson))
  49. return
  50. }
  51. token, adminWx, err := services.WxLogin(utils.WxPlatform, item, wxUserInfo)
  52. if err != nil {
  53. c.FailWithMessage("微信登录失败", "微信登录失败,err:"+err.Error())
  54. return
  55. }
  56. userInfo := adminResp.LoginResp{
  57. RealName: adminWx.RealName,
  58. AdminName: adminWx.AdminName,
  59. RoleName: adminWx.RoleName,
  60. RoleTypeCode: adminWx.RoleTypeCode,
  61. AdminId: adminWx.AdminId,
  62. DepartmentName: adminWx.DepartmentName,
  63. GroupName: adminWx.GroupName,
  64. }
  65. userInfo.AdminId = adminWx.AdminId
  66. var productName string
  67. productId := services.GetProductId(adminWx.RoleTypeCode)
  68. if productId == 1 {
  69. productName = utils.COMPANY_PRODUCT_FICC_NAME
  70. } else if productId == 2 {
  71. productName = utils.COMPANY_PRODUCT_RAI_NAME
  72. } else {
  73. productName = "admin"
  74. }
  75. userInfo.ProductName = productName
  76. userInfo.Authority = adminWx.Authority
  77. resp := wechat.WxLoginResp{
  78. AdminId: adminWx.AdminId,
  79. Code: 0,
  80. Authorization: token,
  81. Headimgurl: adminWx.Headimgurl,
  82. RealName: adminWx.RealName,
  83. UserInfo: userInfo,
  84. }
  85. //登录日志
  86. {
  87. returnResult, err := json.Marshal(resp)
  88. if err != nil {
  89. utils.FileLog.Info(c.Ctx.Input.URI() + " Err:%s" + err.Error())
  90. }
  91. utils.FileLog.Info(c.Ctx.Input.URI()+" code: %s , return data: %s", code, string(returnResult))
  92. }
  93. c.OkDetailed(resp, "登录成功")
  94. }
  95. // WxAppLogin
  96. // @Title 微信登录接口
  97. // @Description 微信登录接口
  98. // @Param Code query string true "微信唯一编码code"
  99. // @Success 200 {object} wechat.WxLoginResp
  100. // @router /wxapp/login [get]
  101. func (c *WeChatCommon) WxAppLogin() {
  102. code := c.GetString("Code")
  103. if code == "" {
  104. c.FailWithMessage("参数错误", "Code 为空")
  105. return
  106. }
  107. wxInfo, err := weapp.Login(utils.WxAppId2, utils.WxAppSecret2, code)
  108. if err != nil {
  109. c.FailWithMessage("获取用户信息失败", "获取用户信息失败,Err:"+err.Error())
  110. return
  111. }
  112. if err = wxInfo.GetResponseError(); err != nil {
  113. c.FailWithMessage("获取用户信息失败", "获取用户信息失败,code:"+strconv.Itoa(wxInfo.ErrCode)+",msg:"+wxInfo.ErrMSG)
  114. return
  115. }
  116. item := &services.WxAccessToken{
  117. //AccessToken :wxInfo.
  118. //ExpiresIn :wxInfo.,
  119. //RefreshToken : wxInfo,
  120. Openid: wxInfo.OpenID,
  121. Unionid: wxInfo.UnionID,
  122. //Scope :wxInfo,
  123. Errcode: wxInfo.ErrCode,
  124. Errmsg: wxInfo.ErrMSG,
  125. }
  126. wxUserInfo := new(services.WxUserInfo)
  127. wxUserInfo.Unionid = wxInfo.UnionID
  128. wxUserInfo.Openid = wxInfo.OpenID
  129. wxUserInfo.Unionid = wxInfo.UnionID
  130. wxUserInfo.Errcode = wxInfo.ErrCode
  131. wxUserInfo.Errmsg = wxInfo.ErrMSG
  132. wxUserInfo.SessionKey = wxInfo.SessionKey
  133. fmt.Println("openId", wxInfo.OpenID)
  134. fmt.Println("unionId", wxInfo.UnionID)
  135. token, adminWx, err := services.WxLogin(utils.WxPlatform2, item, wxUserInfo)
  136. if err != nil {
  137. c.FailWithMessage("微信登录失败", "微信登录失败,err:"+err.Error())
  138. return
  139. }
  140. userInfo := adminResp.LoginResp{
  141. RealName: adminWx.RealName,
  142. AdminName: adminWx.AdminName,
  143. RoleName: adminWx.RoleName,
  144. RoleTypeCode: adminWx.RoleTypeCode,
  145. AdminId: adminWx.AdminId,
  146. DepartmentName: adminWx.DepartmentName,
  147. GroupName: adminWx.GroupName,
  148. }
  149. userInfo.AdminId = adminWx.AdminId
  150. var productName string
  151. productId := services.GetProductId(adminWx.RoleTypeCode)
  152. if productId == 1 {
  153. productName = utils.COMPANY_PRODUCT_FICC_NAME
  154. } else if productId == 2 {
  155. productName = utils.COMPANY_PRODUCT_RAI_NAME
  156. } else {
  157. productName = "admin"
  158. }
  159. userInfo.ProductName = productName
  160. userInfo.Authority = adminWx.Authority
  161. resp := wechat.WxLoginResp{
  162. AdminId: adminWx.AdminId,
  163. Code: 0,
  164. Authorization: token,
  165. Headimgurl: adminWx.Headimgurl,
  166. RealName: adminWx.RealName,
  167. UserInfo: userInfo,
  168. }
  169. //登录日志
  170. {
  171. returnResult, err := json.Marshal(resp)
  172. if err != nil {
  173. utils.FileLog.Info(c.Ctx.Input.URI() + " Err:%s" + err.Error())
  174. }
  175. utils.FileLog.Info(c.Ctx.Input.URI()+" code: %s , return data: %s", code, string(returnResult))
  176. }
  177. c.OkDetailed(resp, "登录成功")
  178. }
  179. // @Title 小程序获取用户信息
  180. // @Description 小程序获取用户信息接口(需要登录)
  181. // @Param request body models.WxGetUserInfoReq true "type json string"
  182. // @Success 200 {object} models.WxGetUserInfoResp
  183. // @router /getUserInfo [post]
  184. //func (c *WechatController) GetUserInfo() {
  185. // br := new(models.BaseResponse).Init()
  186. // defer func() {
  187. // c.Data["json"] = br
  188. // c.ServeJSON()
  189. // }()
  190. // var req models.WxGetUserInfoReq
  191. // err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  192. // if err != nil {
  193. // br.Msg = "参数解析异常!"
  194. // br.ErrMsg = "参数解析失败,Err:" + err.Error()
  195. // return
  196. // }
  197. // if req.RawData == "" || req.EncryptedData == "" || req.Signature == "" || req.Iv == "" {
  198. // br.Msg = "参数错误"
  199. // return
  200. // }
  201. // user := c.User
  202. // if user == nil {
  203. // br.Msg = "请登陆"
  204. // br.Ret = 408
  205. // return
  206. // }
  207. // userId := user.UserId
  208. // sessionKey := user.SessionKey
  209. // fmt.Println("sessionKey:", sessionKey)
  210. // fmt.Println(sessionKey, req.RawData, req.EncryptedData, req.Signature, req.Iv)
  211. // userInfo, err := weapp.DecryptUserInfo(sessionKey, req.RawData, req.EncryptedData, req.Signature, req.Iv)
  212. // fmt.Println("weapp.DecryptUserInfo ", err)
  213. //
  214. // if err != nil {
  215. // br.Msg = "解析用户信息失败"
  216. // br.ErrMsg = "解析用户信息失败,DecryptUserInfo Err:" + err.Error()
  217. // return
  218. // }
  219. // //修改用户微信信息
  220. // err = models.ModifyUserRecordByDetail(userInfo.OpenID, userInfo.UnionID, userInfo.Nickname, userInfo.Avatar, userInfo.City, userInfo.Province, userInfo.Country, userInfo.Gender, userId)
  221. // if err != nil {
  222. // br.Msg = "授权失败"
  223. // br.ErrMsg = "授权失败,修改用户信息失败:" + err.Error()
  224. // return
  225. // }
  226. // var token string
  227. // tokenItem, err := models.GetTokenByOpenId(userInfo.OpenID)
  228. // if err != nil && err.Error() != utils.ErrNoRow() {
  229. // br.Msg = "授权失败"
  230. // br.ErrMsg = "授权失败,获取token失败:" + err.Error()
  231. // return
  232. // }
  233. //
  234. // if tokenItem == nil || (err != nil && err.Error() == utils.ErrNoRow()) {
  235. // timeUnix := time.Now().Unix()
  236. // timeUnixStr := strconv.FormatInt(timeUnix, 10)
  237. // token = utils.MD5(userInfo.OpenID) + utils.MD5(timeUnixStr)
  238. // //新增session
  239. // {
  240. // session := new(models.CygxSession)
  241. // session.OpenId = userInfo.OpenID
  242. // session.UnionId = userInfo.UnionID
  243. // session.UserId = userId
  244. // session.CreatedTime = time.Now()
  245. // session.LastUpdatedTime = time.Now()
  246. // session.ExpireTime = time.Now().AddDate(0, 3, 0)
  247. // session.AccessToken = token
  248. // err = models.AddSession(session)
  249. // if err != nil {
  250. // br.Msg = "授权失败"
  251. // br.ErrMsg = "授权失败,新增用户session信息失败:" + err.Error()
  252. // return
  253. // }
  254. // }
  255. // } else {
  256. // token = tokenItem.AccessToken
  257. // }
  258. // resp := new(models.WxGetUserInfoResp)
  259. // resp.Authorization = token
  260. // br.Msg = "获取成功!"
  261. // br.Ret = 200
  262. // br.Success = true
  263. //}