73 lines
1.6 KiB
Go
73 lines
1.6 KiB
Go
package engine
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/deluan/navidrome/log"
|
|
"github.com/deluan/navidrome/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(ctx).Exists(id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if exist {
|
|
return r.ds.Annotation(ctx).SetRating(rating, getUserID(ctx), model.AlbumItemType, id)
|
|
}
|
|
return r.ds.Annotation(ctx).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(ctx).Exists(id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if exist {
|
|
err = tx.Annotation(ctx).SetStar(star, userId, model.AlbumItemType, ids...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
exist, err = r.ds.Artist(ctx).Exists(id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if exist {
|
|
err = tx.Annotation(ctx).SetStar(star, userId, model.ArtistItemType, ids...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
err = tx.Annotation(ctx).SetStar(star, userId, model.MediaItemType, ids...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
}
|