subsonic-tui/internal/tui/views/layout.go

package views

import (
	"time"

	"git.dayanhub.com/sagi/subsonic-tui/internal/client"
	"git.dayanhub.com/sagi/subsonic-tui/internal/config"
	"git.dayanhub.com/sagi/subsonic-tui/internal/playback"
	"github.com/delucks/go-subsonic"
	"github.com/gdamore/tcell/v2"
	"github.com/rivo/tview"
)

var _ View = &layout{}

type layout struct {
	view               *tview.Pages
	player             View
	currentFocusedView tview.Primitive
	smallView          *tview.Flex
	largeView          *tview.Grid
	status             *statusLine
	mainView           *main
}

//gocyclo:ignore
func NewLayout(client *client.Client, playbackCtl *playback.Controller, refreshUI func()) *layout {
	layout := &layout{}

	largeView := tview.NewGrid().SetRows(0, 4, 1)
	largeView.SetBackgroundColor(config.ColorBackground)

	smallView := tview.NewFlex().SetDirection(tview.FlexRow)
	smallView.SetBackgroundColor(config.ColorBackground)

	layout.largeView = largeView
	layout.smallView = smallView
	pages := tview.NewPages()
	pages.SetBackgroundColor(config.ColorBackground)

	// Status / command / search
	statusLine := NewStatusLine()
	layout.status = statusLine

	// Side Panel (Artist/Albums/Playlist)
	albums := NewAlbums(client)
	artists := NewArtists(client)
	playlists := NewPlaylists(client)

	sidePanel := tview.NewGrid().SetRows(0, 0, 0)
	sidePanel.AddItem(artists.GetView(), 0, 0, 1, 1, 0, 0, false)
	sidePanel.AddItem(albums.GetView(), 1, 0, 1, 1, 0, 0, false)
	sidePanel.AddItem(playlists.GetView(), 2, 0, 1, 1, 0, 0, false)

	// main pane
	main := NewMainView(client, statusLine.Log)
	layout.mainView = main
	// Queue
	queue := NewQueue()

	// Main view
	mainView := tview.NewFlex()
	mainView.SetBackgroundColor(config.ColorBackground)
	mainView.AddItem(sidePanel, 0, 1, false)
	mainView.AddItem(main.GetView(), 0, 2, false)
	mainView.AddItem(queue.GetView(), 0, 1, false)
	largeView.AddItem(mainView, 0, 0, 1, 3, 0, 0, false)
	// Player
	player := NewPlayer(client)
	largeView.AddItem(player.GetView(), 1, 0, 1, 3, 0, 0, false)

	// Add status line
	largeView.AddItem(statusLine.GetView(), 2, 0, 1, 3, 0, 0, false)

	// Callbacks
	artists.SetSelectArtistFunc(func(artistId string) {
		artists.view.Blur()
		statusLine.Log("Fetching artist's albums...")

		go func() {
			a, _ := client.GetArtist(artistId)
			albums.SetAlbums(a.Album)
			albums.GetView().Focus(nil)
			refreshUI()
		}()
	})
	artists.SetOpenArtistFunc(func(artistId string) {
		artists.view.Blur()
		statusLine.Log("Fetching artists...")

		layout.currentFocusedView = main.GetView()
		layout.rebuildSmallView()

		go func() {
			a, _ := client.GetArtist(artistId)
			main.SetArtist(a)
			main.GetView().Focus(nil)
			refreshUI()
		}()
	})

	albums.SetCallback(func(albumID string) {
		albums.view.Blur()
		statusLine.Log("Fetching album...")

		layout.currentFocusedView = main.GetView()

		layout.rebuildSmallView()

		go func() {
			a, _ := client.GetAlbum(albumID)
			main.SetAlbum(a)
			main.view.Focus(nil)
			refreshUI()
		}()
	})

	playlists.SetCallback(func(p *subsonic.Playlist) {
		playlists.view.Blur()
		statusLine.Log("Fetching playlist...")

		layout.currentFocusedView = main.GetView()
		layout.rebuildSmallView()

		go func() {
			playlist, _ := client.GetPlaylist(p.ID)
			main.SetPlaylist(playlist)
			main.view.Focus(nil)
			refreshUI()
		}()
	})

	main.SetPlayAllFunc(func(songs ...*subsonic.Child) {
		statusLine.Log("Loaded #%d songs.", len(songs))
		playbackCtl.SetQueue(songs)
		queue.Update(songs, playbackCtl.GetQueuePosition())
	})
	main.SetPlayAddSongFunc(func(song ...*subsonic.Child) {
		statusLine.Log("Added #%d songs to queue.", len(song))
		playbackCtl.AddToQueue(song)
		queue.Update(playbackCtl.GetQueue(), playbackCtl.GetQueuePosition())
	})

	playbackCtl.SetSongElapsedFunc(func(song *subsonic.Child, elapsed time.Duration) {
		player.SetSongInfo(song)
		player.UpdateProgress(elapsed)
		queue.Update(playbackCtl.GetQueue(), playbackCtl.GetQueuePosition())
		refreshUI()
	})

	playbackCtl.SetSongEndedFunc(func(song *subsonic.Child) {
		statusLine.Log("")
		queue.Update(playbackCtl.GetQueue(), playbackCtl.GetQueuePosition())

		if config.ScrobbleEnabled() {
			// Scrobble
			_ = client.Scrobble(song.ID)
		}
	})

	queue.SetPlayFunc(func(position int) {
		playbackCtl.SetQueuePosition(position)
		queue.Update(playbackCtl.GetQueue(), playbackCtl.GetQueuePosition())
	})

	statusLine.SetOnUpdateFunc(func() {
		refreshUI()
	})

	statusLine.SetSearchFunc(func(quary string) {
		if len(quary) == 0 {
			layout.currentFocusedView.Focus(nil)
			statusLine.Log("Search canceled")

			return
		}
		// Search...
		statusLine.Log("Searching for '%s'....", quary)
		statusLine.view.Blur()

		go func() {
			result, _ := client.Search(quary)

			layout.currentFocusedView.Blur()
			main.SetSearch(result, quary)
			layout.currentFocusedView = main.GetView()
			layout.rebuildSmallView()
			refreshUI()
		}()
	})

	// Key Bindings
	pages.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
		if statusLine.Mode() == StatusModeSearch {
			return event
		}

		if event.Rune() == '1' {
			// Focus Artists
			artists.view.Blur()
			albums.view.Blur()
			playlists.view.Blur()
			main.view.Blur()
			queue.view.Blur()
			artists.view.Focus(nil)
			layout.currentFocusedView = artists.GetView()
			layout.rebuildSmallView()

			return nil
		} else if event.Rune() == '2' {
			// Focus Albums
			artists.view.Blur()
			albums.view.Blur()
			playlists.view.Blur()
			main.view.Blur()
			queue.view.Blur()
			albums.view.Focus(nil)
			layout.currentFocusedView = albums.GetView()
			layout.rebuildSmallView()

			return nil
		} else if event.Rune() == '3' {
			// Focus Playlists
			artists.view.Blur()
			albums.view.Blur()
			playlists.view.Blur()
			main.view.Blur()
			queue.view.Blur()
			playlists.view.Focus(nil)
			layout.currentFocusedView = playlists.GetView()
			layout.rebuildSmallView()

			return nil
		} else if event.Rune() == '`' {
			// Focus Songs
			artists.view.Blur()
			albums.view.Blur()
			playlists.view.Blur()
			main.view.Blur()
			queue.view.Blur()
			main.view.Focus(nil)
			layout.currentFocusedView = main.GetView()
			layout.rebuildSmallView()

			return nil
		} else if event.Rune() == '4' {
			// Focus Queue
			artists.view.Blur()
			albums.view.Blur()
			playlists.view.Blur()
			main.view.Blur()
			queue.view.Blur()
			queue.view.Focus(nil)
			layout.currentFocusedView = queue.GetView()
			layout.rebuildSmallView()

			return nil
		} else if event.Rune() == 'n' {
			playbackCtl.Next()
			queue.Update(playbackCtl.GetQueue(), playbackCtl.GetQueuePosition())

			return nil
		} else if event.Rune() == 'N' {
			playbackCtl.Prev()
			queue.Update(playbackCtl.GetQueue(), playbackCtl.GetQueuePosition())

			return nil
		} else if event.Rune() == 's' {
			playbackCtl.Stop()
			queue.Update(playbackCtl.GetQueue(), playbackCtl.GetQueuePosition())

			return nil
		} else if event.Rune() == 'p' {
			if playbackCtl.State() == playback.PlaybackStateStopped {
				song := playbackCtl.GetCurrentSong()
				if song != nil {
					playbackCtl.Play(song)
				}

				queue.Update(playbackCtl.GetQueue(), playbackCtl.GetQueuePosition())

				return nil
			}

			playbackCtl.TogglePlayPause()

			return nil
		} else if event.Rune() == 'c' {
			playbackCtl.ClearQueue()
			queue.Update(playbackCtl.GetQueue(), playbackCtl.GetQueuePosition())

			return nil
		} else if event.Rune() == '/' {
			layout.currentFocusedView.Blur()
			statusLine.Search()
			refreshUI()

			return nil
		} else if event.Rune() == 'L' {
			if layout.currentFocusedView == albums.GetView() {
				layout.currentFocusedView.Blur()
				main.view.Focus(nil)
				layout.currentFocusedView = main.GetView()
				layout.rebuildSmallView()
			} else if layout.currentFocusedView == artists.GetView() {
				layout.currentFocusedView.Blur()
				main.view.Focus(nil)
				layout.currentFocusedView = main.GetView()
				layout.rebuildSmallView()
			} else if layout.currentFocusedView == playlists.GetView() {
				layout.currentFocusedView.Blur()
				main.view.Focus(nil)
				layout.currentFocusedView = main.GetView()
				layout.rebuildSmallView()
			} else if layout.currentFocusedView == main.GetView() {
				layout.currentFocusedView.Blur()
				queue.view.Focus(nil)
				layout.currentFocusedView = queue.GetView()
				layout.rebuildSmallView()
			}
		} else if event.Rune() == 'H' {
			if layout.currentFocusedView == queue.GetView() {
				layout.currentFocusedView.Blur()
				main.view.Focus(nil)
				layout.currentFocusedView = main.GetView()
				layout.rebuildSmallView()
			} else if layout.currentFocusedView == main.GetView() {
				layout.currentFocusedView.Blur()
				artists.view.Focus(nil)
				layout.currentFocusedView = artists.GetView()
				layout.rebuildSmallView()
			}
		} else if event.Rune() == 'J' {
			if layout.currentFocusedView == albums.GetView() {
				layout.currentFocusedView.Blur()
				playlists.view.Focus(nil)
				layout.currentFocusedView = playlists.GetView()
				layout.rebuildSmallView()
			} else if layout.currentFocusedView == artists.GetView() {
				layout.currentFocusedView.Blur()
				albums.view.Focus(nil)
				layout.currentFocusedView = albums.GetView()
				layout.rebuildSmallView()
			} else if layout.currentFocusedView == playlists.GetView() {
				layout.currentFocusedView.Blur()
				artists.view.Focus(nil)
				layout.currentFocusedView = artists.GetView()
				layout.rebuildSmallView()
			}
		} else if event.Rune() == 'K' {
			if layout.currentFocusedView == albums.GetView() {
				layout.currentFocusedView.Blur()
				artists.view.Focus(nil)
				layout.currentFocusedView = artists.GetView()
				layout.rebuildSmallView()
			} else if layout.currentFocusedView == artists.GetView() {
				layout.currentFocusedView.Blur()
				playlists.view.Focus(nil)
				layout.currentFocusedView = playlists.GetView()
				layout.rebuildSmallView()
			} else if layout.currentFocusedView == playlists.GetView() {
				layout.currentFocusedView.Blur()
				albums.view.Focus(nil)
				layout.currentFocusedView = albums.GetView()
				layout.rebuildSmallView()
			}
		}

		return event
	})

	largeView.SetFocusFunc(func() {
		largeView.Blur()
		layout.currentFocusedView.Focus(nil)
	})
	smallView.SetFocusFunc(func() {
		smallView.Blur()
		layout.currentFocusedView.Focus(nil)
	})

	statusLine.Log("Press '?' for help")

	pages.AddPage("small", smallView, true, false)
	pages.AddPage("large", largeView, true, true)

	layout.view = pages
	layout.player = player
	// Auto focus on artists
	artists.GetView().Focus(nil)
	layout.currentFocusedView = artists.GetView()
	layout.rebuildSmallView()

	return layout
}

func (l *layout) rebuildSmallView() {
	l.smallView.Clear()
	l.smallView.AddItem(l.currentFocusedView, 0, 1, false)
	l.smallView.AddItem(l.player.GetView(), 4, 0, false)
	l.smallView.AddItem(l.status.GetView(), 1, 0, false)
}

func (l *layout) Mode() Statusmode {
	return l.status.Mode()
}

func (l *layout) GetView() tview.Primitive {
	return l.view
}

func (l *layout) Update() {
	_, _, w, h := l.view.GetRect()
	page, _ := l.view.GetFrontPage()
	smallView := w < 100 || h < 30

	if smallView {
		if page != "small" {
			l.mainView.SetMiniView(true)
			go l.view.SwitchToPage("small")

			return
		}
	} else {
		if page != "large" {
			l.mainView.SetMiniView(false)
			go l.view.SwitchToPage("large")

			return
		}
	}

	l.player.Update()
}