mirror of
https://github.com/silenceper/wechat.git
synced 2026-02-04 12:52:27 +08:00
71c8ab58fb
* feat: enhance WorkAccessToken to include AgentID for improved token management - Added AgentID field to WorkAccessToken struct. - Updated NewWorkAccessToken function to accept AgentID as a parameter. - Modified access token cache key to incorporate AgentID, ensuring unique cache entries per agent. This change improves the handling of access tokens in a multi-agent environment. * refactor: enhance WorkAccessToken to improve cache key handling - Updated the AgentID field in WorkAccessToken struct to clarify its optional nature for distinguishing applications. - Modified the access token cache key construction to support both new and legacy formats based on the presence of AgentID. - Added comments for better understanding of the cache key logic and its compatibility with historical versions. This change improves the flexibility and clarity of access token management in multi-agent scenarios. * feat: enhance WorkAccessToken with new constructor for AgentID support - Introduced NewWorkAccessTokenWithAgentID function to maintain backward compatibility while allowing for AgentID usage. - Updated NewWorkAccessToken to call the new constructor, ensuring seamless integration. - Improved error handling in GetAccessTokenContext by checking for cache availability and handling potential errors during cache operations. This change enhances the flexibility of access token management, particularly in multi-agent scenarios, while ensuring compatibility with existing implementations.
94 lines
2.5 KiB
Go
94 lines
2.5 KiB
Go
package work
|
|
|
|
import (
|
|
"github.com/silenceper/wechat/v2/credential"
|
|
"github.com/silenceper/wechat/v2/work/addresslist"
|
|
"github.com/silenceper/wechat/v2/work/appchat"
|
|
"github.com/silenceper/wechat/v2/work/checkin"
|
|
"github.com/silenceper/wechat/v2/work/config"
|
|
"github.com/silenceper/wechat/v2/work/context"
|
|
"github.com/silenceper/wechat/v2/work/externalcontact"
|
|
"github.com/silenceper/wechat/v2/work/invoice"
|
|
"github.com/silenceper/wechat/v2/work/kf"
|
|
"github.com/silenceper/wechat/v2/work/material"
|
|
"github.com/silenceper/wechat/v2/work/message"
|
|
"github.com/silenceper/wechat/v2/work/msgaudit"
|
|
"github.com/silenceper/wechat/v2/work/oauth"
|
|
"github.com/silenceper/wechat/v2/work/robot"
|
|
)
|
|
|
|
// Work 企业微信
|
|
type Work struct {
|
|
ctx *context.Context
|
|
}
|
|
|
|
// NewWork init work
|
|
func NewWork(cfg *config.Config) *Work {
|
|
defaultAkHandle := credential.NewWorkAccessToken(cfg.CorpID, cfg.CorpSecret, cfg.AgentID, credential.CacheKeyWorkPrefix, cfg.Cache)
|
|
ctx := &context.Context{
|
|
Config: cfg,
|
|
AccessTokenHandle: defaultAkHandle,
|
|
}
|
|
return &Work{ctx: ctx}
|
|
}
|
|
|
|
// GetContext get Context
|
|
func (wk *Work) GetContext() *context.Context {
|
|
return wk.ctx
|
|
}
|
|
|
|
// GetOauth get oauth
|
|
func (wk *Work) GetOauth() *oauth.Oauth {
|
|
return oauth.NewOauth(wk.ctx)
|
|
}
|
|
|
|
// GetMsgAudit get msgAudit
|
|
func (wk *Work) GetMsgAudit() (*msgaudit.Client, error) {
|
|
return msgaudit.NewClient(wk.ctx.Config)
|
|
}
|
|
|
|
// GetKF get kf
|
|
func (wk *Work) GetKF() (*kf.Client, error) {
|
|
return kf.NewClient(wk.ctx.Config)
|
|
}
|
|
|
|
// GetExternalContact get external_contact
|
|
func (wk *Work) GetExternalContact() *externalcontact.Client {
|
|
return externalcontact.NewClient(wk.ctx)
|
|
}
|
|
|
|
// GetAddressList get address_list
|
|
func (wk *Work) GetAddressList() *addresslist.Client {
|
|
return addresslist.NewClient(wk.ctx)
|
|
}
|
|
|
|
// GetMaterial get material
|
|
func (wk *Work) GetMaterial() *material.Client {
|
|
return material.NewClient(wk.ctx)
|
|
}
|
|
|
|
// GetRobot get robot
|
|
func (wk *Work) GetRobot() *robot.Client {
|
|
return robot.NewClient(wk.ctx)
|
|
}
|
|
|
|
// GetMessage 获取发送应用消息接口实例
|
|
func (wk *Work) GetMessage() *message.Client {
|
|
return message.NewClient(wk.ctx)
|
|
}
|
|
|
|
// GetAppChat 获取应用发送消息到群聊会话接口实例
|
|
func (wk *Work) GetAppChat() *appchat.Client {
|
|
return appchat.NewClient(wk.ctx)
|
|
}
|
|
|
|
// GetInvoice get invoice
|
|
func (wk *Work) GetInvoice() *invoice.Client {
|
|
return invoice.NewClient(wk.ctx)
|
|
}
|
|
|
|
// GetCheckin 获取打卡接口实例
|
|
func (wk *Work) GetCheckin() *checkin.Client {
|
|
return checkin.NewClient(wk.ctx)
|
|
}
|