142 lines
3.5 KiB
Go
142 lines
3.5 KiB
Go
package worker
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
AppId = "101827468"
|
|
AppKey = "0d2d856e48e0ebf6b98e0d0c879fe74d"
|
|
RedirectURI = "https://www.ljsea.top/qq_callback.php"
|
|
)
|
|
|
|
type PrivateInfo struct {
|
|
AccessToken string `json:"access_token"`
|
|
ExpiresIn string `json:"expires_in"`
|
|
RefreshToken string `json:"refresh_token"`
|
|
OpenId string `json:"openid"`
|
|
}
|
|
|
|
//func main() {
|
|
// http.HandleFunc("/toLogin", GetAuthCode)
|
|
// http.HandleFunc("/qqLogin", GetToken)
|
|
//
|
|
// fmt.Println("started...")
|
|
// err := http.ListenAndServe(":9090", nil)
|
|
// if err != nil {
|
|
// panic(err)
|
|
// }
|
|
//}
|
|
|
|
type GetCodeResponse struct {
|
|
}
|
|
type GetCodeRequest struct {
|
|
ResponseType string `json:"response_type"`
|
|
ClientID string `json:"client_id"`
|
|
RedirectURI string `json:"redirect_uri"`
|
|
State string `json:"state"`
|
|
Scope string `json:"scope,omitempty"`
|
|
Display string `json:"display,omitempty"`
|
|
}
|
|
|
|
type QQAccessTokenRequest struct {
|
|
GrantType string `json:"grant_type"`
|
|
ClientID string `json:"client_id"`
|
|
ClientSecret string `json:"client_secret"`
|
|
Code string `json:"code"`
|
|
RedirectURI string `json:"redirect_uri"`
|
|
Fmt string `json:"fmt,omitempty"`
|
|
NeedOpenID string `json:"need_openid,omitempty"`
|
|
}
|
|
|
|
type QQAccessTokenResponse struct {
|
|
AccessToken string `json:"access_token"`
|
|
ExpiresIn int `json:"expires_in"`
|
|
RefreshToken string `json:"refresh_token"`
|
|
}
|
|
|
|
type QQRefreshTokenRequest struct {
|
|
GrantType string `json:"grant_type"`
|
|
ClientID string `json:"client_id"`
|
|
ClientSecret string `json:"client_secret"`
|
|
RefreshToken string `json:"refresh_token"`
|
|
Fmt string `json:"fmt,omitempty"`
|
|
}
|
|
|
|
// 2. Get Access Token
|
|
func GetToken(w http.ResponseWriter, r *http.Request) {
|
|
code := r.FormValue("code")
|
|
params := url.Values{}
|
|
params.Add("grant_type", "authorization_code")
|
|
params.Add("client_id", AppId)
|
|
params.Add("client_secret", AppKey)
|
|
params.Add("code", code)
|
|
str := fmt.Sprintf("%s&redirect_uri=%s", params.Encode(), RedirectURI)
|
|
loginURL := fmt.Sprintf("%s?%s", "https://graph.qq.com/oauth2.0/token", str)
|
|
|
|
response, err := http.Get(loginURL)
|
|
if err != nil {
|
|
w.Write([]byte(err.Error()))
|
|
}
|
|
defer response.Body.Close()
|
|
|
|
bs, _ := ioutil.ReadAll(response.Body)
|
|
body := string(bs)
|
|
|
|
resultMap := convertToMap(body)
|
|
|
|
info := &PrivateInfo{}
|
|
info.AccessToken = resultMap["access_token"]
|
|
info.RefreshToken = resultMap["refresh_token"]
|
|
info.ExpiresIn = resultMap["expires_in"]
|
|
|
|
GetOpenId(info, w)
|
|
}
|
|
|
|
// 3. Get OpenId
|
|
func GetOpenId(info *PrivateInfo, w http.ResponseWriter) {
|
|
resp, err := http.Get(fmt.Sprintf("%s?access_token=%s", "https://graph.qq.com/oauth2.0/me", info.AccessToken))
|
|
if err != nil {
|
|
w.Write([]byte(err.Error()))
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
bs, _ := ioutil.ReadAll(resp.Body)
|
|
body := string(bs)
|
|
info.OpenId = body[45:77]
|
|
|
|
GetUserInfo(info, w)
|
|
}
|
|
|
|
// 4. Get User info
|
|
func GetUserInfo(info *PrivateInfo, w http.ResponseWriter) {
|
|
params := url.Values{}
|
|
params.Add("access_token", info.AccessToken)
|
|
params.Add("openid", info.OpenId)
|
|
params.Add("oauth_consumer_key", AppId)
|
|
|
|
uri := fmt.Sprintf("https://graph.qq.com/user/get_user_info?%s", params.Encode())
|
|
resp, err := http.Get(uri)
|
|
if err != nil {
|
|
w.Write([]byte(err.Error()))
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
bs, _ := ioutil.ReadAll(resp.Body)
|
|
w.Write(bs)
|
|
}
|
|
|
|
func convertToMap(str string) map[string]string {
|
|
var resultMap = make(map[string]string)
|
|
values := strings.Split(str, "&")
|
|
for _, value := range values {
|
|
vs := strings.Split(value, "=")
|
|
resultMap[vs[0]] = vs[1]
|
|
}
|
|
return resultMap
|
|
}
|