gospt/internal/tui/main.go

470 lines
13 KiB
Go
Raw Normal View History

2023-01-09 04:42:02 +00:00
package tui
import (
"fmt"
2023-01-11 17:45:29 +00:00
"time"
2023-01-09 04:42:02 +00:00
"gospt/internal/commands"
"gospt/internal/gctx"
2023-01-11 17:45:29 +00:00
"github.com/charmbracelet/bubbles/key"
2023-01-09 04:42:02 +00:00
"github.com/charmbracelet/bubbles/list"
tea "github.com/charmbracelet/bubbletea"
2023-01-11 17:45:29 +00:00
"github.com/charmbracelet/lipgloss"
2023-01-09 04:42:02 +00:00
"github.com/zmb3/spotify/v2"
)
2023-01-11 17:45:29 +00:00
var (
currentlyPlaying string
main_updates chan *mainModel
page = 1
docStyle = lipgloss.NewStyle().Margin(1, 2)
)
2023-01-11 04:46:15 +00:00
2023-01-09 04:42:02 +00:00
type mainItem struct {
Name string
2023-01-11 17:45:29 +00:00
Duration string
Artist spotify.SimpleArtist
ID spotify.ID
2023-01-09 04:42:02 +00:00
Desc string
SpotifyItem any
2023-01-11 17:45:29 +00:00
Device spotify.PlayerDevice
spotify.SavedTrack
2023-01-09 04:42:02 +00:00
}
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 {
2023-01-11 17:45:29 +00:00
list list.Model
ctx *gctx.Context
client *spotify.Client
mode string
playlist spotify.SimplePlaylist
2023-01-09 04:42:02 +00:00
}
func (m mainModel) Init() tea.Cmd {
2023-01-11 17:45:29 +00:00
main_updates = make(chan *mainModel)
2023-01-09 04:42:02 +00:00
return nil
}
2023-01-11 17:45:29 +00:00
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)
2023-01-11 21:30:55 +00:00
if playing.Playing {
currentlyPlaying = "Now playing " + playing.Item.Name + " by " + playing.Item.Artists[0].Name
}
2023-01-11 17:45:29 +00:00
case <-quit:
ticker.Stop()
return
}
}
}()
}
func HandlePlay(ctx *gctx.Context, client *spotify.Client, uri *spotify.URI, pos int) {
var err error
err = commands.PlaySongInPlaylist(ctx, client, uri, pos)
2023-01-11 04:46:15 +00:00
if err != nil {
2023-01-11 21:30:55 +00:00
return
2023-01-11 04:46:15 +00:00
}
2023-01-11 17:45:29 +00:00
}
func HandleRadio(ctx *gctx.Context, client *spotify.Client, id spotify.ID) {
err := commands.RadioGivenSong(ctx, client, id, 0)
if err != nil {
2023-01-11 21:30:55 +00:00
return
}
}
func HandlePlaylistRadio(ctx *gctx.Context, client *spotify.Client, playlist spotify.SimplePlaylist) {
err := commands.RadioFromPlaylist(ctx, client, playlist)
if err != nil {
return
}
}
func HandleLibraryRadio(ctx *gctx.Context, client *spotify.Client) {
err := commands.RadioFromSavedTracks(ctx, client)
if err != nil {
fmt.Println(err.Error())
return
2023-01-11 04:46:15 +00:00
}
2023-01-11 17:45:29 +00:00
}
func (m *mainModel) LoadMoreItems() {
switch m.mode {
case "main":
playlists, err := commands.Playlists(m.ctx, m.client, (page + 1))
page++
if err != nil {
return
}
items := []list.Item{}
for _, playlist := range playlists.Playlists {
items = append(items, mainItem{
Name: playlist.Name,
Desc: playlist.Description,
SpotifyItem: playlist,
})
}
for _, item := range items {
m.list.InsertItem(len(m.list.Items())+1, item)
}
main_updates <- m
return
case "playlist":
tracks, err := commands.PlaylistTracks(m.ctx, m.client, m.playlist.ID, (page + 1))
page++
if err != nil {
return
}
items := []mainItem{}
for _, track := range tracks.Tracks {
items = append(items, mainItem{
Name: track.Track.Name,
Artist: track.Track.Artists[0],
Duration: track.Track.TimeDuration().Round(time.Second).String(),
ID: track.Track.ID,
Desc: track.Track.Artists[0].Name + " - " + track.Track.TimeDuration().Round(time.Second).String(),
})
}
for _, item := range items {
m.list.InsertItem(len(m.list.Items())+1, item)
}
main_updates <- m
return
case "tracks":
tracks, err := commands.TrackList(m.ctx, m.client, (page + 1))
page++
if err != nil {
return
}
page++
items := []list.Item{}
for _, track := range tracks.Tracks {
items = append(items, mainItem{
Name: track.Name,
Artist: track.Artists[0],
Duration: track.TimeDuration().Round(time.Second).String(),
ID: track.ID,
Desc: track.Artists[0].Name + " - " + track.TimeDuration().Round(time.Second).String(),
})
}
for _, item := range items {
m.list.InsertItem(len(m.list.Items())+1, item)
}
main_updates <- m
return
}
}
func HandlePlayLikedSong(ctx *gctx.Context, client *spotify.Client, position int) {
err := commands.PlayLikedSongs(ctx, client, position)
if err != nil {
fmt.Println(err.Error())
2023-01-11 21:30:55 +00:00
return
2023-01-11 04:46:15 +00:00
}
}
2023-01-09 04:42:02 +00:00
func (m mainModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
2023-01-11 17:45:29 +00:00
m.list.NewStatusMessage(currentlyPlaying)
2023-01-11 04:46:15 +00:00
select {
2023-01-11 17:45:29 +00:00
case update := <-main_updates:
m.list.SetItems(update.list.Items())
2023-01-11 04:46:15 +00:00
default:
}
2023-01-11 17:45:29 +00:00
if m.list.Paginator.Page == m.list.Paginator.TotalPages-2 && m.list.Cursor() == 0 {
2023-01-11 04:46:15 +00:00
// if last request was still full request more
2023-01-09 04:42:02 +00:00
if len(m.list.Items())%50 == 0 {
2023-01-11 04:46:15 +00:00
go m.LoadMoreItems()
2023-01-09 04:42:02 +00:00
}
}
switch msg := msg.(type) {
case tea.KeyMsg:
2023-01-11 17:45:29 +00:00
if msg.String() == "d" {
m.mode = "devices"
new_items, err := DeviceView(m.ctx, m.client)
if err != nil {
fmt.Println(err.Error())
2023-01-11 21:30:55 +00:00
return m, tea.Quit
2023-01-11 17:45:29 +00:00
}
m.list.SetItems(new_items)
m.list.ResetSelected()
m.list.NewStatusMessage("Setting view to devices")
2023-01-09 04:42:02 +00:00
}
2023-01-11 17:45:29 +00:00
if msg.String() == "backspace" {
if m.mode == "playlist" || m.mode == "tracks" || m.mode == "devices" {
m.mode = "main"
m.list.NewStatusMessage("Setting view to main")
new_items, err := MainView(m.ctx, m.client)
2023-01-09 18:54:21 +00:00
if err != nil {
2023-01-11 17:45:29 +00:00
fmt.Println(err.Error())
2023-01-09 18:54:21 +00:00
}
2023-01-11 17:45:29 +00:00
m.list.SetItems(new_items)
} else {
return m, tea.Quit
}
m.list.ResetSelected()
}
if msg.String() == "ctrl+c" {
return m, tea.Quit
}
if msg.String() == "enter" || msg.String() == "spacebar" {
switch m.mode {
case "main":
switch m.list.SelectedItem().(mainItem).SpotifyItem.(type) {
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())
2023-01-11 21:30:55 +00:00
return m, tea.Quit
2023-01-11 17:45:29 +00:00
}
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())
2023-01-11 21:30:55 +00:00
return m, tea.Quit
2023-01-11 17:45:29 +00:00
}
m.list.SetItems(new_items)
m.list.ResetSelected()
m.list.NewStatusMessage("Setting view to tracks")
2023-01-09 18:54:21 +00:00
}
2023-01-11 17:45:29 +00:00
case "playlist":
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":
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).Device)
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)
2023-01-09 18:54:21 +00:00
if err != nil {
2023-01-11 17:45:29 +00:00
fmt.Println(err.Error())
2023-01-09 18:54:21 +00:00
}
2023-01-11 17:45:29 +00:00
m.list.SetItems(new_items)
}
}
if msg.String() == "ctrl+r" {
switch m.mode {
case "main":
switch m.list.SelectedItem().(mainItem).SpotifyItem.(type) {
case spotify.SimplePlaylist:
2023-01-11 21:30:55 +00:00
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))
2023-01-11 17:45:29 +00:00
case *spotify.SavedTrackPage:
2023-01-11 21:30:55 +00:00
currentlyPlaying = m.list.SelectedItem().FilterValue()
m.list.NewStatusMessage("Starting radio for " + currentlyPlaying)
go HandleLibraryRadio(m.ctx, m.client)
2023-01-09 18:54:21 +00:00
}
2023-01-11 17:45:29 +00:00
case "playlist":
currentlyPlaying = m.list.SelectedItem().FilterValue()
m.list.NewStatusMessage("Starting radio for " + currentlyPlaying)
go HandleRadio(m.ctx, m.client, m.list.SelectedItem().(mainItem).ID)
case "tracks":
currentlyPlaying = m.list.SelectedItem().FilterValue()
m.list.NewStatusMessage("Playing " + currentlyPlaying)
go HandleRadio(m.ctx, m.client, m.list.SelectedItem().(mainItem).ID)
2023-01-09 04:42:02 +00:00
}
}
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)
}
var cmd tea.Cmd
m.list, cmd = m.list.Update(msg)
return m, cmd
}
func (m mainModel) View() string {
return docStyle.Render(m.list.View())
}
func DisplayMain(ctx *gctx.Context, client *spotify.Client) error {
items := []list.Item{}
saved_items, err := commands.TrackList(ctx, client, 1)
items = append(items, mainItem{
Name: "Saved Tracks",
Desc: fmt.Sprintf("%d saved songs", saved_items.Total),
SpotifyItem: saved_items,
})
playlists, err := commands.Playlists(ctx, client, 1)
if err != nil {
return err
}
for _, playlist := range playlists.Playlists {
items = append(items, mainItem{
Name: playlist.Name,
Desc: playlist.Description,
SpotifyItem: playlist,
})
}
m := mainModel{
list: list.New(items, list.NewDefaultDelegate(), 0, 0),
ctx: ctx,
client: client,
2023-01-11 17:45:29 +00:00
mode: "main",
2023-01-09 04:42:02 +00:00
}
2023-01-09 06:50:27 +00:00
m.list.Title = "GOSPT"
2023-01-11 17:45:29 +00:00
go m.Tick()
2023-01-09 04:42:02 +00:00
p := tea.NewProgram(m, tea.WithAltScreen(), tea.WithMouseCellMotion())
if _, err := p.Run(); err != nil {
fmt.Println("Error running program:", err)
2023-01-11 21:30:55 +00:00
return err
2023-01-09 04:42:02 +00:00
}
return nil
}
2023-01-09 18:54:21 +00:00
2023-01-11 17:45:29 +00:00
func PlaylistView(ctx *gctx.Context, client *spotify.Client, playlist spotify.SimplePlaylist) ([]list.Item, error) {
items := []list.Item{}
tracks, err := commands.PlaylistTracks(ctx, client, playlist.ID, 1)
if err != nil {
return nil, err
}
for _, track := range tracks.Tracks {
items = append(items, mainItem{
Name: track.Track.Name,
Artist: track.Track.Artists[0],
Duration: track.Track.TimeDuration().Round(time.Second).String(),
ID: track.Track.ID,
Desc: track.Track.Artists[0].Name + " - " + track.Track.TimeDuration().Round(time.Second).String(),
})
}
return items, nil
}
func SavedTracksView(ctx *gctx.Context, client *spotify.Client) ([]list.Item, error) {
items := []list.Item{}
tracks, err := commands.TrackList(ctx, client, 1)
if err != nil {
return nil, err
}
for _, track := range tracks.Tracks {
items = append(items, mainItem{
Name: track.Name,
Artist: track.Artists[0],
Duration: track.TimeDuration().Round(time.Second).String(),
ID: track.ID,
Desc: track.Artists[0].Name + " - " + track.TimeDuration().Round(time.Second).String(),
})
}
return items, err
}
func MainView(ctx *gctx.Context, client *spotify.Client) ([]list.Item, error) {
2023-01-09 18:54:21 +00:00
items := []list.Item{}
saved_items, err := commands.TrackList(ctx, client, 1)
items = append(items, mainItem{
Name: "Saved Tracks",
Desc: fmt.Sprintf("%d saved songs", saved_items.Total),
SpotifyItem: saved_items,
})
playlists, err := commands.Playlists(ctx, client, 1)
if err != nil {
return nil, err
}
for _, playlist := range playlists.Playlists {
items = append(items, mainItem{
Name: playlist.Name,
Desc: playlist.Description,
SpotifyItem: playlist,
})
}
2023-01-11 17:45:29 +00:00
return items, nil
}
func InitMain(ctx *gctx.Context, client *spotify.Client, mode string) (tea.Model, error) {
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
}
}
2023-01-09 18:54:21 +00:00
m := mainModel{
list: list.New(items, list.NewDefaultDelegate(), 0, 0),
ctx: ctx,
client: client,
2023-01-11 17:45:29 +00:00
mode: mode,
2023-01-09 18:54:21 +00:00
}
m.list.Title = "GOSPT"
2023-01-11 17:45:29 +00:00
go m.Tick()
m.list.AdditionalShortHelpKeys = func() []key.Binding {
return []key.Binding{
key.NewBinding(key.WithKeys("ctrl"+"r"), key.WithHelp("ctrl+r", "start radio")),
key.NewBinding(key.WithKeys("d"), key.WithHelp("d", "select device")),
}
}
2023-01-09 18:54:21 +00:00
return m, nil
}
2023-01-11 17:45:29 +00:00
func DeviceView(ctx *gctx.Context, client *spotify.Client) ([]list.Item, error) {
items := []list.Item{}
devices, err := client.PlayerDevices(ctx)
if err != nil {
return nil, err
}
for _, device := range devices {
items = append(items, mainItem{
Name: device.Name,
Desc: fmt.Sprintf("%s - active: %t", device.ID, device.Active),
Device: device,
})
}
return items, nil
}
func HandleSetDevice(ctx *gctx.Context, client *spotify.Client, player spotify.PlayerDevice) {
fmt.Println("WHOA")
var err error
err = commands.SetDevice(ctx, client, player)
if err != nil {
fmt.Println(err.Error())
2023-01-11 21:30:55 +00:00
return
2023-01-11 17:45:29 +00:00
}
}