work.go 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  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/kf"
  7. "github.com/silenceper/wechat/v2/work/msgaudit"
  8. "github.com/silenceper/wechat/v2/work/oauth"
  9. )
  10. // Work 企业微信
  11. type Work struct {
  12. ctx *context.Context
  13. }
  14. //NewWork init work
  15. func NewWork(cfg *config.Config) *Work {
  16. defaultAkHandle := credential.NewWorkAccessToken(cfg.CorpID, cfg.CorpSecret, credential.CacheKeyWorkPrefix, cfg.Cache)
  17. ctx := &context.Context{
  18. Config: cfg,
  19. AccessTokenHandle: defaultAkHandle,
  20. }
  21. return &Work{ctx: ctx}
  22. }
  23. //GetContext get Context
  24. func (wk *Work) GetContext() *context.Context {
  25. return wk.ctx
  26. }
  27. //GetOauth get oauth
  28. func (wk *Work) GetOauth() *oauth.Oauth {
  29. return oauth.NewOauth(wk.ctx)
  30. }
  31. // GetMsgAudit get msgAudit
  32. func (wk *Work) GetMsgAudit() (*msgaudit.Client, error) {
  33. return msgaudit.NewClient(wk.ctx.Config)
  34. }
  35. // GetKF get kf
  36. func (wk *Work) GetKF() (*kf.Client, error) {
  37. return kf.NewClient(wk.ctx.Config)
  38. }