Skip to content

Commit

Permalink
Merge pull request #107 from lixh00/master
Browse files Browse the repository at this point in the history
✨ 新增自定义设备Id接口
  • Loading branch information
eatmoreapple authored Sep 20, 2022
2 parents 879be7a + 8b7d66a commit 1b926f3
Showing 1 changed file with 27 additions and 10 deletions.
37 changes: 27 additions & 10 deletions bot.go
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ type Bot struct {
Storage *Storage
HotReloadStorage HotReloadStorage
uuid string
deviceId string // 设备Id
}

// Alive 判断当前用户是否正常在线
Expand All @@ -44,12 +45,21 @@ func (b *Bot) Alive() bool {
}
}

// SetDeviceId
// @description: 设置设备Id
// @receiver b
// @param deviceId
func (b *Bot) SetDeviceId(deviceId string) {
b.deviceId = deviceId
}

// GetCurrentUser 获取当前的用户
// self, err := bot.GetCurrentUser()
// if err != nil {
// return
// }
// fmt.Println(self.NickName)
//
// self, err := bot.GetCurrentUser()
// if err != nil {
// return
// }
// fmt.Println(self.NickName)
func (b *Bot) GetCurrentUser() (*Self, error) {
if b.self == nil {
return nil, errors.New("user not login")
Expand All @@ -59,9 +69,10 @@ func (b *Bot) GetCurrentUser() (*Self, error) {

// HotLogin 热登录,可实现重复登录,
// retry设置为true可在热登录失效后进行普通登录行为
// Storage := NewJsonFileHotReloadStorage("Storage.json")
// err := bot.HotLogin(Storage, true)
// fmt.Println(err)
//
// Storage := NewJsonFileHotReloadStorage("Storage.json")
// err := bot.HotLogin(Storage, true)
// fmt.Println(err)
func (b *Bot) HotLogin(storage HotReloadStorage, retry ...bool) error {
b.isHot = true
b.HotReloadStorage = storage
Expand Down Expand Up @@ -172,12 +183,17 @@ func (b *Bot) HandleLogin(data []byte) error {
// 将LoginInfo存到storage里面
b.Storage.LoginInfo = info

// 处理设备Id
if b.deviceId == "" {
b.deviceId = GetRandomDeviceId()
}

// 构建BaseRequest
request := &BaseRequest{
Uin: info.WxUin,
Sid: info.WxSid,
Skey: info.SKey,
DeviceID: GetRandomDeviceId(),
DeviceID: b.deviceId,
}

// 将BaseRequest存到storage里面方便后续调用
Expand Down Expand Up @@ -374,7 +390,8 @@ func NewBot() *Bot {

// DefaultBot 默认的Bot的构造方法,
// mode不传入默认为 openwechat.Desktop,详情见mode
// bot := openwechat.DefaultBot(openwechat.Desktop)
//
// bot := openwechat.DefaultBot(openwechat.Desktop)
func DefaultBot(modes ...Mode) *Bot {
bot := NewBot()
if len(modes) > 0 {
Expand Down

0 comments on commit 1b926f3

Please sign in to comment.