add Jellyfin albumiterator

This commit is contained in:
Drew Weymouth
2023-11-14 09:05:02 -08:00
parent 08ca09d686
commit 8aed1e5234
6 changed files with 86 additions and 21 deletions
@@ -49,9 +49,11 @@ func (r *baseIter) Next() *mediaprovider.Album {
return nil
}
r.serverPos += len(albums)
albums = sharedutil.FilterSlice(albums, func(al *mediaprovider.Album) bool {
return filterMatches(r.filter, al)
})
if !r.filter.IsNil() {
albums = sharedutil.FilterSlice(albums, func(al *mediaprovider.Album) bool {
return filterMatches(r.filter, al)
})
}
r.prefetched = albums
if len(albums) > 0 {
break
@@ -1,6 +1,11 @@
package jellyfin
import "github.com/dweymouth/supersonic/backend/mediaprovider"
import (
"github.com/dweymouth/go-jellyfin"
"github.com/dweymouth/supersonic/backend/mediaprovider"
"github.com/dweymouth/supersonic/backend/mediaprovider/helpers"
"github.com/dweymouth/supersonic/sharedutil"
)
const (
AlbumSortRecentlyAdded string = "Recently Added"
@@ -26,9 +31,64 @@ func (j *jellyfinMediaProvider) AlbumSortOrders() []string {
}
}
func (s *jellyfinMediaProvider) IterateAlbums(sortOrder string, filter mediaprovider.AlbumFilter) mediaprovider.AlbumIterator {
return nil
// TODO: unimplemented
func (j *jellyfinMediaProvider) IterateAlbums(sortOrder string, filter mediaprovider.AlbumFilter) mediaprovider.AlbumIterator {
var jfFilt jellyfin.Filter
if filter.ExcludeUnfavorited {
jfFilt.Favorite = true
filter.ExcludeUnfavorited = false
}
var jfSort jellyfin.Sort
switch sortOrder {
case AlbumSortRecentlyAdded:
jfSort.Field = "DateCreated"
jfSort.Mode = jellyfin.SortDesc
case AlbumSortFrequentlyPlayed:
jfSort.Field = "PlayCount"
jfSort.Mode = jellyfin.SortDesc
case AlbumSortRandom:
jfSort.Field = "Random"
case AlbumSortArtistAZ:
jfSort.Field = "AlbumArtist"
jfSort.Mode = jellyfin.SortAsc
case AlbumSortTitleAZ:
jfSort.Field = "SortName"
jfSort.Mode = jellyfin.SortAsc
case AlbumSortRecentlyPlayed:
jfSort.Field = "DatePlayed"
jfSort.Mode = jellyfin.SortDesc
case AlbumSortYearAscending:
jfSort.Field = "ProductionYear"
jfSort.Mode = jellyfin.SortAsc
case AlbumSortYearDescending:
jfSort.Field = "ProductionYear"
jfSort.Mode = jellyfin.SortDesc
}
fetcher := func(offs, limit int) ([]*mediaprovider.Album, error) {
al, err := j.client.GetAlbums(jellyfin.QueryOpts{
Sort: jfSort,
Filter: jfFilt,
Paging: jellyfin.Paging{StartIndex: offs, Limit: limit},
})
if err != nil {
return nil, err
}
return sharedutil.MapSlice(al, toAlbum), nil
}
if sortOrder == AlbumSortRandom {
determFetcher := func(offs, limit int) ([]*mediaprovider.Album, error) {
al, err := j.client.GetAlbums(jellyfin.QueryOpts{
Sort: jellyfin.Sort{Field: "SortName", Mode: jellyfin.SortAsc},
Filter: jfFilt,
Paging: jellyfin.Paging{StartIndex: offs, Limit: limit},
})
if err != nil {
return nil, err
}
return sharedutil.MapSlice(al, toAlbum), nil
}
return helpers.NewRandomIter(determFetcher, fetcher, filter, j.prefetchCoverCB)
}
return helpers.NewBaseIter(fetcher, filter, j.prefetchCoverCB)
}
func (s *jellyfinMediaProvider) SearchAlbums(searchQuery string, filter mediaprovider.AlbumFilter) mediaprovider.AlbumIterator {
@@ -9,12 +9,21 @@ import (
"time"
"github.com/dweymouth/go-jellyfin"
jellyfinCli "github.com/dweymouth/go-jellyfin"
"github.com/dweymouth/supersonic/backend/mediaprovider"
"github.com/dweymouth/supersonic/sharedutil"
)
const cacheValidDurationSeconds = 60
type JellyfinServer struct {
jellyfinCli.Client
}
func (j *JellyfinServer) MediaProvider() mediaprovider.MediaProvider {
return newJellyfinMediaProvider(&j.Client)
}
var _ mediaprovider.MediaProvider = (*jellyfinMediaProvider)(nil)
type jellyfinMediaProvider struct {
@@ -1,14 +0,0 @@
package jellyfin
import (
jellyfinCli "github.com/dweymouth/go-jellyfin"
"github.com/dweymouth/supersonic/backend/mediaprovider"
)
type JellyfinServer struct {
jellyfinCli.Client
}
func (j *JellyfinServer) MediaProvider() mediaprovider.MediaProvider {
return newJellyfinMediaProvider(&j.Client)
}
+7
View File
@@ -14,6 +14,13 @@ type AlbumFilter struct {
ExcludeUnfavorited bool // mut. exc. with ExcludeFavorited
}
// Returns true if the filter is the nil filter - i.e. matches everything
func (a *AlbumFilter) IsNil() bool {
return a.MinYear == 0 && a.MaxYear == 0 &&
len(a.Genres) == 0 &&
!a.ExcludeFavorited && !a.ExcludeUnfavorited
}
type AlbumIterator interface {
Next() *Album
}
@@ -74,6 +74,7 @@ func (s *subsonicMediaProvider) IterateAlbums(sortOrder string, filter mediaprov
return helpers.NewBaseIter(makeFetchFn(fetchFn), filter, s.prefetchCoverCB)
}
if sortOrder == "" && filter.ExcludeUnfavorited {
filter.ExcludeUnfavorited = false // we're already filtering by this
return s.baseIterFromSimpleSortOrder("starred", filter)
}
if sortOrder == "" {