Clear interactive frame on exit

commit: c55f92fb9eac251006d9db375bab9a6b2fbdad63
author: patriceckhart · 2mo ago · 2026-08-11 07:10 UTC
committer: brice · 2mo ago · 2026-08-12 12:07 UTC
parent: 84a9305
signature: GPG VerifiedCD88F9E564E0102646FB6077A1A07C7C7C3E349D · arch-gpg Verified for the committer email

2 files · +53 -1

M packages/agent/modes/interactive.go +5 -1


@@ -1092,7 +1092,11 @@ func (i *Interactive) Run(ctx context.Context) error {
1092	1092		if seq := tui.ReportCWD(i.cfg.CWD); seq != "" {
1093	1093		_, _ = term.Write([]byte(seq))
1094	1094		}
1095		-	defer term.Write([]byte(tui.SeqResetScrollRegion + tui.SeqDeleteKittyImages + tui.SeqEnhancedKeyboardOff + tui.SeqBracketedPasteOff + tui.ResetCursorColor() + tui.ResetCursorShape() + tui.SeqShowCursor))
	1095	+	// Erase the live frame on exit without moving the cursor. The shell can
	1096	+	// then draw its prompt on the row where zot ended instead of leaving the
	1097	+	// inactive TUI visible above it. Do not erase scrollback: users should
	1098	+	// still be able to review the session after closing zot.
	1099	+	defer term.Write([]byte(tui.SeqResetScrollRegion + tui.SeqDeleteKittyImages + tui.SeqEnhancedKeyboardOff + tui.SeqBracketedPasteOff + tui.ResetCursorColor() + tui.ResetCursorShape() + tui.SeqClearScreenNoHome + tui.SeqShowCursor))
1096	1100		i.applyInputCursorColor()
1097	1101
1098	1102		// Streaming pacer: drains buffered text deltas at a steady rate

 	if seq := tui.ReportCWD(i.cfg.CWD); seq != "" {
 		_, _ = term.Write([]byte(seq))
 	}
-	defer term.Write([]byte(tui.SeqResetScrollRegion + tui.SeqDeleteKittyImages + tui.SeqEnhancedKeyboardOff + tui.SeqBracketedPasteOff + tui.ResetCursorColor() + tui.ResetCursorShape() + tui.SeqShowCursor))
+	// Erase the live frame on exit without moving the cursor. The shell can
+	// then draw its prompt on the row where zot ended instead of leaving the
+	// inactive TUI visible above it. Do not erase scrollback: users should
+	// still be able to review the session after closing zot.
+	defer term.Write([]byte(tui.SeqResetScrollRegion + tui.SeqDeleteKittyImages + tui.SeqEnhancedKeyboardOff + tui.SeqBracketedPasteOff + tui.ResetCursorColor() + tui.ResetCursorShape() + tui.SeqClearScreenNoHome + tui.SeqShowCursor))
 	i.applyInputCursorColor()
 	// Streaming pacer: drains buffered text deltas at a steady rate

A packages/agent/modes/interactive_terminal_test.go +48 -0

@@ -0,0 +1,48 @@
	1	+	package modes
	2	+
	3	+	import (
	4	+	"bytes"
	5	+	"context"
	6	+	"errors"
	7	+	"io"
	8	+	"strings"
	9	+	"testing"
	10	+	"time"
	11	+
	12	+	"github.com/bnema/zut/packages/tui"
	13	+	)
	14	+
	15	+	type cleanupTestTerminal struct {
	16	+	bytes.Buffer
	17	+	}
	18	+
	19	+	func (t *cleanupTestTerminal) Size() (int, int) { return 80, 24 }
	20	+	func (t *cleanupTestTerminal) OnResize(func()) {}
	21	+	func (t *cleanupTestTerminal) SetNonblock(bool) error { return nil }
	22	+	func (t *cleanupTestTerminal) ReadByte() (byte, error) { return 0, io.EOF }
	23	+	func (t *cleanupTestTerminal) EnterRaw() (func() error, error) {
	24	+	return func() error { return nil }, nil
	25	+	}
	26	+	func (t *cleanupTestTerminal) PeekByteTimeout(time.Duration) (byte, bool, error) {
	27	+	return 0, false, io.EOF
	28	+	}
	29	+
	30	+	func TestInteractiveRunClearsVisibleFrameOnExit(t *testing.T) {
	31	+	term := &cleanupTestTerminal{}
	32	+	interactive := NewInteractive(InteractiveConfig{Terminal: term})
	33	+	ctx, cancel := context.WithCancel(context.Background())
	34	+	cancel()
	35	+
	36	+	err := interactive.Run(ctx)
	37	+	if !errors.Is(err, context.Canceled) {
	38	+	t.Fatalf("Run error = %v, want context.Canceled", err)
	39	+	}
	40	+
	41	+	cleanup := tui.SeqResetScrollRegion + tui.SeqDeleteKittyImages +
	42	+	tui.SeqEnhancedKeyboardOff + tui.SeqBracketedPasteOff +
	43	+	tui.ResetCursorColor() + tui.ResetCursorShape() +
	44	+	tui.SeqClearScreenNoHome + tui.SeqShowCursor
	45	+	if got := term.String(); !strings.HasSuffix(got, cleanup) {
	46	+	t.Fatalf("terminal output does not end with frame cleanup\ngot suffix: %q\nwant: %q", got, cleanup)
	47	+	}
	48	+	}

@@ -0,0 +1,48 @@
	1	+	package modes
	2	+
	3	+	import (
	4	+	"bytes"
	5	+	"context"
	6	+	"errors"
	7	+	"io"
	8	+	"strings"
	9	+	"testing"
	10	+	"time"
	11	+
	12	+	"github.com/bnema/zut/packages/tui"
	13	+	)
	14	+
	15	+	type cleanupTestTerminal struct {
	16	+	bytes.Buffer
	17	+	}
	18	+
	19	+	func (t *cleanupTestTerminal) Size() (int, int) { return 80, 24 }
	20	+	func (t *cleanupTestTerminal) OnResize(func()) {}
	21	+	func (t *cleanupTestTerminal) SetNonblock(bool) error { return nil }
	22	+	func (t *cleanupTestTerminal) ReadByte() (byte, error) { return 0, io.EOF }
	23	+	func (t *cleanupTestTerminal) EnterRaw() (func() error, error) {
	24	+	return func() error { return nil }, nil
	25	+	}
	26	+	func (t *cleanupTestTerminal) PeekByteTimeout(time.Duration) (byte, bool, error) {
	27	+	return 0, false, io.EOF
	28	+	}
	29	+
	30	+	func TestInteractiveRunClearsVisibleFrameOnExit(t *testing.T) {
	31	+	term := &cleanupTestTerminal{}
	32	+	interactive := NewInteractive(InteractiveConfig{Terminal: term})
	33	+	ctx, cancel := context.WithCancel(context.Background())
	34	+	cancel()
	35	+
	36	+	err := interactive.Run(ctx)
	37	+	if !errors.Is(err, context.Canceled) {
	38	+	t.Fatalf("Run error = %v, want context.Canceled", err)
	39	+	}
	40	+
	41	+	cleanup := tui.SeqResetScrollRegion + tui.SeqDeleteKittyImages +
	42	+	tui.SeqEnhancedKeyboardOff + tui.SeqBracketedPasteOff +
	43	+	tui.ResetCursorColor() + tui.ResetCursorShape() +
	44	+	tui.SeqClearScreenNoHome + tui.SeqShowCursor
	45	+	if got := term.String(); !strings.HasSuffix(got, cleanup) {
	46	+	t.Fatalf("terminal output does not end with frame cleanup\ngot suffix: %q\nwant: %q", got, cleanup)
	47	+	}
	48	+	}