feat: emit errors to the render

Author: PhearZero

cmd/root.go

@@ -75,8 +75,12 @@ var (
 			)
 			go func() {
 				state.Watch(func(status *internal.StateModel, err error) {
-					cobra.CheckErr(err)
-					p.Send(state)
+					if err == nil {
+						p.Send(state)
+					}
+					if err != nil {
+						p.Send(err)
+					}
 				}, context.Background(), client)
 			}()
 			_, err = p.Run()

internal/state.go

@@ -40,6 +40,7 @@ func (s *StateModel) Watch(cb func(model *StateModel, err error), ctx context.Co
 		}
 		if status.StatusCode() != 200 {
 			cb(nil, errors.New(status.Status()))
+			return
 		}
 
 		// Update Status

ui/error.go

@@ -0,0 +1,48 @@
+package ui
+
+import (
+	"github.com/algorandfoundation/hack-tui/ui/controls"
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/charmbracelet/lipgloss"
+	"strings"
+)
+
+type ErrorViewModel struct {
+	Height   int
+	Width    int
+	controls controls.Model
+	Message  string
+}
+
+func NewErrorViewModel(message string) ErrorViewModel {
+	return ErrorViewModel{
+		Height:   0,
+		Width:    0,
+		controls: controls.New(" Error "),
+	}
+}
+
+func (m ErrorViewModel) Init() tea.Cmd {
+	return nil
+}
+
+func (m ErrorViewModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	return m.HandleMessage(msg)
+}
+
+func (m ErrorViewModel) HandleMessage(msg tea.Msg) (ErrorViewModel, tea.Cmd) {
+	var cmd tea.Cmd
+	switch msg := msg.(type) {
+
+	case tea.WindowSizeMsg:
+		m.Width = msg.Width
+		m.Height = msg.Height - 2
+	}
+	m.controls, cmd = m.controls.HandleMessage(msg)
+	return m, cmd
+}
+
+func (m ErrorViewModel) View() string {
+	pad := strings.Repeat("\n", max(0, m.Height/2-1))
+	return lipgloss.JoinVertical(lipgloss.Center, pad, red.Render(m.Message), pad, m.controls.View())
+}

ui/style.go

@@ -23,6 +23,7 @@ var (
 	cyan    = lipgloss.NewStyle().Foreground(lipgloss.Color("14"))
 	yellow  = lipgloss.NewStyle().Foreground(lipgloss.Color("11"))
 	green   = lipgloss.NewStyle().Foreground(lipgloss.Color("10"))
+	red     = lipgloss.NewStyle().Foreground(lipgloss.Color("9"))
 	Magenta = lipgloss.NewStyle().
 		Foreground(lipgloss.Color("5")).
 		Render

ui/viewport.go

@@ -20,6 +20,7 @@ const (
 	KeysPage        ViewportPage = "keys"
 	GeneratePage    ViewportPage = "generate"
 	TransactionPage ViewportPage = "transaction"
+	ErrorPage       ViewportPage = "error"
 )
 
 type ViewportViewModel struct {
@@ -40,6 +41,10 @@ type ViewportViewModel struct {
 
 	page   ViewportPage
 	client *api.ClientWithResponses
+
+	// Error Handler
+	errorMsg  *string
+	errorPage ErrorViewModel
 }
 
 type DeleteFinished string
@@ -72,6 +77,9 @@ func (m ViewportViewModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	cmds = append(cmds, cmd)
 
 	switch msg := msg.(type) {
+	case error:
+		strMsg := msg.Error()
+		m.errorMsg = &strMsg
 	// When the state updates
 	case internal.StateModel:
 		m.Data = &msg
@@ -158,6 +166,7 @@ func (m ViewportViewModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		m.transactionPage, cmd = m.transactionPage.HandleMessage(pageMsg)
 		cmds = append(cmds, cmd)
 		//}
+		m.errorPage, cmd = m.errorPage.HandleMessage(pageMsg)
 		cmds = append(cmds, cmd)
 		// Avoid triggering commands again
 		return m, tea.Batch(cmds...)
@@ -180,6 +189,13 @@ func (m ViewportViewModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 
 // View renders the viewport.Model
 func (m ViewportViewModel) View() string {
+	errMsg := m.errorMsg
+
+	if errMsg != nil {
+		m.errorPage.Message = *errMsg
+		m.page = ErrorPage
+	}
+
 	// Handle Page render
 	var page tea.Model
 	switch m.page {
@@ -191,6 +207,8 @@ func (m ViewportViewModel) View() string {
 		page = m.keysPage
 	case TransactionPage:
 		page = m.transactionPage
+	case ErrorPage:
+		page = m.errorPage
 	}
 
 	if page == nil {
@@ -235,6 +253,8 @@ func MakeViewportViewModel(state *internal.StateModel, client *api.ClientWithRes
 		page: AccountsPage,
 		// RPC client
 		client: client,
+
+		errorPage: NewErrorViewModel(""),
 	}
 
 	return &m, nil