work.go 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  1. package work
  2. import (
  3. "github.com/silenceper/wechat/v2/credential"
  4. "github.com/silenceper/wechat/v2/work/addresslist"
  5. "github.com/silenceper/wechat/v2/work/appchat"
  6. "github.com/silenceper/wechat/v2/work/checkin"
  7. "github.com/silenceper/wechat/v2/work/config"
  8. "github.com/silenceper/wechat/v2/work/context"
  9. "github.com/silenceper/wechat/v2/work/externalcontact"
  10. "github.com/silenceper/wechat/v2/work/invoice"
  11. "github.com/silenceper/wechat/v2/work/jsapi"
  12. "github.com/silenceper/wechat/v2/work/kf"
  13. "github.com/silenceper/wechat/v2/work/material"
  14. "github.com/silenceper/wechat/v2/work/message"
  15. "github.com/silenceper/wechat/v2/work/msgaudit"
  16. "github.com/silenceper/wechat/v2/work/oauth"
  17. "github.com/silenceper/wechat/v2/work/robot"
  18. )
  19. // Work 企业微信
  20. type Work struct {
  21. ctx *context.Context
  22. }
  23. // NewWork init work
  24. func NewWork(cfg *config.Config) *Work {
  25. defaultAkHandle := credential.NewWorkAccessToken(cfg.CorpID, cfg.CorpSecret, cfg.AgentID, credential.CacheKeyWorkPrefix, cfg.Cache)
  26. ctx := &context.Context{
  27. Config: cfg,
  28. AccessTokenHandle: defaultAkHandle,
  29. }
  30. return &Work{ctx: ctx}
  31. }
  32. // GetContext get Context
  33. func (wk *Work) GetContext() *context.Context {
  34. return wk.ctx
  35. }
  36. // GetOauth get oauth
  37. func (wk *Work) GetOauth() *oauth.Oauth {
  38. return oauth.NewOauth(wk.ctx)
  39. }
  40. // GetMsgAudit get msgAudit
  41. func (wk *Work) GetMsgAudit() (*msgaudit.Client, error) {
  42. return msgaudit.NewClient(wk.ctx.Config)
  43. }
  44. // GetKF get kf
  45. func (wk *Work) GetKF() (*kf.Client, error) {
  46. return kf.NewClient(wk.ctx.Config)
  47. }
  48. // JsSdk get JsSdk
  49. func (wk *Work) JsSdk() *jsapi.Js {
  50. return jsapi.NewJs(wk.ctx)
  51. }
  52. // GetExternalContact get external_contact
  53. func (wk *Work) GetExternalContact() *externalcontact.Client {
  54. return externalcontact.NewClient(wk.ctx)
  55. }
  56. // GetAddressList get address_list
  57. func (wk *Work) GetAddressList() *addresslist.Client {
  58. return addresslist.NewClient(wk.ctx)
  59. }
  60. // GetMaterial get material
  61. func (wk *Work) GetMaterial() *material.Client {
  62. return material.NewClient(wk.ctx)
  63. }
  64. // GetRobot get robot
  65. func (wk *Work) GetRobot() *robot.Client {
  66. return robot.NewClient(wk.ctx)
  67. }
  68. // GetMessage 获取发送应用消息接口实例
  69. func (wk *Work) GetMessage() *message.Client {
  70. return message.NewClient(wk.ctx)
  71. }
  72. // GetAppChat 获取应用发送消息到群聊会话接口实例
  73. func (wk *Work) GetAppChat() *appchat.Client {
  74. return appchat.NewClient(wk.ctx)
  75. }
  76. // GetInvoice get invoice
  77. func (wk *Work) GetInvoice() *invoice.Client {
  78. return invoice.NewClient(wk.ctx)
  79. }
  80. // GetCheckin 获取打卡接口实例
  81. func (wk *Work) GetCheckin() *checkin.Client {
  82. return checkin.NewClient(wk.ctx)
  83. }