2016-03-23 22:46:54 +01:00
|
|
|
package engine
|
|
|
|
|
|
|
|
import (
|
2020-01-09 02:45:07 +01:00
|
|
|
"context"
|
|
|
|
|
2020-01-24 01:44:08 +01:00
|
|
|
"github.com/deluan/navidrome/log"
|
|
|
|
"github.com/deluan/navidrome/model"
|
2016-03-23 22:46:54 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
type Ratings interface {
|
2020-01-09 02:45:07 +01:00
|
|
|
SetStar(ctx context.Context, star bool, ids ...string) error
|
|
|
|
SetRating(ctx context.Context, id string, rating int) error
|
2016-03-23 22:46:54 +01:00
|
|
|
}
|
|
|
|
|
2020-01-19 21:37:41 +01:00
|
|
|
func NewRatings(ds model.DataStore) Ratings {
|
|
|
|
return &ratings{ds}
|
2016-03-23 22:46:54 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
type ratings struct {
|
2020-01-19 21:37:41 +01:00
|
|
|
ds model.DataStore
|
2016-03-23 22:46:54 +01:00
|
|
|
}
|
|
|
|
|
2020-01-09 02:45:07 +01:00
|
|
|
func (r ratings) SetRating(ctx context.Context, id string, rating int) error {
|
2020-01-27 15:41:33 +01:00
|
|
|
exist, err := r.ds.Album(ctx).Exists(id)
|
2020-01-22 05:01:43 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if exist {
|
2020-02-01 03:09:23 +01:00
|
|
|
return r.ds.Album(ctx).SetRating(rating, id)
|
2020-01-22 05:01:43 +01:00
|
|
|
}
|
2020-02-01 03:09:23 +01:00
|
|
|
return r.ds.MediaFile(ctx).SetRating(rating, id)
|
2016-03-24 00:37:48 +01:00
|
|
|
}
|
|
|
|
|
2020-01-09 02:45:07 +01:00
|
|
|
func (r ratings) SetStar(ctx context.Context, star bool, ids ...string) error {
|
2020-01-22 05:01:43 +01:00
|
|
|
if len(ids) == 0 {
|
|
|
|
log.Warn(ctx, "Cannot star/unstar an empty list of ids")
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2020-01-19 21:37:41 +01:00
|
|
|
return r.ds.WithTx(func(tx model.DataStore) error {
|
2020-01-22 05:01:43 +01:00
|
|
|
for _, id := range ids {
|
2020-01-27 15:41:33 +01:00
|
|
|
exist, err := r.ds.Album(ctx).Exists(id)
|
2020-01-22 05:01:43 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if exist {
|
2020-02-01 03:09:23 +01:00
|
|
|
err = tx.Album(ctx).SetStar(star, ids...)
|
2020-01-22 05:01:43 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
continue
|
|
|
|
}
|
2020-01-27 15:41:33 +01:00
|
|
|
exist, err = r.ds.Artist(ctx).Exists(id)
|
2020-01-22 05:01:43 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if exist {
|
2020-02-01 03:09:23 +01:00
|
|
|
err = tx.Artist(ctx).SetStar(star, ids...)
|
2020-01-22 05:01:43 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
continue
|
|
|
|
}
|
2020-02-01 03:09:23 +01:00
|
|
|
err = tx.MediaFile(ctx).SetStar(star, ids...)
|
2020-01-22 05:01:43 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2020-01-19 02:03:52 +01:00
|
|
|
}
|
2020-01-22 05:01:43 +01:00
|
|
|
return nil
|
2020-01-19 21:37:41 +01:00
|
|
|
})
|
2016-03-23 22:46:54 +01:00
|
|
|
}
|