package gomus

import (
	"time"

	tea "github.com/charmbracelet/bubbletea"
	"github.com/faiface/beep/speaker"
)

var (
	termWidth  = 0
	termHeight = 0
)

type ModelArgs struct {
	MusicPath string
}

type Model struct {
	cursor           int
	currentlyPlaying int

	TrackPlayer
	TrackPlayerEffects
	trackIndex

	trackPlayerView
}

func NewModel(args ModelArgs) Model {
	ti := NewDirTrackIndex(args.MusicPath)
	tpv := newTrackPlayerView(ti.tracks)

	return Model{
		cursor:           0,
		currentlyPlaying: 0,

		trackIndex:         ti,
		TrackPlayer:        TrackPlayer{},
		TrackPlayerEffects: newTrackPlayerEffects(),

		trackPlayerView: tpv,
	}
}

func (m Model) Init() tea.Cmd {
	return tea.Batch(tea.EnterAltScreen, m.trackPlayerView.Init())
}

func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	var cmds []tea.Cmd

	switch msg := msg.(type) {
	case tea.WindowSizeMsg:
		termHeight = msg.Height
		termWidth = msg.Width
	case tea.KeyMsg:
		switch msg.String() {
		case "ctrl+c", "q":
			m.TrackPlayer.Close()
			return m, tea.Quit
		case "enter":
			t := m.trackPlayerView.trackList.SelectedItem().(track)

			stream, format, err := t.GetStream()
			check(err)

			speaker.Init(format.SampleRate, format.SampleRate.N(time.Second/10))

			m.TrackPlayer.Play(&stream, &m.TrackPlayerEffects)
			cmds = append(cmds, newTrackChangeCmd(t))
		case " ":
			if m.TrackPlayer.playerCtrl != nil {
				s := m.TrackPlayer.TogglePause()
				cmds = append(cmds, newTrackPauseCmd(s))
			}
		case "-", "=":
			v := &m.TrackPlayerEffects.volume

			if msg.String() == "-" {
				*v -= 0.1
				if *v < minVolume {
					*v = minVolume
				}
			} else {
				*v += 0.1
				if *v > maxVolume {
					*v = maxVolume
				}
			}

			if m.TrackPlayer.playerVol != nil {
				m.TrackPlayer.SetVolume(*v)
			}

			cmds = append(cmds, newTrackVolumeCmd(*v))
		}
	}

	var cmd tea.Cmd
	m.trackPlayerView, cmd = m.trackPlayerView.Update(msg)
	cmds = append(cmds, cmd)

	return m, tea.Batch(cmds...)
}

func (m Model) View() string {
	return m.trackPlayerView.View()
}