cwinfo/matterbridge
5
0
Fork 0
mirror of https://github.com/cwinfo/matterbridge.git synced 2024-11-22 05:50:28 +00:00
Code Releases Activity

Refactor login logic (whatsapp)

Browse Source
This commit is contained in:
Wim 2022-03-13 01:01:42 +01:00
parent 69c74be7bb
commit 320c996a21
1 changed files with 30 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
bridge/whatsapp/whatsapp.go
View File

@ -74,26 +74,42 @@ func (b *Bwhatsapp) Connect() error {
b.wc = whatsmeow.NewClient(device, waLog.Stdout("Client", "INFO", true)) b.wc = whatsmeow.NewClient(device, waLog.Stdout("Client", "INFO", true))
b.wc.AddEventHandler(b.eventHandler) b.wc.AddEventHandler(b.eventHandler)
// No ID stored, new login firstlogin := false
qrChan, err := b.wc.GetQRChannel(context.Background()) var qrChan <-chan whatsmeow.QRChannelItem
// This error means that we're already logged in, so ignore it. if b.wc.Store.ID == nil {
firstlogin = true
qrChan, err = b.wc.GetQRChannel(context.Background())
if err != nil && !errors.Is(err, whatsmeow.ErrQRStoreContainsID) { if err != nil && !errors.Is(err, whatsmeow.ErrQRStoreContainsID) {
return errors.New("failed to to get QR channel:" + err.Error()) return errors.New("failed to to get QR channel:" + err.Error())
} }
}
err = b.wc.Connect() err = b.wc.Connect()
if err != nil { if err != nil {
return errors.New("failed to connect to WhatsApp: " + err.Error()) return errors.New("failed to connect to WhatsApp: " + err.Error())
} }
if b.wc.Store.ID == nil {
for evt := range qrChan { for evt := range qrChan {
if evt.Event == "code" { if evt.Event == "code" {
// Render the QR code here
qrterminal.GenerateHalfBlock(evt.Code, qrterminal.L, os.Stdout) qrterminal.GenerateHalfBlock(evt.Code, qrterminal.L, os.Stdout)
} else { } else {
b.Log.Infof("QR channel result: %s", evt.Event) b.Log.Infof("QR channel result: %s", evt.Event)
} }
} }
}
// disconnect and reconnect on our first login/pairing
// for some reason the GetJoinedGroups in JoinChannel doesn't work on first login
if firstlogin {
b.wc.Disconnect()
time.Sleep(time.Second)
err = b.wc.Connect()
if err != nil {
return errors.New("failed to connect to WhatsApp: " + err.Error())
}
}
b.Log.Infoln("WhatsApp connection successful") b.Log.Infoln("WhatsApp connection successful")
@ -113,7 +129,6 @@ func (b *Bwhatsapp) Connect() error {
} }
// get user avatar asynchronously // get user avatar asynchronously
// go func() {
b.Log.Info("Getting user avatars..") b.Log.Info("Getting user avatars..")
for jid := range b.users { for jid := range b.users {
@ -122,13 +137,14 @@ func (b *Bwhatsapp) Connect() error {
b.Log.Warnf("Could not get profile photo of %s: %v", jid, err) b.Log.Warnf("Could not get profile photo of %s: %v", jid, err)
} else { } else {
b.Lock() b.Lock()
if info != nil {
b.userAvatars[jid] = info.URL b.userAvatars[jid] = info.URL
}
b.Unlock() b.Unlock()
} }
} }
b.Log.Info("Finished getting avatars..") b.Log.Info("Finished getting avatars..")
// }()
return nil return nil
} }