Skip to content

Commit

Permalink
feat(subsonic): update track play stats on scrobble instead of stream
Browse files Browse the repository at this point in the history
  • Loading branch information
sentriz committed Sep 28, 2023
1 parent 6b322e4 commit e0b1603
Show file tree
Hide file tree
Showing 10 changed files with 179 additions and 166 deletions.
3 changes: 2 additions & 1 deletion cmd/gonic/gonic.go
Expand Up @@ -32,6 +32,7 @@ import (
"go.senan.xyz/gonic/podcasts"
"go.senan.xyz/gonic/scanner"
"go.senan.xyz/gonic/scanner/tags"
"go.senan.xyz/gonic/scrobble"
"go.senan.xyz/gonic/server/ctrladmin"
"go.senan.xyz/gonic/server/ctrlbase"
"go.senan.xyz/gonic/server/ctrlsubsonic"
Expand Down Expand Up @@ -235,7 +236,7 @@ func main() {
CacheCoverPath: cacheDirCovers,
LastFMClient: lastfmClient,
ArtistInfoCache: artistInfoCache,
Scrobblers: []ctrlsubsonic.Scrobbler{
Scrobblers: []scrobble.Scrobbler{
lastfmClient,
listenbrainzClient,
},
Expand Down
2 changes: 1 addition & 1 deletion go.mod
Expand Up @@ -5,6 +5,7 @@ go 1.21
require (
github.com/Masterminds/sprig v2.22.0+incompatible
github.com/andybalholm/cascadia v1.3.2
github.com/davecgh/go-spew v1.1.1
github.com/dexterlb/mpvipc v0.0.0-20230829142118-145d6eabdc37
github.com/disintegration/imaging v1.6.2
github.com/dustin/go-humanize v1.0.1
Expand Down Expand Up @@ -40,7 +41,6 @@ require (
github.com/Masterminds/goutils v1.1.1 // indirect
github.com/Masterminds/semver v1.5.0 // indirect
github.com/PuerkitoBio/goquery v1.8.1 // indirect
github.com/davecgh/go-spew v1.1.1 // indirect
github.com/felixge/httpsnoop v1.0.3 // indirect
github.com/go-openapi/jsonpointer v0.19.5 // indirect
github.com/go-openapi/swag v0.21.1 // indirect
Expand Down
42 changes: 18 additions & 24 deletions lastfm/client.go
Expand Up @@ -10,12 +10,11 @@ import (
"net/url"
"sort"
"strconv"
"strings"
"time"

"github.com/andybalholm/cascadia"
"github.com/google/uuid"
"go.senan.xyz/gonic/db"
"go.senan.xyz/gonic/scrobble"
"golang.org/x/net/html"
)

Expand Down Expand Up @@ -185,11 +184,11 @@ func (c *Client) StealArtistImage(artistURL string) (string, error) {
return imageURL, nil
}

func (c *Client) IsUserAuthenticated(user *db.User) bool {
func (c *Client) IsUserAuthenticated(user db.User) bool {
return user.LastFMSession != ""
}

func (c *Client) Scrobble(user *db.User, track *db.Track, stamp time.Time, submission bool) error {
func (c *Client) Scrobble(user db.User, track scrobble.Track, stamp time.Time, submission bool) error {
apiKey, secret, err := c.keySecret()
if err != nil {
return fmt.Errorf("get key and secret: %w", err)
Expand All @@ -198,33 +197,27 @@ func (c *Client) Scrobble(user *db.User, track *db.Track, stamp time.Time, submi
return ErrNoUserSession
}

if track.Album == nil || len(track.Album.Artists) == 0 {
return fmt.Errorf("track has no album artists")
}

params := url.Values{}
if submission {
params.Add("method", "track.Scrobble")
// last.fm wants the timestamp in seconds
params.Add("timestamp", strconv.Itoa(int(stamp.Unix())))
params.Add("timestamp", strconv.Itoa(int(stamp.Unix()))) // last.fm wants the timestamp in seconds
} else {
params.Add("method", "track.updateNowPlaying")
}

params.Add("api_key", apiKey)
params.Add("sk", user.LastFMSession)
params.Add("artist", track.TagTrackArtist)
params.Add("track", track.TagTitle)
params.Add("trackNumber", strconv.Itoa(track.TagTrackNumber))
params.Add("album", track.Album.TagTitle)
params.Add("albumArtist", strings.Join(track.Album.ArtistsStrings(), ", "))
params.Add("duration", strconv.Itoa(track.Length))

// make sure we provide a valid uuid, since some users may have an incorrect mbid in their tags
if _, err := uuid.Parse(track.TagBrainzID); err == nil {
params.Add("mbid", track.TagBrainzID)
params.Add("artist", track.Artist)
params.Add("track", track.Track)
params.Add("trackNumber", strconv.Itoa(int(track.TrackNumber)))
params.Add("album", track.Album)
params.Add("albumArtist", track.AlbumArtist)
params.Add("duration", strconv.Itoa(int(track.Duration.Seconds())))

if track.MusicBrainzID != "" {
params.Add("mbid", track.MusicBrainzID)
}

params.Add("sk", user.LastFMSession)
params.Add("api_key", apiKey)
params.Add("api_sig", GetParamSignature(params, secret))

_, err = c.makeRequest(http.MethodPost, params)
Expand All @@ -236,7 +229,7 @@ func (c *Client) LoveTrack(user *db.User, track *db.Track) error {
if err != nil {
return fmt.Errorf("get key and secret: %w", err)
}
if !c.IsUserAuthenticated(user) {
if !c.IsUserAuthenticated(*user) {
return ErrNoUserSession
}

Expand All @@ -257,7 +250,7 @@ func (c *Client) GetCurrentUser(user *db.User) (User, error) {
if err != nil {
return User{}, fmt.Errorf("get key and secret: %w", err)
}
if !c.IsUserAuthenticated(user) {
if !c.IsUserAuthenticated(*user) {
return User{}, ErrNoUserSession
}

Expand All @@ -277,6 +270,7 @@ func (c *Client) GetCurrentUser(user *db.User) (User, error) {
func (c *Client) makeRequest(method string, params url.Values) (LastFM, error) {
req, _ := http.NewRequest(method, BaseURL, nil)
req.URL.RawQuery = params.Encode()

resp, err := c.httpClient.Do(req)
if err != nil {
return LastFM{}, fmt.Errorf("get: %w", err)
Expand Down
42 changes: 19 additions & 23 deletions lastfm/client_test.go
Expand Up @@ -14,6 +14,7 @@ import (
"go.senan.xyz/gonic/db"
"go.senan.xyz/gonic/lastfm"
"go.senan.xyz/gonic/lastfm/mockclient"
"go.senan.xyz/gonic/scrobble"
)

func TestArtistGetInfo(t *testing.T) {
Expand Down Expand Up @@ -485,26 +486,6 @@ func TestGetSessionClientRequestFails(t *testing.T) {
func TestScrobble(t *testing.T) {
t.Parallel()

user := &db.User{
LastFMSession: "lastFMSession1",
}

track := &db.Track{
Album: &db.Album{
TagTitle: "album1",
Artists: []*db.Artist{{
Name: "artist1",
}},
},
Length: 100,
TagBrainzID: "916b242d-d439-4ae4-a439-556eef99c06e",
TagTitle: "title1",
TagTrackArtist: "trackArtist1",
TagTrackNumber: 1,
}

stamp := time.Date(2023, 8, 12, 12, 34, 1, 200, time.UTC)

client := lastfm.NewClientCustom(
mockclient.New(t, func(w http.ResponseWriter, r *http.Request) {
require.Equal(t, http.MethodPost, r.Method)
Expand Down Expand Up @@ -534,6 +515,21 @@ func TestScrobble(t *testing.T) {
},
)

user := db.User{
LastFMSession: "lastFMSession1",
}
track := scrobble.Track{
Track: "title1",
Artist: "trackArtist1",
Album: "album1",
AlbumArtist: "artist1",
TrackNumber: 1,
Duration: 100 * time.Second,
MusicBrainzID: "916b242d-d439-4ae4-a439-556eef99c06e",
}

stamp := time.Date(2023, 8, 12, 12, 34, 1, 200, time.UTC)

err := client.Scrobble(user, track, stamp, true)
require.NoError(t, err)
}
Expand All @@ -545,22 +541,22 @@ func TestScrobbleErrorsWithoutLastFMSession(t *testing.T) {
return "", "", nil
})

err := client.Scrobble(&db.User{}, &db.Track{}, time.Now(), false)
err := client.Scrobble(db.User{}, scrobble.Track{}, time.Now(), false)
require.ErrorIs(t, err, lastfm.ErrNoUserSession)
}

func TestScrobbleFailsWithoutLastFMAPIKey(t *testing.T) {
t.Parallel()

user := &db.User{
user := db.User{
LastFMSession: "lastFMSession1",
}

scrobbler := lastfm.NewClient(func() (string, string, error) {
return "", "", fmt.Errorf("no keys")
})

err := scrobbler.Scrobble(user, &db.Track{}, time.Now(), false)
err := scrobbler.Scrobble(user, scrobble.Track{}, time.Now(), false)

require.Error(t, err)
}
Expand Down
24 changes: 9 additions & 15 deletions listenbrainz/listenbrainz.go
Expand Up @@ -10,8 +10,8 @@ import (
"net/http/httputil"
"time"

"github.com/google/uuid"
"go.senan.xyz/gonic/db"
"go.senan.xyz/gonic/scrobble"
)

const (
Expand All @@ -36,27 +36,21 @@ func NewClientCustom(httpClient *http.Client) *Client {
return &Client{httpClient: httpClient}
}

func (c *Client) IsUserAuthenticated(user *db.User) bool {
func (c *Client) IsUserAuthenticated(user db.User) bool {
return user.ListenBrainzURL != "" && user.ListenBrainzToken != ""
}

func (c *Client) Scrobble(user *db.User, track *db.Track, stamp time.Time, submission bool) error {
// make sure we provide a valid uuid, since some users may have an incorrect mbid in their tags
var trackMBID string
if _, err := uuid.Parse(track.TagBrainzID); err == nil {
trackMBID = track.TagBrainzID
}

func (c *Client) Scrobble(user db.User, track scrobble.Track, stamp time.Time, submission bool) error {
payload := &Payload{
TrackMetadata: &TrackMetadata{
AdditionalInfo: &AdditionalInfo{
TrackNumber: track.TagTrackNumber,
RecordingMBID: trackMBID,
TrackLength: track.Length,
TrackNumber: int(track.TrackNumber),
RecordingMBID: track.MusicBrainzID,
TrackLength: int(track.Duration.Seconds()),
},
ArtistName: track.TagTrackArtist,
TrackName: track.TagTitle,
ReleaseName: track.Album.TagTitle,
ArtistName: track.Artist,
TrackName: track.Track,
ReleaseName: track.Album,
},
}
scrobble := Scrobble{
Expand Down
13 changes: 7 additions & 6 deletions listenbrainz/listenbrainz_test.go
Expand Up @@ -14,6 +14,7 @@ import (
"github.com/stretchr/testify/require"
"go.senan.xyz/gonic/db"
"go.senan.xyz/gonic/listenbrainz"
"go.senan.xyz/gonic/scrobble"
)

func TestScrobble(t *testing.T) {
Expand All @@ -35,8 +36,8 @@ func TestScrobble(t *testing.T) {
)

err := client.Scrobble(
&db.User{ListenBrainzURL: "https://listenbrainz.org", ListenBrainzToken: "token1"},
&db.Track{Album: &db.Album{TagTitle: "album"}, TagTitle: "title", TagTrackArtist: "artist", TagTrackNumber: 1},
db.User{ListenBrainzURL: "https://listenbrainz.org", ListenBrainzToken: "token1"},
scrobble.Track{Track: "title", Artist: "artist", Album: "album", TrackNumber: 1},
time.Unix(1683804525, 0),
true,
)
Expand All @@ -59,8 +60,8 @@ func TestScrobbleUnauthorized(t *testing.T) {
)

err := client.Scrobble(
&db.User{ListenBrainzURL: "https://listenbrainz.org", ListenBrainzToken: "token1"},
&db.Track{Album: &db.Album{TagTitle: "album"}, TagTitle: "title", TagTrackArtist: "artist", TagTrackNumber: 1},
db.User{ListenBrainzURL: "https://listenbrainz.org", ListenBrainzToken: "token1"},
scrobble.Track{Track: "title", Artist: "artist", Album: "album", TrackNumber: 1},
time.Now(),
true,
)
Expand All @@ -83,8 +84,8 @@ func TestScrobbleServerError(t *testing.T) {
)

err := client.Scrobble(
&db.User{ListenBrainzURL: "https://listenbrainz.org", ListenBrainzToken: "token1"},
&db.Track{Album: &db.Album{TagTitle: "album"}, TagTitle: "title", TagTrackArtist: "artist", TagTrackNumber: 1},
db.User{ListenBrainzURL: "https://listenbrainz.org", ListenBrainzToken: "token1"},
scrobble.Track{Track: "title", Artist: "artist", Album: "album", TrackNumber: 1},
time.Now(),
true,
)
Expand Down
22 changes: 22 additions & 0 deletions scrobble/scrobble.go
@@ -0,0 +1,22 @@
package scrobble

import (
"time"

"go.senan.xyz/gonic/db"
)

type Track struct {
Track string
Artist string
Album string
AlbumArtist string
TrackNumber uint
Duration time.Duration
MusicBrainzID string
}

type Scrobbler interface {
IsUserAuthenticated(user db.User) bool
Scrobble(user db.User, track Track, stamp time.Time, submission bool) error
}
10 changes: 2 additions & 8 deletions server/ctrlsubsonic/ctrl.go
Expand Up @@ -7,12 +7,11 @@ import (
"io"
"log"
"net/http"
"time"

"go.senan.xyz/gonic/db"
"go.senan.xyz/gonic/jukebox"
"go.senan.xyz/gonic/lastfm"
"go.senan.xyz/gonic/podcasts"
"go.senan.xyz/gonic/scrobble"
"go.senan.xyz/gonic/server/ctrlbase"
"go.senan.xyz/gonic/server/ctrlsubsonic/artistinfocache"
"go.senan.xyz/gonic/server/ctrlsubsonic/params"
Expand Down Expand Up @@ -40,19 +39,14 @@ func PathsOf(paths []MusicPath) []string {
return r
}

type Scrobbler interface {
IsUserAuthenticated(user *db.User) bool
Scrobble(user *db.User, track *db.Track, stamp time.Time, submission bool) error
}

type Controller struct {
*ctrlbase.Controller
MusicPaths []MusicPath
PodcastsPath string
CacheAudioPath string
CacheCoverPath string
Jukebox *jukebox.Jukebox
Scrobblers []Scrobbler
Scrobblers []scrobble.Scrobbler
Podcasts *podcasts.Podcasts
Transcoder transcode.Transcoder
LastFMClient *lastfm.Client
Expand Down

0 comments on commit e0b1603

Please sign in to comment.