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 }