gomu/playingbar.go

// Copyright (C) 2020  Raziman

package main

import (
	"fmt"
	"log"
	"strconv"
	"strings"
	"time"

	"github.com/gdamore/tcell"
	"github.com/rivo/tview"
)

type PlayingBar struct {
	*tview.Frame
	full      int
	limit     int
	progress  chan int
	_progress int
	skip      bool
	text      *tview.TextView
}

func (p *PlayingBar) Help() []string {
	return []string{}
}

// Playing bar shows progress of the song and the title of the song
func NewPlayingBar() *PlayingBar {

	textView := tview.NewTextView().SetTextAlign(tview.AlignCenter)

	frame := tview.NewFrame(textView).SetBorders(1, 1, 1, 1, 1, 1)
	frame.SetBorder(true).SetTitle(" Now Playing ")

	p := &PlayingBar{
		Frame:    frame,
		text:     textView,
		progress: make(chan int),
	}

	p.SetDefault()

	textView.SetChangedFunc(func() {
		gomu.App.Draw()

		if !gomu.Player.IsRunning {
			p.SetDefault()
		}
	})

	return p
}

// start processing progress bar
// runs asynchronusly
func (p *PlayingBar) Run() {

	go func() {
		for {

			// stop progressing if song ends or skipped
			if p._progress > p.full || p.skip {
				p.skip = false
				p._progress = 0
				break
			}

			p._progress += <-p.progress

			p.text.Clear()

			start, err := time.ParseDuration(strconv.Itoa(p._progress) + "s")

			if err != nil {
				log.Println(err)
			}

			end, err := time.ParseDuration(strconv.Itoa(p.full) + "s")

			if err != nil {
				log.Println(err)
			}

			x := p._progress * p.limit / p.full
			p.text.SetText(fmt.Sprintf("%s ┃%s%s┫ %s",
				fmtDuration(start),
				strings.Repeat("█", x),
				strings.Repeat("━", p.limit-x),
				fmtDuration(end),
			))

		}
	}()
}

func (p *PlayingBar) SetSongTitle(title string) {
	p.Clear()
	p.AddText(title, true, tview.AlignCenter, tcell.ColorGreen)
}

func (p *PlayingBar) NewProgress(songTitle string, full, limit int) {
	p.full = full
	p.limit = limit
	p._progress = 0
	p.SetSongTitle(songTitle)
}

// sets default title and progress bar
func (p *PlayingBar) SetDefault() {
	p.SetSongTitle("---------:---------")
	text := fmt.Sprintf(
		"%s ┣%s┫ %s", "00:00", strings.Repeat("━", 100), "00:00",
	)
	p.text.SetText(text)
}

func (p *PlayingBar) Stop() {
	p.skip = true
}