gospt/internal/tui/main.go

package tui

import (
	"fmt"
	"time"

	"gospt/internal/gctx"

	"github.com/charmbracelet/bubbles/key"
	"github.com/charmbracelet/bubbles/list"
	"github.com/charmbracelet/bubbles/textinput"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
	"github.com/zmb3/spotify/v2"
)

var (
	P                *tea.Program
	DocStyle         = lipgloss.NewStyle().Margin(0, 2)
	currentlyPlaying string
	main_updates     chan *mainModel
	page             = 1
)

type Mode string

const (
	Album           Mode = "album"
	ArtistAlbum          = "artistalbum"
	Artist               = "artist"
	Artists              = "artists"
	Tracks               = "tracks"
	Albums               = "albums"
	Main                 = "main"
	Playlists            = "playlists"
	Playlist             = "playlist"
	Devices              = "devices"
	Search               = "search"
	SearchAlbums         = "searchalbums"
	SearchArtists        = "searchartists"
	SearchTracks         = "searchtracks"
	SearchPlaylists      = "searchplaylsits"
	SearchArtist         = "searchartist"
	SearchPlaylist       = "searchplaylist"
	SearchAlbum          = "searchalbum"
)

type mainItem struct {
	Name        string
	Duration    string
	Artist      spotify.SimpleArtist
	ID          spotify.ID
	Desc        string
	SpotifyItem any
}

func (i mainItem) Title() string       { return i.Name }
func (i mainItem) Description() string { return i.Desc }
func (i mainItem) FilterValue() string { return i.Title() + i.Desc }

type mainModel struct {
	list       list.Model
	input      textinput.Model
	ctx        *gctx.Context
	client     *spotify.Client
	mode       Mode
	playlist   spotify.SimplePlaylist
	artist     spotify.SimpleArtist
	album      spotify.SimpleAlbum
	search     string
	fromArtist bool
}

func (m *mainModel) GoBack() (tea.Cmd, error) {
	switch m.mode {
	case Main:
		return tea.Quit, nil
	case Albums, Artists, Tracks, Playlist, Devices, Search:
		m.mode = Main
		m.list.NewStatusMessage("Setting view to main")
		new_items, err := MainView(m.ctx, m.client)
		if err != nil {
			fmt.Println(err.Error())
		}
		m.list.SetItems(new_items)

	case Album:
		m.mode = Albums
		m.list.NewStatusMessage("Setting view to albums")
		new_items, err := AlbumsView(m.ctx, m.client)
		if err != nil {
			fmt.Println(err.Error())
			return nil, err
		}
		m.list.SetItems(new_items)
		m.list.ResetSelected()

	case Artist:
		m.mode = Artists
		m.fromArtist = false
		m.list.NewStatusMessage("Setting view to artists")
		new_items, err := ArtistsView(m.ctx, m.client)
		if err != nil {
			return nil, err
		}
		m.list.SetItems(new_items)
		m.list.ResetSelected()

	case ArtistAlbum:
		m.mode = Artist
		m.fromArtist = true
		m.list.NewStatusMessage("Opening " + m.artist.Name)
		new_items, err := ArtistAlbumsView(m.ctx, m.artist.ID, m.client)
		if err != nil {
			return nil, err
		}
		m.list.SetItems(new_items)
		m.list.ResetSelected()

	case SearchArtists, SearchTracks, SearchAlbums, SearchPlaylists:
		m.mode = Search
		m.list.NewStatusMessage("Setting view to search for " + m.input.Value())
		items, err := SearchView(m.ctx, m.client, m.search)
		if err != nil {
			return nil, err
		}
		m.list.SetItems(items)
	case SearchArtist:
		m.mode = SearchArtists

	case SearchAlbum:
		m.mode = SearchAlbums

	case SearchPlaylist:
		m.mode = SearchPlaylists

	default:
		m.list.ResetSelected()
		page = 0
	}
	return nil, nil
}

func (m mainModel) Init() tea.Cmd {
	main_updates = make(chan *mainModel)
	return nil
}

func (m *mainModel) Tick() {
	ticker := time.NewTicker(5 * time.Second)
	quit := make(chan struct{})
	go func() {
		for {
			select {
			case <-ticker.C:
				playing, _ := m.client.PlayerCurrentlyPlaying(m.ctx)
				if playing.Playing {
					currentlyPlaying = "Now playing " + playing.Item.Name + " by " + playing.Item.Artists[0].Name
				}
			case <-quit:
				ticker.Stop()
				return
			}
		}
	}()
}

func (m mainModel) View() string {
	if m.input.Focused() {
		return DocStyle.Render(m.list.View() + "\n" + m.input.View())
	}
	return DocStyle.Render(m.list.View() + "\n")
}

func (m *mainModel) Typing(msg tea.KeyMsg) (bool, tea.Cmd) {
	if msg.String() == "enter" {
		m.list.NewStatusMessage("Setting view to search for " + m.input.Value())
		items, err := SearchView(m.ctx, m.client, m.input.Value())
		if err != nil {
			fmt.Println(err.Error())
			return false, tea.Quit
		}
		m.search = m.input.Value()
		m.list.SetItems(items)
		m.list.ResetSelected()
		m.input.SetValue("")
		m.input.Blur()
		return true, nil
	}
	if msg.String() == "esc" {
		m.input.SetValue("")
		m.input.Blur()
		return false, nil
	}
	m.input, _ = m.input.Update(msg)
	return false, nil
}

func (m mainModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	m.list.NewStatusMessage(currentlyPlaying)
	select {
	case update := <-main_updates:
		m.list.SetItems(update.list.Items())
	default:
	}
	if m.list.Paginator.Page == m.list.Paginator.TotalPages-2 && m.list.Cursor() == 0 {
		// if last request was still full request more
		if len(m.list.Items())%50 == 0 {
			go m.LoadMoreItems()
		}
	}
	switch msg := msg.(type) {
	case tea.KeyMsg:
		if msg.String() == "ctrl+c" {
			return m, tea.Quit
		}
		if m.input.Focused() {
			search, cmd := m.Typing(msg)
			if search {
				m.mode = "search"
			}
			return m, cmd
		}
		if msg.String() == "s" {
			m.input.Focus()
		}
		if msg.String() == "d" {
			if !m.input.Focused() {
				m.mode = Devices
				new_items, err := DeviceView(m.ctx, m.client)
				if err != nil {
					fmt.Println(err.Error())
					return m, tea.Quit
				}
				m.list.SetItems(new_items)
				m.list.ResetSelected()
				m.list.NewStatusMessage("Setting view to devices")
			}
		}
		if msg.String() == "backspace" || msg.String() == "esc" || msg.String() == "q" {
			msg, err := m.GoBack()
			if err != nil {
				fmt.Println(err)
			}
			return m, msg
		}

		if msg.String() == "enter" || msg.String() == "spacebar" {
			switch m.mode {
			case SearchArtists:
				m.mode = SearchArtist
				m.fromArtist = true
				m.artist = m.list.SelectedItem().(mainItem).SpotifyItem.(spotify.SimpleArtist)
				m.list.NewStatusMessage("Opening " + m.artist.Name)
				new_items, err := ArtistAlbumsView(m.ctx, m.artist.ID, m.client)
				if err != nil {
					fmt.Println(err.Error())
					return m, tea.Quit
				}
				m.list.SetItems(new_items)
				m.list.ResetSelected()
			case SearchAlbums:
				m.mode = SearchAlbum
				m.album = m.list.SelectedItem().(mainItem).SpotifyItem.(spotify.SimpleAlbum)
				m.list.NewStatusMessage("Opening " + m.album.Name)
				new_items, err := AlbumTracksView(m.ctx, m.album.ID, m.client)
				if err != nil {
					fmt.Println(err.Error())
					return m, tea.Quit
				}
				m.list.SetItems(new_items)
				m.list.ResetSelected()
			case SearchPlaylists:
				m.mode = SearchPlaylist
				playlist := m.list.SelectedItem().(mainItem).SpotifyItem.(spotify.SimplePlaylist)
				m.playlist = playlist
				m.list.NewStatusMessage("Setting view to playlist " + playlist.Name)
				new_items, err := PlaylistView(m.ctx, m.client, playlist)
				if err != nil {
					fmt.Println(err.Error())
					return m, tea.Quit
				}
				m.list.SetItems(new_items)
				m.list.ResetSelected()
			case Search:
				switch m.list.SelectedItem().(mainItem).SpotifyItem.(type) {
				case *spotify.FullArtistPage:
					m.mode = SearchArtists
					m.list.NewStatusMessage("Setting view to artists")
					new_items, err := SearchArtistsView(m.ctx, m.client, m.list.SelectedItem().(mainItem).SpotifyItem.(*spotify.FullArtistPage))
					if err != nil {
						fmt.Println(err.Error())
						return m, tea.Quit
					}
					m.list.SetItems(new_items)
					m.list.ResetSelected()
				case *spotify.SimpleAlbumPage:
					m.mode = SearchAlbums
					m.list.NewStatusMessage("Setting view to albums")
					new_items, err := SearchAlbumsView(m.ctx, m.client, m.list.SelectedItem().(mainItem).SpotifyItem.(*spotify.SimpleAlbumPage))
					if err != nil {
						fmt.Println(err.Error())
						return m, tea.Quit
					}
					m.list.SetItems(new_items)
					m.list.ResetSelected()
				case *spotify.SimplePlaylistPage:
					m.mode = SearchPlaylists
					playlists := m.list.SelectedItem().(mainItem).SpotifyItem.(*spotify.SimplePlaylistPage)
					m.list.NewStatusMessage("Setting view to playlist")
					new_items, err := SearchPlaylistsView(m.ctx, m.client, playlists)
					if err != nil {
						fmt.Println(err.Error())
						return m, tea.Quit
					}
					m.list.SetItems(new_items)
					m.list.ResetSelected()
				case *spotify.FullTrackPage:
					m.mode = SearchTracks
					m.list.NewStatusMessage("Setting view to tracks")
					new_items, err := SearchTracksView(m.ctx, m.client, m.list.SelectedItem().(mainItem).SpotifyItem.(*spotify.FullTrackPage))
					if err != nil {
						fmt.Println(err.Error())
						return m, tea.Quit
					}
					m.list.SetItems(new_items)
					m.list.ResetSelected()
					m.list.NewStatusMessage("Setting view to tracks")
				}
			case "main":
				switch m.list.SelectedItem().(mainItem).SpotifyItem.(type) {
				case *spotify.FullArtistCursorPage:
					m.mode = Artists
					m.list.NewStatusMessage("Setting view to artists")
					new_items, err := ArtistsView(m.ctx, m.client)
					if err != nil {
						fmt.Println(err.Error())
						return m, tea.Quit
					}
					m.list.SetItems(new_items)
					m.list.ResetSelected()
				case *spotify.SavedAlbumPage:
					m.mode = Albums
					m.list.NewStatusMessage("Setting view to albums")
					new_items, err := AlbumsView(m.ctx, m.client)
					if err != nil {
						fmt.Println(err.Error())
						return m, tea.Quit
					}
					m.list.SetItems(new_items)
					m.list.ResetSelected()
				case spotify.SimplePlaylist:
					m.mode = Playlist
					playlist := m.list.SelectedItem().(mainItem).SpotifyItem.(spotify.SimplePlaylist)
					m.playlist = playlist
					m.list.NewStatusMessage("Setting view to playlist " + playlist.Name)
					new_items, err := PlaylistView(m.ctx, m.client, playlist)
					if err != nil {
						fmt.Println(err.Error())
						return m, tea.Quit
					}
					m.list.SetItems(new_items)
					m.list.ResetSelected()
				case *spotify.SavedTrackPage:
					m.mode = Tracks
					m.list.NewStatusMessage("Setting view to saved tracks")
					new_items, err := SavedTracksView(m.ctx, m.client)
					if err != nil {
						fmt.Println(err.Error())
						return m, tea.Quit
					}
					m.list.SetItems(new_items)
					m.list.ResetSelected()
					m.list.NewStatusMessage("Setting view to tracks")
				}
			case Albums:
				m.mode = Album
				m.album = m.list.SelectedItem().(mainItem).SpotifyItem.(spotify.SimpleAlbum)
				m.list.NewStatusMessage("Opening " + m.album.Name)
				new_items, err := AlbumTracksView(m.ctx, m.album.ID, m.client)
				if err != nil {
					fmt.Println(err.Error())
					return m, tea.Quit
				}
				m.list.SetItems(new_items)
				m.list.ResetSelected()
			case Artist:
				m.mode = ArtistAlbum
				m.album = m.list.SelectedItem().(mainItem).SpotifyItem.(spotify.SimpleAlbum)
				m.list.NewStatusMessage("Opening " + m.album.Name)
				new_items, err := AlbumTracksView(m.ctx, m.album.ID, m.client)
				if err != nil {
					fmt.Println(err.Error())
					return m, tea.Quit
				}
				m.list.SetItems(new_items)
				m.list.ResetSelected()
			case Artists:
				m.mode = Artist
				m.fromArtist = true
				m.artist = m.list.SelectedItem().(mainItem).SpotifyItem.(spotify.SimpleArtist)
				m.list.NewStatusMessage("Opening " + m.artist.Name)
				new_items, err := ArtistAlbumsView(m.ctx, m.artist.ID, m.client)
				if err != nil {
					fmt.Println(err.Error())
					return m, tea.Quit
				}
				m.list.SetItems(new_items)
				m.list.ResetSelected()
			case Album, ArtistAlbum:
				currentlyPlaying = m.list.SelectedItem().FilterValue()
				m.list.NewStatusMessage("Playing " + currentlyPlaying)
				go HandlePlay(m.ctx, m.client, &m.album.URI, m.list.Cursor()+(m.list.Paginator.Page*m.list.Paginator.TotalPages))
			case Playlist, SearchPlaylist:
				currentlyPlaying = m.list.SelectedItem().FilterValue()
				m.list.NewStatusMessage("Playing " + currentlyPlaying)
				go HandlePlay(m.ctx, m.client, &m.playlist.URI, m.list.Cursor()+(m.list.Paginator.Page*m.list.Paginator.PerPage))
			case Tracks, SearchTracks:
				currentlyPlaying = m.list.SelectedItem().FilterValue()
				m.list.NewStatusMessage("Playing " + currentlyPlaying)
				go HandlePlayLikedSong(m.ctx, m.client, m.list.Cursor()+(m.list.Paginator.Page*m.list.Paginator.PerPage))
			case Devices:
				go HandleSetDevice(m.ctx, m.client, m.list.SelectedItem().(mainItem).SpotifyItem.(spotify.PlayerDevice))
				m.list.NewStatusMessage("Setting device to " + m.list.SelectedItem().FilterValue())
				m.mode = "main"
				m.list.NewStatusMessage("Setting view to main")
				new_items, err := MainView(m.ctx, m.client)
				if err != nil {
					fmt.Println(err.Error())
				}
				m.list.SetItems(new_items)
			}
		}
		if msg.String() == "ctrl+r" {
			switch m.list.SelectedItem().(mainItem).SpotifyItem.(type) {
			case spotify.SimplePlaylist:
				currentlyPlaying = m.list.SelectedItem().FilterValue()
				m.list.NewStatusMessage("Starting radio for " + currentlyPlaying)
				go HandlePlaylistRadio(m.ctx, m.client, m.list.SelectedItem().(mainItem).SpotifyItem.(spotify.SimplePlaylist))
			case *spotify.SavedTrackPage:
				currentlyPlaying = m.list.SelectedItem().FilterValue()
				m.list.NewStatusMessage("Starting radio for " + currentlyPlaying)
				go HandleLibraryRadio(m.ctx, m.client)
			case *spotify.SimpleAlbum, *spotify.FullAlbum:
				currentlyPlaying = m.list.SelectedItem().FilterValue()
				m.list.NewStatusMessage("Stating radio for" + currentlyPlaying)
				go HandleAlbumRadio(m.ctx, m.client, m.list.SelectedItem().(mainItem).SpotifyItem.(spotify.SimpleAlbum).ID)
			default:
				currentlyPlaying = m.list.SelectedItem().FilterValue()
				m.list.NewStatusMessage("Playing " + currentlyPlaying)
				go HandleRadio(m.ctx, m.client, m.list.SelectedItem().(mainItem).ID)
			}
		}
	case tea.MouseMsg:
		if msg.Type == 5 {
			m.list.CursorUp()
		}
		if msg.Type == 6 {
			m.list.CursorDown()
		}
	case tea.WindowSizeMsg:
		h, v := DocStyle.GetFrameSize()
		m.list.SetSize(msg.Width-h, msg.Height-v-1)
	}
	var cmd tea.Cmd
	m.list, cmd = m.list.Update(msg)
	return m, cmd
}

func InitMain(ctx *gctx.Context, client *spotify.Client, mode Mode) (tea.Model, error) {
	playing, _ := client.PlayerCurrentlyPlaying(ctx)
	if playing.Playing {
		currentlyPlaying = "Now playing " + playing.Item.Name + " by " + playing.Item.Artists[0].Name
	}
	items := []list.Item{}
	var err error
	switch mode {
	case Main:
		items, err = MainView(ctx, client)
		if err != nil {
			return nil, err
		}
	case Devices:
		items, err = DeviceView(ctx, client)
		if err != nil {
			return nil, err
		}
	case Tracks:
		items, err = SavedTracksView(ctx, client)
		if err != nil {
			return nil, err
		}
	}
	m := mainModel{
		list:   list.New(items, list.NewDefaultDelegate(), 0, 0),
		ctx:    ctx,
		client: client,
		mode:   mode,
	}
	m.list.Title = "GOSPT"
	go m.Tick()
	m.list.DisableQuitKeybindings()
	m.list.AdditionalShortHelpKeys = func() []key.Binding {
		return []key.Binding{
			key.NewBinding(key.WithKeys("esc"), key.WithHelp("esc", "back")),
			key.NewBinding(key.WithKeys("s"), key.WithHelp("s", "search")),
			key.NewBinding(key.WithKeys("ctrl+c"), key.WithHelp("ctrl+c", "quit")),
			key.NewBinding(key.WithKeys("ctrl"+"r"), key.WithHelp("ctrl+r", "start radio")),
			key.NewBinding(key.WithKeys("d"), key.WithHelp("d", "select device")),
		}
	}
	m.list.AdditionalFullHelpKeys = func() []key.Binding {
		return []key.Binding{
			key.NewBinding(key.WithKeys("esc"), key.WithHelp("esc", "back")),
			key.NewBinding(key.WithKeys("s"), key.WithHelp("s", "search")),
			key.NewBinding(key.WithKeys("ctrl+c"), key.WithHelp("ctrl+c", "quit")),
			key.NewBinding(key.WithKeys("ctrl"+"r"), key.WithHelp("ctrl+r", "start radio")),
			key.NewBinding(key.WithKeys("d"), key.WithHelp("d", "select device")),
		}
	}
	input := textinput.New()
	input.Prompt = "$ "
	input.Placeholder = "Search..."
	input.CharLimit = 250
	input.Width = 50
	m.input = input
	return m, nil
}