ior: route default mode through tui and add plain flag

1 files changed, 6 insertions, 1 deletions

diff --git a/internal/tui/tui.go b/internal/tui/tui.go
index 758213d..49d2365 100644
--- a/internal/tui/tui.go
+++ b/internal/tui/tui.go
@@ -28,7 +28,12 @@ type TraceStarter func(context.Context) error
 
 // Run starts the TUI program in alternate screen mode.
 func Run() error {
-	model := NewModel(flags.Get().PidFilter, defaultTraceStarter)
+	return RunWithTraceStarter(defaultTraceStarter)
+}
+
+// RunWithTraceStarter starts the TUI program with a custom trace starter.
+func RunWithTraceStarter(starter TraceStarter) error {
+	model := NewModel(flags.Get().PidFilter, starter)
 	program := tea.NewProgram(model, tea.WithAltScreen())
 	_, err := program.Run()
 	return err