157 lines
3.0 KiB
Go
157 lines
3.0 KiB
Go
package ui
|
|
|
|
import (
|
|
"context"
|
|
"image"
|
|
"supersonic/backend"
|
|
"supersonic/ui/widgets"
|
|
|
|
"fyne.io/fyne/v2"
|
|
"fyne.io/fyne/v2/widget"
|
|
"github.com/dweymouth/go-subsonic"
|
|
)
|
|
|
|
type ImageFetcher func(string) (image.Image, error)
|
|
|
|
type AlbumGrid struct {
|
|
widget.BaseWidget
|
|
|
|
grid *widget.GridWrapList
|
|
albums []*subsonic.AlbumID3
|
|
iter backend.AlbumIterator
|
|
fetching bool
|
|
done bool
|
|
|
|
imageFetcher ImageFetcher
|
|
OnPlayAlbum func(string)
|
|
OnShowArtistPage func(string)
|
|
}
|
|
|
|
var _ fyne.Widget = (*AlbumGrid)(nil)
|
|
|
|
func NewFixedAlbumGrid(albums []*subsonic.AlbumID3, fetch ImageFetcher) *AlbumGrid {
|
|
ag := &AlbumGrid{
|
|
albums: albums,
|
|
done: true,
|
|
imageFetcher: fetch,
|
|
}
|
|
ag.ExtendBaseWidget(ag)
|
|
ag.createGridWrapList()
|
|
return ag
|
|
}
|
|
|
|
func NewAlbumGrid(iter backend.AlbumIterator, fetch ImageFetcher) *AlbumGrid {
|
|
ag := &AlbumGrid{
|
|
iter: iter,
|
|
imageFetcher: fetch,
|
|
}
|
|
ag.ExtendBaseWidget(ag)
|
|
|
|
ag.createGridWrapList()
|
|
|
|
// fetch initial albums
|
|
ag.fetchMoreAlbums(36)
|
|
return ag
|
|
}
|
|
|
|
func (ag *AlbumGrid) Clear() {
|
|
ag.albums = nil
|
|
ag.done = true
|
|
}
|
|
|
|
func (ag *AlbumGrid) Reset(iter backend.AlbumIterator) {
|
|
ag.albums = nil
|
|
ag.fetching = false
|
|
ag.done = false
|
|
ag.iter = iter
|
|
ag.fetchMoreAlbums(36)
|
|
}
|
|
|
|
func (ag *AlbumGrid) createGridWrapList() {
|
|
g := widget.NewGridWrapList(
|
|
func() int {
|
|
return len(ag.albums)
|
|
},
|
|
// create func
|
|
func() fyne.CanvasObject {
|
|
ac := widgets.NewAlbumCard()
|
|
ac.OnPlay = func() {
|
|
if ag.OnPlayAlbum != nil {
|
|
ag.OnPlayAlbum(ac.AlbumID())
|
|
}
|
|
}
|
|
ac.OnShowArtistPage = func() {
|
|
if ag.OnShowArtistPage != nil {
|
|
ag.OnShowArtistPage(ac.ArtistID())
|
|
}
|
|
}
|
|
return ac
|
|
},
|
|
// update func
|
|
func(itemID int, obj fyne.CanvasObject) {
|
|
ac := obj.(*widgets.AlbumCard)
|
|
ag.doUpdateAlbumCard(itemID, ac)
|
|
},
|
|
)
|
|
ag.grid = g
|
|
}
|
|
|
|
func (ag *AlbumGrid) doUpdateAlbumCard(albumIdx int, ac *widgets.AlbumCard) {
|
|
album := ag.albums[albumIdx]
|
|
if ac.PrevAlbumID == album.ID {
|
|
// nothing to do
|
|
return
|
|
}
|
|
ac.Update(album)
|
|
ac.PrevAlbumID = album.ID
|
|
// TODO: set image to a placeholder before spinning off async fetch
|
|
// cancel any previous image fetch
|
|
if ac.ImgLoadCancel != nil {
|
|
ac.ImgLoadCancel()
|
|
ac.ImgLoadCancel = nil
|
|
}
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
go func(ctx context.Context) {
|
|
i, err := ag.imageFetcher(album.ID)
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
default:
|
|
if err == nil {
|
|
ac.Cover.SetImage(i)
|
|
ac.Cover.Refresh()
|
|
}
|
|
}
|
|
}(ctx)
|
|
ac.ImgLoadCancel = cancel
|
|
|
|
// TODO: remove magic number 10
|
|
if !ag.done && !ag.fetching && albumIdx > len(ag.albums)-10 {
|
|
ag.fetchMoreAlbums(10)
|
|
}
|
|
}
|
|
|
|
func (a *AlbumGrid) fetchMoreAlbums(count int) {
|
|
if a.iter == nil {
|
|
a.done = true
|
|
}
|
|
i := 0
|
|
a.fetching = true
|
|
a.iter.NextN(count, func(al *subsonic.AlbumID3) {
|
|
if al == nil {
|
|
a.done = true
|
|
return
|
|
}
|
|
a.albums = append(a.albums, al)
|
|
i++
|
|
if i == count {
|
|
a.fetching = false
|
|
}
|
|
a.Refresh()
|
|
})
|
|
}
|
|
|
|
func (a *AlbumGrid) CreateRenderer() fyne.WidgetRenderer {
|
|
return widget.NewSimpleRenderer(a.grid)
|
|
}
|