96 lines
2.5 KiB
Go
96 lines
2.5 KiB
Go
package spotify
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/navidrome/navidrome/conf"
|
|
"github.com/navidrome/navidrome/consts"
|
|
"github.com/navidrome/navidrome/core/agents"
|
|
"github.com/navidrome/navidrome/log"
|
|
"github.com/navidrome/navidrome/model"
|
|
"github.com/navidrome/navidrome/utils"
|
|
"github.com/xrash/smetrics"
|
|
)
|
|
|
|
const spotifyAgentName = "spotify"
|
|
|
|
type spotifyAgent struct {
|
|
ds model.DataStore
|
|
id string
|
|
secret string
|
|
client *Client
|
|
}
|
|
|
|
func spotifyConstructor(ds model.DataStore) agents.Interface {
|
|
l := &spotifyAgent{
|
|
ds: ds,
|
|
id: conf.Server.Spotify.ID,
|
|
secret: conf.Server.Spotify.Secret,
|
|
}
|
|
hc := &http.Client{
|
|
Timeout: consts.DefaultHttpClientTimeOut,
|
|
}
|
|
chc := utils.NewCachedHTTPClient(hc, consts.DefaultHttpClientTimeOut)
|
|
l.client = NewClient(l.id, l.secret, chc)
|
|
return l
|
|
}
|
|
|
|
func (s *spotifyAgent) AgentName() string {
|
|
return spotifyAgentName
|
|
}
|
|
|
|
func (s *spotifyAgent) GetImages(ctx context.Context, id, name, mbid string) ([]agents.ArtistImage, error) {
|
|
a, err := s.searchArtist(ctx, name)
|
|
if err != nil {
|
|
if errors.Is(err, model.ErrNotFound) {
|
|
log.Warn(ctx, "Artist not found in Spotify", "artist", name)
|
|
} else {
|
|
log.Error(ctx, "Error calling Spotify", "artist", name, err)
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
var res []agents.ArtistImage
|
|
for _, img := range a.Images {
|
|
res = append(res, agents.ArtistImage{
|
|
URL: img.URL,
|
|
Size: img.Width,
|
|
})
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func (s *spotifyAgent) searchArtist(ctx context.Context, name string) (*Artist, error) {
|
|
artists, err := s.client.SearchArtists(ctx, name, 40)
|
|
if err != nil || len(artists) == 0 {
|
|
return nil, model.ErrNotFound
|
|
}
|
|
name = strings.ToLower(name)
|
|
|
|
// Sort results, prioritizing artists with images, with similar names and with high popularity, in this order
|
|
sort.Slice(artists, func(i, j int) bool {
|
|
ai := fmt.Sprintf("%-5t-%03d-%04d", len(artists[i].Images) == 0, smetrics.WagnerFischer(name, strings.ToLower(artists[i].Name), 1, 1, 2), 1000-artists[i].Popularity)
|
|
aj := fmt.Sprintf("%-5t-%03d-%04d", len(artists[j].Images) == 0, smetrics.WagnerFischer(name, strings.ToLower(artists[j].Name), 1, 1, 2), 1000-artists[j].Popularity)
|
|
return ai < aj
|
|
})
|
|
|
|
// If the first one has the same name, that's the one
|
|
if strings.ToLower(artists[0].Name) != name {
|
|
return nil, model.ErrNotFound
|
|
}
|
|
return &artists[0], err
|
|
}
|
|
|
|
func init() {
|
|
conf.AddHook(func() {
|
|
if conf.Server.Spotify.ID != "" && conf.Server.Spotify.Secret != "" {
|
|
agents.Register(spotifyAgentName, spotifyConstructor)
|
|
}
|
|
})
|
|
}
|