execution_manager.go

package prompt

import (
	"github.com/aschey/bubbleprompt/executor"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
)

type ExecutorFinishedMsg tea.Model

type executionManager struct {
	inner          tea.Model
	errorTextStyle lipgloss.Style
	err            error
}

func newExecutorManager(
	inner tea.Model,
	errorTextStyle lipgloss.Style,
	err error,
) *executionManager {
	return &executionManager{
		inner:          inner,
		errorTextStyle: errorTextStyle,
		err:            err,
	}
}

func (m executionManager) Init() tea.Cmd {
	return m.inner.Init()
}

func (m executionManager) Update(msg tea.Msg) (executionManager, tea.Cmd) {
	inner, cmd := m.inner.Update(msg)
	m.inner = inner
	if msg, ok := msg.(executor.ErrorMsg); ok {
		m.err = error(msg)
		return m, tea.Quit
	} else {
		return m, cmd
	}
}

func (m executionManager) View() string {
	if m.err != nil {
		return m.errorTextStyle.Render(m.err.Error()) + "\n"
	} else {
		return m.inner.View()
	}
}