Function runTUIWrapped

cmd/root/new.go:99–142

runTUIWrapped is runTUI with an optional model wrapper, used by --record to interpose the input recorder between the terminal and the real model.

(ctx context.Context, rt runtime.Runtime, sess *session.Session, spawner tui.SessionSpawner, cleanup func(), tuiOpts []tui.Option, wrap func(tea.Model) tea.Model, opts ...app.Opt)

// runTUIWrapped is runTUI with an optional model wrapper, used by --record to
// interpose the input recorder between the terminal and the real model.
func runTUIWrapped(ctx context.Context, rt runtime.Runtime, sess *session.Session, spawner tui.SessionSpawner, cleanup func(), tuiOpts []tui.Option, wrap func(tea.Model) tea.Model, opts ...app.Opt) error {
	if gen := rt.TitleGenerator(ctx); gen != nil {
		opts = append(opts, app.WithTitleGenerator(gen))
	}

	a := app.New(ctx, rt, sess, opts...)

	coalescer := tuiinput.NewWheelCoalescer()
	filter := func(model tea.Model, msg tea.Msg) tea.Msg {
		wheelMsg, ok := msg.(tea.MouseWheelMsg)
		if !ok {
			return msg
		}
		if coalescer.Handle(wheelMsg) {
			return nil
		}
		return msg
	}

	if cleanup == nil {
		cleanup = func() {}
	}
	// Prefer the session's working directory so the TUI (and features keyed
	// off it, like /shell) operate where the tools do — e.g. the worktree
	// created by --worktree, not the process CWD it was launched from.
	wd := sess.WorkingDir
	if wd == "" {
		wd, _ = os.Getwd()
	}
	model := tui.New(ctx, spawner, a, wd, cleanup, tuiOpts...)
	if wrap != nil {
		model = wrap(model)
	}

	p := tea.NewProgram(model, tea.WithContext(ctx), tea.WithFilter(filter))
	coalescer.SetSender(p.Send)

	if m, ok := model.(interface{ SetProgram(p *tea.Program) }); ok {
		m.SetProgram(p)
	}

	_, err := p.Run()
	return err
}