package src

import (
	"github.com/gdamore/tcell/v2"
	"github.com/rivo/tview"
)

func (a *app) setupMusicControlKeys(p *tview.Box) {
	// Add 'k' and 'l' key bindings
	p.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
		if event.Rune() == 'l' {
			a.playQueue.Next()
			return nil
		}

		if event.Rune() == 'p' {
			a.playQueue.TogglePause()
			return nil
		}

		if a.tv.GetFocus() == a.playQueueList {
			if event.Rune() == 'd' {
				sel := a.playQueueList.GetCurrentItem()
				err := a.playQueue.RemoveSong(sel)
				if err != nil {
					a.alert("Error: %v", err)
				}
			} else if event.Rune() == 'k' {
				sel := a.playQueueList.GetCurrentItem()
				if sel == a.playQueueList.GetItemCount()-1 {
					return nil
				}
				err := a.playQueue.Switch(sel, sel+1)
				if err != nil {
					a.alert("Error: %v", err)
				}

				a.playQueueList.SetCurrentItem(sel + 1)

				return nil
			} else if event.Rune() == 'j' {
				sel := a.playQueueList.GetCurrentItem()
				if sel == 0 {
					return nil
				}
				err := a.playQueue.Switch(sel, sel-1)
				if err != nil {
					a.alert("Error: %v", err)
				}

				a.playQueueList.SetCurrentItem(sel - 1)

				return nil
			} else if event.Rune() == 'r' {
				a.playQueue.Stop()
				songs := a.playQueue.GetSongs()
				a.playQueue.Clear()
				for _, s := range randomize(songs) {
					a.playQueue.Append(s)
				}

				a.playQueue.Play()
				a.updatePageQueue()
			}
		}

		if a.tv.GetFocus() == a.songsList {
			if event.Rune() == 'e' {
				// Add to end
				sel := a.songsList.GetCurrentItem()
				a.playQueue.Append(a.currentSongs[sel])

				a.updatePageQueue()
			} else if event.Rune() == 'n' {
				// Add next
				sel := a.songsList.GetCurrentItem()
				a.playQueue.Insert(1, a.currentSongs[sel])

				a.updatePageQueue()
			}
		} else if a.tv.GetFocus() == a.artistsTree {
			if event.Rune() == 'e' {
				// Add to end
				sel := a.artistsTree.GetCurrentNode()
				ref := sel.GetReference()
				if ref == nil {
					return nil
				}

				if ref.(selection).entryType != "album" {
					return nil
				}

				id := ref.(selection).id
				album, err := a.sub.GetMusicDirectory(id)
				if err != nil {
					a.alert("Error: %v", err)
					return nil
				}

				for _, s := range album.Child {
					a.playQueue.Append(s)
				}

				a.updatePageQueue()
			} else if event.Rune() == 'n' {
				// Add next
				sel := a.artistsTree.GetCurrentNode()
				ref := sel.GetReference()
				if ref == nil {
					return nil
				}

				if ref.(selection).entryType != "album" {
					return nil
				}

				id := ref.(selection).id
				album, err := a.sub.GetMusicDirectory(id)
				if err != nil {
					a.alert("Error: %v", err)
					return nil
				}

				for i := len(album.Child) - 1; i >= 0; i-- {
					a.playQueue.Insert(1, album.Child[i])
				}

				a.updatePageQueue()
			}
		}

		if a.tv.GetFocus() == a.playlistsList {
			if event.Rune() == 'e' {
				sel := a.playlistsList.GetCurrentItem()
				pl, err := a.sub.GetPlaylist(a.allPlaylists[sel].ID)
				if err != nil {
					a.alert("Error: %v", err)
					return nil
				}

				for _, s := range pl.Entry {
					a.playQueue.Append(s)
				}

				a.updatePageQueue()
			} else if event.Rune() == 'n' {
				sel := a.playlistsList.GetCurrentItem()
				pl, err := a.sub.GetPlaylist(a.allPlaylists[sel].ID)
				if err != nil {
					a.alert("Error: %v", err)
					return nil
				}

				for i := len(pl.Entry) - 1; i >= 0; i-- {
					a.playQueue.Insert(1, pl.Entry[i])
				}

				a.updatePageQueue()
			} else if event.Rune() == 'r' {
				sel := a.playlistsList.GetCurrentItem()
				pl, err := a.sub.GetPlaylist(a.allPlaylists[sel].ID)
				if err != nil {
					a.alert("Error: %v", err)
					return nil
				}

				a.playQueue.Clear()
				for _, s := range randomize(pl.Entry) {
					a.playQueue.Append(s)
				}
				a.playQueue.Play()

				a.updatePageQueue()
			}
		}

		if a.tv.GetFocus() == a.playlistSongs {
			if event.Rune() == 'e' {
				sel := a.playlistSongs.GetCurrentItem()
				a.playQueue.Append(a.currentPlaylist.Entry[sel])

				a.updatePageQueue()
			} else if event.Rune() == 'n' {
				sel := a.playlistSongs.GetCurrentItem()
				a.playQueue.Insert(1, a.currentPlaylist.Entry[sel])

				a.updatePageQueue()
			}
		}

		return event
	})
}