liulei/baseapi
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

- refactor weixin, remove weixinsns.

Browse Source
This commit is contained in:
gazebo
2018-12-27 16:14:16 +08:00
parent 189a6f8b98
commit babc357d23
4 changed files with 139 additions and 243 deletions
Download Patch File Download Diff File Expand all files Collapse all files

133
platformapi/weixinapi/weixinapi.go
View File

@@ -1,6 +1,8 @@
package weixinapi
import (
"crypto/aes"
"crypto/cipher"
"net/http"
"strings"
"sync"
@@ -10,7 +12,7 @@ import (
)
const (
prodURL = "https://api.weixin.qq.com/cgi-bin"
prodURL = "https://api.weixin.qq.com"
)
const (
@@ -41,6 +43,32 @@ type API struct {
locker sync.RWMutex
}
type SNSTokenInfo struct {
AccessToken string `json:"access_token"`
ExpiresIn int `json:"expires_in"`
RefreshToken string `json:"refresh_token"`
OpenID string `json:"openid"`
Scope string `json:"scope"`
}
type SNSUserInfo struct {
OpenID string `json:"openid"`
NickName string `json:"nickname"`
Sex int `json:"sex"`
Province string `json:"province"`
City string `json:"city"`
Country string `json:"country"`
HeadImgURL string `json:"headimgurl"`
Privilege interface{} `json:"privilege"`
UnionID string `json:"unionid"`
}
type SessionInfo struct {
OpenID string `json:"openid"`
SessionKey string `json:"session_key"`
UnionID string `json:"unionid"`
}
func New(appID, secret string, config ...*platformapi.APIConfig) *API {
curConfig := platformapi.DefAPIConfig
if len(config) > 0 {
@@ -79,6 +107,10 @@ func (a *API) GetToken() string {
return a.token
}
func isSNSAction(action string) bool {
return strings.Index(action, "sns/") == 0
}
func (a *API) AccessAPI(action string, params map[string]interface{}, body string) (retVal map[string]interface{}, err error) {
if params != nil && body != "" {
panic("params and body can not all non-empty")
@@ -87,16 +119,18 @@ func (a *API) AccessAPI(action string, params map[string]interface{}, body strin
for k, v := range params {
params2[k] = v
}
if action == actionGetToken {
if params2["grant_type"] != nil {
params2["appid"] = a.appID
params2["secret"] = a.secret
} else {
} else if !isSNSAction(action) {
accessToken := a.GetToken()
if accessToken == "" {
panic("token is empty!")
}
params2["access_token"] = accessToken
}
fullURL := utils.GenerateGetURL(prodURL, action, params2)
// baseapi.SugarLogger.Debug(fullURL)
@@ -143,7 +177,7 @@ func (a *API) AccessAPI(action string, params map[string]interface{}, body strin
}
func (a *API) RefreshToken() (tokenInfo *TokenInfo, err error) {
result, err := a.AccessAPI(actionGetToken, utils.Params2Map("grant_type", "client_credential"), "")
result, err := a.AccessAPI("cgi-bin/token", utils.Params2Map("grant_type", "client_credential"), "")
if err != nil {
return nil, err
}
@@ -169,6 +203,95 @@ func (a *API) MessageTemplateSend(userOpneID, templateID, downloadURL string, mi
if miniProgram != nil {
bodyJson["miniprogram"] = miniProgram
}
_, err = a.AccessAPI("message/template/send", nil, string(utils.MustMarshal(bodyJson)))
_, err = a.AccessAPI("cgi-bin/message/template/send", nil, string(utils.MustMarshal(bodyJson)))
return err
}
func mapData2SNSToken(result map[string]interface{}) *SNSTokenInfo {
return &SNSTokenInfo{
AccessToken: utils.Interface2String(result["access_token"]),
ExpiresIn: int(utils.MustInterface2Int64(result["expires_in"])),
RefreshToken: utils.Interface2String(result["refresh_token"]),
OpenID: utils.Interface2String(result["openid"]),
Scope: utils.Interface2String(result["scope"]),
}
}
func (a *API) SNSGetToken(code string) (tokenInfo *SNSTokenInfo, err error) {
result, err := a.AccessAPI("sns/oauth2/access_token", utils.Params2Map("grant_type", "authorization_code", "code", code), "")
if err != nil {
return nil, err
}
return mapData2SNSToken(result), nil
}
func (a *API) SNSRefreshToken(refreshToken string) (tokenInfo *SNSTokenInfo, err error) {
result, err := a.AccessAPI("sns/oauth2/refresh_token", utils.Params2Map("grant_type", "refresh_token", "refresh_token", refreshToken), "")
if err != nil {
return nil, err
}
return mapData2SNSToken(result), nil
}
func (a *API) SNSGetUserInfo(accessToken, openid string) (*SNSUserInfo, error) {
result, err := a.AccessAPI("sns/userinfo", map[string]interface{}{
"access_token": accessToken,
"openid": openid,
}, "")
if err == nil {
retVal := &SNSUserInfo{
OpenID: utils.Interface2String(result["openid"]),
NickName: utils.Interface2String(result["nickname"]),
Sex: int(utils.MustInterface2Int64(result["sex"])),
Province: utils.Interface2String(result["province"]),
City: utils.Interface2String(result["city"]),
Country: utils.Interface2String(result["country"]),
HeadImgURL: utils.Interface2String(result["headimgurl"]),
Privilege: result["privilege"],
UnionID: utils.Interface2String(result["unionid"]),
}
return retVal, nil
}
return nil, err
}
func (a *API) SNSIsOpenIDValid(accessToken, openid string) (bool, error) {
_, err := a.AccessAPI("sns/auth", map[string]interface{}{
"access_token": accessToken,
"openid": openid,
}, "")
if err == nil {
return true, nil
}
return false, err
}
func (a *API) SNSCode2Session(code string) (sessionInfo *SessionInfo, err error) {
result, err := a.AccessAPI("sns/jscode2session", map[string]interface{}{
"js_code": code,
"grant_type": "authorization_code",
}, "")
if err == nil {
return &SessionInfo{
OpenID: utils.Interface2String(result["openid"]),
SessionKey: utils.Interface2String(result["session_key"]),
UnionID: utils.Interface2String(result["unionid"]),
}, nil
}
return nil, err
}
func (a *API) SNSDecodeMiniProgramData(encryptedData, sessionKey, iv string) (decryptedData []byte, err error) {
decodedDataList, err := utils.Base64DecodeMultiString(encryptedData, sessionKey, iv)
if err != nil {
return nil, err
}
c, err := aes.NewCipher(decodedDataList[1])
if err != nil {
return nil, err
}
cfbdec := cipher.NewCFBDecrypter(c, decodedDataList[2])
decryptedData = make([]byte, len(decodedDataList[0]))
cfbdec.XORKeyStream(decryptedData, decodedDataList[0])
return decryptedData, nil
}