add caching of artist images

This commit is contained in:
Drew Weymouth
2023-03-19 11:01:20 -07:00
parent 005191c9c3
commit 2380caab35
4 changed files with 95 additions and 23 deletions
+74 -13
View File
@@ -1,6 +1,7 @@
package backend
import (
"bytes"
"context"
"fmt"
"image"
@@ -11,9 +12,12 @@ import (
"path/filepath"
"time"
"fyne.io/fyne/v2"
"github.com/20after4/configdir"
)
const CachedImageValidTime = 24 * time.Hour
type ImageManager struct {
s *ServerManager
baseCacheDir string
@@ -77,24 +81,60 @@ func (i *ImageManager) GetFullSizeAlbumCover(albumID string) (image.Image, error
return im, nil
}
func (i *ImageManager) GetCachedArtistImage(artistID string) (image.Image, bool) {
return i.loadLocalImage(i.filePathForArtistImage(artistID))
}
func (i *ImageManager) FetchAndCacheArtistImage(artistID string, imgURL string) (image.Image, error) {
im, err := i.fetchRemoteArtistImage(imgURL)
if err != nil {
return nil, err
}
_ = i.writeJpeg(im, i.filePathForArtistImage(artistID))
return im, nil
}
func (i *ImageManager) RefreshCachedArtistImageIfExpired(artistID string, imgURL string) error {
stat, err := os.Stat(i.filePathForArtistImage(artistID))
if err == nil && time.Since(stat.ModTime()) > CachedImageValidTime {
_, err = i.FetchAndCacheArtistImage(artistID, imgURL)
}
return err
}
func (i *ImageManager) ensureCoverCacheDir() string {
path := path.Join(i.baseCacheDir, i.s.ServerID.String(), "covers")
configdir.MakePath(path)
return path
}
func (i *ImageManager) ensureArtistCoverCacheDir() string {
path := path.Join(i.baseCacheDir, i.s.ServerID.String(), "artistimages")
configdir.MakePath(path)
return path
}
func (i *ImageManager) fetchRemoteArtistImage(url string) (image.Image, error) {
res, err := fyne.LoadResourceFromURLString(url)
if err == nil {
im, _, err := image.Decode(bytes.NewReader(res.Content()))
if err == nil {
return im, nil
}
return nil, err
}
return nil, err
}
func (i *ImageManager) fetchAndCacheCoverFromDiskOrServer(albumID string, ttl time.Duration) (image.Image, error) {
// on disc cache
path := i.filePathForCover(albumID)
if i.ensureCoverCacheDir() != "" {
if s, err := os.Stat(path); err == nil {
go i.checkRefreshLocalCover(s, albumID, ttl)
if f, err := os.Open(path); err == nil {
defer f.Close()
if img, _, err := image.Decode(f); err == nil {
i.thumbnailCache.SetWithTTL(albumID, img, ttl)
return img, nil
}
if img, ok := i.loadLocalImage(path); ok {
i.thumbnailCache.SetWithTTL(albumID, img, ttl)
return img, nil
}
}
}
@@ -109,19 +149,14 @@ func (i *ImageManager) fetchAndCacheCoverFromServer(albumID string, ttl time.Dur
}
if i.ensureCoverCacheDir() != "" {
path := i.filePathForCover(albumID)
if f, err := os.Create(path); err == nil {
defer f.Close()
if err := jpeg.Encode(f, img, nil /*options*/); err != nil {
log.Printf("failed to cache image: %s", err.Error())
}
}
_ = i.writeJpeg(img, path)
}
i.thumbnailCache.SetWithTTL(albumID, img, ttl)
return img, nil
}
func (i *ImageManager) checkRefreshLocalCover(stat os.FileInfo, albumID string, ttl time.Duration) {
if time.Since(stat.ModTime()) > 24*time.Hour {
if time.Since(stat.ModTime()) > CachedImageValidTime {
i.fetchAndCacheCoverFromServer(albumID, ttl)
}
}
@@ -129,3 +164,29 @@ func (i *ImageManager) checkRefreshLocalCover(stat os.FileInfo, albumID string,
func (i *ImageManager) filePathForCover(albumID string) string {
return filepath.Join(i.ensureCoverCacheDir(), fmt.Sprintf("%s.jpg", albumID))
}
func (i *ImageManager) filePathForArtistImage(id string) string {
return filepath.Join(i.ensureArtistCoverCacheDir(), fmt.Sprintf("%s.jpg", id))
}
func (i *ImageManager) writeJpeg(img image.Image, path string) error {
f, err := os.Create(path)
if err == nil {
defer f.Close()
if err := jpeg.Encode(f, img, nil /*options*/); err != nil {
log.Printf("failed to cache image: %s", err.Error())
return err
}
}
return err
}
func (i *ImageManager) loadLocalImage(path string) (image.Image, bool) {
if f, err := os.Open(path); err == nil {
defer f.Close()
if img, _, err := image.Decode(f); err == nil {
return img, true
}
}
return nil, false
}