di-tui/main.go

package main

import (
	"fmt"
	"os"
	"time"

	"github.com/acaloiaro/di-tui/app"
	"github.com/acaloiaro/di-tui/config"
	"github.com/acaloiaro/di-tui/context"
	"github.com/acaloiaro/di-tui/difm"
	"github.com/acaloiaro/di-tui/mpris"
	"github.com/acaloiaro/di-tui/views"
	"github.com/gdamore/tcell/v2"
	"github.com/rivo/tview"
	"github.com/spf13/pflag"
	"github.com/spf13/viper"
)

var ctx *context.AppContext

func main() {
	pflag.String("username", "", "your di.fm username")
	pflag.String("password", "", "your di.fm password")

	pflag.Parse()
	viper.BindPFlags(pflag.CommandLine)

	username := viper.GetString("username")
	password := viper.GetString("password")

	var token string
	if len(username) > 0 && len(password) > 0 {
		difm.Authenticate(ctx, username, password)
	}

	token = config.GetToken()
	if token == "" {
		fmt.Println("Authenticate by running: di-tui --username USER --password PASSWORD\n\n" +
			"Or, visit https://github.com/acaloiaro/di-tui#authenticate for other options.")
		os.Exit(1)
	}

	ctx = context.CreateAppContext(views.CreateViewContext())
	ctx.DifmToken = token

	run()
}

func run() {
	configureEventHandling()
	updateScreenLayout()
	FetchFavoritesAndChannels()

	ctx.View.Keybindings.Bindings = views.GetKeybindings()

	err := ctx.View.App.Run()
	if err != nil {
		panic(err)
	}
}

func updateScreenLayout() {
	focusView := ctx.View.App.GetFocus()

	main := tview.NewFlex()
	main.SetDirection(tview.FlexRow)

	favsAndChannels := tview.NewFlex().
		AddItem(ctx.View.FavoriteList, 0, 1, false).
		AddItem(ctx.View.ChannelList, 0, 2, false).
		SetDirection(tview.FlexRow)

	primaryView := tview.NewFlex()
	primaryView.
		AddItem(favsAndChannels, 30, 0, false).
		AddItem(ctx.View.NowPlaying, 0, 4, false)

	if ctx.ShowStatus {
		main.AddItem(ctx.View.Status, 4, 0, false)
	}

	main.
		AddItem(primaryView, 0, 3, false).
		AddItem(ctx.View.Keybindings, 4, 0, false)

	if focusView == nil {
		focusView = ctx.View.FavoriteList
	}

	ctx.View.App.
		SetRoot(main, true).
		SetFocus(focusView)
}

// configureEventHandling handles key press events, and regular UI updates such as the currently playing track
func configureEventHandling() {
	ctx.View.App.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
		focus := ctx.View.App.GetFocus().(*tview.List)

		switch event.Key() {
		case tcell.KeyEnter:
			current := focus.GetCurrentItem()
			if focus != ctx.View.ChannelList {
				highlightedFavorite := ctx.FavoriteList[current]
				ctx.HighlightedChannel = difm.FavoriteItemChannel(ctx, highlightedFavorite)
			} else {
				ctx.HighlightedChannel = &ctx.ChannelList[current]
			}
			app.Play(ctx)
		case tcell.KeyRune:
			switch event.Rune() {
			case 'c':
				if focus != ctx.View.ChannelList {
					ctx.View.App.SetFocus(ctx.View.ChannelList)
				}
			case 'f':
				if focus != ctx.View.FavoriteList {
					ctx.View.App.SetFocus(ctx.View.FavoriteList)
				}
			case 'F':
				current := focus.GetCurrentItem()
				if focus == ctx.View.ChannelList {
					ctx.HighlightedChannel = &ctx.ChannelList[current]
				} else {
					highlightedFavorite := ctx.FavoriteList[current]
					ctx.HighlightedChannel = difm.FavoriteItemChannel(ctx, highlightedFavorite)
				}
				difm.ToggleFavorite(ctx)
				FetchFavoritesAndChannels()
			case 'q':
				ctx.View.App.Stop()
			case 'j': // scroll down
				current := focus.GetCurrentItem() + 1
				focus.SetCurrentItem(current)
			case 'k': // scroll up
				current := focus.GetCurrentItem()
				if current > 0 {
					focus.SetCurrentItem(current - 1)
				}
			case 'p', 32: // tcell has no constant for the space bar rune (32)
				app.TogglePause(ctx)
			}
		}

		return event
	})

	// keep 'now playing' up to date second-by-second
	go func() {
		c := time.Tick(1 * time.Second)
		for range c {
			elapsed := time.Since(ctx.View.NowPlaying.Track.StartTime)

			// If the current time is past the end of the track, then a new track is playing and the now playing track needs
			// to be refreshed.
			if ctx.View.NowPlaying.Track.Duration > 0 && ctx.View.NowPlaying.Track.Duration < elapsed.Seconds() {
				app.UpdateNowPlaying(ctx, ctx.CurrentChannel)
			}

			if ctx.CurrentChannel != nil && elapsed.Seconds() > 0 {
				ctx.View.NowPlaying.Elapsed = elapsed.Seconds()
			}

			ctx.View.App.QueueUpdateDraw(func() {})
		}
	}()

	// display the status pane when new status messages arrive
	go func() {
		for {
			status := <-ctx.StatusChannel
			ctx.View.Status.Message = status.Message
			updateScreenLayout() // add the status pane to the screen

			<-time.Tick(status.Duration)
			ctx.ShowStatus = false
			updateScreenLayout() // remove the status pane from the screen
		}
	}()

	// Start the mpris server for d-bus support
	mpris.Start(ctx)
}

func FetchFavoritesAndChannels() {
	ctx.View.ChannelList.Clear()
	ctx.View.FavoriteList.Clear()

	channels := difm.ListChannels(ctx)
	for _, chn := range channels {
		ctx.View.ChannelList.AddItem(chn.Name, "", 0, func() {
		})
	}

	favorites := difm.ListFavorites(ctx)
	for _, fav := range favorites {
		ctx.View.FavoriteList.AddItem(fav.Name, "", 0, func() {})
	}
	ctx.ChannelList = channels
	ctx.FavoriteList = favorites

	// default the highlighted channel to the first favorite; even before users select a channel manually. This way,
	// when di-tui starts and the user presses the "Play" media key, di-tui will start playing the first favorite
	// instead of requiring them to choose the channel to be played
	highlightedFavorite := ctx.FavoriteList[0]
	ctx.HighlightedChannel = difm.FavoriteItemChannel(ctx, highlightedFavorite)
}