Refactor login logic (whatsapp)

bridge/whatsapp/whatsapp.go

@@ -74,26 +74,42 @@ func (b *Bwhatsapp) Connect() error {
 	b.wc = whatsmeow.NewClient(device, waLog.Stdout("Client", "INFO", true))
 	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) {
 			return errors.New("failed to to get QR channel:" + err.Error())
 		}
+	}
 
 	err = b.wc.Connect()
 	if err != nil {
 		return errors.New("failed to connect to WhatsApp: " + err.Error())
 	}
 
+	if b.wc.Store.ID == nil {
 		for evt := range qrChan {
 			if evt.Event == "code" {
-			// Render the QR code here
 				qrterminal.GenerateHalfBlock(evt.Code, qrterminal.L, os.Stdout)
 			} else {
 				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")
 
@@ -113,7 +129,6 @@ func (b *Bwhatsapp) Connect() error {
 	}
 
 	// get user avatar asynchronously
-	//	go func() {
 	b.Log.Info("Getting user avatars..")
 
 	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)
 		} else {
 			b.Lock()
+			if info != nil {
 				b.userAvatars[jid] = info.URL
+			}
 			b.Unlock()
 		}
 	}
 
 	b.Log.Info("Finished getting avatars..")
-	//	}()
 
 	return nil
 }