navidrome/server/subsonic/media_annotation.go

package subsonic

import (
	"net/http"
	"time"

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

type MediaAnnotationController struct {
	scrobbler engine.Scrobbler
	ratings   engine.Ratings
}

func NewMediaAnnotationController(scrobbler engine.Scrobbler, ratings engine.Ratings) *MediaAnnotationController {
	return &MediaAnnotationController{
		scrobbler: scrobbler,
		ratings:   ratings,
	}
}

func (c *MediaAnnotationController) SetRating(w http.ResponseWriter, r *http.Request) (*responses.Subsonic, error) {
	id, err := RequiredParamString(r, "id", "Required id parameter is missing")
	if err != nil {
		return nil, err
	}
	rating, err := RequiredParamInt(r, "rating", "Required rating parameter is missing")
	if err != nil {
		return nil, err
	}

	log.Debug(r, "Setting rating", "rating", rating, "id", id)
	err = c.ratings.SetRating(r.Context(), id, rating)

	switch {
	case err == model.ErrNotFound:
		log.Error(r, err)
		return nil, NewError(responses.ErrorDataNotFound, "ID not found")
	case err != nil:
		log.Error(r, err)
		return nil, NewError(responses.ErrorGeneric, "Internal Error")
	}

	return NewResponse(), nil
}

func (c *MediaAnnotationController) getIds(r *http.Request) ([]string, error) {
	ids := ParamStrings(r, "id")
	albumIds := ParamStrings(r, "albumId")
	artistIds := ParamStrings(r, "artistId")

	if len(ids)+len(albumIds)+len(artistIds) == 0 {
		return nil, NewError(responses.ErrorMissingParameter, "Required id parameter is missing")
	}

	ids = append(ids, albumIds...)
	ids = append(ids, artistIds...)
	return ids, nil
}

func (c *MediaAnnotationController) Star(w http.ResponseWriter, r *http.Request) (*responses.Subsonic, error) {
	ids, err := c.getIds(r)
	if err != nil {
		return nil, err
	}
	log.Debug(r, "Starring items", "ids", ids)
	err = c.ratings.SetStar(r.Context(), true, ids...)
	switch {
	case err == model.ErrNotFound:
		log.Error(r, err)
		return nil, NewError(responses.ErrorDataNotFound, "ID not found")
	case err != nil:
		log.Error(r, err)
		return nil, NewError(responses.ErrorGeneric, "Internal Error")
	}

	return NewResponse(), nil
}

func (c *MediaAnnotationController) Unstar(w http.ResponseWriter, r *http.Request) (*responses.Subsonic, error) {
	ids, err := c.getIds(r)
	if err != nil {
		return nil, err
	}
	log.Debug(r, "Unstarring items", "ids", ids)
	err = c.ratings.SetStar(r.Context(), false, ids...)
	switch {
	case err == model.ErrNotFound:
		log.Error(r, err)
		return nil, NewError(responses.ErrorDataNotFound, "Directory not found")
	case err != nil:
		log.Error(r, err)
		return nil, NewError(responses.ErrorGeneric, "Internal Error")
	}

	return NewResponse(), nil
}

func (c *MediaAnnotationController) Scrobble(w http.ResponseWriter, r *http.Request) (*responses.Subsonic, error) {
	ids, err := RequiredParamStrings(r, "id", "Required id parameter is missing")
	if err != nil {
		return nil, err
	}
	times := ParamTimes(r, "time")
	if len(times) > 0 && len(times) != len(ids) {
		return nil, NewError(responses.ErrorGeneric, "Wrong number of timestamps: %d, should be %d", len(times), len(ids))
	}
	submission := ParamBool(r, "submission", true)
	playerId := 1 // TODO Multiple players, based on playerName/username/clientIP(?)
	playerName := ParamString(r, "c")
	username := ParamString(r, "u")

	log.Debug(r, "Scrobbling tracks", "ids", ids, "times", times, "submission", submission)
	for i, id := range ids {
		var t time.Time
		if len(times) > 0 {
			t = times[i]
		} else {
			t = time.Now()
		}
		if submission {
			mf, err := c.scrobbler.Register(r.Context(), playerId, id, t)
			if err != nil {
				log.Error(r, "Error scrobbling track", "id", id, err)
				continue
			}
			log.Info(r, "Scrobbled", "id", id, "title", mf.Title, "timestamp", t)
		} else {
			mf, err := c.scrobbler.NowPlaying(r.Context(), playerId, playerName, id, username)
			if err != nil {
				log.Error(r, "Error setting current song", "id", id, err)
				continue
			}
			log.Info(r, "Now Playing", "id", id, "title", mf.Title, "timestamp", t)
		}
	}
	return NewResponse(), nil
}