From 04c82f05dea372ca74adcd974bba94d8fba6b792 Mon Sep 17 00:00:00 2001
From: Marcin Kulik <m@ku1ik.com>
Date: Sun, 31 Aug 2025 22:43:42 +0200
Subject: [PATCH] Fix reading output from PTY after child's death

---
 src/session.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/session.rs b/src/session.rs
index e3fb4e1..0cc2f27 100644
--- a/src/session.rs
+++ b/src/session.rs
@@ -202,8 +202,16 @@ impl<N: Notifier> Session<N> {
             }
         }
 
+        while let Ok(n) = pty.read(&mut output_buf).await {
+            if n > 0 {
+                self.handle_output(&output_buf[..n]).await;
+                output.extend_from_slice(&output_buf[0..n]);
+            } else {
+                break;
+            }
+        }
+
         if !output.is_empty() {
-            self.handle_output(&output).await;
             let _ = tty.write_all(&output).await;
         }