1 files changed, 5 insertions, 0 deletions
diff --git a/src/termio/stream_handler.zig b/src/termio/stream_handler.zig
index 2d90831f2..b2b2af3d0 100644
--- a/src/termio/stream_handler.zig
+++ b/src/termio/stream_handler.zig
@@ -1054,6 +1054,11 @@ pub const StreamHandler = struct {
 
     pub inline fn endOfInput(self: *StreamHandler) !void {
         self.terminal.markSemanticPrompt(.command);
+        self.surfaceMessageWriter(.start_command_timer);
+    }
+
+    pub inline fn endOfCommand(self: *StreamHandler, exit_code: ?u8) !void {
+        self.surfaceMessageWriter(.{ .stop_command_timer = exit_code });
     }
 
     pub fn reportPwd(self: *StreamHandler, url: []const u8) !void {