/
authUtils.go
149 lines (137 loc) · 4.16 KB
/
authUtils.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
package utils
import (
"net/http"
"fmt"
"io/ioutil"
"encoding/json"
"bytes"
"strconv"
)
type AuthUserInfo struct {
Openid string
Nickname string
Gender int
Avatar string
}
type wxAuthVerifyResult struct {
ErrCode int `json:"errcode"`
ErrMsg string `json:"errmsg"`
}
type wxUserInfo struct {
UnionId string `json:"unionid"`
Nickname string `json:"nickname"`
Gender int `json:"sex"`
Avatar string `json:"headimgurl"`
}
type wbAuthVerifyResult struct {
Uid int64 `json:"uid"`
AppKey string `json:"appkey"`
}
type wbUserInfo struct {
Id int64 `json:"id"`
Nickname string `json:"screen_name"`
Gender string `json:"gender"`
Avatar string `json:"avatar_large"`
}
type qqUserInfo struct {
Ret int `json:"ret"`
Nickname string `json:"nickname"`
Gender string `json:"gender"`
Avatar string `json:"figureurl_qq_1"`
}
func AuthWithWeiXin(wx_openid, wx_token string) (*AuthUserInfo, int) {
resp, err := http.Get(fmt.Sprintf("https://api.weixin.qq.com/sns/auth?access_token=%s&openid=%s&lang=zh_CN",
wx_token, wx_openid))
if err != nil {
return nil, ERROR_CODE_AUTH_WEIXIN_AUTH_FAIL
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return nil, ERROR_CODE_AUTH_WEIXIN_AUTH_FAIL
}
var verifyResult wxAuthVerifyResult
err = json.Unmarshal(body, &verifyResult)
if err != nil || verifyResult.ErrCode != 0 {
return nil, ERROR_CODE_AUTH_WEIXIN_AUTH_FAIL
}
resp, err = http.Get(fmt.Sprintf("https://api.weixin.qq.com/sns/userinfo?access_token=%s&openid=%s&lang=zh_CN",
wx_token, wx_openid))
if err != nil {
return nil, ERROR_CODE_AUTH_WEIXIN_AUTH_FAIL
}
defer resp.Body.Close()
body, err = ioutil.ReadAll(resp.Body)
if err != nil {
return nil, ERROR_CODE_AUTH_WEIXIN_AUTH_FAIL
}
var userInfo wxUserInfo
err = json.Unmarshal(body, &userInfo)
if err != nil {
return nil, ERROR_CODE_AUTH_WEIXIN_AUTH_FAIL
}
return &AuthUserInfo{Openid: userInfo.UnionId, Nickname: userInfo.Nickname, Gender: userInfo.Gender, Avatar: userInfo.Avatar}, 0
}
func AuthWithWeiBo(wb_token string) (*AuthUserInfo, int) {
resp, err := http.Post("https://api.weibo.com/oauth2/get_token_info",
"application/x-www-form-urlencoded", bytes.NewBuffer([]byte("access_token=" + wb_token)))
if err != nil {
return nil, ERROR_CODE_AUTH_WEIBO_AUTH_FAIL
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return nil, ERROR_CODE_AUTH_WEIBO_AUTH_FAIL
}
var verifyResult wbAuthVerifyResult
err = json.Unmarshal(body, &verifyResult)
if err != nil || verifyResult.Uid <= 0 {
return nil, ERROR_CODE_AUTH_WEIBO_AUTH_FAIL
}
resp, err = http.Get(fmt.Sprintf("https://api.weibo.com/2/users/show.json?access_token=%s&uid=%d",
wb_token, verifyResult.Uid))
if err != nil {
return nil, ERROR_CODE_AUTH_WEIBO_AUTH_FAIL
}
defer resp.Body.Close()
body, err = ioutil.ReadAll(resp.Body)
if err != nil {
return nil, ERROR_CODE_AUTH_WEIBO_AUTH_FAIL
}
var userInfo wbUserInfo
err = json.Unmarshal(body, &userInfo)
if err != nil {
return nil, ERROR_CODE_AUTH_WEIBO_AUTH_FAIL
}
gender := GENDER_UNKNOWN
if userInfo.Gender == "m" {
gender = GENDER_MALE
} else if userInfo.Gender == "f" {
gender = GENDER_FEMALE
}
return &AuthUserInfo{Openid: strconv.FormatInt(userInfo.Id, 10), Nickname: userInfo.Nickname, Gender: gender, Avatar: userInfo.Avatar}, 0
}
func AuthWithQQ(qq_openid, qq_token, qq_oauth_consumer_key string) (*AuthUserInfo, int) {
resp, err := http.Get(fmt.Sprintf("https://graph.qq.com/user/get_user_info?oauth_consumer_key=%s&access_token=%s&openid=%s&format=json",
qq_oauth_consumer_key, qq_token, qq_openid))
if err != nil {
return nil, ERROR_CODE_AUTH_QQ_AUTH_FAIL
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return nil, ERROR_CODE_AUTH_QQ_AUTH_FAIL
}
var userInfo qqUserInfo
err = json.Unmarshal(body, &userInfo)
if err != nil || userInfo.Ret != 0 {
return nil, ERROR_CODE_AUTH_QQ_AUTH_FAIL
}
gender := GENDER_UNKNOWN
if userInfo.Gender == "男" {
gender = GENDER_MALE
} else if userInfo.Gender == "女" {
gender = GENDER_FEMALE
}
return &AuthUserInfo{Openid: qq_openid, Nickname: userInfo.Nickname, Gender: gender, Avatar: userInfo.Avatar}, 0
}