Redirect Output From Program

main.go

package main

import (
	"bytes"
	"fmt"
	"os"
	"os/exec"

	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
)

type editorFinishedMsg struct {
	err error
	out bytes.Buffer
}

func openEditor() tea.Cmd {
	c := exec.Command("hyperfine", "--style", "full", "sleep 0.3")
	var out bytes.Buffer
	c.Stdout = &out
	return tea.ExecProcess(c, func(err error) tea.Msg {
		return editorFinishedMsg{err, out}
	})
}

type model struct {
	altscreenActive bool
	output          []byte
	err             error
}

func (m model) Init() tea.Cmd {
	return nil
}

func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch msg.String() {
		case "a":
			m.altscreenActive = !m.altscreenActive
			cmd := tea.EnterAltScreen
			if !m.altscreenActive {
				cmd = tea.ExitAltScreen
			}
			return m, cmd
		case "e":
			return m, openEditor()
		case "ctrl+c", "q":
			return m, tea.Quit
		}
	case editorFinishedMsg:
		m.output = msg.out.Bytes()
		if msg.err != nil {
			m.err = msg.err
			return m, tea.Quit
		}
	}
	return m, nil
}

func (m model) View() string {
	if m.err != nil {
		return "Error: " + m.err.Error() + "\n"
	}
	return lipgloss.JoinVertical(lipgloss.Left, "Press 'e' to open your EDITOR.", "Press 'a' to toggle the altscreen", string(m.output), "Press 'q' to quit.")
}

func main() {
	m := model{}
	if _, err := tea.NewProgram(m).Run(); err != nil {
		fmt.Println("Error running program:", err)
		os.Exit(1)
	}
}