fix: console dropping some errors messages

block · Dec 20, 2024 · 7b4b93a · 7b4b93a
1 parent c0b8c4e
commit 7b4b93a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/frontend/cli/main.go b/frontend/cli/main.go
@@ -128,7 +128,6 @@ func main() {
 		sm := terminal.NewStatusManager(ctx)
 		csm.statusManager = optional.Some(sm)
 		ctx = sm.IntoContext(ctx)
-		defer sm.Close()
 	}
 	rpc.InitialiseClients(cli.Authenticators, cli.Insecure)
 
@@ -171,7 +170,14 @@ func main() {
 	ctx = bindContext(ctx, kctx)
 
 	err = kctx.Run(ctx)
-	kctx.FatalIfErrorf(err)
+
+	if sm, ok := csm.statusManager.Get(); ok {
+		sm.Close()
+	}
+
+	if err != nil {
+		kctx.FatalIfErrorf(err)
+	}
 }
 
 func createKongApplication(cli any, csm *currentStatusManager) *kong.Kong {