work.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. package work
  2. import (
  3. "github.com/silenceper/wechat/v2/credential"
  4. "github.com/silenceper/wechat/v2/work/config"
  5. "github.com/silenceper/wechat/v2/work/context"
  6. "github.com/silenceper/wechat/v2/work/externalcontact"
  7. "github.com/silenceper/wechat/v2/work/kf"
  8. "github.com/silenceper/wechat/v2/work/msgaudit"
  9. "github.com/silenceper/wechat/v2/work/oauth"
  10. )
  11. // Work 企业微信
  12. type Work struct {
  13. ctx *context.Context
  14. }
  15. // NewWork init work
  16. func NewWork(cfg *config.Config) *Work {
  17. defaultAkHandle := credential.NewWorkAccessToken(cfg.CorpID, cfg.CorpSecret, credential.CacheKeyWorkPrefix, cfg.Cache)
  18. ctx := &context.Context{
  19. Config: cfg,
  20. AccessTokenHandle: defaultAkHandle,
  21. }
  22. return &Work{ctx: ctx}
  23. }
  24. // GetContext get Context
  25. func (wk *Work) GetContext() *context.Context {
  26. return wk.ctx
  27. }
  28. // GetOauth get oauth
  29. func (wk *Work) GetOauth() *oauth.Oauth {
  30. return oauth.NewOauth(wk.ctx)
  31. }
  32. // GetMsgAudit get msgAudit
  33. func (wk *Work) GetMsgAudit() (*msgaudit.Client, error) {
  34. return msgaudit.NewClient(wk.ctx.Config)
  35. }
  36. // GetKF get kf
  37. func (wk *Work) GetKF() (*kf.Client, error) {
  38. return kf.NewClient(wk.ctx.Config)
  39. }
  40. // GetExternalContact get external_contact
  41. func (wk *Work) GetExternalContact() *externalcontact.Client {
  42. return externalcontact.NewClient(wk.ctx)
  43. }