gomp/ui/progressBar.go

package ui

import (
	"fmt"
	"strconv"
	"strings"

	"github.com/fhs/gompd/v2/mpd"
	"github.com/gdamore/tcell/v2"

	"github.com/aditya-K2/gomp/utils"
	"github.com/aditya-K2/tview"
)

var (
	CurrentSong string = ""
	CONN        *mpd.Client
	RENDERER    interface{ Send(string) }
)

func SetConnection(c *mpd.Client) {
	CONN = c
}

func ConnectRenderer(r interface{ Send(string) }) {
	RENDERER = r
}

// ProgressBar is a two-lined Box. First line is the BarTitle
// Second being the actual progress done.
// Use SetProgressFunc to provide the callback which provides the Fields each
// time the ProgressBar will be Drawn.
// The progressFunc must return (BarTitle, BarTopTitle, BarText, percentage)
// respectively
type ProgressBar struct {
	*tview.Box
	BarTitle     string
	BarText      string
	BarTopTitle  string
	progressFunc func() (BarTitle string,
		BarTopTitle string,
		BarText string,
		percentage float64)
}

func (self *ProgressBar) SetProgressFunc(
	pfunc func() (string, string, string, float64),
) (
	*ProgressBar,
) {
	self.progressFunc = pfunc
	return self
}

func NewProgressBar() *ProgressBar {
	return &ProgressBar{
		Box: tview.NewBox(),
	}
}

func GetProgressGlyph(width, percentage float64, btext string) string {
	q := "[default:gray:b]"
	var a string
	a += strings.Repeat(" ", int(width)-len(btext))
	a = utils.InsertAt(a, btext, int(width/2)-10)
	a = utils.InsertAt(a, "[-:-:-]", int(width*percentage/100))
	q += a
	return q
}

func (self *ProgressBar) Draw(screen tcell.Screen) {
	var (
		OFFSET int = 1
	)
	self.Box.SetBorder(false)
	self.Box.SetBorderPadding(1, 1, 1, 1)
	self.Box.SetBackgroundColor(tcell.ColorBlack)
	var percentage float64
	self.BarTitle, _, self.BarText, percentage =
		self.progressFunc()
	self.DrawForSubclass(screen, self.Box)
	self.Box.SetTitle(self.BarTopTitle)
	self.Box.SetTitleAlign(tview.AlignRight)
	x, y, _width, _ := self.Box.GetInnerRect()
	tview.Print(screen,
		self.BarTitle,
		x+OFFSET, y, _width, tview.AlignLeft, tcell.ColorWhite)
	tview.Print(screen,
		GetProgressGlyph(float64(_width-OFFSET-1),
			percentage,
			self.BarText),
		x, y+2, _width-OFFSET, tview.AlignRight, tcell.ColorWhite)
}

func ProgressFunction() (
	song string,
	top string,
	text string,
	percentage float64,
) {
	// song info
	currentAttributes, err := CONN.CurrentSong()
	if err == nil {
		song += "[green::bi]" +
			currentAttributes["Title"] + "[-:-:-] - " + "[blue::b]" +
			currentAttributes["Artist"] + "\n"
		if len(currentAttributes) == 0 && CurrentSong != "" {
			CurrentSong = ""
			RENDERER.Send("stop")
		} else if song != CurrentSong && len(currentAttributes) != 0 {
			RENDERER.Send(currentAttributes["file"])
			CurrentSong = song
		}
	} else {
		utils.Print("RED", "Error Retrieving Current Song\n")
		panic(err)
	}

	// status
	status, err := CONN.Status()
	el, err1 := strconv.ParseFloat(status["elapsed"], 8)
	du, err := strconv.ParseFloat(status["duration"], 8)
	song += fmt.Sprintf(
		"[gray::-] | " +
		"[[-:]%s[gray::-]] " +
		"[[-:]shuf: %s[gray::-]] " +
		"[[-:]rep: %s[gray::-]] " +
		"[[-:]vol: %s[gray::-]] ",
		utils.FormatString(status["state"]),
		utils.FormatString(status["random"]),
		utils.FormatString(status["repeat"]),
		status["volume"])

	// bar text
	if du != 0 {
		percentage = el / du * 100
		if err == nil && err1 == nil {
			text = utils.StrTime(el) + " / " + utils.StrTime(du) +
				" (" + strconv.FormatFloat(
					percentage, 'f', 2, 32) + "%" + ")"
		} else {
			text = ""
		}
	} else {
		text = "   ---:---"
		percentage = 0
	}

	// percentage
	if percentage > 100 {
		percentage = 0
	}
	
	song = top + song
	return
}