client.go 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247
  1. package wechat
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "net/url"
  6. "time"
  7. "git.chuangxin1.com/cx/myth"
  8. )
  9. func key(appid string) string {
  10. return "wechat:client:" + appid
  11. }
  12. // NewClient new client
  13. func NewClient(appID, appSecret, token, encodingAESKey string) *Client {
  14. key := key(appID)
  15. if v, ok := cache.Load(key); ok {
  16. return v.(*Client)
  17. }
  18. c := &Client{AppID: appID, AppSecret: appSecret, Token: token, EncodingAESKey: encodingAESKey}
  19. cache.Store(key, c)
  20. return c
  21. }
  22. // getToken get token
  23. func (wc *Client) getToken() (token string, err error) {
  24. now := time.Now().Unix()
  25. if wc.LastTokenTime > 0 {
  26. if now-wc.LastTokenTime < TokenExpires {
  27. token = wc.AccessToken
  28. return
  29. }
  30. }
  31. uri := BaseURL + "/cgi-bin/token?"
  32. args := url.Values{}
  33. args.Add("grant_type", "client_credential")
  34. args.Add("appid", wc.AppID)
  35. args.Add("secret", wc.AppSecret)
  36. uri += args.Encode()
  37. var res Response
  38. if res, err = getJSON(uri); err == nil {
  39. wc.LastTokenTime = now
  40. wc.AccessToken = res.AccessToken
  41. token = wc.AccessToken
  42. key := key(wc.AppID)
  43. cache.Store(key, wc)
  44. }
  45. return
  46. }
  47. // GetCodeURL get code 授权获取 OpenID URL
  48. // /connect/oauth2/authorize?appid=APPID&redirect_uri=REDIRECT_URI&response_type=code&scope=SCOPE&state=STATE#wechat_redirect
  49. // 转向到
  50. // redirect_uri/?code=CODE&state=STATE
  51. func (wc Client) GetCodeURL(redirectURL, state string) (uri string) {
  52. uri = OpenURL + "/connect/oauth2/authorize?"
  53. args := url.Values{}
  54. args.Add("appid", wc.AppID)
  55. args.Add("redirect_uri", redirectURL)
  56. args.Add("response_type", "code")
  57. args.Add("scope", "snsapi_base")
  58. args.Add("state", state)
  59. uri += args.Encode()
  60. uri += "#wechat_redirect"
  61. return
  62. }
  63. // GetOpenID 获取 OpenID
  64. // /sns/oauth2/access_token?appid=APPID&secret=SECRET&code=CODE&grant_type=authorization_code
  65. func (wc Client) GetOpenID(code string) (res Response, err error) {
  66. uri := BaseURL + "/sns/oauth2/access_token?"
  67. if wc.AccessToken, err = wc.getToken(); err != nil {
  68. return
  69. }
  70. args := url.Values{}
  71. args.Add("access_token", wc.AccessToken)
  72. args.Add("appid", wc.AppID)
  73. args.Add("secret", wc.AppSecret)
  74. args.Add("code", code)
  75. args.Add("grant_type", "authorization_code")
  76. uri += args.Encode()
  77. res, err = getJSON(uri)
  78. return
  79. }
  80. // GetUserInfo user info
  81. func (wc Client) GetUserInfo(openid string) (res UserInfo, err error) {
  82. uri := BaseURL + "/cgi-bin/user/info?"
  83. if wc.AccessToken, err = wc.getToken(); err != nil {
  84. return
  85. }
  86. args := url.Values{}
  87. args.Add("access_token", wc.AccessToken)
  88. args.Add("openid", openid)
  89. args.Add("lang", "zh_CN")
  90. uri += args.Encode()
  91. var body []byte
  92. body, err = getBody(uri)
  93. if err == nil {
  94. err = json.Unmarshal(body, &res)
  95. }
  96. return
  97. }
  98. // GetUserList user list
  99. func (wc Client) GetUserList(nextOpenID string) (res UserList, err error) {
  100. uri := BaseURL + "/cgi-bin/user/get?"
  101. if wc.AccessToken, err = wc.getToken(); err != nil {
  102. return
  103. }
  104. args := url.Values{}
  105. args.Add("access_token", wc.AccessToken)
  106. args.Add("next_openid", nextOpenID)
  107. uri += args.Encode()
  108. var body []byte
  109. body, err = getBody(uri)
  110. if err == nil {
  111. err = json.Unmarshal(body, &res)
  112. }
  113. return
  114. }
  115. // GetMaterial 永久资料
  116. func (wc Client) GetMaterial(mtype string, offset, count int) (res Material, err error) {
  117. uri := BaseURL + "/cgi-bin/material/batchget_material?"
  118. if wc.AccessToken, err = wc.getToken(); err != nil {
  119. return
  120. }
  121. args := url.Values{}
  122. data := make(map[string]interface{})
  123. args.Add("access_token", wc.AccessToken)
  124. data["type"] = mtype
  125. data["offset"] = offset
  126. data["count"] = count
  127. uri += args.Encode()
  128. var body []byte
  129. params, err := json.Marshal(data)
  130. body, err = postJSONBody(uri, params)
  131. if err == nil {
  132. err = json.Unmarshal(body, &res)
  133. }
  134. return
  135. }
  136. func (wc *Client) getTicket(now int64) (ticket string, err error) {
  137. var res Response
  138. if wc.LastTicketTime > 0 {
  139. if now-wc.LastTokenTime < TokenExpires {
  140. ticket = wc.Ticket
  141. return
  142. }
  143. }
  144. args := url.Values{}
  145. args.Add("access_token", wc.AccessToken)
  146. args.Add("type", "jsapi")
  147. uri := BaseURL + "/cgi-bin/ticket/getticket?"
  148. uri += args.Encode()
  149. if res, err = getJSON(uri); err == nil {
  150. fmt.Println(res)
  151. wc.LastTicketTime = now
  152. wc.Ticket = res.Ticket
  153. key := key(wc.AppID)
  154. cache.Store(key, wc)
  155. ticket = res.Ticket
  156. }
  157. return
  158. }
  159. // GetSignPackage JS 签名
  160. // uri 当前 URL
  161. // nonceStr 随机字符串
  162. func (wc Client) GetSignPackage(uri, nonceStr string) (sign SignPackage, err error) {
  163. if wc.AccessToken, err = wc.getToken(); err != nil {
  164. return
  165. }
  166. var (
  167. s string
  168. unix int64
  169. )
  170. unix = time.Now().Unix()
  171. if wc.Ticket, err = wc.getTicket(unix); err != nil {
  172. return
  173. }
  174. //fmt.Println(wc.Ticket)
  175. s = `jsapi_ticket=` + wc.Ticket
  176. s += `&noncestr=` + nonceStr
  177. s += `&timestamp=` + fmt.Sprintf("%d", unix)
  178. s += `&url=` + uri
  179. //fmt.Println(s)
  180. sign.AppID = wc.AppID
  181. sign.NonceStr = nonceStr
  182. sign.Signature = myth.SHA1(s)
  183. sign.Timestamp = unix
  184. sign.URL = uri
  185. return
  186. }
  187. // SendTemplateMessage send template message
  188. // POST /cgi-bin/message/template/send?access_token=ACCESS_TOKEN
  189. func (wc Client) SendTemplateMessage(template TemplateMessage) (res Response, err error) {
  190. uri := BaseURL + "/cgi-bin/message/template/send?"
  191. if wc.AccessToken, err = wc.getToken(); err != nil {
  192. return
  193. }
  194. args := url.Values{}
  195. args.Add("access_token", wc.AccessToken)
  196. uri += args.Encode()
  197. data, err := json.Marshal(template)
  198. if err != nil {
  199. return
  200. }
  201. res, err = postJSON(uri, data)
  202. return
  203. }