mirror of https://github.com/gabehf/Koito.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
135 lines
3.4 KiB
135 lines
3.4 KiB
// package imagesrc defines interfaces for album and artist image providers
|
|
package images
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"sync"
|
|
|
|
"github.com/gabehf/koito/internal/logger"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type ImageSource struct {
|
|
deezerEnabled bool
|
|
deezerC *DeezerClient
|
|
subsonicEnabled bool
|
|
subsonicC *SubsonicClient
|
|
caaEnabled bool
|
|
}
|
|
type ImageSourceOpts struct {
|
|
UserAgent string
|
|
EnableCAA bool
|
|
EnableDeezer bool
|
|
EnableSubsonic bool
|
|
}
|
|
|
|
var once sync.Once
|
|
var imgsrc ImageSource
|
|
|
|
type ArtistImageOpts struct {
|
|
Aliases []string
|
|
}
|
|
|
|
type AlbumImageOpts struct {
|
|
Artists []string
|
|
Album string
|
|
ReleaseMbzID *uuid.UUID
|
|
ReleaseGroupMbzID *uuid.UUID
|
|
}
|
|
|
|
const caaBaseUrl = "https://coverartarchive.org"
|
|
|
|
// all functions are no-op if no providers are enabled
|
|
func Initialize(opts ImageSourceOpts) {
|
|
once.Do(func() {
|
|
if opts.EnableCAA {
|
|
imgsrc.caaEnabled = true
|
|
}
|
|
if opts.EnableDeezer {
|
|
imgsrc.deezerEnabled = true
|
|
imgsrc.deezerC = NewDeezerClient()
|
|
}
|
|
if opts.EnableSubsonic {
|
|
imgsrc.subsonicEnabled = true
|
|
imgsrc.subsonicC = NewSubsonicClient()
|
|
}
|
|
})
|
|
}
|
|
|
|
func Shutdown() {
|
|
imgsrc.deezerC.Shutdown()
|
|
}
|
|
|
|
func GetArtistImage(ctx context.Context, opts ArtistImageOpts) (string, error) {
|
|
l := logger.FromContext(ctx)
|
|
if imgsrc.subsonicEnabled {
|
|
img, err := imgsrc.subsonicC.GetArtistImage(ctx, opts.Aliases[0])
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if img != "" {
|
|
return img, nil
|
|
}
|
|
l.Debug().Msg("Could not find artist image from Subsonic")
|
|
}
|
|
if imgsrc.deezerC != nil {
|
|
img, err := imgsrc.deezerC.GetArtistImages(ctx, opts.Aliases)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return img, nil
|
|
}
|
|
l.Warn().Msg("GetArtistImage: No image providers are enabled")
|
|
return "", nil
|
|
}
|
|
func GetAlbumImage(ctx context.Context, opts AlbumImageOpts) (string, error) {
|
|
l := logger.FromContext(ctx)
|
|
if imgsrc.subsonicEnabled {
|
|
img, err := imgsrc.subsonicC.GetAlbumImage(ctx, opts.Artists[0], opts.Album)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if img != "" {
|
|
return img, nil
|
|
}
|
|
l.Debug().Msg("Could not find album cover from Subsonic")
|
|
}
|
|
if imgsrc.caaEnabled {
|
|
l.Debug().Msg("Attempting to find album image from CoverArtArchive")
|
|
if opts.ReleaseMbzID != nil && *opts.ReleaseMbzID != uuid.Nil {
|
|
url := fmt.Sprintf(caaBaseUrl+"/release/%s/front", opts.ReleaseMbzID.String())
|
|
resp, err := http.DefaultClient.Head(url)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if resp.StatusCode == 200 {
|
|
return url, nil
|
|
}
|
|
l.Debug().Str("url", url).Str("status", resp.Status).Msg("Could not find album cover from CoverArtArchive with MusicBrainz release ID")
|
|
}
|
|
if opts.ReleaseGroupMbzID != nil && *opts.ReleaseGroupMbzID != uuid.Nil {
|
|
url := fmt.Sprintf(caaBaseUrl+"/release-group/%s/front", opts.ReleaseGroupMbzID.String())
|
|
resp, err := http.DefaultClient.Head(url)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if resp.StatusCode == 200 {
|
|
return url, nil
|
|
}
|
|
l.Debug().Str("url", url).Str("status", resp.Status).Msg("Could not find album cover from CoverArtArchive with MusicBrainz release group ID")
|
|
}
|
|
}
|
|
if imgsrc.deezerEnabled {
|
|
l.Debug().Msg("Attempting to find album image from Deezer")
|
|
img, err := imgsrc.deezerC.GetAlbumImages(ctx, opts.Artists, opts.Album)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return img, nil
|
|
}
|
|
l.Warn().Msg("GetAlbumImage: No image providers are enabled")
|
|
return "", nil
|
|
}
|