forked from golang/hotime
77 lines
2.0 KiB
Go
77 lines
2.0 KiB
Go
|
package wechat
|
||
|
|
||
|
import (
|
||
|
"os"
|
||
|
|
||
|
"github.com/silenceper/wechat/v2/cache"
|
||
|
"github.com/silenceper/wechat/v2/miniprogram"
|
||
|
miniConfig "github.com/silenceper/wechat/v2/miniprogram/config"
|
||
|
"github.com/silenceper/wechat/v2/officialaccount"
|
||
|
offConfig "github.com/silenceper/wechat/v2/officialaccount/config"
|
||
|
"github.com/silenceper/wechat/v2/openplatform"
|
||
|
openConfig "github.com/silenceper/wechat/v2/openplatform/config"
|
||
|
"github.com/silenceper/wechat/v2/pay"
|
||
|
payConfig "github.com/silenceper/wechat/v2/pay/config"
|
||
|
"github.com/silenceper/wechat/v2/work"
|
||
|
workConfig "github.com/silenceper/wechat/v2/work/config"
|
||
|
log "github.com/sirupsen/logrus"
|
||
|
)
|
||
|
|
||
|
func init() {
|
||
|
// Log as JSON instead of the default ASCII formatter.
|
||
|
log.SetFormatter(&log.TextFormatter{})
|
||
|
|
||
|
// Output to stdout instead of the default stderr
|
||
|
// Can be any io.Writer, see below for File example
|
||
|
log.SetOutput(os.Stdout)
|
||
|
|
||
|
// Only log the warning severity or above.
|
||
|
log.SetLevel(log.DebugLevel)
|
||
|
}
|
||
|
|
||
|
// Wechat struct
|
||
|
type Wechat struct {
|
||
|
cache cache.Cache
|
||
|
}
|
||
|
|
||
|
// NewWechat init
|
||
|
func NewWechat() *Wechat {
|
||
|
return &Wechat{}
|
||
|
}
|
||
|
|
||
|
// SetCache 设置cache
|
||
|
func (wc *Wechat) SetCache(cahce cache.Cache) {
|
||
|
wc.cache = cahce
|
||
|
}
|
||
|
|
||
|
// GetOfficialAccount 获取微信公众号实例
|
||
|
func (wc *Wechat) GetOfficialAccount(cfg *offConfig.Config) *officialaccount.OfficialAccount {
|
||
|
if cfg.Cache == nil {
|
||
|
cfg.Cache = wc.cache
|
||
|
}
|
||
|
return officialaccount.NewOfficialAccount(cfg)
|
||
|
}
|
||
|
|
||
|
// GetMiniProgram 获取小程序的实例
|
||
|
func (wc *Wechat) GetMiniProgram(cfg *miniConfig.Config) *miniprogram.MiniProgram {
|
||
|
if cfg.Cache == nil {
|
||
|
cfg.Cache = wc.cache
|
||
|
}
|
||
|
return miniprogram.NewMiniProgram(cfg)
|
||
|
}
|
||
|
|
||
|
// GetPay 获取微信支付的实例
|
||
|
func (wc *Wechat) GetPay(cfg *payConfig.Config) *pay.Pay {
|
||
|
return pay.NewPay(cfg)
|
||
|
}
|
||
|
|
||
|
// GetOpenPlatform 获取微信开放平台的实例
|
||
|
func (wc *Wechat) GetOpenPlatform(cfg *openConfig.Config) *openplatform.OpenPlatform {
|
||
|
return openplatform.NewOpenPlatform(cfg)
|
||
|
}
|
||
|
|
||
|
// GetWork 获取企业微信的实例
|
||
|
func (wc *Wechat) GetWork(cfg *workConfig.Config) *work.Work {
|
||
|
return work.NewWork(cfg)
|
||
|
}
|