Subsonic: annotation endpoints (star, unstar, setRating, scrobble)

internal/subsonic/annotation.go

@@ -0,0 +1,124 @@
+package subsonic
+
+import (
+	"git.noroutes.net/NoRoutes/vibetunes/internal/database"
+)
+
+func init() {
+	register("star", handleStar)
+	register("unstar", handleUnstar)
+	register("setRating", handleSetRating)
+	register("scrobble", handleScrobble)
+}
+
+func handleStar(rc *Context) *apiError {
+	return applyStarUnstar(rc, true)
+}
+
+func handleUnstar(rc *Context) *apiError {
+	return applyStarUnstar(rc, false)
+}
+
+func applyStarUnstar(rc *Context, star bool) *apiError {
+	db := rc.Server.DB
+	ids := rc.Params.GetAll("id")
+	albumIDs := rc.Params.GetAll("albumId")
+	artistIDs := rc.Params.GetAll("artistId")
+	if len(ids) == 0 && len(albumIDs) == 0 && len(artistIDs) == 0 {
+		return errMissingParameter("id, albumId, or artistId")
+	}
+
+	apply := func(itemType, itemID string) error {
+		if star {
+			return database.Star(db, rc.User.ID, itemType, itemID)
+		}
+		return database.Unstar(db, rc.User.ID, itemType, itemID)
+	}
+	for _, id := range ids {
+		if err := apply("song", id); err != nil {
+			return newError(ErrGeneric, err.Error())
+		}
+	}
+	for _, id := range albumIDs {
+		if err := apply("album", id); err != nil {
+			return newError(ErrGeneric, err.Error())
+		}
+	}
+	for _, id := range artistIDs {
+		if err := apply("artist", id); err != nil {
+			return newError(ErrGeneric, err.Error())
+		}
+	}
+
+	rc.Response = newResponse()
+	return nil
+}
+
+func handleSetRating(rc *Context) *apiError {
+	id, aerr := rc.Params.Required("id")
+	if aerr != nil {
+		return aerr
+	}
+	rating, rerr := rc.Params.RequiredInt("rating")
+	if rerr != nil {
+		return rerr
+	}
+	if rating < 0 || rating > 5 {
+		return newError(ErrGeneric, "rating must be between 0 and 5")
+	}
+
+	itemType := "song"
+	db := rc.Server.DB
+	if _, err := database.GetSong(db, id); err != nil {
+		if _, err := database.GetAlbum(db, id); err == nil {
+			itemType = "album"
+		} else if _, err := database.GetArtist(db, id); err == nil {
+			itemType = "artist"
+		} else {
+			return errNotFound("Item")
+		}
+	}
+
+	if err := database.SetRating(db, rc.User.ID, itemType, id, rating); err != nil {
+		return newError(ErrGeneric, err.Error())
+	}
+	rc.Response = newResponse()
+	return nil
+}
+
+// handleScrobble records playback history and, when submission=true (the
+// default), a scrobble record for potential Last.fm-style submission
+// later. VibeSonic doesn't itself talk to Last.fm in Phase 2; it stores
+// scrobbles so a future integration (or the user's own reporting) has
+// real data rather than inventing play counts elsewhere.
+func handleScrobble(rc *Context) *apiError {
+	ids := rc.Params.GetAll("id")
+	if len(ids) == 0 {
+		return errMissingParameter("id")
+	}
+	submission := rc.Params.Bool("submission", true)
+	client := rc.Params.Get("c")
+	db := rc.Server.DB
+
+	for _, id := range ids {
+		song, err := database.GetSong(db, id)
+		if err != nil {
+			continue // unknown song id: skip rather than fail the whole batch
+		}
+		if submission {
+			if err := database.RecordPlayback(db, rc.User.ID, song.ID, 0); err != nil {
+				return newError(ErrGeneric, err.Error())
+			}
+			if err := database.RecordScrobble(db, rc.User.ID, song.ID, client, false); err != nil {
+				return newError(ErrGeneric, err.Error())
+			}
+		} else {
+			if err := database.SetNowPlaying(db, rc.User.ID, song.ID, client, 0); err != nil {
+				return newError(ErrGeneric, err.Error())
+			}
+		}
+	}
+
+	rc.Response = newResponse()
+	return nil
+}