setRating.view implemented

engine/ratings.go

@@ -4,10 +4,12 @@ import (
 	"github.com/astaxie/beego"
 	"github.com/deluan/gosonic/domain"
 	"github.com/deluan/gosonic/itunesbridge"
+	"github.com/deluan/gosonic/utils"
 )
 
 type Ratings interface {
 	SetStar(star bool, ids ...string) error
+	SetRating(id string, rating int) error
 }
 
 func NewRatings(itunes itunesbridge.ItunesControl, mr domain.MediaFileRepository, alr domain.AlbumRepository, ar domain.ArtistRepository) Ratings {
@@ -21,6 +23,35 @@ type ratings struct {
 	artistRepo domain.ArtistRepository
 }
 
+func (r ratings) SetRating(id string, rating int) error {
+	rating = utils.MinInt(rating, 5) * 20
+
+	isAlbum, _ := r.albumRepo.Exists(id)
+	if isAlbum {
+		mfs, _ := r.mfRepo.FindByAlbum(id)
+		if len(mfs) > 0 {
+			beego.Debug("SetRating:", rating, "Album:", mfs[0].Album)
+			if err := r.itunes.SetAlbumRating(mfs[0].Id, rating); err != nil {
+				return err
+			}
+		}
+		return nil
+	}
+
+	mf, err := r.mfRepo.Get(id)
+	if err != nil {
+		return err
+	}
+	if mf != nil {
+		beego.Debug("SetRating:", rating, "Song:", mf.Title)
+		if err := r.itunes.SetTrackRating(mf.Id, rating); err != nil {
+			return err
+		}
+		return nil
+	}
+	return domain.ErrNotFound
+}
+
 func (r ratings) SetStar(star bool, ids ...string) error {
 	for _, id := range ids {
 		isAlbum, _ := r.albumRepo.Exists(id)