package engine_test

import (
	"testing"
	"time"

	"github.com/deluan/gosonic/engine"
	"github.com/deluan/gosonic/itunesbridge"
	"github.com/deluan/gosonic/persistence"
	. "github.com/deluan/gosonic/tests"
	. "github.com/smartystreets/goconvey/convey"
	"github.com/syndtr/goleveldb/leveldb/errors"
)

func TestScrobbler(t *testing.T) {

	Init(t, false)

	mfRepo := persistence.CreateMockMediaFileRepo()
	npRepo := engine.CreateMockNowPlayingRepo()
	itCtrl := &mockItunesControl{}

	scrobbler := engine.NewScrobbler(itCtrl, mfRepo, npRepo)

	Convey("Given a DB with one song", t, func() {
		mfRepo.SetData(`[{"Id":"2","Title":"Hands Of Time"}]`, 1)

		Convey("When I scrobble an existing song", func() {
			now := time.Now()
			mf, err := scrobbler.Register(1, "2", now)

			Convey("Then I get the scrobbled song back", func() {
				So(err, ShouldBeNil)
				So(mf.Title, ShouldEqual, "Hands Of Time")
			})

			Convey("And iTunes is notified", func() {
				So(itCtrl.played, ShouldContainKey, "2")
				So(itCtrl.played["2"].Equal(now), ShouldBeTrue)
			})

		})

		Convey("When the ID is not in the DB", func() {
			_, err := scrobbler.Register(1, "3", time.Now())

			Convey("Then I receive an error", func() {
				So(err, ShouldNotBeNil)
			})

			Convey("And iTunes is not notified", func() {
				So(itCtrl.played, ShouldNotContainKey, "3")
			})
		})

		Convey("When I inform the song that is now playing", func() {
			mf, err := scrobbler.NowPlaying(1, "DSub", "2", "deluan")

			Convey("Then I get the song for that id back", func() {
				So(err, ShouldBeNil)
				So(mf.Title, ShouldEqual, "Hands Of Time")
			})

			Convey("And it saves the song as the one current playing", func() {
				info, _ := npRepo.Head(1)
				So(info.TrackId, ShouldEqual, "2")
				So(info.Start, ShouldHappenBefore, time.Now())
				So(info.Username, ShouldEqual, "deluan")
				So(info.PlayerName, ShouldEqual, "DSub")
			})

			Convey("And iTunes is not notified", func() {
				So(itCtrl.played, ShouldNotContainKey, "2")
			})
		})

		Reset(func() {
			itCtrl.played = make(map[string]time.Time)
			itCtrl.skipped = make(map[string]time.Time)
		})

	})
	Convey("Given a DB with three songs", t, func() {
		mfRepo.SetData(`[{"Id":"1","Title":"Femme Fatale"},{"Id":"2","Title":"Here She Comes Now"},{"Id":"3","Title":"Lady Godiva's Operation"}]`, 3)
		itCtrl.skipped = make(map[string]time.Time)
		npRepo.ClearAll()
		Convey("When I play one song", func() {
			scrobbler.NowPlaying(1, "DSub", "1", "deluan")
			Convey("And I play the other song without scrobbling the first one", func() {
				scrobbler.NowPlaying(1, "DSub", "2", "deluan")
				mf, err := scrobbler.Register(1, "2", time.Now())
				Convey("Then the first song should be marked as skipped", func() {
					So(mf.Id, ShouldEqual, "2")
					So(itCtrl.skipped, ShouldContainKey, "1")
					So(err, ShouldBeNil)
				})
			})
			Convey("And I scrobble it before starting to play the other song", func() {
				mf, err := scrobbler.Register(1, "1", time.Now())
				Convey("Then the first song should NOT marked as skipped", func() {
					So(mf.Id, ShouldEqual, "1")
					So(itCtrl.skipped, ShouldBeEmpty)
					So(err, ShouldBeNil)
				})
			})
		})
		Convey("When the NowPlaying for the next song happens before the Scrobble", func() {
			scrobbler.NowPlaying(1, "DSub", "2", "deluan")
			scrobbler.Register(1, "1", time.Now())
			Convey("Then the NowPlaying info is not not changed", func() {
				np, _ := npRepo.GetAll()
				So(np, ShouldHaveLength, 1)
				So(np[0].TrackId, ShouldEqual, "2")
			})
		})
	})
}

type mockItunesControl struct {
	itunesbridge.ItunesControl
	played  map[string]time.Time
	skipped map[string]time.Time
	error   bool
}

func (m *mockItunesControl) MarkAsPlayed(id string, playDate time.Time) error {
	if m.error {
		return errors.New("ID not found")
	}
	if m.played == nil {
		m.played = make(map[string]time.Time)
	}
	m.played[id] = playDate
	return nil
}

func (m *mockItunesControl) MarkAsSkipped(id string, skipDate time.Time) error {
	if m.error {
		return errors.New("ID not found")
	}
	if m.skipped == nil {
		m.skipped = make(map[string]time.Time)
	}
	m.skipped[id] = skipDate
	return nil
}