feat: working on spinner

pkg/chtsht/commands.go

@@ -2,12 +2,68 @@ package chtsht
 
 import (
 	"bufio"
+	"fmt"
+	"github.com/charmbracelet/bubbles/spinner"
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/charmbracelet/lipgloss"
 	"log"
 	"os"
 	"os/exec"
 	"strings"
 )
 
+type errMsg error
+
+type model struct {
+	spinner  spinner.Model
+	quitting bool
+	err      error
+}
+
+func initialModel() model {
+	s := spinner.New()
+	s.Spinner = spinner.Dot
+	s.Style = lipgloss.NewStyle().Foreground(lipgloss.Color("205"))
+	return model{spinner: s}
+}
+
+func (m model) Init() tea.Cmd {
+	return m.spinner.Tick
+}
+
+func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	switch msg := msg.(type) {
+	case tea.KeyMsg:
+		switch msg.String() {
+		case "q", "esc", "ctrl+c":
+			m.quitting = true
+			return m, tea.Quit
+		default:
+			return m, nil
+		}
+
+	case errMsg:
+		m.err = msg
+		return m, nil
+
+	default:
+		var cmd tea.Cmd
+		m.spinner, cmd = m.spinner.Update(msg)
+		return m, cmd
+	}
+}
+
+func (m model) View() string {
+	if m.err != nil {
+		return m.err.Error()
+	}
+	str := fmt.Sprintf("\n   %s Loading information...press q to quit\n", m.spinner.View())
+	if m.quitting {
+		return str + "\n"
+	}
+	return str
+}
+
 func SelectFromList(items []string) (string, error) {
 
 	cmd := exec.Command("fzf")
@@ -22,23 +78,35 @@ func SelectFromList(items []string) (string, error) {
 }
 
 func DisplayOutput(url string) {
+	p := tea.NewProgram(initialModel())
+
 	cmd := exec.Command("curl", "-s", url)
 	cmd.Stderr = os.Stderr
 
 	lessCmd := exec.Command("less")
 	lessCmd.Stdin, _ = cmd.StdoutPipe()
 	lessCmd.Stdout = os.Stdout
 
-	if err := cmd.Start(); err != nil {
+	go func() {
+		if _, err := p.Run(); err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+	}()
+	if err := cmd.Run(); err != nil {
 		log.Fatalf("Error while querying: %s", err)
 	}
 
-	if err := lessCmd.Run(); err != nil {
-		log.Fatalf("Error while piping into $PAGER: %s", err)
-	}
+	go func() {
+		if err := lessCmd.Run(); err != nil {
+			log.Fatalf("Error while piping into $PAGER: %s", err)
+		}
+		p.Quit()
+	}()
 
-	cmd.Wait()
-	return
+	if err := cmd.Run(); err != nil {
+		log.Fatalf("Error while querying: %s", err)
+	}
 }
 
 func ChtReadOptions() ([]string, error) {