81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
package dao
|
||
|
||
import (
|
||
"context"
|
||
"net/url"
|
||
|
||
"go-common/app/admin/ep/marthe/model"
|
||
)
|
||
|
||
const (
|
||
_qyWechatURL = "https://qyapi.weixin.qq.com"
|
||
_corpID = "wx0833ac9926284fa5" // 企业微信：Bilibili的企业ID
|
||
_departmentID = "12" // 公司统一用部门ID
|
||
_corpsecret = "WveODxk3xpT9box48wcxkmArx3mu6d4vJHdJkNy_iTk"
|
||
|
||
_getToken = "/cgi-bin/gettoken"
|
||
_userList = "/cgi-bin/user/list"
|
||
)
|
||
|
||
// WechatAccessToken query access token with the specified secret 企业微信api获取公司token
|
||
func (d *Dao) WechatAccessToken(c context.Context) (token string, err error) {
|
||
var (
|
||
u string
|
||
params = url.Values{}
|
||
res struct {
|
||
ErrCode int `json:"errcode"`
|
||
ErrMsg string `json:"errmsg"`
|
||
AccessToken string `json:"access_token"`
|
||
ExpiresIn int32 `json:"expires_in"`
|
||
}
|
||
)
|
||
u = _qyWechatURL + _getToken
|
||
params.Set("corpid", _corpID)
|
||
params.Set("corpsecret", _corpsecret)
|
||
|
||
if err = d.httpClient.Get(c, u, "", params, &res); err != nil {
|
||
return
|
||
}
|
||
|
||
if res.ErrCode != 0 {
|
||
return
|
||
}
|
||
|
||
token = res.AccessToken
|
||
return
|
||
}
|
||
|
||
// WechatContacts Wechat Contacts 获取用户信息列表
|
||
func (d *Dao) WechatContacts(c context.Context) (contacts []*model.WechatContact, err error) {
|
||
var (
|
||
token string
|
||
u string
|
||
params = url.Values{}
|
||
res struct {
|
||
ErrCode int `json:"errcode"`
|
||
ErrMsg string `json:"errmsg"`
|
||
UserList []*model.WechatContact `json:"userlist"`
|
||
}
|
||
)
|
||
//get token
|
||
if token, err = d.WechatAccessToken(c); err != nil {
|
||
return
|
||
}
|
||
|
||
u = _qyWechatURL + _userList
|
||
params.Set("access_token", token)
|
||
params.Set("department_id", _departmentID)
|
||
params.Set("fetch_child", "1")
|
||
|
||
if err = d.httpClient.Get(c, u, "", params, &res); err != nil {
|
||
return
|
||
}
|
||
|
||
if res.ErrCode != 0 {
|
||
return
|
||
}
|
||
|
||
contacts = res.UserList
|
||
return
|
||
}
|