From 1fcee07a875aefce5105bb1abb49b0f7786115e2 Mon Sep 17 00:00:00 2001 From: vtxyer Date: Wed, 30 Sep 2026 16:56:19 +0800 Subject: [PATCH] fix(cli): switch to SimpleWaiter to unblock cross-DC downloads --- cmd/tg/app.go | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/cmd/tg/app.go b/cmd/tg/app.go index 9a8529c..6e20ca5 100644 --- a/cmd/tg/app.go +++ b/cmd/tg/app.go @@ -61,7 +61,7 @@ type app struct { cfg Config log *zap.Logger - waiter *floodwait.Waiter + waiter *floodwait.SimpleWaiter printer *output.Printer // active is the account currently being operated on (set per run iteration). @@ -80,7 +80,7 @@ func newApp() *app { } return &app{ - waiter: floodwait.NewWaiter(), + waiter: floodwait.NewSimpleWaiter(), log: defaultLog, printer: output.New(output.Text, nil), } @@ -225,10 +225,8 @@ func (a *app) connectWith( } client := telegram.NewClient(appID, appHash, a.optionsFor(st, rp, d)) - if err := a.waiter.Run(ctx, func(ctx context.Context) error { - return client.Run(ctx, func(ctx context.Context) error { - return f(ctx, client, d) - }) + if err := client.Run(ctx, func(ctx context.Context) error { + return f(ctx, client, d) }); err != nil && !errors.Is(err, context.Canceled) { return err }