-
-
Notifications
You must be signed in to change notification settings - Fork 975
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
17fbbd3
commit fc438ab
Showing
6 changed files
with
289 additions
and
98 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,209 @@ | ||
| package openwechat | ||
|
|
||
| // BotLogin 定义了一个Login的接口 | ||
| type BotLogin interface { | ||
| Login(bot *Bot) error | ||
| } | ||
|
|
||
| // SacnLogin 扫码登录 | ||
| type SacnLogin struct{} | ||
|
|
||
| // Login 实现了 BotLogin 接口 | ||
| func (s *SacnLogin) Login(bot *Bot) error { | ||
| uuid, err := bot.Caller.GetLoginUUID() | ||
| if err != nil { | ||
| return err | ||
| } | ||
| return s.checkLogin(bot, uuid) | ||
| } | ||
|
|
||
| // checkLogin 该方法会一直阻塞，直到用户扫码登录，或者二维码过期 | ||
| func (s *SacnLogin) checkLogin(bot *Bot, uuid string) error { | ||
| bot.uuid = uuid | ||
| // 二维码获取回调 | ||
| if bot.UUIDCallback != nil { | ||
| bot.UUIDCallback(uuid) | ||
| } | ||
| for { | ||
| // 长轮询检查是否扫码登录 | ||
| resp, err := bot.Caller.CheckLogin(uuid, "0") | ||
| if err != nil { | ||
| return err | ||
| } | ||
| switch resp.Code { | ||
| case StatusSuccess: | ||
| // 判断是否有登录回调，如果有执行它 | ||
| if err = bot.HandleLogin(resp.Raw); err != nil { | ||
| return err | ||
| } | ||
| if bot.LoginCallBack != nil { | ||
| bot.LoginCallBack(resp.Raw) | ||
| } | ||
| return nil | ||
| case StatusScanned: | ||
| // 执行扫码回调 | ||
| if bot.ScanCallBack != nil { | ||
| bot.ScanCallBack(resp.Raw) | ||
| } | ||
| case StatusTimeout: | ||
| return ErrLoginTimeout | ||
| case StatusWait: | ||
| continue | ||
| } | ||
| } | ||
| } | ||
|
|
||
| type hotLoginOption struct { | ||
| withRetry bool | ||
| _ struct{} | ||
| } | ||
|
|
||
| type HotLoginOptionFunc func(o *hotLoginOption) | ||
|
|
||
| func HotLoginWithRetry(flag bool) HotLoginOptionFunc { | ||
| return func(o *hotLoginOption) { | ||
| o.withRetry = flag | ||
| } | ||
| } | ||
|
|
||
| // HotLogin 热登录模式 | ||
| type HotLogin struct { | ||
| storage HotReloadStorage | ||
| opt hotLoginOption | ||
| } | ||
|
|
||
| // Login 实现了 BotLogin 接口 | ||
| func (h *HotLogin) Login(bot *Bot) error { | ||
| err := h.login(bot) | ||
| if err != nil && h.opt.withRetry { | ||
| scanLogin := SacnLogin{} | ||
| return scanLogin.Login(bot) | ||
| } | ||
| return err | ||
| } | ||
|
|
||
| func (h *HotLogin) login(bot *Bot) error { | ||
| if err := h.hotLoginInit(bot); err != nil { | ||
| return err | ||
| } | ||
| return bot.WebInit() | ||
| } | ||
|
|
||
| func (h *HotLogin) hotLoginInit(bot *Bot) error { | ||
| bot.hotReloadStorage = h.storage | ||
| return bot.reload() | ||
| } | ||
|
|
||
| type pushLoginOption struct { | ||
| withoutUUIDCallback bool | ||
| withoutScanCallback bool | ||
| withoutLoginCallback bool | ||
| withRetry bool | ||
| } | ||
|
|
||
| type PushLoginOptionFunc func(o *pushLoginOption) | ||
|
|
||
| // PushLoginWithoutUUIDCallback 设置 PushLogin 不执行二维码回调 | ||
| func PushLoginWithoutUUIDCallback(flag bool) PushLoginOptionFunc { | ||
| return func(o *pushLoginOption) { | ||
| o.withoutUUIDCallback = flag | ||
| } | ||
| } | ||
|
|
||
| // PushLoginWithoutScanCallback 设置 PushLogin 不执行扫码回调 | ||
| func PushLoginWithoutScanCallback(flag bool) PushLoginOptionFunc { | ||
| return func(o *pushLoginOption) { | ||
| o.withoutScanCallback = flag | ||
| } | ||
| } | ||
|
|
||
| // PushLoginWithoutLoginCallback 设置 PushLogin 不执行登录回调 | ||
| func PushLoginWithoutLoginCallback(flag bool) PushLoginOptionFunc { | ||
| return func(o *pushLoginOption) { | ||
| o.withoutLoginCallback = flag | ||
| } | ||
| } | ||
|
|
||
| // PushLoginWithRetry 设置 PushLogin 失败后执行扫码登录 | ||
| func PushLoginWithRetry(flag bool) PushLoginOptionFunc { | ||
| return func(o *pushLoginOption) { | ||
| o.withRetry = flag | ||
| } | ||
| } | ||
|
|
||
| // defaultPushLoginOpts 默认的 PushLogin | ||
| var defaultPushLoginOpts = [...]PushLoginOptionFunc{ | ||
| PushLoginWithoutUUIDCallback(true), | ||
| PushLoginWithoutScanCallback(true), | ||
| } | ||
|
|
||
| // PushLogin 免扫码登录模式 | ||
| type PushLogin struct { | ||
| storage HotReloadStorage | ||
| opt pushLoginOption | ||
| } | ||
|
|
||
| // Login 实现了 BotLogin 接口 | ||
| func (p PushLogin) Login(bot *Bot) error { | ||
| if err := p.pushLoginInit(bot); err != nil { | ||
| return err | ||
| } | ||
| resp, err := bot.Caller.WebWxPushLogin(bot.Storage.LoginInfo.WxUin) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| if err = resp.Err(); err != nil { | ||
| return err | ||
| } | ||
| err = p.checkLogin(bot, resp.UUID, "1") | ||
| if err != nil && p.opt.withRetry { | ||
| scanLogin := SacnLogin{} | ||
| return scanLogin.Login(bot) | ||
| } | ||
| return err | ||
| } | ||
|
|
||
| func (p PushLogin) pushLoginInit(bot *Bot) error { | ||
| bot.hotReloadStorage = p.storage | ||
| return bot.reload() | ||
| } | ||
|
|
||
| // checkLogin 登录检查 | ||
| func (p PushLogin) checkLogin(bot *Bot, uuid, tip string) error { | ||
| // todo 将checkLogin剥离出来 | ||
| bot.uuid = uuid | ||
| // 二维码获取回调 | ||
| if bot.UUIDCallback != nil && !p.opt.withoutUUIDCallback { | ||
| bot.UUIDCallback(uuid) | ||
| } | ||
| for { | ||
| // 长轮询检查是否扫码登录 | ||
| resp, err := bot.Caller.CheckLogin(uuid, tip) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| if tip == "1" { | ||
| tip = "0" | ||
| } | ||
| switch resp.Code { | ||
| case StatusSuccess: | ||
| // 判断是否有登录回调，如果有执行它 | ||
| if err = bot.HandleLogin(resp.Raw); err != nil { | ||
| return err | ||
| } | ||
| if bot.LoginCallBack != nil && !p.opt.withoutLoginCallback { | ||
| bot.LoginCallBack(resp.Raw) | ||
| } | ||
| return nil | ||
| case StatusScanned: | ||
| // 执行扫码回调 | ||
| if bot.ScanCallBack != nil && !p.opt.withoutScanCallback { | ||
| bot.ScanCallBack(resp.Raw) | ||
| } | ||
| case StatusTimeout: | ||
| return ErrLoginTimeout | ||
| case StatusWait: | ||
| continue | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.