You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
1.9 KiB
76 lines
1.9 KiB
package wechat
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
)
|
|
|
|
var Wechat = &wechat{}
|
|
|
|
type wechat struct {
|
|
appid string
|
|
secret string
|
|
}
|
|
|
|
type AuthAccessToken struct {
|
|
AccessToken string `json:"access_token"`
|
|
ExpiresIn int64 `json:"expires_in"`
|
|
Openid string `json:"openid"`
|
|
RefreshToken string `json:"refresh_token"`
|
|
Scope string `json:"scope"`
|
|
}
|
|
|
|
// InitWechat @Title 初始化
|
|
func InitWechat(appid, secret string) *wechat {
|
|
Wechat.appid = appid
|
|
Wechat.secret = secret
|
|
return Wechat
|
|
}
|
|
|
|
// GetOpenId @Title 获取openId
|
|
func (w *wechat) GetOpenId(code string) string {
|
|
accessToken, err := w.GetAccessToken(code)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
return accessToken.Openid
|
|
}
|
|
|
|
// GetAccessToken @Title 获取微信验证结构体
|
|
func (w *wechat) GetAccessToken(code string) (*AuthAccessToken, error) {
|
|
res, err := request(get, fmt.Sprintf("https://api.weixin.qq.com/sns/oauth2/access_token?appid=%v&secret=%v&code=%v&grant_type=authorization_code", w.appid, w.secret, code), "")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result := AuthAccessToken{}
|
|
if err := json.Unmarshal(res, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|
|
|
|
type UserInfo struct {
|
|
City string `json:"city"`
|
|
Country string `json:"country"`
|
|
Headimgurl string `json:"headimgurl"`
|
|
Nickname string `json:"nickname"`
|
|
Openid string `json:"openid"`
|
|
Privilege []string `json:"privilege"`
|
|
Province string `json:"province"`
|
|
Sex uint `json:"sex"`
|
|
Unionid string `json:"unionid"`
|
|
}
|
|
|
|
// GetUserInfo @Title 获取微信登录会员信息
|
|
func (w *wechat) GetUserInfo(accessToken, openId string) (*UserInfo, error) {
|
|
res, err := request(get, fmt.Sprintf("https://api.weixin.qq.com/sns/userinfo?access_token=%s&openid=%s&lang=zh_CN", accessToken, openId), "")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result := UserInfo{}
|
|
if err := json.Unmarshal(res, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|