From 0313388b59c85c5c5e829630c6c6abed3d1a90bc Mon Sep 17 00:00:00 2001
From: Rustam Gilyazov <16064414+rusq@users.noreply.github.com>
Date: Sat, 25 Nov 2023 15:14:58 +1000
Subject: [PATCH] we don`t need chats for batch mode in main

---
 main.go | 30 +++++++++++++++---------------
 1 file changed, 15 insertions(+), 15 deletions(-)

diff --git a/main.go b/main.go
index 6748b44..d950c46 100644
--- a/main.go
+++ b/main.go
@@ -182,6 +182,9 @@ func run(ctx context.Context, p Params) error {
 		return err
 	}
 
+	ctx, stop := signal.NotifyContext(ctx, os.Interrupt, syscall.SIGTERM)
+	defer stop()
+
 	dlog.Println("Connecting to telegram . . .")
 	if err := cl.Start(ctx); err != nil {
 		return err
@@ -192,27 +195,24 @@ func run(ctx context.Context, p Params) error {
 		}
 	}()
 
-	ctx, stop := signal.NotifyContext(ctx, os.Interrupt, syscall.SIGTERM)
-	defer stop()
-
-	done, finished := fakeProgress("Getting chats . . .", 0)
-	chats, err := cl.GetChats(ctx)
-	close(done)
-	<-finished
-	if err != nil {
-		return err
-	}
-	sort.Slice(chats, func(i, j int) bool {
-		return chats[i].GetTitle() < chats[j].GetTitle()
-	})
-	dlog.Printf("got %d chats", len(chats))
-
 	if p.List {
 		return waipu.List(ctx, os.Stdout, cl)
 	} else if len(p.Batch) > 0 {
 		return waipu.Batch(ctx, cl, []int64(p.Batch))
 	} else {
 		// run UI
+		done, finished := fakeProgress("Getting chats . . .", 0)
+		chats, err := cl.GetChats(ctx)
+		close(done)
+		<-finished
+		if err != nil {
+			return err
+		}
+		sort.Slice(chats, func(i, j int) bool {
+			return chats[i].GetTitle() < chats[j].GetTitle()
+		})
+		dlog.Printf("got %d chats", len(chats))
+
 		tva := tui.New(ctx, cl)
 		if err := tva.Run(ctx, chats); err != nil {
 			return err