navidrome/engine/ratings.go

package engine

import (
	"context"

	"github.com/cloudsonic/sonic-server/log"
	"github.com/cloudsonic/sonic-server/model"
)

type Ratings interface {
	SetStar(ctx context.Context, star bool, ids ...string) error
	SetRating(ctx context.Context, id string, rating int) error
}

func NewRatings(ds model.DataStore) Ratings {
	return &ratings{ds}
}

type ratings struct {
	ds model.DataStore
}

func (r ratings) SetRating(ctx context.Context, id string, rating int) error {
	exist, err := r.ds.Album().Exists(id)
	if err != nil {
		return err
	}
	if exist {
		return r.ds.Annotation().SetRating(rating, getUserID(ctx), model.AlbumItemType, id)
	}
	return r.ds.Annotation().SetRating(rating, getUserID(ctx), model.MediaItemType, id)
}

func (r ratings) SetStar(ctx context.Context, star bool, ids ...string) error {
	if len(ids) == 0 {
		log.Warn(ctx, "Cannot star/unstar an empty list of ids")
		return nil
	}
	userId := getUserID(ctx)

	return r.ds.WithTx(func(tx model.DataStore) error {
		for _, id := range ids {
			exist, err := r.ds.Album().Exists(id)
			if err != nil {
				return err
			}
			if exist {
				err = tx.Annotation().SetStar(star, userId, model.AlbumItemType, ids...)
				if err != nil {
					return err
				}
				continue
			}
			exist, err = r.ds.Artist().Exists(id)
			if err != nil {
				return err
			}
			if exist {
				err = tx.Annotation().SetStar(star, userId, model.ArtistItemType, ids...)
				if err != nil {
					return err
				}
				continue
			}
			err = tx.Annotation().SetStar(star, userId, model.MediaItemType, ids...)
			if err != nil {
				return err
			}
		}
		return nil
	})
}