From 8c0c901fb1c508ba878aeaf371084daeebe52a24 Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Mon, 16 Oct 2023 14:16:25 +0200
Subject: Avoid `run_loop` flag

---
 src/ui.rs | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

(limited to 'src/ui.rs')

diff --git a/src/ui.rs b/src/ui.rs
index 8fd44b1..e6c8e1b 100644
--- a/src/ui.rs
+++ b/src/ui.rs
@@ -8,7 +8,7 @@ use crossterm::{
     terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
     ExecutableCommand,
 };
-use nix::unistd::Pid;
+use nix::{sys::wait::waitpid, unistd::Pid};
 use ratatui::{prelude::*, widgets::*};
 use std::io::{self, stdout};
 
@@ -66,12 +66,16 @@ impl UI {
         let mut have_to_trace = !args.command.is_some();
         let mut should_quit = false;
 
-        let lines = trace(pid, &args, args.command.is_some())?;
-        if lines.len() > 1 {
-            for line in lines {
-                self.add_line(line);
+        if args.command.is_some() {
+            let registers = trace(pid, &args)?;
+            for register in registers {
+                self.add_line(register);
             }
+        } else {
+            // First wait for the parent process
+            _ = waitpid(pid, None)?;
         }
+
         while !should_quit {
             if have_to_trace {
                 if let Some(reg) = trace_next(pid)? {
-- 
cgit v1.2.3-18-g5258