a flag for hiding progress bar

Signed-off-by: Omri Bornstein <omribor@gmail.com>

Author: AppleGamer22

commands/root.go

@@ -10,8 +10,11 @@ import (
 	"github.com/spf13/cobra"
 )
 
-var duration time.Duration
-var pid int
+var (
+	duration time.Duration
+	pid      int
+	quiet    bool
+)
 
 var RootCommand = &cobra.Command{
 	Use:     "cocainate",
@@ -31,7 +34,7 @@ var RootCommand = &cobra.Command{
 			return err
 		}
 
-		if err := s.Wait(); err != nil {
+		if err := s.Wait(quiet); err != nil {
 			return err
 		}
 
@@ -46,6 +49,7 @@ var RootCommand = &cobra.Command{
 func init() {
 	RootCommand.Flags().DurationVarP(&duration, "duration", "d", 0, "duration with units ns, us (or µs), ms, s, m, h")
 	RootCommand.Flags().IntVarP(&pid, "pid", "p", 0, "a running process ID, duration (used as polling interval) must be provided")
+	RootCommand.Flags().BoolVarP(&quiet, "quiet", "q", false, "hide progress bar")
 	// RootCommand.MarkFlagsRequiredTogether("pid", "duration")
 	RootCommand.SetVersionTemplate("{{.Version}}\n")
 }

session/session.go

@@ -10,6 +10,7 @@ import (
 
 	"github.com/AppleGamer22/cocainate/progress_bar"
 	"github.com/AppleGamer22/cocainate/ps"
+	tea "github.com/charmbracelet/bubbletea"
 )
 
 /*
@@ -34,7 +35,7 @@ Wait will block further execution until the user send an interrupt signal, or un
 
 A non-nil error is returned if the un-inhabitation call fails.
 */
-func (s *Session) Wait() error {
+func (s *Session) Wait(quiet bool) error {
 	if !s.Active() {
 		return errors.New("Wait can be called only after Start has been called successfully")
 	}
@@ -50,14 +51,19 @@ func (s *Session) Wait() error {
 		case <-s.Signals:
 		}
 	} else if s.Duration > 0 {
-		program := progress_bar.New(s.Duration, s.Signals)
+		var program *tea.Program
+		if !quiet {
+			program = progress_bar.New(s.Duration, s.Signals)
+		}
 		// https://pkg.go.dev/time#After
 		timer := time.NewTimer(s.Duration)
 		select {
 		case <-timer.C:
 		case <-s.Signals:
 			timer.Stop()
-			program.Kill()
+			if !quiet {
+				program.Kill()
+			}
 		}
 	} else {
 		<-s.Signals

session/session_test.go

@@ -16,7 +16,7 @@ func TestDuration(t *testing.T) {
 	err := s.Start()
 	assert.NoError(t, err)
 
-	err = s.Wait()
+	err = s.Wait(true)
 	assert.NoError(t, err)
 }
 
@@ -29,7 +29,7 @@ func TestInterrupt(t *testing.T) {
 	err = s.Kill()
 	assert.NoError(t, err)
 
-	err = s.Wait()
+	err = s.Wait(true)
 	assert.NoError(t, err)
 }
 
@@ -43,7 +43,7 @@ func TestKill(t *testing.T) {
 	wg.Add(2)
 
 	go func() {
-		err := s.Wait()
+		err := s.Wait(false)
 		assert.NoError(t, err)
 		wg.Done()
 	}()
@@ -70,7 +70,7 @@ func TestStop(t *testing.T) {
 // Test for when Wait is called before Start
 func TestErrors(t *testing.T) {
 	s := session.New(0, 0)
-	err := s.Wait()
+	err := s.Wait(false)
 	assert.Error(t, err)
 
 	err = s.Kill()