Preserve tool block colors across ANSI resets

1 files changed, 3 insertions, 1 deletions

diff --git a/src/tui_components.zig b/src/tui_components.zig
index b00098a..6f298c3 100644
--- a/src/tui_components.zig
+++ b/src/tui_components.zig
@@ -2506,7 +2506,9 @@ pub const ToolUse = struct {
                 const rp = try ctx.a.alloc(u8, right_pad_n);
                 defer ctx.a.free(rp);
                 @memset(rp, ' ');
-                return std.fmt.allocPrint(ctx.a, "{s}{s}{s}{s}{s}{s}", .{ ctx.bg.open(), fg.open(), indent, text, rp, ansi_reset });
+                const inner = try reapplyAfterReset(ctx.a, text, ctx.bg.open(), fg.open());
+                defer ctx.a.free(inner);
+                return std.fmt.allocPrint(ctx.a, "{s}{s}{s}{s}{s}{s}", .{ ctx.bg.open(), fg.open(), indent, inner, rp, ansi_reset });
             }
             fn blank(ctx: @This()) ![]u8 {
                 return ctx.line(.{ .open_seq = "", .is_plain = true }, "");