client: Wait for updates manager to finish on disconnect

pkg/connector/client.go

@@ -64,6 +64,7 @@ type TelegramClient struct {
 	userLogin      *bridgev2.UserLogin
 	client         *telegram.Client
 	updatesManager *updates.Manager
+	updatesCloseC  chan struct{}
 	clientCtx      context.Context
 	clientCancel   context.CancelFunc
 	clientCloseC   <-chan struct{}
@@ -504,12 +505,22 @@ func (t *TelegramClient) Connect(ctx context.Context) {
 		t.sendBadCredentialsOrUnknownError(err)
 		return
 	}
+	t.updatesCloseC = make(chan struct{})
 	go func() {
-		err = t.updatesManager.Run(t.clientCtx, t.client.API(), t.telegramUserID, updates.AuthOptions{})
+		defer close(t.updatesCloseC)
-		if err != nil && !errors.Is(err, context.Canceled) {
+		for {
-			zerolog.Ctx(t.clientCtx).Err(err).Msg("failed to run updates manager")
+			err = t.updatesManager.Run(t.clientCtx, t.client.API(), t.telegramUserID, updates.AuthOptions{})
-			t.Disconnect()
+			if err == nil || errors.Is(err, context.Canceled) {
-			t.Connect(t.main.Bridge.Log.WithContext(context.Background()))
+				return
+			}
+
+			zerolog.Ctx(t.clientCtx).Err(err).Msg("failed to run updates manager, retrying")
+
+			select {
+			case <-t.clientCtx.Done():
+				return
+			case <-time.After(5 * time.Second):
+			}
 		}
 	}()
 
@@ -542,6 +553,9 @@ func (t *TelegramClient) Disconnect() {
 	if t.clientCloseC != nil {
 		<-t.clientCloseC
 	}
+	if t.updatesCloseC != nil {
+		<-t.updatesCloseC
+	}
 }
 
 func (t *TelegramClient) getInputUser(ctx context.Context, id int64) (*tg.InputUser, error) {