// Copyright (C) 2020 Raziman package main import ( "fmt" "strconv" "strings" "time" "github.com/gdamore/tcell" "github.com/rivo/tview" ) func PlayingBar(app *tview.Application) *Progress { textView := tview.NewTextView(). SetChangedFunc(func () { app.Draw() }) progress := InitProgressBar(textView) return progress } type Progress struct { textView *tview.TextView full int limit int progress chan int frame *tview.Frame _progress int } // full is the maximum amount of value can be sent to channel // limit is the progress bar size func InitProgressBar(txt *tview.TextView) *Progress { p := &Progress{textView: txt} p.progress = make(chan int) p.textView.SetTextAlign(tview.AlignCenter) p.frame = tview.NewFrame(p.textView).SetBorders(1, 1, 1, 1, 1, 1) p.frame.SetBorder(true).SetTitle("Now Playing") p.SetSongTitle("-") p.textView.SetText(fmt.Sprintf("%s %s %s", "0s", strings.Repeat("□", 100), "0s")) return p } func (p *Progress) Run() { go func() { // Simple channel status gauge (progress bar) for { p._progress += <-p.progress p.textView.Clear() if p._progress > p.full { p._progress = 0 break } start, err := time.ParseDuration(strconv.Itoa(p._progress) + "s") if err != nil { panic(err) } end, err := time.ParseDuration(strconv.Itoa(p.full) + "s") if err != nil { panic(err) } x := p._progress * p.limit / p.full p.textView.SetText(fmt.Sprintf("%s %s%s %s", start.String(), strings.Repeat("■", x), strings.Repeat("□", p.limit-x), end.String(), )) } }() } func (p *Progress) SetSongTitle(title string) { p.frame.Clear() p.frame.AddText(title, true, tview.AlignCenter, tcell.ColorGreen) } func (p *Progress) NewProgress(songTitle string, full, limit int) { p.full = full p.limit = limit p._progress = 0 p.SetSongTitle(songTitle) }