122 lines
3.7 KiB
Go
122 lines
3.7 KiB
Go
package jellyfin
|
|
|
|
import (
|
|
"slices"
|
|
|
|
"github.com/dweymouth/go-jellyfin"
|
|
"github.com/dweymouth/supersonic/backend/mediaprovider"
|
|
"github.com/dweymouth/supersonic/backend/mediaprovider/helpers"
|
|
"github.com/dweymouth/supersonic/sharedutil"
|
|
)
|
|
|
|
func (j *jellyfinMediaProvider) ArtistSortOrders() []string {
|
|
return []string{
|
|
mediaprovider.ArtistSortAlbumCount,
|
|
mediaprovider.ArtistSortNameAZ,
|
|
mediaprovider.ArtistSortRandom,
|
|
}
|
|
}
|
|
|
|
func (j *jellyfinMediaProvider) IterateArtists(sortOrder string, filter mediaprovider.ArtistFilter) mediaprovider.ArtistIterator {
|
|
var jfSort jellyfin.Sort
|
|
var disablePagination bool
|
|
var sortFn func([]*jellyfin.Artist) []*jellyfin.Artist
|
|
|
|
if sortOrder == "" {
|
|
sortOrder = mediaprovider.ArtistSortNameAZ // default
|
|
}
|
|
switch sortOrder {
|
|
case mediaprovider.ArtistSortAlbumCount:
|
|
// Pagination needs to be disabled, to retrieve all results in a single request, and correctly sort them.
|
|
disablePagination = true
|
|
sortFn = func(artists []*jellyfin.Artist) []*jellyfin.Artist {
|
|
slices.SortStableFunc(artists, func(a, b *jellyfin.Artist) int {
|
|
return b.AlbumCount - a.AlbumCount
|
|
})
|
|
return artists
|
|
}
|
|
case mediaprovider.ArtistSortNameAZ:
|
|
jfSort.Field = jellyfin.SortByName
|
|
jfSort.Mode = jellyfin.SortAsc
|
|
case mediaprovider.ArtistSortRandom:
|
|
jfSort.Field = jellyfin.SortByRandom
|
|
}
|
|
|
|
fetcher := makeArtistFetchFn(
|
|
func(offs, limit int) ([]*jellyfin.Artist, error) {
|
|
if disablePagination && offs > 0 {
|
|
return nil, nil
|
|
}
|
|
var paging jellyfin.Paging
|
|
if !disablePagination {
|
|
paging = jellyfin.Paging{StartIndex: offs, Limit: limit}
|
|
}
|
|
return j.client.GetAlbumArtists(jellyfin.QueryOpts{
|
|
Sort: jfSort,
|
|
Paging: paging,
|
|
Filter: jellyfin.Filter{ParentID: j.currentLibraryID},
|
|
})
|
|
},
|
|
sortFn,
|
|
)
|
|
|
|
return helpers.NewArtistIterator(fetcher, filter, j.prefetchCoverCB)
|
|
}
|
|
|
|
func (j *jellyfinMediaProvider) SearchArtists(searchQuery string, filter mediaprovider.ArtistFilter) mediaprovider.ArtistIterator {
|
|
// TODO: Jellyfin API is not returning search results for artists.
|
|
// Uncomment the following code once the issue is resolved.
|
|
// Related issue: https://github.com/jellyfin/jellyfin/issues/8222
|
|
// fetcher := makeArtistFetchFn(
|
|
// func(offs, limit int) ([]*jellyfin.Artist, error) {
|
|
// log.Printf("Searching for artists: %s", searchQuery)
|
|
// sr, err := j.client.Search(searchQuery, jellyfin.TypeArtist, jellyfin.Paging{StartIndex: offs, Limit: limit})
|
|
// if err != nil {
|
|
// return nil, err
|
|
// }
|
|
// log.Printf("Found %d artists", len(sr.Artists))
|
|
// for _, a := range sr.Artists {
|
|
// log.Printf("Artist: %s", a.Name)
|
|
// }
|
|
// return sr.Artists, nil
|
|
// },
|
|
// nil,
|
|
// )
|
|
// return helpers.NewArtistIterator(fetcher, filter, j.prefetchCoverCB)
|
|
|
|
modifiedFilter := filter.Clone()
|
|
modifiedOptions := modifiedFilter.Options()
|
|
modifiedOptions.SearchQuery = searchQuery
|
|
modifiedFilter.SetOptions(modifiedOptions)
|
|
|
|
fetcher := makeArtistFetchFn(
|
|
func(offs, limit int) ([]*jellyfin.Artist, error) {
|
|
return j.client.GetAlbumArtists(jellyfin.QueryOpts{
|
|
Sort: jellyfin.Sort{
|
|
Field: jellyfin.SortByName,
|
|
Mode: jellyfin.SortAsc,
|
|
},
|
|
Paging: jellyfin.Paging{StartIndex: offs, Limit: limit},
|
|
})
|
|
},
|
|
nil,
|
|
)
|
|
return helpers.NewArtistIterator(fetcher, modifiedFilter, j.prefetchCoverCB)
|
|
}
|
|
|
|
func makeArtistFetchFn(
|
|
fetchFn func(offset, limit int) ([]*jellyfin.Artist, error),
|
|
sortFn func([]*jellyfin.Artist) []*jellyfin.Artist,
|
|
) helpers.ArtistFetchFn {
|
|
return func(offset, limit int) ([]*mediaprovider.Artist, error) {
|
|
ar, err := fetchFn(offset, limit)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if sortFn != nil {
|
|
ar = sortFn(ar)
|
|
}
|
|
return sharedutil.MapSlice(ar, toArtist), nil
|
|
}
|
|
}
|