103 lines
3.3 KiB
Go
103 lines
3.3 KiB
Go
package weixinapi
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"git.rosy.net.cn/baseapi/utils"
|
|
)
|
|
|
|
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) SNSRetrieveToken(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(jsCode string) (sessionInfo *SessionInfo, err error) {
|
|
result, err := a.AccessAPI("sns/jscode2session", map[string]interface{}{
|
|
"js_code": jsCode,
|
|
"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
|
|
}
|
|
decryptedData, err = utils.AESCBCDecpryt(decodedDataList[0], decodedDataList[1], decodedDataList[2])
|
|
return decryptedData, err
|
|
}
|
|
|
|
func PKCSUnPadding(origData []byte) []byte {
|
|
length := len(origData)
|
|
unpadding := int(origData[length-1])
|
|
return origData[:(length - unpadding)]
|
|
}
|
|
|
|
func PKCSPadding(ciphertext []byte, blockSize int) []byte {
|
|
padding := blockSize - len(ciphertext)%blockSize
|
|
padtext := bytes.Repeat([]byte{byte(padding)}, padding)
|
|
return append(ciphertext, padtext...)
|
|
}
|